feat: move agent_id from query param to path variable and remove unus… (#1094)

Co-authored-by: cpacker <packercharles@gmail.com>

memgpt/server/rest_api/agents/memory.py

@@ -1,7 +1,7 @@
 import uuid
 from functools import partial
 
-from fastapi import APIRouter, Depends, Body, Query
+from fastapi import APIRouter, Depends, Body
 from pydantic import BaseModel, Field
 
 from memgpt.server.rest_api.auth_token import get_current_user
@@ -16,10 +16,6 @@ class CoreMemory(BaseModel):
     persona: str | None = Field(None, description="Persona element of the core memory.")
 
 
-class GetAgentMemoryRequest(BaseModel):
-    agent_id: str = Field(..., description="The unique identifier of the agent.")
-
-
 class GetAgentMemoryResponse(BaseModel):
     core_memory: CoreMemory = Field(..., description="The state of the agent's core memory.")
     recall_memory: int = Field(..., description="Size of the agent's recall memory.")
@@ -41,9 +37,9 @@ class UpdateAgentMemoryResponse(BaseModel):
 def setup_agents_memory_router(server: SyncServer, interface: QueuingInterface, password: str):
     get_current_user_with_server = partial(partial(get_current_user, server), password)
 
-    @router.get("/agents/memory", tags=["agents"], response_model=GetAgentMemoryResponse)
+    @router.get("/agents/{agent_id}/memory", tags=["agents"], response_model=GetAgentMemoryResponse)
     def get_agent_memory(
-        agent_id: str = Query(..., description="The unique identifier of the agent."),
+        agent_id: uuid.UUID,
         user_id: uuid.UUID = Depends(get_current_user_with_server),
     ):
         """
@@ -51,17 +47,13 @@ def setup_agents_memory_router(server: SyncServer, interface: QueuingInterface,
 
         This endpoint fetches the current memory state of the agent identified by the user ID and agent ID.
         """
-        # Validate with the Pydantic model (optional)
-        request = GetAgentMemoryRequest(agent_id=agent_id)
-
-        agent_id = uuid.UUID(request.agent_id) if request.agent_id else None
-
         interface.clear()
         memory = server.get_agent_memory(user_id=user_id, agent_id=agent_id)
         return GetAgentMemoryResponse(**memory)
 
-    @router.post("/agents/memory", tags=["agents"], response_model=UpdateAgentMemoryResponse)
+    @router.post("/agents/{agent_id}/memory", tags=["agents"], response_model=UpdateAgentMemoryResponse)
     def update_agent_memory(
+        agent_id: uuid.UUID,
         request: UpdateAgentMemoryRequest = Body(...),
         user_id: uuid.UUID = Depends(get_current_user_with_server),
     ):