ref: Make load_flow_from_json async (#5057)

* Make load_flow_from_json async

* Add back sync load_flow_from_json for backward compatibility
This commit is contained in:
Christophe Bornet 2024-12-11 18:53:33 +01:00 • committed by GitHub
commit abd3321677
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
6 changed files with 74 additions and 23 deletions

View file

@ -1,8 +1,6 @@
import asyncio
from langflow.graph import Graph
from langflow.initial_setup.setup import load_starter_projects
from langflow.load import load_flow_from_json
from langflow.load import aload_flow_from_json
# TODO: UPDATE BASIC EXAMPLE
# def test_load_flow_from_json():
@ -24,6 +22,6 @@ async def test_load_flow_from_json_object():
"""Test loading a flow from a json file and applying tweaks."""
result = await load_starter_projects()
project = result[0][1]
loaded = await asyncio.to_thread(load_flow_from_json, project)
loaded = await aload_flow_from_json(project)
assert loaded is not None
assert isinstance(loaded, Graph)