fixing a psycho error

Author: naji247

pyproject.toml

@@ -18,7 +18,7 @@ classifiers = [
     "Programming Language :: Python :: 3.12",
 ]
 dependencies = [
-    "mcp==1.2.1",
+    "mcp==1.4.1",
     "mcpcat-api==0.1.3",
     "pydantic>=2.0.0",
 ]

tests/test_tool_context.py

@@ -507,7 +507,7 @@ async def test_original_functionality_preserved(self):
             # Complete a todo
             complete_result = await client.call_tool(
                 "complete_todo",
-                {"id": "1", "context": "Completing the first todo"}
+                {"id": 1, "context": "Completing the first todo"}
             )
 
             assert "Completed todo" in complete_result.content[0].text

tests/test_utils/todo_server.py

@@ -13,7 +13,7 @@
 class Todo:
     """Todo item."""
 
-    def __init__(self, id: str, text: str, completed: bool = False):
+    def __init__(self, id: int, text: str, completed: bool = False):
         self.id = id
         self.text = text
         self.completed = completed
@@ -33,7 +33,7 @@ def create_todo_server():
     def add_todo(text: str) -> str:
         """Add a new todo item."""
         nonlocal next_id
-        todo = Todo(str(next_id), text)
+        todo = Todo(next_id, text)
         todos.append(todo)
         next_id += 1
         return f'Added todo: "{text}" with ID {todo.id}'
@@ -52,7 +52,7 @@ def list_todos() -> str:
         return "\n".join(todo_list)
 
     @server.tool()
-    def complete_todo(id: str) -> str:
+    def complete_todo(id: int) -> str:
         """Mark a todo item as completed."""
         for todo in todos:
             if todo.id == id: