Browse Source

修改位置

xueyiming 2 days ago
parent
commit
2596b434d9
4 changed files with 2 additions and 40 deletions
  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):
     def __init__(self):
         self.api_key = "sk-1022fe8e15ff4e0e9abc20541b281165"
         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(
     def search_and_chat(
         self,
         self,
         model="qwen3-max",
         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.resource import get_resource_manager
 from applications.utils.chat import RAGChatAgent
 from applications.utils.chat import RAGChatAgent
 from applications.utils.mysql import ChatResult
 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 applications.utils.spider.study import study
 from routes.blueprint import query_search
 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.search import HybridSearch
 from applications.utils.chat import RAGChatAgent
 from applications.utils.chat import RAGChatAgent
 from applications.utils.mysql import Dataset, Contents, ContentChunks, ChatResult
 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
 from applications.utils.spider.study import study
 
 
 server_bp = Blueprint("api", __name__, url_prefix="/api")
 server_bp = Blueprint("api", __name__, url_prefix="/api")