Update app.py

app.py

@@ -83,6 +83,48 @@ def enviar_ubicacion(access_token: str, phone_number_id: str, recipient_number:
     response = requests.post(url, headers=headers, json=data)
     return response.json()    
 
+def handle_tool_call(tool_call, contexto):
+    """
+    Genera los mensajes necesarios tras una llamada a una tool:
+    - assistant con tool_call
+    - tool con respuesta de la función
+
+    Retorna una lista con ambos mensajes para agregarlos al historial.
+    """
+    tool_call_id = tool_call.id
+    function_name = tool_call.function.name
+
+    # Asegurar que los argumentos son string JSON, no dict
+    arguments = (
+        json.dumps(tool_call.function.arguments)
+        if isinstance(tool_call.function.arguments, dict)
+        else tool_call.function.arguments
+    )
+
+    return [
+        {
+            "role": "assistant",
+            "content": None,
+            "tool_calls": [
+                {
+                    "id": tool_call_id,
+                    "type": "function",
+                    "function": {
+                        "name": function_name,
+                        "arguments": arguments
+                    }
+                }
+            ]
+        },
+        {
+            "role": "tool",
+            "tool_call_id": tool_call_id,
+            "name": function_name,
+            "content": contexto
+        }
+    ]
+
+
 # Modelo de datos para la solicitud
 class ChatRequest(BaseModel):
     message: str
@@ -166,31 +208,12 @@ async def chat(request: ChatRequest):
 
                 if name == "obtener_extractos":
                     contexto = obtener_extractos(args["query"])
-
-                    # Añadir la respuesta de la tool al historial de mensajes
-                    tool_call_id = tool_call.id
-                    messages.append({
-                        "role": "assistant",
-                        "content": None,
-                        "tool_calls": [
-                            {
-                                "id": tool_call_id,
-                                "type": "function",
-                                "function": {
-                                    "name": name,
-                                    "arguments": tool_call.function.arguments  # debe ser string (JSON serializado)
-                                }
-                            }
-                        ]
-                    })
-                    messages.append({
-                        "role": "tool",
-                        "tool_call_id": tool_call_id,
-                        "name": name,
-                        "content": contexto
-                    })
-
-                    # Segunda llamada a OpenAI ahora con contexto
+                
+                    # Generar los mensajes necesarios
+                    tool_messages = handle_tool_call(tool_call, contexto)
+                    messages.extend(tool_messages)
+                
+                    # Segunda llamada con nuevo contexto
                     response = client.chat.completions.create(
                         model="gpt-4o-mini",
                         messages=messages,
@@ -198,7 +221,7 @@ async def chat(request: ChatRequest):
                         temperature=request.temperature,
                         top_p=request.top_p
                     )
-
+                
                     return {"response": response.choices[0].message.content, "context": contexto}
 
                 elif name == "enviar_contacto":