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]