Apply suggestions from code review

Co-authored-by: Copilot <[email protected]>

Author: abrookins

agent_memory_server/mcp.py

@@ -417,14 +417,12 @@ async def memory_prompt(
         A list of messages, including memory context and the user's query
     """
     _session_id = session_id.eq if session_id and session_id.eq else None
-    _namespace = namespace.eq if namespace and namespace.eq else None
-    _params: dict[str, SessionMemoryRequest | SearchRequest] = {}
     session = None
 
     if _session_id is not None:
         session = SessionMemoryRequest(
             session_id=_session_id,
-            namespace=_namespace,
+            namespace=namespace.eq if namespace and namespace.eq else None,
             window_size=window_size,
             model_name=model_name,
             context_window_max=context_window_max,

agent_memory_server/models.py

@@ -317,4 +317,4 @@ class SystemMessage(base.Message):
 class UserMessage(base.Message):
     """A user message"""
 
-    role: Literal["system"] = "system"
+    role: Literal["user"] = "user"

tests/test_mcp.py

@@ -102,8 +102,6 @@ async def test_memory_prompt(self, session, mcp_test_setup):
             assert "messages" in messages
             assert len(messages["messages"]) == 5
 
-            print(messages)
-
             # The returned messages structure is:
             # 0: system (summary)
             # 1: user ("Hello")