highperfocused/danswer
1
0
Fork 0
mirror of https://github.com/danswer-ai/danswer.git synced 2025-10-10 13:15:18 +02:00
Code Issues Packages Projects Releases Wiki Activity

Ensure relevance functions for selected docs (#2063)

Browse Source 
* ensure relevance functions for selected docs

* remove logs

* remove log
This commit is contained in:
pablodanswer
2024-08-06 14:06:44 -07:00
committed by GitHub
parent 57aec7d02a
commit e0f0cfd92e
2 changed files with 25 additions and 13 deletions
Download Patch File Download Diff File Expand all files Collapse all files

14
backend/danswer/tools/search/search_tool.py
View File

@@ -10,6 +10,7 @@ from danswer.chat.chat_utils import llm_doc_from_inference_section
from danswer.chat.models import DanswerContext from danswer.chat.models import DanswerContext
from danswer.chat.models import DanswerContexts from danswer.chat.models import DanswerContexts
from danswer.chat.models import LlmDoc from danswer.chat.models import LlmDoc
from danswer.chat.models import SectionRelevancePiece
from danswer.db.models import Persona from danswer.db.models import Persona
from danswer.db.models import User from danswer.db.models import User
from danswer.dynamic_configs.interface import JSON_ro from danswer.dynamic_configs.interface import JSON_ro
@@ -193,9 +194,20 @@ class SearchTool(Tool):
recency_bias_multiplier=1.0, recency_bias_multiplier=1.0,
), ),
) )
# Build selected sections for specified documents
selected_sections = [
SectionRelevancePiece(
relevant=True,
document_id=section.center_chunk.document_id,
chunk_id=section.center_chunk.chunk_id,
)
for section in self.selected_sections
]
yield ToolResponse( yield ToolResponse(
id=SECTION_RELEVANCE_LIST_ID, id=SECTION_RELEVANCE_LIST_ID,
response=[i for i in range(len(self.selected_sections))], response=selected_sections,
) )
final_context_sections = prune_and_merge_sections( final_context_sections = prune_and_merge_sections(