include project_id in scope

Signed-off-by: Arnesh Dadhich <arnesh.dadhich@unifyapps.com>

Author: isuniverseok-ua

src/client/acontext-py/src/acontext/resources/sessions.py

@@ -503,23 +503,29 @@ def search(
         *,
         query: str,
         user_id: str,
+        project_id: str,
         limit: int | None = None,
     ) -> SessionSearchResult:
         """Search for sessions by semantic similarity to a query string.
 
         Args:
             query: The search query text.
             user_id: The User ID to search within.
+            project_id: The Project ID to search within.
             limit: Maximum number of results to return (1-100, default 10).
 
         Returns:
             SessionSearchResult containing list of matching session UUIDs.
 
         Example:
-            >>> result = client.sessions.search(query="conversations about authentication", user_id="user_123")
+            >>> result = client.sessions.search(
+            ...     query="conversations about authentication",
+            ...     user_id="user_123",
+            ...     project_id="proj_456"
+            ... )
             >>> for session_id in result.session_ids:
             ...     print(session_id)
         """
-        params = build_params(query=query, user_id=user_id, limit=limit)
+        params = build_params(query=query, user_id=user_id, project_id=project_id, limit=limit)
         data = self._requester.request("GET", "/sessions/search", params=params or None)
         return SessionSearchResult.model_validate(data)

src/client/acontext-ts/src/resources/sessions.ts

@@ -445,23 +445,27 @@ export class SessionsAPI {
    * @param options - Options for searching sessions.
    * @param options.query - The search query text.
    * @param options.userId - The User ID to search within.
+   * @param options.projectId - The Project ID to search within.
    * @param options.limit - Maximum number of results to return (1-100, default 10).
    * @returns SessionSearchResult containing list of matching session UUIDs.
    *
    * @example
    * const result = await client.sessions.search({
    *   query: 'conversations about authentication',
-   *   userId: 'user_123'
+   *   userId: 'user_123',
+   *   projectId: 'proj_456'
    * });
    */
   async search(options: {
     query: string;
     userId: string;
+    projectId: string;
     limit?: number | null;
   }): Promise<{ session_ids: string[] }> {
     const params = buildParams({
       query: options.query,
       user_id: options.userId,
+      project_id: options.projectId,
       limit: options.limit ?? null,
     });
     const data = await this.requester.request('GET', '/sessions/search', {

src/server/api/go/internal/infra/httpclient/core.go

@@ -401,7 +401,7 @@ type SessionSearchResponse struct {
 }
 
 // SessionSearch calls the session search endpoint in Python Core
-func (c *CoreClient) SessionSearch(ctx context.Context, userID string, query string, limit int) (*SessionSearchResponse, error) {
+func (c *CoreClient) SessionSearch(ctx context.Context, userID string, projectID string, query string, limit int) (*SessionSearchResponse, error) {
 	endpoint := fmt.Sprintf("%s/api/v1/sessions/search", c.BaseURL)
 
 	httpReq, err := http.NewRequestWithContext(ctx, http.MethodGet, endpoint, nil)
@@ -412,6 +412,7 @@ func (c *CoreClient) SessionSearch(ctx context.Context, userID string, query str
 	// Add query parameters
 	q := httpReq.URL.Query()
 	q.Add("user_id", userID)
+	q.Add("project_id", projectID)
 	q.Add("query", query)
 	q.Add("limit", fmt.Sprintf("%d", limit))
 	httpReq.URL.RawQuery = q.Encode()

src/server/api/go/internal/modules/handler/session.go

@@ -780,9 +780,10 @@ func (h *SessionHandler) PatchConfigs(c *gin.Context) {
 }
 
 type SessionSearchReq struct {
-	Query  string `form:"query" json:"query" binding:"required" example:"find conversations about authentication"`
-	UserID string `form:"user_id" json:"user_id" binding:"required" example:"123e4567-e89b-12d3-a456-426614174000"`
-	Limit  int    `form:"limit,default=10" json:"limit" binding:"omitempty,min=1,max=100" example:"10"`
+	Query     string `form:"query" json:"query" binding:"required" example:"find conversations about authentication"`
+	UserID    string `form:"user_id" json:"user_id" binding:"required" example:"123e4567-e89b-12d3-a456-426614174000"`
+	ProjectID string `form:"project_id" json:"project_id" binding:"required" example:"123e4567-e89b-12d3-a456-426614174001"`
+	Limit     int    `form:"limit,default=10" json:"limit" binding:"omitempty,min=1,max=100" example:"10"`
 }
 
 // SessionSearch godoc
@@ -792,8 +793,10 @@ type SessionSearchReq struct {
 //	@Tags			session
 //	@Accept			json
 //	@Produce		json
-//	@Param			query	query	string	true	"Search query text"
-//	@Param			limit	query	integer	false	"Maximum number of results (1-100, default 10)"
+//	@Param			query		query	string	true	"Search query text"
+//	@Param			user_id		query	string	true	"User ID"
+//	@Param			project_id	query	string	true	"Project ID"
+//	@Param			limit		query	integer	false	"Maximum number of results (1-100, default 10)"
 //	@Security		BearerAuth
 //	@Success		200	{object}	serializer.Response{data=httpclient.SessionSearchResponse}
 //	@Router			/sessions/search [get]
@@ -809,7 +812,7 @@ func (h *SessionHandler) SessionSearch(c *gin.Context) {
 		limit = 10
 	}
 
-	result, err := h.coreClient.SessionSearch(c.Request.Context(), req.UserID, req.Query, limit)
+	result, err := h.coreClient.SessionSearch(c.Request.Context(), req.UserID, req.ProjectID, req.Query, limit)
 	if err != nil {
 		c.JSON(http.StatusInternalServerError, serializer.Err(http.StatusInternalServerError, "failed to search sessions", err))
 		return

src/server/core/acontext_core/service/data/session_search_service.py

@@ -10,6 +10,7 @@
 async def search_sessions_by_task_query(
     db_session: AsyncSession,
     user_id: asUUID,
+    project_id: asUUID,
     query: str,
     topk: int = 10,
     threshold: float = 0.8,
@@ -20,6 +21,7 @@ async def search_sessions_by_task_query(
     Args:
         db_session: Database session
         user_id: User ID to scope the search
+        project_id: Project ID to scope the search
         query: Search query text
         topk: Maximum number of results to return
         threshold: Cosine distance threshold (lower = more similar)
@@ -41,6 +43,7 @@ async def search_sessions_by_task_query(
             FROM tasks t
             JOIN sessions s ON t.session_id = s.id
             WHERE s.user_id = :user_id
+              AND s.project_id = :project_id
               AND t.embedding IS NOT NULL
               AND (t.embedding <=> :query_embedding::vector) < :threshold
             GROUP BY t.session_id
@@ -53,6 +56,7 @@ async def search_sessions_by_task_query(
             {
                 "query_embedding": str(query_embedding),
                 "user_id": str(user_id),
+                "project_id": str(project_id),
                 "threshold": threshold,
                 "topk": topk,
             },

src/server/core/routers/session.py

@@ -37,18 +37,20 @@ class SessionSearchResponse(BaseModel):
 @search_router.get("/search")
 async def session_search(
     user_id: asUUID = Query(..., description="User ID to search within"),
+    project_id: asUUID = Query(..., description="Project ID to search within"),
     query: str = Query(..., description="Search query text"),
     limit: int = Query(10, ge=1, le=100, description="Maximum results to return"),
 ) -> SessionSearchResponse:
     """
     Uses vector embeddings on Tasks to find sessions with relevant context.
     """
-    LOG.info(f"Searching sessions in user {user_id} with query: {query[:50]}...")
+    LOG.info(f"Searching sessions in project {project_id} for user {user_id} with query: {query[:50]}...")
 
     async with DB_CLIENT.get_session_context() as db_session:
         result = await search_sessions_by_task_query(
             db_session,
             user_id,
+            project_id,
             query,
             topk=limit,
         )