Update app.py

app.py

@@ -17,6 +17,10 @@ from typing_extensions import TypedDict
 
 def override(_, new): return new
 
+def merge_dicts(dict1: Dict, dict2: Dict) -> Dict:
+    """Merge two dictionaries, with values from dict2 taking precedence."""
+    return {**dict1, **dict2}
+
 print("trial")
 # (Keep Constants as is)
 # --- Constants ---
@@ -52,8 +56,8 @@ class AgentState(TypedDict):
     history: Annotated[List[Dict[str, str]], operator.add]
     needs_more_info: Annotated[bool, override]
     search_query: Annotated[str, override]
-    task_id: Annotated[str, override]  # Add task_id to state
-    logs: Annotated[Dict[str, Any], operator.add]  # Add logs to state
+    task_id: Annotated[str, override]
+    logs: Annotated[Dict[str, Any], merge_dicts]  # Use merge_dicts instead of operator.add
 
 class BasicAgent:
     def __init__(self):