|
@@ -4,7 +4,7 @@ Uses RAG retrieval from health knowledge base.
|
|
|
"""
|
|
"""
|
|
|
from typing import TypedDict
|
|
from typing import TypedDict
|
|
|
from langgraph.graph import StateGraph, START, END
|
|
from langgraph.graph import StateGraph, START, END
|
|
|
-from langgraph.checkpoint.memory import MemorySaver
|
|
|
|
|
|
|
+from app.checkpointer import get_checkpointer
|
|
|
from langchain_openai import ChatOpenAI
|
|
from langchain_openai import ChatOpenAI
|
|
|
from langchain_core.messages import SystemMessage, HumanMessage
|
|
from langchain_core.messages import SystemMessage, HumanMessage
|
|
|
from app.rag.retriever import RagRetriever
|
|
from app.rag.retriever import RagRetriever
|
|
@@ -187,7 +187,7 @@ def create_health_coach_graph():
|
|
|
builder.add_edge("generate_answer", "save_memory")
|
|
builder.add_edge("generate_answer", "save_memory")
|
|
|
builder.add_edge("save_memory", END)
|
|
builder.add_edge("save_memory", END)
|
|
|
|
|
|
|
|
- checkpointer = MemorySaver()
|
|
|
|
|
|
|
+ checkpointer = get_checkpointer()
|
|
|
graph = builder.compile(checkpointer=checkpointer)
|
|
graph = builder.compile(checkpointer=checkpointer)
|
|
|
|
|
|
|
|
return graph
|
|
return graph
|