Update function use langgraph.

muazhari · Jun 5, 2024 · 5d1ca02 · 5d1ca02
1 parent d7ba8ed
commit 5d1ca02
Show file tree

Hide file tree

Showing 4 changed files with 23 additions and 23 deletions.
diff --git a/.idea/workspace.xml b/.idea/workspace.xml
diff --git a/apps/inners/use_cases/graphs/long_form_qa_graph.py b/apps/inners/use_cases/graphs/long_form_qa_graph.py
@@ -416,47 +416,47 @@ def _compile(self) -> CompiledGraph:
         graph: StateGraph = StateGraph(LongFormQaGraphState)
 
         graph.add_node(
-            key=self.node_get_llm_model.__name__,
+            node=self.node_get_llm_model.__name__,
             action=self.node_get_llm_model
         )
         graph.add_node(
-            key=self.node_prepare_get_categorized_documents.__name__,
+            node=self.node_prepare_get_categorized_documents.__name__,
             action=self.node_prepare_get_categorized_documents
         )
         graph.add_node(
-            key=self.node_get_categorized_documents.__name__,
+            node=self.node_get_categorized_documents.__name__,
             action=self.node_get_categorized_documents
         )
         graph.add_node(
-            key=self.node_prepare_embed.__name__,
+            node=self.node_prepare_embed.__name__,
             action=self.node_prepare_embed
         )
         graph.add_node(
-            key=self.node_embed.__name__,
+            node=self.node_embed.__name__,
             action=self.node_embed
         )
         graph.add_node(
-            key=self.node_get_relevant_documents.__name__,
+            node=self.node_get_relevant_documents.__name__,
             action=self.node_get_relevant_documents
         )
         graph.add_node(
-            key=self.node_get_re_ranked_documents.__name__,
+            node=self.node_get_re_ranked_documents.__name__,
             action=self.node_get_re_ranked_documents
         )
         graph.add_node(
-            key=self.node_generate_answer.__name__,
+            node=self.node_generate_answer.__name__,
             action=self.node_generate_answer
         )
         graph.add_node(
-            key=self.node_grade_hallucination.__name__,
+            node=self.node_grade_hallucination.__name__,
             action=self.node_grade_hallucination
         )
         graph.add_node(
-            key=self.node_grade_answer_relevancy.__name__,
+            node=self.node_grade_answer_relevancy.__name__,
             action=self.node_grade_answer_relevancy
         )
         graph.add_node(
-            key=self.node_transform_question.__name__,
+            node=self.node_transform_question.__name__,
             action=self.node_transform_question
         )
 

diff --git a/apps/inners/use_cases/graphs/passage_search_graph.py b/apps/inners/use_cases/graphs/passage_search_graph.py
@@ -376,31 +376,31 @@ def _compile(self) -> CompiledGraph:
         graph: StateGraph = StateGraph(PassageSearchGraphState)
 
         graph.add_node(
-            key=self.node_get_llm_model.__name__,
+            node=self.node_get_llm_model.__name__,
             action=self.node_get_llm_model
         )
         graph.add_node(
-            key=self.node_prepare_get_categorized_documents.__name__,
+            node=self.node_prepare_get_categorized_documents.__name__,
             action=self.node_prepare_get_categorized_documents
         )
         graph.add_node(
-            key=self.node_get_categorized_documents.__name__,
+            node=self.node_get_categorized_documents.__name__,
             action=self.node_get_categorized_documents
         )
         graph.add_node(
-            key=self.node_prepare_embed.__name__,
+            node=self.node_prepare_embed.__name__,
             action=self.node_prepare_embed
         )
         graph.add_node(
-            key=self.node_embed.__name__,
+            node=self.node_embed.__name__,
             action=self.node_embed
         )
         graph.add_node(
-            key=self.node_get_relevant_documents.__name__,
+            node=self.node_get_relevant_documents.__name__,
             action=self.node_get_relevant_documents
         )
         graph.add_node(
-            key=self.node_get_re_ranked_documents.__name__,
+            node=self.node_get_re_ranked_documents.__name__,
             action=self.node_get_re_ranked_documents
         )
 

diff --git a/apps/inners/use_cases/graphs/preparation_graph.py b/apps/inners/use_cases/graphs/preparation_graph.py
@@ -288,15 +288,15 @@ def _compile_graph(self):
         graph: StateGraph = StateGraph(PreparationGraphState)
 
         graph.add_node(
-            key=self.node_get_llm_model.__name__,
+            node=self.node_get_llm_model.__name__,
             action=self.node_get_llm_model
         )
         graph.add_node(
-            key=self.node_prepare_get_categorized_documents.__name__,
+            node=self.node_prepare_get_categorized_documents.__name__,
             action=self.node_prepare_get_categorized_documents
         )
         graph.add_node(
-            key=self.node_get_categorized_documents.__name__,
+            node=self.node_get_categorized_documents.__name__,
             action=self.node_get_categorized_documents
         )