feat(rag): include references

Fixes #20

Author: sanjayankur31

data-sources/neuroml/nml.py

@@ -56,18 +56,16 @@ def setup(self):
         # extract model name
         if self.embedding_model.lower().startswith("huggingface:"):
             # strip suffix/prefix
-            self.embedding_model = (
-                self.embedding_model.replace("huggingface:", "")
-                .replace(":cheapest", "")
-                .replace(":fastest", "")
-            )
+            self.embedding_model = self.embedding_model.replace("huggingface:", "")
             # strip collection name
             splits = self.embedding_model.split("/")
             self.embedding_model = "".join(splits[1:])
         elif self.embedding_model.lower().startswith("ollama:"):
             # strip prefix
             self.embedding_model = self.embedding_model.replace("ollama:", "")
 
+        self.embedding_model.replace(":cheapest", "").replace(":fastest", "")
+
     def create(self):
         """Create the vector store"""
         assert self.embeddings

rag_pkg/gen_rag/api/chat.py

@@ -19,6 +19,7 @@ async def query(request: Request, query: str):
     try:
         result = await rag.run_graph_invoke(query)
     except Exception as e:
+        print(e)
         result = HTTPException(status_code=500, detail=str(e))
 
     return {"result": result}