bugfix: fix get_extra_readers() return None. (#505)

modelscope · Jun 26, 2024 · f6a658c · f6a658c
1 parent aa0948c
commit f6a658c
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/modelscope_agent/rag/knowledge.py b/modelscope_agent/rag/knowledge.py
@@ -243,11 +243,13 @@ def get_extra_readers(
             )
             return extra_readers
 
-        return {
+        extra_readers.update({
             '.pb': PandasCSVReader(),
             '.html': HTMLTagReader(),
             '.txt': FlatReader()
-        }.update(extra_readers)
+        })
+
+        return extra_readers
 
     def read(self,
              knowledge_source: Union[str, List[str]],