Update agents/executor.py

agents/executor.py

@@ -25,7 +25,7 @@ def executor_agent(anthropic_client, db, state: Dict[str, Any]) -> Dict[str, Any
     messages = state.get("messages", [])
     sql_queries = state.get("sql_queries", [])
     
-    # Add agent-specific instructions
+    # Add agent-specific instructions as system parameter
     system_message = """
     You are an AI assistant specializing in executing and validating pharmaceutical data pipelines.
     Your job is to execute the SQL queries and report on the results.
@@ -47,17 +47,29 @@ 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="system", content=system_message),
-        *[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 state messages to the format expected by Anthropic API
+    # Excluding any messages with role "system"
+    anthropic_messages = []
+    for msg in messages:
+        if msg["role"] != "system":  # Skip system messages
+            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
+    # Call Claude API with system message as separate parameter
     response = anthropic_client.messages.create(
         model="claude-3-7-sonnet-20250219",
-        messages=prompt_messages,
+        system=system_message,
+        messages=anthropic_messages,
         max_tokens=2000
     )