From 2f31de5ed570abb4373bd48391064479b42142a5 Mon Sep 17 00:00:00 2001 From: Debanjum Singh Solanky Date: Fri, 7 Jul 2023 12:29:36 -0700 Subject: [PATCH] Set no. of references to use for chat configurable in Chat API --- 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 9fe68db9..f99a991d 100644 --- a/src/khoj/routers/api.py +++ b/src/khoj/routers/api.py @@ -436,6 +436,7 @@ def chat_init( async def chat( request: Request, q: Optional[str] = None, + n: Optional[int] = 5, client: Optional[str] = None, user_agent: Optional[str] = Header(None), referer: Optional[str] = Header(None), @@ -495,7 +496,7 @@ async def chat( result_list = [] for query in inferred_queries: result_list.extend( - await search(query, request=request, n=5, r=False, score_threshold=-5.0, dedupe=False) + await search(query, request=request, n=n, r=False, score_threshold=-5.0, dedupe=False) ) compiled_references = [item.additional["compiled"] for item in result_list]