Accepting custom loading of index param.

Author: sriroopar

ols/src/rag_index/index_loader.py

@@ -56,13 +56,17 @@ class QueryFusionRetrieverCustom(QueryFusionRetriever):  # pylint: disable=W0612
 
         def __init__(self, **kwargs):
             """Initialize custom query fusion class."""
+            # Extract custom parameters before passing to parent
+            retriever_weights = kwargs.pop("retriever_weights", None)
+            index_configs = kwargs.pop("index_configs", None)
+            retrievers = kwargs.get("retrievers", [])
+
             super().__init__(**kwargs)
 
-            retriever_weights = kwargs.get("retriever_weights", None)
             if not retriever_weights:
-                retriever_weights = [1.0] * len(kwargs["retrievers"])
+                retriever_weights = [1.0] * len(retrievers)
             self._custom_retriever_weights = retriever_weights
-            self._index_configs = kwargs.get("index_configs", None)
+            self._index_configs = index_configs
 
         def _simple_fusion(self, results):
             """Override internal method and apply weighted score."""