From 29081f44294f9d6eca8da3d29e039076f83a7044 Mon Sep 17 00:00:00 2001 From: sabaimran Date: Thu, 27 Jul 2023 22:22:09 -0700 Subject: [PATCH] Adjust parameters for offline chat --- src/khoj/routers/api.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/src/khoj/routers/api.py b/src/khoj/routers/api.py index d68404b3..c2103b8d 100644 --- a/src/khoj/routers/api.py +++ b/src/khoj/routers/api.py @@ -723,8 +723,9 @@ async def extract_references_and_questions( with timer("Searching knowledge base took", logger): result_list = [] for query in inferred_queries: + n_items = n if state.processor_config.conversation.openai_model else min(n, 3) result_list.extend( - await search(query, request=request, n=n, r=True, score_threshold=-5.0, dedupe=False) + await search(query, request=request, n=n_items, r=True, score_threshold=-5.0, dedupe=False) ) compiled_references = [item.additional["compiled"] for item in result_list]