Update app.py

app.py

@@ -14,37 +14,34 @@ import io
 
 class GroqLLM:
     """Compatible LLM interface for smolagents CodeAgent"""
-    def __init__(self, model_name="llama-3.1-8B-Instant"):
+    def __init__(self, model_name="llama3-8b-8192"):
         self.client = Groq(api_key=os.environ.get("GROQ_API_KEY"))
         self.model_name = model_name
-    
+
     def __call__(self, prompt: Union[str, dict, List[Dict]]) -> str:
-        """Make the class callable as required by smolagents"""
+        """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"""
         try:
-            # Handle different prompt formats
             if isinstance(prompt, (dict, list)):
                 prompt_str = str(prompt)
             else:
                 prompt_str = str(prompt)
-            
-            # Create a properly formatted message
+
             completion = self.client.chat.completions.create(
                 model=self.model_name,
-                messages=[{
-                    "role": "user",
-                    "content": prompt_str
-                }],
+                messages=[{"role": "user", "content": prompt_str}],
                 temperature=0.7,
                 max_tokens=1024,
                 stream=False
             )
-            
-            return completion.choices[0].message.content if completion.choices else "Error: No response generated"
-            
+
+            return completion.choices[0].message.content.strip() if completion.choices else "Error: No response generated"
         except Exception as e:
-            error_msg = f"Error generating response: {str(e)}"
-            print(error_msg)
-            return error_msg
+            return f"Error generating response: {str(e)}"
+
 
 class DataAnalysisAgent(CodeAgent):
     """Extended CodeAgent with dataset awareness"""