临时屏蔽问题重构

src/controller/api.py

@@ -123,12 +123,13 @@ def classify(chat_request: GeoAgentRateRequest):
     llm = agent_manager.get_router_llm()
     re_llm = agent_manager.get_re_rewriter_llm()
     try:
-        history = re_llm.extend_history(history=chat_request.history)
-        rewrite = re_llm.invoke(chat_request.query, history)
-        answer = llm.invoke(rewrite.rewriter)
-        res = {}
-        res["datasource"] =answer.datasource
-        res["rewrite"] = rewrite.rewriter
+        # history = re_llm.extend_history(history=chat_request.history)
+        # rewrite = re_llm.invoke(chat_request.query, history)
+        answer = llm.invoke(chat_request.query)
+        res = {
+            "datasource": answer.datasource,
+            "rewrite": chat_request.query
+        }
     except Exception as e:
         print(f"分类失败, 错误信息: {str(e)}，请重新提问")
         return {