Spaces:

neoguojing
/

rag

Sleeping

guoerjun commited on Jun 25, 2024

Commit

0e15749

1 Parent(s): 61eb5cb

fix

Files changed (1) hide show

retriever.py CHANGED Viewed

@@ -155,7 +155,7 @@ class KnowledgeBaseManager:
         return results
     def get_db_files(self,name):
-        data = self.db_files_map[name]
         return data
     def get_bases(self):

         return results
     def get_db_files(self,name):
+        data = self.db_files_map.get(name)
         return data
     def get_bases(self):