From 396d0cb7a674f0a34fe2b7a462b2a430e3b2fa63 Mon Sep 17 00:00:00 2001 From: Alex Co Date: Fri, 24 May 2024 14:04:45 +0800 Subject: [PATCH] make NUM_RETURNED_HITS configurable (#32) Signed-off-by: Alex Co --- backend/danswer/configs/chat_configs.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/backend/danswer/configs/chat_configs.py b/backend/danswer/configs/chat_configs.py index 42553bfead3..66cb8adceb2 100644 --- a/backend/danswer/configs/chat_configs.py +++ b/backend/danswer/configs/chat_configs.py @@ -4,7 +4,7 @@ PROMPTS_YAML = "./danswer/chat/prompts.yaml" PERSONAS_YAML = "./danswer/chat/personas.yaml" -NUM_RETURNED_HITS = 50 +NUM_RETURNED_HITS = os.environ.get("TOOL_SEARCH_NUM_RETURNED_HITS") or 50 NUM_RERANKED_RESULTS = 15 # May be less depending on model