Update agents.py

agents.py

@@ -50,9 +50,33 @@ Format your final answer clearly and concisely."""
             if response.get("tool_calls"):
                 print(f"\nIteration {iteration + 1}: Tool calls requested")
                 
+                # Create assistant message with tool calls
+                assistant_msg = {
+                    "role": "assistant",
+                    "content": response.get("content") or "Using tools...",
+                }
+                
+                # Add tool_calls if the model supports it
+                if "groq" in str(self.model.model_id).lower():
+                    # For Groq, we need to include tool_calls in the message
+                    assistant_msg["tool_calls"] = [
+                        {
+                            "id": tc.get("id", f"call_{tc['name']}"),
+                            "type": "function",
+                            "function": {
+                                "name": tc["name"],
+                                "arguments": str(tc["arguments"])
+                            }
+                        }
+                        for tc in response["tool_calls"]
+                    ]
+                
+                messages.append(assistant_msg)
+                
                 for tool_call in response["tool_calls"]:
                     tool_name = tool_call["name"]
                     tool_args = tool_call["arguments"]
+                    tool_id = tool_call.get("id", f"call_{tool_name}")
                     
                     print(f"  → {tool_name}({tool_args})")
                     
@@ -62,13 +86,9 @@ Format your final answer clearly and concisely."""
                             print(f"  ← Result: {str(result)[:100]}...")
                             
                             # Add tool result to messages
-                            messages.append({
-                                "role": "assistant",
-                                "content": f"Used {tool_name}",
-                                "tool_calls": [tool_call]
-                            })
                             messages.append({
                                 "role": "tool",
+                                "tool_call_id": tool_id,
                                 "name": tool_name,
                                 "content": str(result)
                             })
@@ -76,11 +96,18 @@ Format your final answer clearly and concisely."""
                             print(f"  ← Error: {e}")
                             messages.append({
                                 "role": "tool",
+                                "tool_call_id": tool_id,
                                 "name": tool_name,
                                 "content": f"Error: {str(e)}"
                             })
                     else:
                         print(f"  ← Tool not found!")
+                        messages.append({
+                            "role": "tool",
+                            "tool_call_id": tool_id,
+                            "name": tool_name,
+                            "content": "Tool not found"
+                        })
             else:
                 # Model provided final answer
                 answer = response.get("content", "Unknown")