Enable Surrounding Context (#2210)

This commit is contained in:
Yuhong Sun
2024-08-22 09:59:13 -07:00
committed by GitHub
parent 8afcb03f3c
commit 1a9e5da7c0
15 changed files with 137 additions and 67 deletions

View File

@@ -106,8 +106,9 @@ def handle_simplified_chat_message(
search_doc_ids=chat_message_req.search_doc_ids,
retrieval_options=retrieval_options,
query_override=chat_message_req.query_override,
chunks_above=chat_message_req.chunks_above,
chunks_below=chat_message_req.chunks_below,
# Currently only applies to search flow not chat
chunks_above=0,
chunks_below=0,
full_doc=chat_message_req.full_doc,
)
@@ -232,8 +233,8 @@ def handle_send_message_simple_with_history(
search_doc_ids=None,
retrieval_options=req.retrieval_options,
query_override=rephrased_query,
chunks_above=req.chunks_above,
chunks_below=req.chunks_below,
chunks_above=0,
chunks_below=0,
full_doc=req.full_doc,
)

View File

@@ -54,6 +54,7 @@ def handle_search_request(
logger.notice(f"Received document search query: {query}")
llm, fast_llm = get_default_llms()
search_pipeline = SearchPipeline(
search_request=SearchRequest(
query=query,