Update app.py

app.py

@@ -14,24 +14,39 @@ import io
 
 class GroqLLM:
     """Compatible LLM interface for smolagents CodeAgent"""
-    def __init__(self, model_name="llama2-70b-4096"):
+    def __init__(self, model_name="llama-3.1-8B-Instant"):
         self.client = Groq(api_key=os.environ.get("GROQ_API_KEY"))
         self.model_name = model_name
     
     def __call__(self, prompt: str) -> str:
         """Make the class callable as required by smolagents"""
-        messages = [{"role": "user", "content": prompt}]
+        # Ensure the prompt is a string
+        if not isinstance(prompt, str):
+            return "Error: Prompt must be a string"
+            
         try:
-            response = self.client.chat.completions.create(
+            # Create a properly formatted message
+            completion = self.client.chat.completions.create(
                 model=self.model_name,
-                messages=messages,
+                messages=[{
+                    "role": "user",
+                    "content": str(prompt)  # Ensure content is string
+                }],
                 temperature=0.7,
                 max_tokens=1024,
                 stream=False
             )
-            return response.choices[0].message.content
+            
+            # Extract and return the response content
+            if completion.choices and len(completion.choices) > 0:
+                return completion.choices[0].message.content
+            return "Error: No response generated"
+            
         except Exception as e:
-            return f"Error generating response: {str(e)}"
+            # Provide more detailed error handling
+            error_msg = f"Error generating response: {str(e)}"
+            print(error_msg)  # Log the error
+            return error_msg
 
 @tool
 def analyze_basic_stats(data: pd.DataFrame) -> str: