updated langgraph version

emarco177 · emarco177 · commit 06a8e5ef807f · 2024-06-29T14:13:10.000+03:00
diff --git a/README.md b/README.md
@@ -53,7 +53,7 @@ Install dependencies
 Start the flask server
 
 ```bash
-  poetry run app.py
+  poetry run main.py
 ```
 
 
diff --git a/graph/graph.py b/graph/graph.py
@@ -7,8 +7,13 @@
 from graph.consts import GENERATE, GRADE_DOCUMENTS, RETRIEVE, WEBSEARCH
 from graph.nodes import generate, grade_documents, retrieve, web_search
 from graph.state import GraphState
+from langgraph.checkpoint.sqlite import SqliteSaver
+from langgraph.checkpoint import MemorySaver
+
 
 load_dotenv()
+memory = SqliteSaver.from_conn_string(":memory:")
+memory = MemorySaver()
 
 
 def decide_to_generate(state):
@@ -96,6 +101,6 @@ def route_question(state: GraphState) -> str:
 )
 
 
-app = workflow.compile()
+app = workflow.compile(checkpointer=memory)
 
 app.get_graph().draw_mermaid_png(output_file_path="graph.png")
diff --git a/langgraph.json b/langgraph.json
@@ -0,0 +1,7 @@
+{
+  "graphs": {
+    "agent": "./graph/graph.py:app"
+  },
+  "env": ".env",
+  "dependencies": ["."]
+}
diff --git a/main.py b/main.py
@@ -8,7 +8,7 @@
 question1 = "What are the types of agent memory?"
 inputs = {"question": question1}
 
-for output in app.stream(inputs):
+for output in app.stream(inputs, config={"configurable": {"thread_id": "2"}}):
     for key, value in output.items():
         pprint(f"Finished running: {key}:")
 pprint(value["generation"])
diff --git a/poetry.lock b/poetry.lock
diff --git a/pyproject.toml b/pyproject.toml
