issue #1118: HYBRID as constant

smclab · Nov 25, 2024 · c446dcd · c446dcd
1 parent d2a6361
commit c446dcd
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/ai-packages/rag-module/app/rag/retriever.py b/ai-packages/rag-module/app/rag/retriever.py
@@ -9,6 +9,7 @@
 
 TOKEN_SIZE = 3.5
 MAX_CONTEXT_WINDOW_PERCENTAGE = 0.85
+HYBRID_RETRIEVE_TYPE = "HYBRID"
 
 
 class OpenSearchRetriever(BaseRetriever):
@@ -67,7 +68,7 @@ def _get_relevant_documents(
         query = query_data.query
         index_name = list(query_data.indexName)
         query_parameters = query_data.queryParameters
-        params = query_parameters if self.retrieve_type == "HYBRID" else None
+        params = query_parameters if self.retrieve_type == HYBRID_RETRIEVE_TYPE else None
 
         documents = []