set default similarity_top_k to 5

safevideo · Nov 8, 2023 · bc2ee49 · bc2ee49
1 parent 159afe4
commit bc2ee49
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/autollm/auto/query_engine.py b/autollm/auto/query_engine.py
@@ -44,7 +44,7 @@ def create_query_engine(
         "vector_store_type": "LanceDBVectorStore"
     } if vector_store_params is None else vector_store_params
     service_context_params = {} if service_context_params is None else service_context_params
-    query_engine_params = {} if query_engine_params is None else query_engine_params
+    query_engine_params = {"similarity_top_k": 5} if query_engine_params is None else query_engine_params
 
     llm = AutoLiteLLM.from_defaults(**llm_params)
     service_context = AutoServiceContext.from_defaults(