liuzhiheng 10 hours ago
parent
commit
90afb6a243
1 changed files with 3 additions and 10 deletions
  1. 3 10
      knowledge_v2/function_knowledge.py

+ 3 - 10
knowledge_v2/function_knowledge.py

@@ -78,11 +78,7 @@ class FunctionKnowledge:
         
         try:
             prompt_template = self._load_prompt("function_generate_query_prompt.md")
-            prompt = prompt_template.format(
-                question=question,
-                post_info=post_info,
-                persona_info=persona_info
-            )
+            prompt = prompt_template.replace("{question}", question)
             
             detail_info["prompt"] = prompt
             
@@ -124,7 +120,7 @@ class FunctionKnowledge:
                 return cached_tool, detail_info
         
         try:
-            all_tool_infos = get_all_tool_infos()
+            all_tool_infos = self._load_prompt("all_tools_infos.md")
             if not all_tool_infos:
                 logger.info("  工具库为空,无可用工具")
                 return "None", detail_info
@@ -134,10 +130,7 @@ class FunctionKnowledge:
             logger.info(f"  当前可用工具数: {tool_count}")
                 
             prompt_template = self._load_prompt("function_knowledge_select_tools_prompt.md")
-            prompt = prompt_template.format(
-                query=query,
-                tool_infos=all_tool_infos
-            )
+            prompt = prompt_template.replace("{all_tool_infos}", all_tool_infos)
             
             detail_info["prompt"] = prompt
             detail_info["tool_infos"] = all_tool_infos