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

Handle saved search docs in eval flow (#2075)

Browse Source
This commit is contained in:
pablodanswer
2024-08-07 16:18:34 -07:00
committed by GitHub
parent c4e1c62c00
commit 5097c7f284
4 changed files with 29 additions and 40 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@ -1,5 +1,3 @@
from typing import cast
from fastapi import APIRouter
from fastapi import Depends
from fastapi import HTTPException
@ -11,9 +9,7 @@ from danswer.configs.danswerbot_configs import DANSWER_BOT_TARGET_CHUNK_PERCENTA
from danswer.danswerbot.slack.handlers.handle_standard_answers import (
oneoff_standard_answers,
)
from danswer.db.chat import translate_db_search_doc_to_server_search_doc
from danswer.db.engine import get_session
from danswer.db.models import SearchDoc
from danswer.db.models import User
from danswer.db.persona import get_persona_by_id
from danswer.llm.answering.prompts.citations_prompt import (
@ -31,7 +27,7 @@ from danswer.search.models import SearchRequest
from danswer.search.pipeline import SearchPipeline
from danswer.search.utils import dedupe_documents
from danswer.search.utils import drop_llm_indices
from danswer.search.utils import relevant_documents_to_indices
from danswer.search.utils import relevant_sections_to_indices
from danswer.utils.logger import setup_logger
from ee.danswer.server.query_and_chat.models import DocumentSearchRequest
from ee.danswer.server.query_and_chat.models import StandardAnswerRequest
@ -113,12 +109,8 @@ def handle_search_request(
if search_request.retrieval_options.dedupe_docs:
deduped_docs, dropped_inds = dedupe_documents(top_docs)
llm_indices = relevant_documents_to_indices(
relevance_sections=relevance_sections,
search_docs=[
translate_db_search_doc_to_server_search_doc(cast(SearchDoc, doc))
for doc in deduped_docs
],
llm_indices = relevant_sections_to_indices(
relevance_sections=relevance_sections, items=deduped_docs
)
if dropped_inds: