Update app.py

app.py

@@ -544,7 +544,13 @@ def get_agent_response(prompt: str, task_id: Optional[str]=None, thread_id: Opti
     agent_name_get = type(AGENT_INSTANCE).__name__
     logger.info(f"Agent ({agent_name_get}) processing. Task: {task_id or 'N/A'}. Thread: {thread_id_to_use}. Prompt: {prompt[:100]}...")
     
-    is_langgraph_agent_get = LANGGRAPH_FLAVOR_AVAILABLE and AGENT_INSTANCE and hasattr(AGENT_INSTANCE, 'graph') and hasattr(AGENT_INSTANCE, 'config_schema')
+    # Detect LangGraph agent - either custom or from langgraph.prebuilt.create_react_agent
+    # Both return CompiledStateGraph with invoke() method that accepts messages
+    is_langgraph_agent_get = (
+        agent_name_get == 'CompiledStateGraph' or 
+        (LANGGRAPH_FLAVOR_AVAILABLE and AGENT_INSTANCE and hasattr(AGENT_INSTANCE, 'graph') and hasattr(AGENT_INSTANCE, 'config_schema')) or
+        (hasattr(AGENT_INSTANCE, 'invoke') and hasattr(AGENT_INSTANCE, 'get_state'))
+    )
     
     try:
         if is_langgraph_agent_get: