vrr/open-notebook
2
0
Fork 0
mirror of https://github.com/lfnovo/open-notebook.git synced 2026-05-01 04:50:01 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

organize data folders

Browse source
This commit is contained in:
LUIS NOVO 2024-10-26 18:54:54 -03:00
parent 86eb9d7440
commit 1002c6b4eb
4 changed files with 24 additions and 9 deletions
Download patch file Download diff file Expand all files Collapse all files

4
open_notebook/graphs/chat.py
View file

@ -1,4 +1,3 @@
import os
import sqlite3
from typing import Annotated, Optional
@ -10,6 +9,7 @@ from langgraph.graph import END, START, StateGraph
from langgraph.graph.message import add_messages
from typing_extensions import TypedDict
from open_notebook.config import LANGGRAPH_CHECKPOINT_FILE
from open_notebook.domain import Notebook
from open_notebook.graphs.utils import run_pattern
@ -33,7 +33,7 @@ def call_model_with_messages(state: ThreadState, config: RunnableConfig) -> dict
conn = sqlite3.connect(
os.environ.get("CHECKPOINT_DATA_PATH", "sqlite-db/checkpoints.sqlite"),
LANGGRAPH_CHECKPOINT_FILE,
check_same_thread=False,
)
memory = SqliteSaver(conn)