highperfocused/danswer
1
0
Fork 0
mirror of https://github.com/danswer-ai/danswer.git synced 2025-09-23 12:31:30 +02:00
Code Issues Packages Projects Releases Wiki Activity

Fix limit/offset for document-search endpoint

Browse Source
This commit is contained in:
Weves
2024-01-23 15:26:13 -08:00
committed by Chris Weaver
parent fe975c3357
commit 15934ee268
1 changed files with 13 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

13
backend/ee/danswer/server/query_and_chat/query_backend.py
View File

@@ -4,6 +4,7 @@ from sqlalchemy.orm import Session
from danswer.auth.users import current_user from danswer.auth.users import current_user
from danswer.configs.chat_configs import DISABLE_LLM_CHUNK_FILTER from danswer.configs.chat_configs import DISABLE_LLM_CHUNK_FILTER
from danswer.configs.chat_configs import NUM_RETURNED_HITS
from danswer.db.engine import get_session from danswer.db.engine import get_session
from danswer.db.models import User from danswer.db.models import User
from danswer.document_index.factory import get_default_document_index from danswer.document_index.factory import get_default_document_index
@@ -44,11 +45,23 @@ def handle_search_request(
access_control_list=user_acl_filters, access_control_list=user_acl_filters,
) )
limit = (
search_request.retrieval_options.limit
if search_request.retrieval_options.limit is not None
else NUM_RETURNED_HITS
)
offset = (
search_request.retrieval_options.offset
if search_request.retrieval_options.offset is not None
else 0
)
search_query = SearchQuery( search_query = SearchQuery(
query=query, query=query,
search_type=search_request.search_type, search_type=search_request.search_type,
filters=final_filters, filters=final_filters,
recency_bias_multiplier=search_request.recency_bias_multiplier, recency_bias_multiplier=search_request.recency_bias_multiplier,
num_hits=limit,
offset=offset,
skip_rerank=search_request.skip_rerank, skip_rerank=search_request.skip_rerank,
skip_llm_chunk_filter=disable_llm_chunk_filter, skip_llm_chunk_filter=disable_llm_chunk_filter,
) )