From 7dfba1f38f013f916fadb2a200f211dbfa0f3ab8 Mon Sep 17 00:00:00 2001 From: shamoon <4887959+shamoon@users.noreply.github.com> Date: Wed, 1 Apr 2026 10:18:26 -0700 Subject: [PATCH] Wire the simple searches to view --- src/documents/views.py | 30 +++++++++++++++++++++--------- 1 file changed, 21 insertions(+), 9 deletions(-) diff --git a/src/documents/views.py b/src/documents/views.py index 80c83d1d7..4308a3ec6 100644 --- a/src/documents/views.py +++ b/src/documents/views.py @@ -2001,6 +2001,12 @@ class ChatStreamingView(GenericAPIView): location=OpenApiParameter.QUERY, description="Simple text search query string", ), + OpenApiParameter( + name="title_search", + type=OpenApiTypes.STR, + location=OpenApiParameter.QUERY, + description="Simple title-only search query string", + ), OpenApiParameter( name="query", type=OpenApiTypes.STR, @@ -2040,6 +2046,7 @@ class UnifiedSearchViewSet(DocumentViewSet): def _is_search_request(self): return ( "text" in self.request.query_params + or "title_search" in self.request.query_params or "query" in self.request.query_params or "more_like_id" in self.request.query_params ) @@ -2059,15 +2066,20 @@ class UnifiedSearchViewSet(DocumentViewSet): user = None if request.user.is_superuser else request.user - if "text" in request.query_params or "query" in request.query_params: - search_mode = ( - SearchMode.TEXT - if "text" in request.query_params - else SearchMode.QUERY - ) - query_str = ( - request.query_params.get("text") or request.query_params["query"] - ) + if ( + "text" in request.query_params + or "title_search" in request.query_params + or "query" in request.query_params + ): + if "text" in request.query_params: + search_mode = SearchMode.TEXT + query_str = request.query_params["text"] + elif "title_search" in request.query_params: + search_mode = SearchMode.TITLE + query_str = request.query_params["title_search"] + else: + search_mode = SearchMode.QUERY + query_str = request.query_params["query"] results = backend.search( query_str, user=user,