rwayz commited on
Commit
4058190
·
verified ·
1 Parent(s): 9e29bb4

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -14,7 +14,7 @@ from huggingface_hub import InferenceClient
14
  from langgraph.graph import StateGraph, END
15
  from langgraph.prebuilt import ToolNode
16
  from langchain_core.tools import tool
17
- from langgraph.checkpoint.sqlite import SqliteSaver
18
  import gradio as gr
19
  from uuid import uuid4
20
 
@@ -213,7 +213,7 @@ builder.add_edge("generate_sql", "execute_sql")
213
  builder.add_edge("execute_sql", "refine")
214
  builder.add_edge("refine", END)
215
 
216
- memory = SqliteSaver("graph_checkpoint")
217
  graph = builder.compile(checkpointer=memory)
218
 
219
  async def async_chatbot_response(question, model_name):
 
14
  from langgraph.graph import StateGraph, END
15
  from langgraph.prebuilt import ToolNode
16
  from langchain_core.tools import tool
17
+ from langgraph.checkpoint.sqlite.aio import AsyncSqliteSaver
18
  import gradio as gr
19
  from uuid import uuid4
20
 
 
213
  builder.add_edge("execute_sql", "refine")
214
  builder.add_edge("refine", END)
215
 
216
+ memory = AsyncSqliteSaver.from_conn_string("graph_checkpoint.db")
217
  graph = builder.compile(checkpointer=memory)
218
 
219
  async def async_chatbot_response(question, model_name):