AI/dify
1
0
Fork 0
mirror of https://git.mirrors.martin98.com/https://github.com/langgenius/dify.git synced 2025-08-18 07:15:52 +08:00
Code Issues Packages Projects Releases Wiki Activity

fix: Validates session factory type in repository (#18497)

Browse Source 
Signed-off-by: -LAN- <laipz8200@outlook.com>
Co-authored-by: Copilot <175728472+Copilot@users.noreply.github.com>
This commit is contained in:
-LAN- 2025-04-22 17:04:06 +09:00 committed by GitHub
parent 21412a8c55
commit 6b7dfee88b
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
3 changed files with 7 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
api/repositories/workflow_node_execution/sqlalchemy_repository.py
View File

@ -37,8 +37,12 @@ class SQLAlchemyWorkflowNodeExecutionRepository:
# If an engine is provided, create a sessionmaker from it # If an engine is provided, create a sessionmaker from it
if isinstance(session_factory, Engine): if isinstance(session_factory, Engine):
self._session_factory = sessionmaker(bind=session_factory, expire_on_commit=False) self._session_factory = sessionmaker(bind=session_factory, expire_on_commit=False)
else: elif isinstance(session_factory, sessionmaker):
self._session_factory = session_factory self._session_factory = session_factory
else:
raise ValueError(
f"Invalid session_factory type {type(session_factory).__name__}; expected sessionmaker or Engine"
)
self._tenant_id = tenant_id self._tenant_id = tenant_id
self._app_id = app_id self._app_id = app_id

2
api/services/workflow_run_service.py
View File

@ -133,7 +133,7 @@ class WorkflowRunService:
params={ params={
"tenant_id": app_model.tenant_id, "tenant_id": app_model.tenant_id,
"app_id": app_model.id, "app_id": app_model.id,
"session_factory": db.session.get_bind, "session_factory": db.session.get_bind(),
} }
) )

2
api/tasks/remove_app_and_related_data_task.py
View File

@ -193,7 +193,7 @@ def _delete_app_workflow_node_executions(tenant_id: str, app_id: str):
params={ params={
"tenant_id": tenant_id, "tenant_id": tenant_id,
"app_id": app_id, "app_id": app_id,
"session_factory": db.session.get_bind, "session_factory": db.session.get_bind(),
} }
) )