make NUM_RETURNED_HITS configurable (#32)

Signed-off-by: Alex Co <alex.tuan@mindvalley.com>
mindvalley · May 24, 2024 · 396d0cb · 396d0cb
1 parent 7e87921
commit 396d0cb
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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