feat: support session_id in current chat search

core/api.py

@@ -10,7 +10,7 @@ from urllib.parse import parse_qsl, quote, urlsplit, urlunsplit
 import httpx
 from fastapi import APIRouter, FastAPI, File, Form, HTTPException, Request, UploadFile
 from pydantic import ValidationError
-from pydantic import BaseModel, Field, field_validator
+from pydantic import BaseModel, Field, field_validator, model_validator
 from starlette.datastructures import UploadFile as StarletteUploadFile
 from starlette.responses import Response
 
@@ -43,6 +43,7 @@ class SearchMemoriesRequest(BaseModel):
     user_id: str = Field(min_length=1)
     user_key: str = Field(min_length=1)
     agent_id: str | None = Field(default=None, min_length=1)
+    session_id: str | None = Field(default=None, min_length=1)
     conversation_id: str | None = None
     query: str = Field(min_length=1)
     scope: list[Literal["current_chat", "resources", "all_user_memory"]] = Field(
@@ -64,6 +65,16 @@ class SearchMemoriesRequest(BaseModel):
             raise ValueError("top_k must be -1 or in 1..100")
         return value
 
+    @model_validator(mode="after")
+    def validate_current_chat_session_aliases(self) -> SearchMemoriesRequest:
+        if self.session_id and self.conversation_id:
+            expected_session_id = f"chat:{self.conversation_id}"
+            if self.session_id != expected_session_id:
+                raise ValueError(
+                    "session_id must match chat:{conversation_id} when both are provided"
+                )
+        return self
+
 
 class AddMemoryMessage(BaseModel):
     sender_id: str = Field(min_length=1)
@@ -487,6 +498,7 @@ def create_app(
             user_id=request.user_id,
             agent_id=request.agent_id,
             query=request.query,
+            session_id=request.session_id,
             conversation_id=request.conversation_id,
             scope=request.scope,
             method=request.method,