Fix tests

Author: abrookins

agent-memory-client/agent_memory_client/client.py

@@ -1719,7 +1719,7 @@ def parse_tool_call(tool_call: dict[str, Any]) -> UnifiedToolCall:
                 id=tool_call.get("id"),
                 name=tool_call.get("name", ""),
                 arguments=tool_call.get("args", {}),
-                provider="langchain",
+                provider="generic",
             )
 
         # Generic format - assume it's already in a usable format
@@ -2104,9 +2104,9 @@ async def _resolve_delete_long_term_memories(
         result = await self.delete_long_term_memories(memory_ids=memory_ids)
         # Handle both dict-like and model responses
         try:
-            status = result.get("status")  # type: ignore[call-arg]
-        except Exception:
             status = getattr(result, "status", None)
+        except Exception:
+            status = None
         if not status:
             status = "Deleted memories successfully"
         return {"status": status}

agent_memory_server/mcp.py

@@ -1030,7 +1030,18 @@ async def edit_long_term_memory(
 
     # Filter out None values to only include fields that should be updated
     update_dict = {k: v for k, v in update_dict.items() if v is not None}
-    updates = EditMemoryRecordRequest(**update_dict)
+    updates = EditMemoryRecordRequest(
+        text=text if text is not None else None,
+        topics=topics if topics is not None else None,
+        entities=entities if entities is not None else None,
+        memory_type=memory_type if memory_type is not None else None,
+        namespace=namespace if namespace is not None else None,
+        user_id=user_id if user_id is not None else None,
+        session_id=session_id if session_id is not None else None,
+        event_date=update_dict.get("event_date")
+        if "event_date" in update_dict
+        else None,
+    )
 
     return await core_update_long_term_memory(memory_id=memory_id, updates=updates)
 

agent_memory_server/models.py

@@ -4,6 +4,7 @@
 from typing import Literal
 
 from mcp.server.fastmcp.prompts import base
+from mcp.types import AudioContent, EmbeddedResource, ImageContent, TextContent
 from pydantic import BaseModel, Field
 from ulid import ULID
 
@@ -434,10 +435,11 @@ class MemoryPromptRequest(BaseModel):
     long_term_search: SearchRequest | bool | None = None
 
 
-class SystemMessage(base.Message):
+class SystemMessage(BaseModel):
     """A system message"""
 
     role: Literal["system"] = "system"
+    content: str | TextContent | ImageContent | AudioContent | EmbeddedResource
 
 
 class UserMessage(base.Message):
@@ -453,7 +455,7 @@ class MemoryPromptResponse(BaseModel):
 class LenientMemoryRecord(ExtractedMemoryRecord):
     """A memory record that can be created without an ID"""
 
-    id: str | None = Field(default_factory=lambda: str(ULID()))
+    id: str = Field(default_factory=lambda: str(ULID()))
 
 
 class DeleteMemoryRecordRequest(BaseModel):

examples/memory_editing_agent.py

@@ -156,7 +156,7 @@ async def _handle_multiple_function_calls(
         # Execute all tool calls
         for i, tool_call in enumerate(tool_calls):
             function_name = tool_call.get("name", "unknown")
-            print(f"🔧 Using {function_name} tool ({i+1}/{len(tool_calls)})...")
+            print(f"🔧 Using {function_name} tool ({i + 1}/{len(tool_calls)})...")
 
             # Use the client's unified tool call resolver
             result = await client.resolve_tool_call(
@@ -312,7 +312,7 @@ async def _handle_multiple_function_calls(
             for i, tool_call in enumerate(followup_calls):
                 fname = tool_call.get("name", "unknown")
                 print(
-                    f"   🔧 Follow-up using {fname} tool ({i+1}/{len(followup_calls)})..."
+                    f"   🔧 Follow-up using {fname} tool ({i + 1}/{len(followup_calls)})..."
                 )
                 res = await client.resolve_tool_call(
                     tool_call=tool_call,