ALwrity Version 0.5.0 (Fastapi + React )

backend/services/llm_providers/deepseek_provider.py

@@ -0,0 +1,105 @@
+"""DeepSeek Provider Service for ALwrity Backend.
+
+This service handles DeepSeek API integrations,
+migrated from the legacy lib/gpt_providers/text_generation/deepseek_text_gen.py
+"""
+
+import os
+import json
+import time
+from typing import Dict, Any, Tuple
+from loguru import logger
+from tenacity import (
+    retry,
+    stop_after_attempt,
+    wait_random_exponential,
+)
+
+# Import APIKeyManager
+from ..api_key_manager import APIKeyManager
+
+try:
+    import openai
+except ImportError:
+    openai = None
+    logger.warning("OpenAI library not available. Install with: pip install openai")
+
+async def test_deepseek_api_key(api_key: str) -> Tuple[bool, str]:
+    """
+    Test if the provided DeepSeek API key is valid.
+    
+    Args:
+        api_key (str): The DeepSeek API key to test
+        
+    Returns:
+        tuple[bool, str]: A tuple containing (is_valid, message)
+    """
+    if not openai:
+        return False, "OpenAI library not available"
+    
+    try:
+        # Create DeepSeek client with the provided key
+        client = openai.OpenAI(
+            api_key=api_key,
+            base_url="https://api.deepseek.com/v1"
+        )
+        
+        # Try to generate a simple response as a test
+        response = client.chat.completions.create(
+            model="deepseek-chat",
+            messages=[{"role": "user", "content": "Hello"}],
+            max_tokens=10,
+            temperature=0.1
+        )
+        
+        # If we get here, the key is valid
+        return True, "DeepSeek API key is valid"
+        
+    except openai.AuthenticationError:
+        return False, "Invalid DeepSeek API key"
+    except openai.RateLimitError:
+        return False, "Rate limit exceeded. Please try again later."
+    except Exception as e:
+        return False, f"Error testing DeepSeek API key: {str(e)}"
+
+@retry(wait=wait_random_exponential(min=1, max=60), stop=stop_after_attempt(6))
+def deepseek_text_response(prompt: str, model: str = "deepseek-chat", 
+                          temperature: float = 0.7, max_tokens: int = 4000, 
+                          system_prompt: str = None) -> str:
+    """Get response from DeepSeek."""
+    if not openai:
+        logger.error("OpenAI library not available")
+        return "OpenAI library not available. Please install openai package."
+    
+    try:
+        # Use APIKeyManager instead of direct environment variable access
+        api_key_manager = APIKeyManager()
+        api_key = api_key_manager.get_api_key("deepseek")
+        
+        if not api_key:
+            raise ValueError("DeepSeek API key not found. Please configure it in the onboarding process.")
+        
+        client = openai.OpenAI(
+            api_key=api_key,
+            base_url="https://api.deepseek.com/v1"
+        )
+        
+        # Prepare messages
+        messages = []
+        if system_prompt:
+            messages.append({"role": "system", "content": system_prompt})
+        messages.append({"role": "user", "content": prompt})
+        
+        response = client.chat.completions.create(
+            model=model,
+            messages=messages,
+            max_tokens=max_tokens,
+            temperature=temperature
+        )
+        
+        logger.info(f"[deepseek_text_response] Generated response with {len(response.choices[0].message.content)} characters")
+        return response.choices[0].message.content
+        
+    except Exception as err:
+        logger.error(f"Failed to get response from DeepSeek: {err}. Retrying.")
+        raise