Update ai_providers.py

ai_providers.py

@@ -1,21 +1,10 @@
-from abc import ABC, abstractmethod
 from typing import Dict, Any, Optional
 import os
 from dotenv import load_dotenv
 import time
 
-# Define the abstract base class
-class AIProvider(ABC):
-    @abstractmethod
-    def get_completion(self, prompt: str, **kwargs) -> str:
-        pass
-
-    @abstractmethod
-    def get_config(self) -> Dict[str, Any]:
-        pass
-
-class GroqProvider(AIProvider):
-    def __init__(self, api_key: str, model: str = "deepseek-ai/deepseek-math-7b-instruct"):
+class GroqProvider:
+    def __init__(self, api_key: str, model: str = "deepseek-ai/deepseek-r1-distill-llama-70b"):
         self.api_key = api_key
         self.model = model
         
@@ -60,76 +49,13 @@ class GroqProvider(AIProvider):
             }]
         }
 
-class OpenAIProvider(AIProvider):
-    def __init__(self, api_key: str, model: str = "gpt-4"):
-        self.api_key = api_key
-        self.model = model
-        
-    def get_completion(self, prompt: str, **kwargs) -> str:
-        from openai import OpenAI  # Updated import
-        
-        client = OpenAI(api_key=self.api_key)  # Updated client initialization
-        
-        response = client.chat.completions.create(
-            model=self.model,
-            messages=[{"role": "user", "content": prompt}],
-            **kwargs
-        )
-        return response.choices[0].message.content
+def create_provider(api_key: Optional[str] = None, model: Optional[str] = None) -> GroqProvider:
+    if not api_key:
+        load_dotenv()
+        api_key = os.getenv("GROQ_API_KEY")
         
-    def get_config(self) -> Dict[str, Any]:
-        return {
-            "config_list": [{
-                "model": self.model,
-                "api_key": self.api_key,
-                "temperature": 0.7,
-                "max_tokens": 4000
-            }]
-        }
-
-class AnthropicProvider(AIProvider):
-    def __init__(self, api_key: str, model: str = "claude-3-opus-20240229"):
-        self.api_key = api_key
-        self.model = model
-        
-    def get_completion(self, prompt: str, **kwargs) -> str:
-        import anthropic
-        
-        client = anthropic.Client(api_key=self.api_key)
-        response = client.messages.create(
-            model=self.model,
-            messages=[{"role": "user", "content": prompt}],
-            **kwargs
-        )
-        return response.content[0].text
-        
-    def get_config(self) -> Dict[str, Any]:
-        return {
-            "config_list": [{
-                "model": self.model,
-                "api_key": self.api_key,
-                "temperature": 0.7,
-                "max_tokens": 4000
-            }]
-        }
-
-class AIProviderFactory:
-    @staticmethod
-    def create_provider(provider_type: str, api_key: Optional[str] = None, model: Optional[str] = None) -> AIProvider:
-        if not api_key:
-            load_dotenv()
-            
-        providers = {
-            "openai": (OpenAIProvider, os.getenv("OPENAI_API_KEY"), "gpt-4"),
-            "groq": (GroqProvider, os.getenv("GROQ_API_KEY"), "deepseek-ai/deepseek-r1-distill-llama-70b"),
-            "anthropic": (AnthropicProvider, os.getenv("ANTHROPIC_API_KEY"), "claude-3-opus-20240229")
-        }
+    if not api_key:
+        raise ValueError("GROQ_API_KEY must be set in environment variables")
         
-        if provider_type not in providers:
-            raise ValueError(f"Unsupported provider: {provider_type}")
-            
-        ProviderClass, default_api_key, default_model = providers[provider_type]
-        return ProviderClass(
-            api_key=api_key or default_api_key,
-            model=model or default_model
-        )
+    default_model = "deepseek-ai/deepseek-r1-distill-llama-70b"
+    return GroqProvider(api_key=api_key, model=model or default_model)