Update index_retriever.py

index_retriever.py

@@ -65,24 +65,26 @@ def rerank_nodes(query, nodes, reranker, top_k=25, min_score_threshold=0.5):
         log_message(f"Ошибка переранжировки: {str(e)}")
         return nodes[:top_k]
 
-def create_query_engine(vector_index):
+# MODIFIED: Update create_query_engine function signature
+def create_query_engine(vector_index, vector_top_k=50, bm25_top_k=50, 
+                       similarity_cutoff=0.55, hybrid_top_k=100):
     try:
         from config import CUSTOM_PROMPT
         
         bm25_retriever = BM25Retriever.from_defaults(
             docstore=vector_index.docstore, 
-            similarity_top_k=60 
+            similarity_top_k=bm25_top_k  # NOW PARAMETERIZED
         )
         
         vector_retriever = VectorIndexRetriever(
             index=vector_index, 
-            similarity_top_k=60,
-            similarity_cutoff=0.45
+            similarity_top_k=vector_top_k,  # NOW PARAMETERIZED
+            similarity_cutoff=similarity_cutoff  # NOW PARAMETERIZED
         )
         
         hybrid_retriever = QueryFusionRetriever(
             [vector_retriever, bm25_retriever],
-            similarity_top_k=120,
+            similarity_top_k=hybrid_top_k,  # NOW PARAMETERIZED
             num_queries=1
         )
         
@@ -97,7 +99,9 @@ def create_query_engine(vector_index):
             response_synthesizer=response_synthesizer
         )
         
-        log_message("Query engine успешно создан")
+        log_message(f"Query engine created: vector_top_k={vector_top_k}, "
+                   f"bm25_top_k={bm25_top_k}, similarity_cutoff={similarity_cutoff}, "
+                   f"hybrid_top_k={hybrid_top_k}")
         return query_engine
         
     except Exception as e: