Update agents/sql_generator.py

agents/sql_generator.py

@@ -23,7 +23,7 @@ def sql_generator_agent(anthropic_client, state: Dict[str, Any]) -> Dict[str, An
     messages = state.get("messages", [])
     pipeline_plan = state.get("pipeline_plan", {})
     
-    # Add agent-specific instructions
+    # Add agent-specific instructions as system parameter
     system_message = """
     You are an AI assistant specializing in generating SQL for pharmaceutical data pipelines.
     Your job is to transform the pipeline plan into executable SQL queries.
@@ -48,17 +48,29 @@ def sql_generator_agent(anthropic_client, state: Dict[str, Any]) -> Dict[str, An
     - Data products: DP_SALES_DASHBOARD, DP_HCP_TARGETING
     """
     
-    # 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"Based on this pipeline plan, generate the SQL queries needed. {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"Based on this pipeline plan, generate the SQL queries needed. {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=3000
     )