Update app.py
Browse files
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
|
| 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 =
|
| 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):
|