Spaces:
Running
Running
fix: SqliteSaver context manager error
Browse files
graph.py
CHANGED
|
@@ -12,13 +12,15 @@ from langchain_core.runnables import RunnableConfig
|
|
| 12 |
from langchain_openrouter import ChatOpenRouter
|
| 13 |
from typing import TypedDict, Annotated
|
| 14 |
|
|
|
|
| 15 |
from langgraph.checkpoint.sqlite import SqliteSaver
|
| 16 |
from config import OPENROUTER_API_KEY, DB_PATH
|
| 17 |
import prompts
|
| 18 |
|
| 19 |
|
| 20 |
# --- Checkpointer (SQLite — no external DB needed) ---
|
| 21 |
-
|
|
|
|
| 22 |
|
| 23 |
|
| 24 |
# --- Model config ---
|
|
|
|
| 12 |
from langchain_openrouter import ChatOpenRouter
|
| 13 |
from typing import TypedDict, Annotated
|
| 14 |
|
| 15 |
+
import sqlite3
|
| 16 |
from langgraph.checkpoint.sqlite import SqliteSaver
|
| 17 |
from config import OPENROUTER_API_KEY, DB_PATH
|
| 18 |
import prompts
|
| 19 |
|
| 20 |
|
| 21 |
# --- Checkpointer (SQLite — no external DB needed) ---
|
| 22 |
+
_conn = sqlite3.connect(DB_PATH, check_same_thread=False)
|
| 23 |
+
checkpointer = SqliteSaver(conn=_conn)
|
| 24 |
|
| 25 |
|
| 26 |
# --- Model config ---
|