jyapayne/langflow
1
0
Fork 0
Code Issues Pull requests Projects Releases 1 Packages Wiki Activity Actions

fix: "Flow" object has no field "last_tested_version" when loading flows from dir (#2221)

Browse source
This commit is contained in:
Nicolò Boschi 2024-06-19 18:15:50 +02:00 • committed by GitHub
commit e3656a6a43
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
3 changed files with 12 additions and 7 deletions
Download patch file Download diff file Expand all files Collapse all files

4
src/backend/base/langflow/initial_setup/setup.py
View file

@ -341,8 +341,8 @@ def load_flows_from_directory():
if existing: if existing:
logger.info(f"Updating existing flow: {flow_id} with endpoint name {flow_endpoint_name}") logger.info(f"Updating existing flow: {flow_id} with endpoint name {flow_endpoint_name}")
for key, value in flow.items(): for key, value in flow.items():
if key == "last_tested_version": if hasattr(existing, key):
continue # flow dict from json and db representation are not 100% the same
setattr(existing, key, value) setattr(existing, key, value)
existing.updated_at = datetime.utcnow() existing.updated_at = datetime.utcnow()
existing.user_id = user_id existing.user_id = user_id

3
tests/data/basic_example.json
View file

@ -505,5 +505,6 @@
"y": 0, "y": 0,
"zoom": 0.5 "zoom": 0.5
} }
} },
"last_tested_version": "0.6.2"
} }

8
tests/unit/test_database.py
View file

@ -6,7 +6,7 @@ from fastapi.testclient import TestClient
from sqlmodel import Session from sqlmodel import Session
from langflow.api.v1.schemas import FlowListCreate from langflow.api.v1.schemas import FlowListCreate
from langflow.initial_setup.setup import load_starter_projects from langflow.initial_setup.setup import load_starter_projects, load_flows_from_directory
from langflow.services.database.models.base import orjson_dumps from langflow.services.database.models.base import orjson_dumps
from langflow.services.database.models.flow import Flow, FlowCreate, FlowUpdate from langflow.services.database.models.flow import Flow, FlowCreate, FlowUpdate
from langflow.services.database.utils import session_getter from langflow.services.database.utils import session_getter
@ -271,7 +271,11 @@ def test_read_only_starter_projects(client: TestClient, active_user, logged_in_h
@pytest.mark.load_flows @pytest.mark.load_flows
def test_load_flows(client: TestClient, load_flows_dir): def test_load_flows(client: TestClient, load_flows_dir):
client.get("/api/v1/auto_login") response = client.get("api/v1/flows/c54f9130-f2fa-4a3e-b22a-3856d946351b")
assert response.status_code == 200
assert response.json()["name"] == "BasicExample"
# re-run to ensure updates work well
load_flows_from_directory()
response = client.get("api/v1/flows/c54f9130-f2fa-4a3e-b22a-3856d946351b") response = client.get("api/v1/flows/c54f9130-f2fa-4a3e-b22a-3856d946351b")
assert response.status_code == 200 assert response.status_code == 200
assert response.json()["name"] == "BasicExample" assert response.json()["name"] == "BasicExample"