cope with warnings

Author: ks6088ts

template_langgraph/agents/demo_agents/deep_agent_core/state.py

@@ -1,7 +1,7 @@
-from typing import Annotated, Literal, NotRequired
+from typing import Annotated, Literal
 
 from langgraph.prebuilt.chat_agent_executor import AgentState
-from typing_extensions import TypedDict
+from typing_extensions import NotRequired, TypedDict
 
 
 class Todo(TypedDict):

template_langgraph/agents/demo_agents/deep_agent_core/sub_agent.py

@@ -1,11 +1,11 @@
-from typing import Annotated, Any, NotRequired
+from typing import Annotated, Any
 
 from langchain.chat_models import init_chat_model
 from langchain_core.messages import ToolMessage
 from langchain_core.tools import BaseTool, InjectedToolCallId, tool
 from langgraph.prebuilt import InjectedState, create_react_agent
 from langgraph.types import Command
-from typing_extensions import TypedDict
+from typing_extensions import NotRequired, TypedDict
 
 from template_langgraph.agents.demo_agents.deep_agent_core.prompts import (
     TASK_DESCRIPTION_PREFIX,

template_langgraph/agents/demo_agents/deep_agent_core/tools.py

@@ -105,22 +105,53 @@ def edit_file(
     mock_filesystem = state.get("files", {})
     # Check if file exists in mock filesystem
     if file_path not in mock_filesystem:
-        return f"Error: File '{file_path}' not found"
+        return Command(
+            update={
+                "messages": [ToolMessage(f"Error: File '{file_path}' not found", tool_call_id=tool_call_id)],
+            }
+        )
 
     # Get current file content
     content = mock_filesystem[file_path]
 
     # Check if old_string exists in the file
     if old_string not in content:
-        return f"Error: String not found in file: '{old_string}'"
+        return Command(
+            update={
+                "messages": [
+                    ToolMessage(
+                        f"Error: String not found in file: '{old_string}'",
+                        tool_call_id=tool_call_id,
+                    )
+                ],
+            }
+        )
 
     # If not replace_all, check for uniqueness
     if not replace_all:
         occurrences = content.count(old_string)
         if occurrences > 1:
-            return f"Error: String '{old_string}' appears {occurrences} times in file. Use replace_all=True to replace all instances, or provide a more specific string with surrounding context."
+            return Command(
+                update={
+                    "messages": [
+                        ToolMessage(
+                            f"Error: String '{old_string}' appears {occurrences} times in file. Use replace_all=True to replace all instances, or provide a more specific string with surrounding context.",
+                            tool_call_id=tool_call_id,
+                        )
+                    ]
+                }
+            )
         elif occurrences == 0:
-            return f"Error: String not found in file: '{old_string}'"
+            return Command(
+                update={
+                    "messages": [
+                        ToolMessage(
+                            f"Error: String not found in file: '{old_string}'",
+                            tool_call_id=tool_call_id,
+                        )
+                    ],
+                }
+            )
 
     # Perform the replacement
     if replace_all: