fix(review): address PR feedback regarding RAG concurrency and config defaults

Signed-off-by: Aryan Patel <aryan.patel7291@gmail.com>

Author: ARYANPATEL-BIT

examples/government_rag/singletask_learning_bench/testalgorithms/basemodel.py

@@ -157,7 +157,7 @@ def get_access_token():
     def preprocess(self, **kwargs):
         print("BaseModel preprocess")
         # input('stop here preprocess')
-        self.rag = GovernmentRAG(base_path=Context.get_parameters("base_path", "/path/ianvs/dataset/gov_rag"), model_name="/home/icyfeather/models/bge-m3", device="cuda", persist_directory="./chroma_db")
+        self.rag = GovernmentRAG(base_path=Context.get_parameters("base_path"), model_name=Context.get_parameters("model_name", "BAAI/bge-large-zh-v1.5"), device="cuda", persist_directory="./chroma_db")
         LOGGER.info("RAG initialized")
 
     def train(self, train_data, valid_data=None, **kwargs):
@@ -174,16 +174,18 @@ def process_query(self, query: str, ground_truth: str, location: str, rag_type:
                 response = self.get_model_response(query)
             else:
                 with self.gpu_lock:
+                    base_path = Context.get_parameters("base_path")
+                    model_name = Context.get_parameters("model_name", "BAAI/bge-large-zh-v1.5")
+                    
                     if rag_type == "[global]":
-                        if self.rag is None:
-                            self.rag = GovernmentRAG(base_path=Context.get_parameters("base_path", "/path/ianvs/dataset/gov_rag"), model_name="/home/icyfeather/models/bge-m3", device="cuda", persist_directory="./chroma_db")
+                        rag = GovernmentRAG(base_path=base_path, model_name=model_name, device="cuda", persist_directory="./chroma_db")
                     elif rag_type == "[local]":
-                        self.rag = GovernmentRAG(base_path=Context.get_parameters("base_path", "/path/ianvs/dataset/gov_rag"), model_name="/home/icyfeather/models/bge-m3", device="cuda", persist_directory="./chroma_db", provinces=[location])
+                        rag = GovernmentRAG(base_path=base_path, model_name=model_name, device="cuda", persist_directory="./chroma_db", provinces=[location])
                     else:  # [other]
-                        all_locations = set(self.all_locations)
-                        self.rag = GovernmentRAG(base_path=Context.get_parameters("base_path", "/path/ianvs/dataset/gov_rag"), model_name="/home/icyfeather/models/bge-m3", device="cuda", persist_directory="./chroma_db", provinces=list(all_locations - set([location])))
+                        all_locations = set(getattr(self, "all_locations", []))
+                        rag = GovernmentRAG(base_path=base_path, model_name=model_name, device="cuda", persist_directory="./chroma_db", provinces=list(all_locations - set([location])))
 
-                    relevant_docs = self.rag.query(query, k=1)
+                    relevant_docs = rag.query(query, k=1)
 
                     # Clear GPU cache after query
                     if torch.cuda.is_available():