Update model.py

model.py

@@ -16,9 +16,11 @@ class LiteLLMModel:
     def __init__(self, model_id: str):
         self.model_id = model_id
         
-        if "gemini" in model_id.lower():
-            if not os.getenv("GEMINI_API_KEY"):
-                print("⚠️ GEMINI_API_KEY not set in environment")
+        # Check for Groq API key
+        if "groq" in model_id.lower():
+            if not os.getenv("GROQ_API_KEY"):
+                print("⚠️ GROQ_API_KEY not set in environment")
+                raise RuntimeError("GROQ_API_KEY not set. Please add it to your Space secrets.")
     
     def generate(self, messages: List[Dict], tools: Optional[List] = None) -> Dict:
         if not litellm:
@@ -39,12 +41,13 @@ class LiteLLMModel:
                     for tool in tools
                 ]
             
-            if "gemini" in self.model_id.lower():
-                api_key = os.getenv("GEMINI_API_KEY")
+            # Groq configuration
+            if "groq" in self.model_id.lower():
+                api_key = os.getenv("GROQ_API_KEY")
                 if not api_key:
-                    raise RuntimeError("GEMINI_API_KEY not set in environment")
+                    raise RuntimeError("GROQ_API_KEY not set in environment")
                 
-                print(f"DEBUG: Using model id: {self.model_id}")
+                print(f"DEBUG: Using Groq model: {self.model_id}")
                 
                 response = litellm.completion(
                     model=self.model_id,
@@ -54,6 +57,7 @@ class LiteLLMModel:
                     temperature=0.1
                 )
             else:
+                # Generic model support
                 response = litellm.completion(
                     model=self.model_id,
                     messages=messages,