This commit is contained in:
@@ -1 +1,19 @@
|
||||
"""主图节点"""
|
||||
"""
|
||||
主图节点模块导出
|
||||
"""
|
||||
|
||||
from .router import should_continue
|
||||
from .llm_call import create_llm_call_node
|
||||
from .tool_call import create_tool_call_node
|
||||
from .retrieve_memory import create_retrieve_memory_node
|
||||
from .summarize import create_summarize_node
|
||||
from .finalize import finalize_node
|
||||
|
||||
__all__ = [
|
||||
"should_continue",
|
||||
"create_llm_call_node",
|
||||
"create_tool_call_node",
|
||||
"create_retrieve_memory_node",
|
||||
"create_summarize_node",
|
||||
"finalize_node",
|
||||
]
|
||||
|
||||
@@ -8,8 +8,8 @@ from app.main_graph.config import get_stream_writer
|
||||
|
||||
# 本地模块
|
||||
from app.main_graph.state import MessagesState
|
||||
from ..utils.logging import log_state_change
|
||||
from ..logger import info, error
|
||||
from app.utils.logging import log_state_change
|
||||
from app.logger import info, error
|
||||
|
||||
from langchain_core.runnables.config import RunnableConfig
|
||||
|
||||
|
||||
@@ -10,9 +10,9 @@ from langchain_core.messages import AIMessage
|
||||
|
||||
# 本地模块
|
||||
from app.main_graph.state import MessagesState
|
||||
from ..agent.prompts import create_system_prompt
|
||||
from ..utils.logging import log_state_change
|
||||
from ..logger import debug, info, error
|
||||
from app.agent.prompts import create_system_prompt
|
||||
from app.utils.logging import log_state_change
|
||||
from app.logger import debug, info, error
|
||||
|
||||
def create_llm_call_node(llm: BaseLLM, tools: list):
|
||||
"""
|
||||
|
||||
@@ -1,8 +1,8 @@
|
||||
from typing import Any, Dict
|
||||
from langchain_core.runnables.config import RunnableConfig
|
||||
from app.main_graph.state import MessagesState
|
||||
from ..memory.mem0_client import Mem0Client
|
||||
from ..logger import info
|
||||
from app.memory.mem0_client import Mem0Client
|
||||
from app.logger import info
|
||||
|
||||
# 全局变量,在 GraphBuilder 中注入
|
||||
_mem0_client: Mem0Client = None
|
||||
|
||||
@@ -11,7 +11,7 @@ import asyncio
|
||||
from typing import Dict, Any, Optional
|
||||
from datetime import datetime
|
||||
|
||||
from .state import MainGraphState, ErrorRecord, ErrorSeverity
|
||||
from app.main_graph.state import MainGraphState, ErrorRecord, ErrorSeverity
|
||||
from .retry_utils import (
|
||||
RetryConfig,
|
||||
RAG_RETRY_CONFIG,
|
||||
@@ -19,8 +19,8 @@ from .retry_utils import (
|
||||
)
|
||||
|
||||
# 真正导入和利用已有 RAG 代码
|
||||
from ..rag.tools import create_rag_tool_sync
|
||||
from ..rag.pipeline import RAGPipeline
|
||||
from app.rag.tools import create_rag_tool_sync
|
||||
from app.rag.pipeline import RAGPipeline
|
||||
|
||||
|
||||
# ========== 全局 RAG 工具实例(延迟初始化)==========
|
||||
|
||||
@@ -22,7 +22,7 @@ from app.core.intent import (
|
||||
ReasoningResult
|
||||
)
|
||||
from app.core.state_base import StateUtils
|
||||
from .state import MainGraphState, ErrorRecord, ErrorSeverity
|
||||
from app.main_graph.state import MainGraphState, ErrorRecord, ErrorSeverity
|
||||
from .retry_utils import (
|
||||
RetryConfig,
|
||||
SUBGRAPH_RETRY_CONFIG
|
||||
|
||||
@@ -6,10 +6,10 @@
|
||||
from typing import Any, Dict
|
||||
|
||||
# 本地模块
|
||||
from .state import MessagesState
|
||||
from ..memory.mem0_client import Mem0Client
|
||||
from ..utils.logging import log_state_change
|
||||
from ..logger import debug
|
||||
from app.main_graph.state import MessagesState
|
||||
from app.memory.mem0_client import Mem0Client
|
||||
from app.utils.logging import log_state_change
|
||||
from app.logger import debug
|
||||
|
||||
def create_retrieve_memory_node(mem0_client: Mem0Client):
|
||||
"""
|
||||
|
||||
@@ -7,9 +7,9 @@ from typing import Literal
|
||||
from langchain_core.messages import AIMessage
|
||||
|
||||
# 本地模块
|
||||
from ..config import ENABLE_GRAPH_TRACE, MEMORY_SUMMARIZE_INTERVAL
|
||||
from app.config import ENABLE_GRAPH_TRACE, MEMORY_SUMMARIZE_INTERVAL
|
||||
from app.main_graph.state import MessagesState
|
||||
from ..logger import info
|
||||
from app.logger import info
|
||||
|
||||
|
||||
def should_continue(state: MessagesState) -> Literal['tool_node', 'summarize', 'finalize']:
|
||||
|
||||
@@ -7,9 +7,9 @@ from typing import Any, Dict
|
||||
|
||||
# 本地模块
|
||||
from app.main_graph.state import MessagesState
|
||||
from ..memory.mem0_client import Mem0Client
|
||||
from ..utils.logging import log_state_change
|
||||
from ..logger import debug, info, error, warning
|
||||
from app.memory.mem0_client import Mem0Client
|
||||
from app.utils.logging import log_state_change
|
||||
from app.logger import debug, info, error, warning
|
||||
|
||||
def create_summarize_node(mem0_client: Mem0Client):
|
||||
"""
|
||||
|
||||
@@ -10,8 +10,8 @@ from app.main_graph.config import get_stream_writer
|
||||
|
||||
# 本地模块
|
||||
from app.main_graph.state import MessagesState
|
||||
from ..utils.logging import log_state_change
|
||||
from ..logger import debug, info
|
||||
from app.utils.logging import log_state_change
|
||||
from app.logger import debug, info
|
||||
|
||||
def create_tool_call_node(tools_by_name: Dict[str, Any]):
|
||||
"""
|
||||
|
||||
Reference in New Issue
Block a user