bug fix

Author: jerrychen1990

snippets/utils.py

@@ -210,7 +210,14 @@ def _read2list(file_path, **kwargs):
     rs = []
     for f in file_path:
         logger.info(f"reading file_path={f}")
-        rs.extend(_read2list(f, **kwargs))
+        tmp = _read2list(f, **kwargs)
+        if isinstance(tmp, list):
+            rs.extend(tmp)
+        else:
+            rs.append(tmp)
+    if len(file_path) == 1 and len(rs)==1 and file_path[0].endswith(".json"):
+        rs = rs[0]            
+
     return rs
 
 

tests/test_utils.py

@@ -67,5 +67,7 @@ def test_load(self):
         data = load("data/sample.*")
         print(len(data))
         self.assertEquals(len(data), 6)
+        data = load("/Users/chenhao/workspace/XAgents/knowledge_base/test_service/config.json")
+        print(data)