feat(rag): use descriptions from vector store conf when available

Author: sanjayankur31

rag_pkg/gen_rag/rag.py

@@ -164,12 +164,17 @@ def _classify_question_domain(self, state: RAGState) -> dict:
         messages = state.messages
         messages.append(HumanMessage(content=state.query))
 
-        domains = self.stores.domains
+        domain_info = self.stores.vs_config.domains
 
         domain_str = ""
 
-        for d in domains:
-            domain_str += f"\n- {d}: if the question is about {d}"
+        for d, info in domain_info:
+            desc = info.get("description", None)
+            if not desc:
+                desc = f"if the question is about {d}"
+            else:
+                desc = f"if the question is about {desc}"
+            domain_str += f"\n- {d}: {desc}"
 
         system_prompt = dedent("""
             You are an expert query classifier.

rag_pkg/gen_rag/stores.py

@@ -130,7 +130,7 @@ def reset_k(self):
 
     def load_all_stores(self):
         """Load all vector stores"""
-        for domain_name in self.domains():
+        for domain_name in self.domains:
             self.load(domain_name)
 
     @property