修复重构后的导入错误和缺失模块

backend/app/main_graph/config.py

@@ -0,0 +1,21 @@
+"""
+Main Graph Configuration - Streaming Writer
+"""
+from typing import Optional, Callable, Any
+
+_stream_writer: Optional[Callable[[Any], None]] = None
+
+def set_stream_writer(writer: Callable[[Any], None]):
+    """Set the global stream writer"""
+    global _stream_writer
+    _stream_writer = writer
+
+def get_stream_writer() -> Callable[[Any], None]:
+    """Get the global stream writer"""
+    global _stream_writer
+    if _stream_writer is None:
+        # Default no-op writer
+        def noop(_):
+            pass
+        return noop
+    return _stream_writer

backend/app/main_graph/graph.py

@@ -0,0 +1,8 @@
+"""
+LangGraph 核心组件重新导出
+统一导入入口，避免直接依赖 langgraph
+"""
+
+from langgraph.graph import StateGraph, START, END, add_messages
+
+__all__ = ["StateGraph", "START", "END", "add_messages"]

backend/app/main_graph/graph_builder.py

@@ -5,8 +5,8 @@ LangGraph 状态图构建模块 - 精简版，仅负责组装图
 
 from langchain_core.language_models import BaseLLM
 from app.main_graph.graph import StateGraph, START, END
-from .state import MessagesState, GraphContext
-from ..nodes import (
+from app.main_graph.state import MessagesState, GraphContext
+from .nodes import (
     should_continue,
     create_llm_call_node,
     create_tool_call_node,
@@ -15,7 +15,7 @@ from ..nodes import (
     finalize_node,
 )
 from app.main_graph.nodes.memory_trigger import memory_trigger_node, set_mem0_client
-from ..memory import Mem0Client
+from app.memory import Mem0Client
 
 
 class GraphBuilder:

backend/app/main_graph/nodes/__init__.py

@@ -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",
+]

backend/app/main_graph/nodes/finalize.py

@@ -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
 

backend/app/main_graph/nodes/llm_call.py

@@ -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):
     """

backend/app/main_graph/nodes/memory_trigger.py

@@ -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

backend/app/main_graph/nodes/rag_nodes.py

@@ -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 工具实例（延迟初始化）==========

backend/app/main_graph/nodes/react_nodes.py

@@ -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

backend/app/main_graph/nodes/retrieve_memory.py

@@ -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):
     """

backend/app/main_graph/nodes/router.py

@@ -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']:

backend/app/main_graph/nodes/summarize.py

@@ -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):
     """

backend/app/main_graph/nodes/tool_call.py

@@ -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]):
     """

backend/app/main_graph/tools/__init__.py

@@ -1 +1,10 @@
 """主图工具"""
+from .graph_tools import AVAILABLE_TOOLS, TOOLS_BY_NAME
+from .subgraph_tools import SUBGRAPH_TOOLS, SUBGRAPH_TOOLS_BY_NAME
+
+__all__ = [
+    "AVAILABLE_TOOLS",
+    "TOOLS_BY_NAME",
+    "SUBGRAPH_TOOLS",
+    "SUBGRAPH_TOOLS_BY_NAME",
+]

backend/app/main_graph/utils/rag_initializer.py

@@ -1,8 +1,8 @@
 # app/rag_initializer.py
-from ..rag.tools import create_rag_tool_sync
+from app.rag.tools import create_rag_tool_sync
 from rag_core import create_parent_retriever
-from ..model_services import get_embedding_service
-from ..logger import info, warning
+from app.model_services import get_embedding_service
+from app.logger import info, warning
 
 async def init_rag_tool(local_llm_creator):
     """初始化 RAG 工具，失败返回 None"""

backend/app/main_graph/utils/retry_utils.py

@@ -287,7 +287,7 @@ def create_retry_wrapper_for_node(
                 time.sleep(delay)
         
         # 所有重试都失败，更新状态错误信息
-        from .state import ErrorRecord, ErrorSeverity
+        from app.main_graph.state import ErrorRecord, ErrorSeverity
         
         error_record = ErrorRecord(
             error_type=f"{node_name}TimeoutError",

backend/app/main_graph/utils/subgraph_builder.py

@@ -6,7 +6,7 @@ Main Graph Builder - Full React Mode with Loop Reasoning
 from app.main_graph.graph import StateGraph, START, END
 from typing import Dict, Any
 
-from .state import MainGraphState, CurrentAction
+from app.main_graph.state import MainGraphState, CurrentAction
 from .react_nodes import (
     init_state_node,
     react_reason_node,
@@ -50,7 +50,7 @@ def wrap_subgraph_for_error_handling(subgraph, name: str):
             
         except Exception as e:
             # 捕获子图错误，传递给主图
-            from .state import ErrorRecord, ErrorSeverity
+            from app.main_graph.state import ErrorRecord, ErrorSeverity
             from datetime import datetime
             
             error_record = ErrorRecord(