修改引用逻辑,修改长期记忆bug
This commit is contained in:
@@ -5,18 +5,17 @@ LangGraph 状态图构建模块 - 精简版,仅负责组装图
|
||||
|
||||
from langchain_core.language_models import BaseLLM
|
||||
from langgraph.graph import StateGraph, START, END
|
||||
|
||||
# 本地模块
|
||||
from app.graph.state import MessagesState, GraphContext
|
||||
from app.nodes import (
|
||||
should_continue,
|
||||
create_llm_call_node,
|
||||
create_tool_call_node,
|
||||
create_retrieve_memory_node,
|
||||
create_summarize_node,
|
||||
should_continue
|
||||
finalize_node,
|
||||
)
|
||||
from app.nodes.memory_trigger import memory_trigger_node, set_mem0_client
|
||||
from app.memory import Mem0Client
|
||||
from app.nodes.finalize import finalize_node
|
||||
|
||||
|
||||
class GraphBuilder:
|
||||
@@ -45,6 +44,9 @@ class GraphBuilder:
|
||||
Returns:
|
||||
StateGraph 实例
|
||||
"""
|
||||
# 注入全局客户端
|
||||
set_mem0_client(self.mem0_client)
|
||||
|
||||
builder = StateGraph(MessagesState, context_schema=GraphContext)
|
||||
|
||||
# ⭐ 通过工厂函数创建节点(依赖注入)
|
||||
@@ -55,6 +57,7 @@ class GraphBuilder:
|
||||
|
||||
# 添加节点
|
||||
builder.add_node("retrieve_memory", retrieve_memory_node)
|
||||
builder.add_node("memory_trigger", memory_trigger_node)
|
||||
builder.add_node("llm_call", llm_call_node)
|
||||
builder.add_node("tool_node", tool_call_node)
|
||||
builder.add_node("summarize", summarize_node)
|
||||
@@ -62,7 +65,8 @@ class GraphBuilder:
|
||||
|
||||
# 添加边
|
||||
builder.add_edge(START, "retrieve_memory")
|
||||
builder.add_edge("retrieve_memory", "llm_call")
|
||||
builder.add_edge("retrieve_memory", "memory_trigger")
|
||||
builder.add_edge("memory_trigger", "llm_call")
|
||||
builder.add_conditional_edges(
|
||||
"llm_call",
|
||||
should_continue,
|
||||
|
||||
Reference in New Issue
Block a user