Update ai_providers.py

ai_providers.py

@@ -1,9 +1,21 @@
+# ai_providers.py
+
+from abc import ABC, abstractmethod
 from typing import Dict, Any, Optional
 import os
 from dotenv import load_dotenv
 import time
 
-class GroqProvider:
+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-r1-distill-llama-70b"):
         self.api_key = api_key
         self.model = model
@@ -49,13 +61,16 @@ class GroqProvider:
             }]
         }
 
-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")
-        
-    if not api_key:
-        raise ValueError("GROQ_API_KEY must be set in environment variables")
-        
-    default_model = "deepseek-ai/deepseek-r1-distill-llama-70b"
-    return GroqProvider(api_key=api_key, model=model or default_model)
+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()
+            
+        if provider_type != "groq":
+            raise ValueError("Only Groq provider is supported.")
+            
+        return GroqProvider(
+            api_key=api_key or os.getenv("GROQ_API_KEY"),
+            model=model or "deepseek-ai/deepseek-r1-distill-llama-70b"
+        )