refactor: Update server and protocol for task input handling

- Refactored the server to replace `ExecutionInput` with `TaskInput` for improved clarity in task management.
- Updated import paths for `LangGraphAdapter` and related components to streamline the code structure.
- Enhanced the `StreamMessage` protocol to include a `task_id` for better tracking of task execution.
- Removed the `ToolStateManager` and related classes to simplify the memory management approach.
- Adjusted comments and logging for better readability and consistency across the codebase.

Author: lihao

examples/components/web/super_agent/src/super_agent/server.py

@@ -26,12 +26,12 @@
 from fastapi.middleware.cors import CORSMiddleware
 from fastapi.responses import StreamingResponse
 from super_agent.agent.crew import SuperAgentCrew
-from langcrew.web.langgraph_adapter import LangGraphAdapter
+from langcrew.web import LangGraphAdapter
 from langcrew.web.protocol import (
     ChatRequest,
     StreamMessage,
     StopRequest,
-    ExecutionInput,
+    TaskInput,
     MessageType,
 )
 from langcrew.web import generate_message_id
@@ -140,17 +140,18 @@ async def generate():
                         timestamp=int(time.time() * 1000),
                         session_id=session_id,
                     )
-                    yield await adapter._format_sse_message(init_message)
+                    yield adapter._format_sse_message(init_message)
 
-                # Create execution input
-                execution_input = ExecutionInput(
+                # Create task input
+                task_input = TaskInput(
                     session_id=session_id,
-                    user_input=request.message,
-                    is_resume=request.interrupt_data is not None,
+                    message=request.message,
+                    language=request.language,
+                    interrupt_data=request.interrupt_data,
                 )
 
                 # Stream execution results
-                async for chunk in adapter.execute(execution_input):
+                async for chunk in adapter.execute(task_input):
                     yield chunk
 
             except asyncio.CancelledError:
@@ -169,7 +170,7 @@ async def generate():
                     timestamp=int(time.time() * 1000),
                     session_id=session_id,
                 )
-                yield await adapter._format_sse_message(error_message)
+                yield adapter._format_sse_message(error_message)
 
             finally:
                 logger.info(f"Session {session_id} completed")

libs/langcrew/langcrew/crew.py

@@ -23,7 +23,7 @@
 from .hitl import HITLConfig
 from .memory import EntityMemory, LongTermMemory, MemoryConfig, ShortTermMemory
 from .memory.storage import get_checkpointer, get_storage
-from .memory.tool_state import ToolStateManager
+
 from .task import Task
 from .types import CrewState
 
@@ -73,7 +73,7 @@ def __init__(
         self._short_term_memory = None
         self._long_term_memory = None
         self._entity_memory = None
-        self._tool_state_manager = None
+
         self._thread_id = None
 
         # Async components for async methods
@@ -85,7 +85,7 @@ def __init__(
         self._async_short_term_memory = None
         self._async_long_term_memory = None
         self._async_entity_memory = None
-        self._async_tool_state_manager = None
+
         self._async_components_initialized = False
 
         # Context managers for async components
@@ -122,7 +122,17 @@ def __init__(
             agent.store = self.store
 
     def _prepare_tools(self, tools: list[BaseTool]) -> list[BaseTool]:
-        """Inject ToolStateManager into E2BBaseToolV2 and HITLBaseTool instances."""
+        """Prepare tools for execution.
+
+        This method can be extended to inject dependencies or modify tools
+        before they are used by agents or tasks.
+
+        Args:
+            tools: List of tools to prepare
+
+        Returns:
+            List of prepared tools
+        """
         return tools
 
     def _sync_subgraph_message_deletions(
@@ -922,9 +932,6 @@ def _setup_crew_memory(self):
                 store=self.store, config=self.memory_config
             )
 
-        # Initialize tool state manager
-        self._tool_state_manager = ToolStateManager(checkpointer=self.checkpointer)
-
         if self.verbose:
             logger.info(
                 f"Memory system initialized with provider: {self.memory_config.provider}"
@@ -1013,12 +1020,6 @@ async def _setup_async_components(self):
                 store=self._async_store, config=self.memory_config
             )
 
-        # Create async tool state manager (only if not already created)
-        if not self._async_tool_state_manager:
-            self._async_tool_state_manager = ToolStateManager(
-                checkpointer=self._async_checkpointer
-            )
-
         # Mark async components as initialized
         self._async_components_initialized = True