Update app.py

app.py

@@ -22,19 +22,23 @@ class GroqLLM:
         """Allows use as callable (legacy compatibility)"""
         return self.generate(prompt)
 
-    def generate(self, prompt: Union[str, dict, List[Dict]]) -> str:
-        """Generate text completion from Groq API"""
+    def generate(self, prompt: Union[str, dict, List[Dict]], **kwargs) -> str:
+        """Generate text using Groq API. Handles optional kwargs from smolagents."""
         try:
             if isinstance(prompt, (dict, list)):
                 prompt_str = str(prompt)
             else:
                 prompt_str = str(prompt)
 
+            temperature = kwargs.get("temperature", 0.7)
+            max_tokens = kwargs.get("max_tokens", 1024)
+            # 'stop_sequences' is not supported by Groq API directly; you can ignore or handle manually.
+
             completion = self.client.chat.completions.create(
                 model=self.model_name,
                 messages=[{"role": "user", "content": prompt_str}],
-                temperature=0.7,
-                max_tokens=1024,
+                temperature=temperature,
+                max_tokens=max_tokens,
                 stream=False
             )