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

fix: Fix flow addition to session and handle IntegrityError for orphaned flows (#4799)

Browse source 
* Handle IntegrityError specifically when assigning orphaned flows to superuser

* Add missing session.add(flow) to ensure flow is added to the session before commit
This commit is contained in:
Gabriel Luiz Freitas Almeida 2024-11-22 19:05:28 -03:00 • committed by GitHub
commit 4620669129
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
2 changed files with 3 additions and 4 deletions
Download patch file Download diff file Expand all files Collapse all files

1
src/backend/base/langflow/services/database/service.py
View file

@ -187,6 +187,7 @@ class DatabaseService(Service):
flow.user_id = superuser.id flow.user_id = superuser.id
flow.name = self._generate_unique_flow_name(flow.name, existing_names) flow.name = self._generate_unique_flow_name(flow.name, existing_names)
existing_names.add(flow.name) existing_names.add(flow.name)
session.add(flow)
# Commit changes # Commit changes
await session.commit() await session.commit()

6
src/backend/base/langflow/services/utils.py
View file

@ -243,9 +243,7 @@ async def initialize_services(*, fix_migration: bool = False) -> None:
await setup_superuser(settings_service, session) await setup_superuser(settings_service, session)
try: try:
await get_db_service().assign_orphaned_flows_to_superuser() await get_db_service().assign_orphaned_flows_to_superuser()
except Exception as exc: except sqlalchemy_exc.IntegrityError as exc:
msg = "Error assigning orphaned flows to the superuser" logger.warning(f"Error assigning orphaned flows to the superuser: {exc!s}")
logger.exception(msg)
raise RuntimeError(msg) from exc
await clean_transactions(settings_service, session) await clean_transactions(settings_service, session)
await clean_vertex_builds(settings_service, session) await clean_vertex_builds(settings_service, session)