youngkingdom/dify
1
0
Fork 0
mirror of https://github.com/langgenius/dify.git synced 2026-05-03 17:08:03 +08:00
Code Issues Packages Projects Releases Wiki Activity

chore(api): migrate import for db

Browse Source
This commit is contained in:
QuantumGhost
2025-08-31 16:46:05 +08:00
parent e9e7d4f3cd
commit 63c035d8a2
8 changed files with 11 additions and 8 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
api/tests/integration_tests/tasks/test_remove_app_and_related_data_task.py
View File

@ -5,7 +5,8 @@ import pytest
from sqlalchemy import delete
from core.variables.segments import StringSegment
from models import Tenant, db
from extensions.ext_database import db
from models import Tenant
from models.enums import CreatorUserRole
from models.model import App, UploadFile
from models.workflow import WorkflowDraftVariable, WorkflowDraftVariableFile
@ -259,6 +260,7 @@ class TestDeleteDraftVariablesWithOffloadIntegration:
# Create WorkflowDraftVariableFile records
from core.variables.types import SegmentType
var_file1 = WorkflowDraftVariableFile(
tenant_id=tenant.id,
app_id=app.id,

2
api/tests/integration_tests/test_offload.py
View File

@ -3,9 +3,9 @@ import uuid
import pytest
from sqlalchemy.orm import Session, joinedload, selectinload
from extensions.ext_database import db
from libs.datetime_utils import naive_utc_now
from libs.uuid_utils import uuidv7
from models import db
from models.enums import CreatorUserRole
from models.model import UploadFile
from models.workflow import WorkflowNodeExecutionModel, WorkflowNodeExecutionOffload, WorkflowNodeExecutionTriggeredFrom