From d4ed1eb629ab1ab2568b9de9d3a0ed508a444f5e Mon Sep 17 00:00:00 2001 From: Gabriel Almeida Date: Wed, 5 Apr 2023 11:37:46 -0300 Subject: [PATCH] fix: added future langchain functionality --- src/backend/langflow/graph/utils.py | 9 ++++++++- 1 file changed, 8 insertions(+), 1 deletion(-) diff --git a/src/backend/langflow/graph/utils.py b/src/backend/langflow/graph/utils.py index e7dc9e593..879b24471 100644 --- a/src/backend/langflow/graph/utils.py +++ b/src/backend/langflow/graph/utils.py @@ -26,7 +26,14 @@ def load_file(file_name, file_content, accepted_types) -> Any: return json.loads(decoded_string) elif suffix in ["yaml", "yml"]: # Return the yaml content - return yaml.safe_load(decoded_string) + loaded_yaml = yaml.load(decoded_string, Loader=yaml.FullLoader) + try: + from langchain.agents.agent_toolkits.openapi.spec import reduce_openapi_spec + + return reduce_openapi_spec(loaded_yaml) + except ImportError: + return loaded_yaml + elif suffix == "csv": # Load the csv content csv_reader = csv.DictReader(io.StringIO(decoded_string))