内容类别改为 工具使用知识

src/agent/query_agent.py

@@ -77,7 +77,7 @@ class QueryGenerationAgent:
         # 设置入口点
         workflow.set_entry_point("classify_question")
         
-        # 条件路由：工具知识 vs 工具使用 vs 内容知识
+        # 条件路由：工具知识 vs 工具使用知识 vs 内容知识
         try:
             workflow.add_conditional_edges(
                 "classify_question",
@@ -117,13 +117,13 @@ class QueryGenerationAgent:
         return workflow.compile()
 
     def _classify_question(self, state: AgentState) -> AgentState:
-        """判断问题知识类型：工具知识 / 工具使用 / 内容知识"""
+        """判断问题知识类型：工具知识 / 工具使用知识 / 内容知识"""
         question = state.get("question", "")
         
         print(f"knowledgeType: {state.get('knowledgeType')}")
-        # 如果已经设置了 knowledgeType 且为"工具使用"，直接使用
-        if state.get("knowledgeType") == "工具使用":
-            logger.info(f"问题类型已设置为: 工具使用")
+        # 如果已经设置了 knowledgeType 且为"工具使用知识"，直接使用
+        if state.get("knowledgeType") == "工具使用知识":
+            logger.info(f"问题类型已设置为: 工具使用知识")
             return state
         
         instruction = (
@@ -151,7 +151,7 @@ class QueryGenerationAgent:
     def _route_after_classify(self, state: AgentState) -> str:
         """根据分类结果路由：工具知识 -> TOOL；工具使用 -> TOOL_USAGE；内容知识 -> CONTENT"""
         knowledge_type = state.get("knowledgeType", "")
-        if knowledge_type == "工具使用":
+        if knowledge_type == "工具使用知识":
             return "TOOL_USAGE"
         elif knowledge_type == "工具知识":
             return "TOOL"
@@ -201,7 +201,6 @@ class QueryGenerationAgent:
     def _generate_tool_usage_queries(self, state: AgentState) -> AgentState:
         """生成工具使用类型的查询词"""
         question = state["question"]
-        print(f"工具使用类型查询词: {question}")
         prompt = ChatPromptTemplate.from_messages([
             SystemMessage(content=TOOL_USAGE_PROMPT),
             HumanMessage(content=question)