Update model.py

model.py

@@ -1,6 +1,7 @@
 """Model wrapper for LiteLLM"""
 
 import os
+import json
 from typing import List, Dict, Any, Optional
 
 try:
@@ -71,13 +72,21 @@ class LiteLLMModel:
             }
             
             if hasattr(message, 'tool_calls') and message.tool_calls:
-                result["tool_calls"] = [
-                    {
+                result["tool_calls"] = []
+                for tc in message.tool_calls:
+                    # Parse arguments if they're a string
+                    args = tc.function.arguments
+                    if isinstance(args, str):
+                        try:
+                            args = json.loads(args)
+                        except:
+                            args = {}
+                    
+                    result["tool_calls"].append({
+                        "id": tc.id if hasattr(tc, 'id') else f"call_{tc.function.name}",
                         "name": tc.function.name,
-                        "arguments": eval(tc.function.arguments) if isinstance(tc.function.arguments, str) else tc.function.arguments
-                    }
-                    for tc in message.tool_calls
-                ]
+                        "arguments": args
+                    })
             
             return result