DeepSeek API

app.py

@@ -6,39 +6,43 @@ import os
 from typing import Optional
 import tempfile
 import requests
+from openai import OpenAI
 
 # DeepSeek API配置
 DEEPSEEK_API_KEY = os.getenv("DEEPSEEK_API_KEY", "")
-DEEPSEEK_BASE_URL = "https://api.deepseek.com/v1/chat/completions"
+DEEPSEEK_BASE_URL = "https://dashscope.aliyuncs.com/compatible-mode/v1"
 
 def check_api_key():
     """检查API Key是否配置"""
     if not DEEPSEEK_API_KEY:
         raise ValueError("⚠️ 请在 Space Settings 中配置 DEEPSEEK_API_KEY！\n\n前往：Settings → Repository secrets → New secret")
 
-def call_deepseek_api(prompt, model="deepseek-chat", temperature=0.1):
-    """使用 requests 调用 DeepSeek API"""
+def call_deepseek_api(prompt, model="deepseek-r1-distill-llama-8b", temperature=0.1, stream=False):
+    """使用 OpenAI 客户端调用 DeepSeek API"""
     check_api_key()
     
-    headers = {
-        "Content-Type": "application/json",
-        "Authorization": f"Bearer {DEEPSEEK_API_KEY}"
-    }
+    client = OpenAI(
+        api_key=DEEPSEEK_API_KEY,
+        base_url=DEEPSEEK_BASE_URL,
+    )
     
-    data = {
-        "model": model,
-        "messages": [
-            {"role": "user", "content": prompt}
-        ],
-        "temperature": temperature,
-        "stream": False  # 简化处理，不使用流式
-    }
+    completion = client.chat.completions.create(
+        model=model,
+        messages=[{"role": "user", "content": prompt}],
+        temperature=temperature,
+        stream=stream
+    )
     
-    response = requests.post(DEEPSEEK_BASE_URL, headers=headers, json=data, timeout=60)
-    response.raise_for_status()
-    
-    result = response.json()
-    return result['choices'][0]['message']['content']
+    if stream:
+        # 流式响应处理
+        response_content = ""
+        for chunk in completion:
+            if chunk.choices and chunk.choices[0].delta.content:
+                response_content += chunk.choices[0].delta.content
+        return response_content
+    else:
+        # 非流式响应
+        return completion.choices[0].message.content
 
 # 预置的Leaderboard数据
 LEADERBOARD_DATA = [