Update agents/executor.py

agents/executor.py

@@ -22,8 +22,8 @@ def executor_agent(anthropic_client, db, state: Dict[str, Any]) -> Dict[str, Any
         Updated state
     """
     # Get current messages and SQL queries
-    messages = state["messages"]
-    sql_queries = state["sql_queries"]
+    messages = state.get("messages", [])
+    sql_queries = state.get("sql_queries", [])
     
     # Add agent-specific instructions
     system_message = """
@@ -47,17 +47,28 @@ def executor_agent(anthropic_client, db, state: Dict[str, Any]) -> Dict[str, Any
     # Prepare context for Claude
     execution_context = json.dumps(execution_results, indent=2)
     
-    # Prepare prompt for Claude
-    prompt_messages = [
-        *[MessageParam(role=m["role"], content=m["content"]) for m in messages],
-        MessageParam(role="user", content=f"Here are the execution results of the SQL queries. Please analyze and report on them.\n\n{execution_context}")
-    ]
+    # Convert messages to the format expected by Anthropic API
+    anthropic_messages = []
+    for msg in messages:
+        if isinstance(msg, dict) and "role" in msg and "content" in msg:
+            anthropic_messages.append(MessageParam(
+                role=msg["role"],
+                content=msg["content"]
+            ))
+    
+    # Add final user message with context
+    anthropic_messages.append(
+        MessageParam(
+            role="user", 
+            content=f"Here are the execution results of the SQL queries. Please analyze and report on them.\n\n{execution_context}"
+        )
+    )
     
     # Call Claude API
     response = anthropic_client.messages.create(
         model="claude-3-7-sonnet-20250219",
         system=system_message,
-        messages=prompt_messages,
+        messages=anthropic_messages,
         max_tokens=2000
     )