fix: jsondata loader on library (run-llama#539)

Tachikoma000 · Sep 27, 2023 · 6a14dd7 · 6a14dd7
1 parent 6bf1153
commit 6a14dd7
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 1 deletion.
diff --git a/llama_hub/jsondata/base.py b/llama_hub/jsondata/base.py
@@ -27,7 +27,7 @@ def _depth_first_yield(json_data: Dict, path: List[str]) -> Generator[str, None,
         yield " ".join(path)
 
 
-class JSONDataReader(BaseReader):
+class JsonDataReader(BaseReader):
     """Json Data reader.
 
     Reads in Json Data.
@@ -52,3 +52,7 @@ def load_data(self, input_data: Union[str, Dict]) -> List[Document]:
         lines = json_output.split("\n")
         useful_lines = [line for line in lines if not re.match(r"^[{}\[\],]*$", line)]
         return [Document(text="\n".join(useful_lines))]
+
+
+# legacy
+JSONDataReader = JsonDataReader
diff --git a/llama_hub/library.json b/llama_hub/library.json
@@ -105,6 +105,10 @@
     "id": "file/json",
     "author": "yisding"
   },
+  "JsonDataReader": {
+    "id": "jsondata",
+    "author": "jerryjliu"
+  },
   "MarkdownReader": {
     "id": "file/markdown",
     "author": "hursh-desai"