Add DeepSeek

config.py

@@ -31,8 +31,8 @@ class Config:
     ]
 
     # DeepSeek model configuration
-    DEEPSEEK_MODEL = "deepseek-chat"
-    DEEPSEEK_API_BASE = "https://api.deepseek.com/v1"
+    DEEPSEEK_MODEL = "deepseek-chat-1.3"
+    DEEPSEEK_API_BASE = "https://api.deepseek.com"
 
     # Chat context settings
     MAX_HISTORY_CHATS = 10

deepseek_client.py

@@ -3,67 +3,49 @@ import time
 import requests
 from typing import List, Dict, Optional
 from config import Config
+from openai import OpenAI
 
-class DeepseekClient:
-    def __init__(self):
-        self.api_key = Config.get_deepseek_key()
+class DeepseekClient(OpenAI):
+    def __init__(self, api_key=None, **kwargs):
+        super().__init__(api_key=api_key or Config.get_deepseek_key(), **kwargs)
         if not self.api_key:
             raise ValueError("Deepseek API key is not set")
         self.base_url = Config.DEEPSEEK_API_BASE
         self.max_retries = 3
         self.retry_delay = 2  # Initial delay in seconds
 
-    def create(self, messages: List[Dict[str, str]], model: str = None, response_format: Optional[Dict] = None) -> str:
+        # Debug info
+        print(f"Initializing DeepseekClient with:")
+        print(f"  api_key: {'*' * 8}{self.api_key[-4:]}" if self.api_key else "  api_key: None")
+        print(f"  base_url: {self.base_url}")
+
+    def create(self, messages: List[Dict[str, str]], model: str = None, response_format: Optional[Dict] = None, **kwargs) -> str:
         if not self.api_key:
             raise ValueError("Deepseek API key is not set")
 
+        # Debug info
+        print("\n[DeepseekClient] Starting create request")
+        print(f"  Using model: {model or Config.DEEPSEEK_MODEL}")
+        print(f"  Messages count: {len(messages)}")
+        if messages:
+            print(f"  First message: {messages[0]['content'][:50]}...")
+
         retries = 0
         last_error = None
 
         while retries < self.max_retries:
             try:
-                headers = {
-                    "Authorization": f"Bearer {self.api_key}",
-                    "HTTP-Referer": "https://replit.com",
-                    "X-Title": "MyChatMe",
-                    "Content-Type": "application/json"
-                }
-
-                data = {
-                    "model": model or Config.AUTO_MODEL,
-                    "messages": messages,
-                    "temperature": 0.7,
-                    "max_tokens": 1000
-                }
-
-                if response_format:
-                    data["response_format"] = response_format
-
-                response = requests.post(
-                    f"{self.base_url}/chat/completions",
-                    headers=headers,
-                    json=data,
-                    timeout=30  # Added timeout
+                print(f"[DeepseekClient] Attempting API request (retry {retries + 1}/{self.max_retries})")
+                response = self.chat.completions.create(
+                    model=model or Config.DEEPSEEK_MODEL,
+                    messages=messages,
+                    temperature=0.7,
+                    max_tokens=1000,
+                    response_format=response_format,
+                    **kwargs
                 )
 
-                if response.status_code == 429:
-                    error_data = response.json()
-                    error_message = error_data.get('error', {}).get('message', 'Rate limit exceeded')
-                    print(f"Rate limit error: {error_message}")
-
-                    # Check if it's a provider-specific rate limit
-                    if 'metadata' in error_data.get('error', {}):
-                        provider = error_data['error']['metadata'].get('provider_name', 'Unknown')
-                        raise Exception(f"Rate limit exceeded for provider: {provider}")
-
-                    wait_time = self.retry_delay * (2 ** retries)
-                    print(f"Rate limit exceeded. Waiting {wait_time} seconds before retry...")
-                    time.sleep(wait_time)
-                    retries += 1
-                    continue
-
-                response.raise_for_status()
-                response_data = response.json()
+                response_data = response.model_dump()
 
                 if "error" in response_data:
                     raise ValueError(f"Deepseek API returned error: {response_data['error']}")
@@ -74,19 +56,25 @@ class DeepseekClient:
                 if not response_data['choices'][0].get('message'):
                     raise ValueError("No message in Deepseek API response choice")
 
-                return response_data['choices'][0]['message']['content']
+                content = response_data['choices'][0]['message']['content']
+                print("[DeepseekClient] API request successful")
+                print(f"[DeepseekClient] Received response content (length: {len(content)})")
+                return content
 
             except requests.exceptions.RequestException as e:
+                print(f"[DeepseekClient] Request failed: {str(e)}")
                 print(f"Deepseek Request Error: {str(e)}")
                 last_error = e
                 retries += 1
                 if retries < self.max_retries:
                     time.sleep(self.retry_delay * (2 ** retries))
             except ValueError as e:
+                print(f"[DeepseekClient] Value error occurred: {str(e)}")
                 print(f"Deepseek Value Error: {str(e)}")
                 last_error = e
                 break
             except Exception as e:
+                print(f"[DeepseekClient] Unexpected error occurred: {str(e)}")
                 print(f"Deepseek Unexpected Error: {str(e)}")
                 last_error = e
                 break

llm_client.py

@@ -25,7 +25,12 @@ class LLMClient:
             if openrouter_key:
                 self.openrouter_client = OpenRouterClient()
             if deepseek_key:
-                self.deepseek_client = DeepseekClient()
+                try:
+                    self.deepseek_client = DeepseekClient(api_key=deepseek_key)
+                except Exception as e:
+                    print(f"Failed to initialize DeepseekClient: {str(e)}")
+                    print("Check your Deepseek API key and endpoint configuration")
+                    raise
         except Exception as e:
             print(f"Error initializing API clients: {str(e)}")
 

locales/en/translation.json

@@ -12,12 +12,14 @@
     "timezone": "Timezone",
     "api_key_openai": "OpenAI API Key",
     "api_key_openrouter": "OpenRouter API Key",
+    "api_key_deepseek": "DeepSeek API Key",
     "save_settings": "Save Settings",
     "error_missing_key": "Please enter API key",
     "error_api_call": "Error calling API",
     "error_model_switch": "Failed to switch to the selected model",
     "error_model_switch_openai": "Failed to switch to GPT-4. Please check your OpenAI API key.",
     "error_model_switch_openrouter": "Failed to switch to model. Please check your OpenRouter API key.",
+    "error_model_switch_deepseek": "Failed to switch to model. Please check your DeepSeek API key.",
     "error_rate_limit": "Rate limit exceeded. Please try again later.",
     "error_network": "Network connection error. Please check your connection.",
     "default_system_prompt": "You are a helpful assistant. Please provide clear and concise responses.",

locales/ja/translation.json

@@ -12,12 +12,14 @@
     "timezone": "タイムゾーン",
     "api_key_openai": "OpenAI APIキー",
     "api_key_openrouter": "OpenRouter APIキー",
+    "api_key_deepseek": "DeepSeek APIキー",
     "save_settings": "設定を保存",
     "error_missing_key": "APIキーを入力してください",
     "error_api_call": "APIの呼び出しでエラーが発生しました",
     "error_model_switch": "モデルの切り替えに失敗しました",
     "error_model_switch_openai": "GPT-4への切り替えに失敗しました。OpenAI APIキーを確認してください。",
     "error_model_switch_openrouter": "モデルの切り替えに失敗しました。OpenRouter APIキーを確認してください。",
+    "error_model_switch_deepseek": "モデルの切り替えに失敗しました。DeepSeek APIキーを確認してください。",
     "error_rate_limit": "レート制限を超えました。しばらく待ってから再試行してください。",
     "error_network": "ネットワーク接続エラーが発生しました。接続を確認してください。",
     "default_system_prompt": "私は役立つアシスタントです。明確で簡潔な応答を提供します。",

ui_components.py

@@ -52,7 +52,7 @@ def render_sidebar(i18n, chat_manager):
         # Model selection - デフォルトをOpenRouter-Autoに設定
         model = st.selectbox(
             i18n.get_text("model_selection"),
-            ["OpenRouter-Auto", "OpenAI", "Claude-3.5", "Gemini-2.0", "DeepSeek-V3"],
+            ["OpenRouter-Auto", "OpenAI", "Claude-3.5", "Gemini-2.0", "deepseek-chat"],
             index=0,
             key="model_selection"
         )