Update agents/state.py

agents/state.py

@@ -5,6 +5,12 @@ Agent state definition for the pharmaceutical data management system.
 import operator
 from typing import Dict, List, Any, Literal, TypedDict, Optional, Annotated
 
+# Define a reducer function for current_agent
+# This function will take the last value in case of conflicts
+def last_value_reducer(existing_value: str, new_value: str) -> str:
+    """Returns the new value, effectively taking the last agent in the chain."""
+    return new_value
+
 class AgentState(TypedDict):
     """
     State definition for the pharmaceutical data management agents.
@@ -28,5 +34,5 @@ class AgentState(TypedDict):
     # Status remains a single-value key
     status: Literal["planning", "executing", "complete", "error", "pending_approval"]
     
-    # Current agent is ALWAYS a string
-    current_agent: str
+    # Use Annotated with last_value_reducer for current_agent to handle concurrent updates
+    current_agent: Annotated[str, last_value_reducer]