Merge pull request #88 from zmx198768/feature_qwen

LLM供应商增加通义千问

Author: sunmh207

biz/llm/client/qwen.py

@@ -0,0 +1,29 @@
+import os
+from typing import Dict, List, Optional
+
+from openai import OpenAI
+
+from biz.llm.client.base import BaseClient
+from biz.llm.types import NotGiven, NOT_GIVEN
+
+
+class QwenClient(BaseClient):
+    def __init__(self, api_key: str = None):
+        self.api_key = api_key or os.getenv("QWEN_API_KEY")
+        self.base_url = os.getenv("QWEN_API_BASE_URL", "https://dashscope.aliyuncs.com/compatible-mode/v1")
+        if not self.api_key:
+            raise ValueError("API key is required. Please provide it or set it in the environment variables.")
+
+        self.client = OpenAI(api_key=self.api_key, base_url=self.base_url)
+        self.default_model = os.getenv("QWEN_API_MODEL", "qwen-coder-plus")
+
+    def completions(self,
+                    messages: List[Dict[str, str]],
+                    model: Optional[str] | NotGiven = NOT_GIVEN,
+                    ) -> str:
+        model = model or self.default_model
+        completion = self.client.chat.completions.create(
+            model=model,
+            messages=messages,
+        )
+        return completion.choices[0].message.content

biz/llm/factory.py

@@ -4,6 +4,7 @@
 from biz.llm.client.deepseek import DeepSeekClient
 from biz.llm.client.ollama_client import OllamaClient
 from biz.llm.client.openai import OpenAIClient
+from biz.llm.client.qwen import QwenClient
 from biz.llm.client.zhipuai import ZhipuAIClient
 from biz.utils.log import logger
 
@@ -16,6 +17,7 @@ def getClient(provider: str = None) -> BaseClient:
             'zhipuai': lambda: ZhipuAIClient(),
             'openai': lambda: OpenAIClient(),
             'deepseek': lambda: DeepSeekClient(),
+            'qwen': lambda: QwenClient(),
             'ollama': lambda : OllamaClient()
         }
 

biz/utils/config_checker.py

@@ -15,14 +15,15 @@
 ]
 
 # 允许的 LLM 供应商
-LLM_PROVIDERS = {"zhipuai", "openai", "deepseek", "ollama"}
+LLM_PROVIDERS = {"zhipuai", "openai", "deepseek", "ollama", "qwen"}
 
 # 每种供应商必须配置的键
 LLM_REQUIRED_KEYS = {
     "zhipuai": ["ZHIPUAI_API_KEY", "ZHIPUAI_API_MODEL"],
     "openai": ["OPENAI_API_KEY", "OPENAI_API_MODEL"],
     "deepseek": ["DEEPSEEK_API_KEY", "DEEPSEEK_API_MODEL"],
     "ollama": ["OLLAMA_API_BASE_URL", "OLLAMA_API_MODEL"],
+    "qwen": ["QWEN_API_KEY", "QWEN_API_MODEL"],
 }
 
 

conf/.env.dist

@@ -5,7 +5,7 @@ SERVER_PORT=5001
 TZ=Asia/Shanghai
 
 #大模型供应商配置,支持 zhipuai , openai , deepseek or ollama
-LLM_PROVIDER=deepseek
+LLM_PROVIDER=qwen
 
 #ZhipuAI settings
 ZHIPUAI_API_KEY=xxxx
@@ -16,6 +16,11 @@ OPENAI_API_KEY=xxxx
 OPENAI_API_BASE_URL=https://api.openai.com
 OPENAI_API_MODEL=gpt-4o-mini
 
+#Qwen settings
+QWEN_API_KEY=sk-xxx
+QWEN_API_BASE_URL=https://dashscope.aliyuncs.com/compatible-mode/v1
+QWEN_API_MODEL=qwen-coder-plus
+
 #DeepSeek settings
 DEEPSEEK_API_KEY=
 DEEPSEEK_API_BASE_URL=https://api.deepseek.com