浏览代码

修改位置

xueyiming 2 天之前
父节点
当前提交
2596b434d9
共有 4 个文件被更改,包括 2 次插入40 次删除
  1. 0 38
      applications/api/qwen.py
  2. 0 0
      applications/utils/search/__init__.py
  3. 1 1
      mcp_server/server.py
  4. 1 1
      routes/blueprint.py

+ 0 - 38
applications/utils/search/qwen.py → applications/api/qwen.py

@@ -5,44 +5,6 @@ class QwenClient:
     def __init__(self):
         self.api_key = "sk-1022fe8e15ff4e0e9abc20541b281165"
 
-    def chat(
-        self,
-        model="qwen3-max",
-        system_prompt="You are a helpful assistant.",
-        user_prompt="",
-    ):
-        """
-        普通聊天,不使用搜索功能
-
-        Args:
-            model: 模型名称,默认为qwen3-max
-            system_prompt: 系统提示词
-            user_prompt: 用户提示词
-
-        Returns:
-            str: AI回复内容
-        """
-        try:
-            messages = [
-                {"role": "system", "content": system_prompt},
-                {"role": "user", "content": user_prompt},
-            ]
-
-            response = dashscope.Generation.call(
-                api_key=self.api_key,
-                model=model,
-                messages=messages,
-                result_format="message",
-            )
-
-            if response.status_code != 200:
-                raise Exception(f"API调用失败: {response.message}")
-
-            return response["output"]["choices"][0]["message"]["content"]
-
-        except Exception as e:
-            raise Exception(f"QwenClient chat失败: {str(e)}")
-
     def search_and_chat(
         self,
         model="qwen3-max",

+ 0 - 0
applications/utils/search/__init__.py


+ 1 - 1
mcp_server/server.py

@@ -8,7 +8,7 @@ from mcp.server.lowlevel import Server
 from applications.resource import get_resource_manager
 from applications.utils.chat import RAGChatAgent
 from applications.utils.mysql import ChatResult
-from applications.utils.search.qwen import QwenClient
+from applications.api.qwen import QwenClient
 from applications.utils.spider.study import study
 from routes.blueprint import query_search
 

+ 1 - 1
routes/blueprint.py

@@ -20,7 +20,7 @@ from applications.resource import get_resource_manager
 from applications.search import HybridSearch
 from applications.utils.chat import RAGChatAgent
 from applications.utils.mysql import Dataset, Contents, ContentChunks, ChatResult
-from applications.utils.search.qwen import QwenClient
+from applications.api.qwen import QwenClient
 from applications.utils.spider.study import study
 
 server_bp = Blueprint("api", __name__, url_prefix="/api")