[Fix] Enhanced security for OpenAPI and JSON Loader Integration (#1122)

2024-01-08 14:37:03 +09:00
parent 1878f8d4fc
commit f7696d1dc1
2 changed files with 2 additions and 2 deletions
--- a/embedchain/loaders/openapi.py
+++ b/embedchain/loaders/openapi.py
@@ -32,7 +32,7 @@ class OpenAPILoader(BaseLoader):
        file_path = content
        data_content = []
        with OpenAPILoader._get_file_content(content=content) as file:
-            yaml_data = yaml.load(file, Loader=yaml.Loader)
+            yaml_data = yaml.load(file, Loader=yaml.SafeLoader)
            for i, (key, value) in enumerate(yaml_data.items()):
                string_data = f"{key}: {value}"
                meta_data = {"url": file_path, "row": i + 1}