Update app.py

app.py

@@ -33,13 +33,17 @@ class BasicAgent:
         """Run the agent and return whatever FINAL_ANSWER the graph produces."""
         print(f"Agent received question: {question}")
         
+        # Create system prompt with task_id included
+        system_prompt_with_task = SYSTEM_PROMPT
+        if task_id:
+            system_prompt_with_task += f"\n\nIMPORTANT: Your current task_id is: {task_id}. When using any tools that require a task_id parameter (audio_transcriber_tool, excel_tool, analyze_code_tool, image_tool), use this exact task_id: {task_id}"
+        
         # Initialize the state properly with all required fields
         init_state = {
             "messages": [
-                SystemMessage(content=SYSTEM_PROMPT),
+                SystemMessage(content=system_prompt_with_task),
                 HumanMessage(content=question)
-            ],
-            "task_id": task_id
+            ]
         }
         
         # IMPORTANT: invoke() returns a **new** state instance (or an AddableValuesDict),
@@ -54,7 +58,6 @@ class BasicAgent:
             if hasattr(last_message, 'content'):
                 content = last_message.content
                 # Look for FINAL ANSWER: pattern
-                print("task_id: ", out_state["task_id"])
                 print("content: ", content)
                 print("\n\n\n\n")
                 if "FINAL ANSWER:" in content: