Merge branch 'feat/memory-orchestration-be-dev-env' into deploy/dev

api/tests/integration_tests/controllers/console/app/test_chat_message_permissions.py

@@ -33,17 +33,19 @@ class TestChatMessageApiPermissions:
     @pytest.fixture
     def mock_account(self, monkeypatch: pytest.MonkeyPatch):
         """Create a mock Account for testing."""
-        account = Account()
-        account.id = str(uuid.uuid4())
-        account.name = "Test User"
-        account.email = "test@example.com"
+
+        account = Account(
+            name="Test User",
+            email="test@example.com",
+        )
         account.last_active_at = naive_utc_now()
         account.created_at = naive_utc_now()
         account.updated_at = naive_utc_now()
+        account.id = str(uuid.uuid4())
 
-        tenant = Tenant()
+        # Create mock tenant
+        tenant = Tenant(name="Test Tenant")
         tenant.id = str(uuid.uuid4())
-        tenant.name = "Test Tenant"
 
         mock_session_instance = mock.Mock()
 

api/tests/integration_tests/controllers/console/app/test_model_config_permissions.py

@@ -32,17 +32,16 @@ class TestModelConfigResourcePermissions:
     @pytest.fixture
     def mock_account(self, monkeypatch: pytest.MonkeyPatch):
         """Create a mock Account for testing."""
-        account = Account()
+
+        account = Account(name="Test User", email="test@example.com")
         account.id = str(uuid.uuid4())
-        account.name = "Test User"
-        account.email = "test@example.com"
         account.last_active_at = naive_utc_now()
         account.created_at = naive_utc_now()
         account.updated_at = naive_utc_now()
 
-        tenant = Tenant()
+        # Create mock tenant
+        tenant = Tenant(name="Test Tenant")
         tenant.id = str(uuid.uuid4())
-        tenant.name = "Test Tenant"
 
         mock_session_instance = mock.Mock()
 

api/tests/integration_tests/tools/api_tool/test_api_tool.py

@@ -36,7 +36,7 @@ def test_api_tool(setup_http_mock):
         entity=ToolEntity(
             identity=ToolIdentity(provider="", author="", name="", label=I18nObject(en_US="test tool")),
         ),
-        api_bundle=ApiToolBundle(**tool_bundle),
+        api_bundle=ApiToolBundle.model_validate(tool_bundle),
         runtime=ToolRuntime(tenant_id="", credentials={"auth_type": "none"}),
         provider_id="test_tool",
     )

api/tests/test_containers_integration_tests/services/test_account_service.py

@@ -16,6 +16,7 @@ from services.errors.account import (
     AccountPasswordError,
     AccountRegisterError,
     CurrentPasswordIncorrectError,
+    TenantNotFoundError,
 )
 from services.errors.workspace import WorkSpaceNotAllowedCreateError, WorkspacesLimitExceededError
 
@@ -1414,7 +1415,7 @@ class TestTenantService:
         )
 
         # Try to get current tenant (should fail)
-        with pytest.raises(AttributeError):
+        with pytest.raises((AttributeError, TenantNotFoundError)):
             TenantService.get_current_tenant_by_account(account)
 
     def test_switch_tenant_success(self, db_session_with_containers, mock_external_service_dependencies):

api/tests/test_containers_integration_tests/services/test_workflow_service.py

@@ -44,27 +44,26 @@ class TestWorkflowService:
             Account: Created test account instance
         """
         fake = fake or Faker()
-        account = Account()
-        account.id = fake.uuid4()
-        account.email = fake.email()
-        account.name = fake.name()
-        account.avatar_url = fake.url()
-        account.tenant_id = fake.uuid4()
-        account.status = "active"
-        account.type = "normal"
-        account.role = "owner"
-        account.interface_language = "en-US"  # Set interface language for Site creation
+        account = Account(
+            email=fake.email(),
+            name=fake.name(),
+            avatar=fake.url(),
+            status="active",
+            interface_language="en-US",  # Set interface language for Site creation
+        )
         account.created_at = fake.date_time_this_year()
+        account.id = fake.uuid4()
         account.updated_at = account.created_at
 
         # Create a tenant for the account
         from models.account import Tenant
 
-        tenant = Tenant()
-        tenant.id = account.tenant_id
-        tenant.name = f"Test Tenant {fake.company()}"
-        tenant.plan = "basic"
-        tenant.status = "active"
+        tenant = Tenant(
+            name=f"Test Tenant {fake.company()}",
+            plan="basic",
+            status="active",
+        )
+        tenant.id = account.current_tenant_id
         tenant.created_at = fake.date_time_this_year()
         tenant.updated_at = tenant.created_at
 
@@ -91,20 +90,21 @@ class TestWorkflowService:
             App: Created test app instance
         """
         fake = fake or Faker()
-        app = App()
-        app.id = fake.uuid4()
-        app.tenant_id = fake.uuid4()
-        app.name = fake.company()
-        app.description = fake.text()
-        app.mode = AppMode.WORKFLOW
-        app.icon_type = "emoji"
-        app.icon = "🤖"
-        app.icon_background = "#FFEAD5"
-        app.enable_site = True
-        app.enable_api = True
-        app.created_by = fake.uuid4()
+        app = App(
+            id=fake.uuid4(),
+            tenant_id=fake.uuid4(),
+            name=fake.company(),
+            description=fake.text(),
+            mode=AppMode.WORKFLOW,
+            icon_type="emoji",
+            icon="🤖",
+            icon_background="#FFEAD5",
+            enable_site=True,
+            enable_api=True,
+            created_by=fake.uuid4(),
+            workflow_id=None,  # Will be set when workflow is created
+        )
         app.updated_by = app.created_by
-        app.workflow_id = None  # Will be set when workflow is created
 
         from extensions.ext_database import db
 
@@ -126,19 +126,20 @@ class TestWorkflowService:
             Workflow: Created test workflow instance
         """
         fake = fake or Faker()
-        workflow = Workflow()
-        workflow.id = fake.uuid4()
-        workflow.tenant_id = app.tenant_id
-        workflow.app_id = app.id
-        workflow.type = WorkflowType.WORKFLOW.value
-        workflow.version = Workflow.VERSION_DRAFT
-        workflow.graph = json.dumps({"nodes": [], "edges": []})
-        workflow.features = json.dumps({"features": []})
-        # unique_hash is a computed property based on graph and features
-        workflow.created_by = account.id
-        workflow.updated_by = account.id
-        workflow.environment_variables = []
-        workflow.conversation_variables = []
+        workflow = Workflow(
+            id=fake.uuid4(),
+            tenant_id=app.tenant_id,
+            app_id=app.id,
+            type=WorkflowType.WORKFLOW.value,
+            version=Workflow.VERSION_DRAFT,
+            graph=json.dumps({"nodes": [], "edges": []}),
+            features=json.dumps({"features": []}),
+            # unique_hash is a computed property based on graph and features
+            created_by=account.id,
+            updated_by=account.id,
+            environment_variables=[],
+            conversation_variables=[],
+        )
 
         from extensions.ext_database import db
 

api/tests/test_containers_integration_tests/tasks/test_delete_segment_from_index_task.py

@@ -48,11 +48,8 @@ class TestDeleteSegmentFromIndexTask:
             Tenant: Created test tenant instance
         """
         fake = fake or Faker()
-        tenant = Tenant()
+        tenant = Tenant(name=f"Test Tenant {fake.company()}", plan="basic", status="active")
         tenant.id = fake.uuid4()
-        tenant.name = f"Test Tenant {fake.company()}"
-        tenant.plan = "basic"
-        tenant.status = "active"
         tenant.created_at = fake.date_time_this_year()
         tenant.updated_at = tenant.created_at
 
@@ -73,16 +70,14 @@ class TestDeleteSegmentFromIndexTask:
             Account: Created test account instance
         """
         fake = fake or Faker()
-        account = Account()
+        account = Account(
+            name=fake.name(),
+            email=fake.email(),
+            avatar=fake.url(),
+            status="active",
+            interface_language="en-US",
+        )
         account.id = fake.uuid4()
-        account.email = fake.email()
-        account.name = fake.name()
-        account.avatar_url = fake.url()
-        account.tenant_id = tenant.id
-        account.status = "active"
-        account.type = "normal"
-        account.role = "owner"
-        account.interface_language = "en-US"
         account.created_at = fake.date_time_this_year()
         account.updated_at = account.created_at
 

api/tests/test_containers_integration_tests/tasks/test_disable_segments_from_index_task.py

@@ -43,27 +43,30 @@ class TestDisableSegmentsFromIndexTask:
             Account: Created test account instance
         """
         fake = fake or Faker()
-        account = Account()
+        account = Account(
+            email=fake.email(),
+            name=fake.name(),
+            avatar=fake.url(),
+            status="active",
+            interface_language="en-US",
+        )
         account.id = fake.uuid4()
-        account.email = fake.email()
-        account.name = fake.name()
-        account.avatar_url = fake.url()
+        # monkey-patch attributes for test setup
         account.tenant_id = fake.uuid4()
-        account.status = "active"
         account.type = "normal"
         account.role = "owner"
-        account.interface_language = "en-US"
         account.created_at = fake.date_time_this_year()
         account.updated_at = account.created_at
 
         # Create a tenant for the account
         from models.account import Tenant
 
-        tenant = Tenant()
+        tenant = Tenant(
+            name=f"Test Tenant {fake.company()}",
+            plan="basic",
+            status="active",
+        )
         tenant.id = account.tenant_id
-        tenant.name = f"Test Tenant {fake.company()}"
-        tenant.plan = "basic"
-        tenant.status = "active"
         tenant.created_at = fake.date_time_this_year()
         tenant.updated_at = tenant.created_at
 
@@ -91,20 +94,21 @@ class TestDisableSegmentsFromIndexTask:
             Dataset: Created test dataset instance
         """
         fake = fake or Faker()
-        dataset = Dataset()
-        dataset.id = fake.uuid4()
-        dataset.tenant_id = account.tenant_id
-        dataset.name = f"Test Dataset {fake.word()}"
-        dataset.description = fake.text(max_nb_chars=200)
-        dataset.provider = "vendor"
-        dataset.permission = "only_me"
-        dataset.data_source_type = "upload_file"
-        dataset.indexing_technique = "high_quality"
-        dataset.created_by = account.id
-        dataset.updated_by = account.id
-        dataset.embedding_model = "text-embedding-ada-002"
-        dataset.embedding_model_provider = "openai"
-        dataset.built_in_field_enabled = False
+        dataset = Dataset(
+            id=fake.uuid4(),
+            tenant_id=account.tenant_id,
+            name=f"Test Dataset {fake.word()}",
+            description=fake.text(max_nb_chars=200),
+            provider="vendor",
+            permission="only_me",
+            data_source_type="upload_file",
+            indexing_technique="high_quality",
+            created_by=account.id,
+            updated_by=account.id,
+            embedding_model="text-embedding-ada-002",
+            embedding_model_provider="openai",
+            built_in_field_enabled=False,
+        )
 
         from extensions.ext_database import db
 
@@ -128,6 +132,7 @@ class TestDisableSegmentsFromIndexTask:
         """
         fake = fake or Faker()
         document = DatasetDocument()
+
         document.id = fake.uuid4()
         document.tenant_id = dataset.tenant_id
         document.dataset_id = dataset.id
@@ -153,7 +158,6 @@ class TestDisableSegmentsFromIndexTask:
         document.archived = False
         document.doc_form = "text_model"  # Use text_model form for testing
         document.doc_language = "en"
-
         from extensions.ext_database import db
 
         db.session.add(document)

api/tests/test_containers_integration_tests/tasks/test_mail_invite_member_task.py

@@ -96,9 +96,9 @@ class TestMailInviteMemberTask:
             password=fake.password(),
             interface_language="en-US",
             status=AccountStatus.ACTIVE.value,
-            created_at=datetime.now(UTC),
-            updated_at=datetime.now(UTC),
         )
+        account.created_at = datetime.now(UTC)
+        account.updated_at = datetime.now(UTC)
         db_session_with_containers.add(account)
         db_session_with_containers.commit()
         db_session_with_containers.refresh(account)
@@ -106,9 +106,9 @@ class TestMailInviteMemberTask:
         # Create tenant
         tenant = Tenant(
             name=fake.company(),
-            created_at=datetime.now(UTC),
-            updated_at=datetime.now(UTC),
         )
+        tenant.created_at = datetime.now(UTC)
+        tenant.updated_at = datetime.now(UTC)
         db_session_with_containers.add(tenant)
         db_session_with_containers.commit()
         db_session_with_containers.refresh(tenant)
@@ -118,8 +118,8 @@ class TestMailInviteMemberTask:
             tenant_id=tenant.id,
             account_id=account.id,
             role=TenantAccountRole.OWNER.value,
-            created_at=datetime.now(UTC),
         )
+        tenant_join.created_at = datetime.now(UTC)
         db_session_with_containers.add(tenant_join)
         db_session_with_containers.commit()
 
@@ -164,9 +164,10 @@ class TestMailInviteMemberTask:
             password="",
             interface_language="en-US",
             status=AccountStatus.PENDING.value,
-            created_at=datetime.now(UTC),
-            updated_at=datetime.now(UTC),
         )
+
+        account.created_at = datetime.now(UTC)
+        account.updated_at = datetime.now(UTC)
         db_session_with_containers.add(account)
         db_session_with_containers.commit()
         db_session_with_containers.refresh(account)
@@ -176,8 +177,8 @@ class TestMailInviteMemberTask:
             tenant_id=tenant.id,
             account_id=account.id,
             role=TenantAccountRole.NORMAL.value,
-            created_at=datetime.now(UTC),
         )
+        tenant_join.created_at = datetime.now(UTC)
         db_session_with_containers.add(tenant_join)
         db_session_with_containers.commit()
 

api/tests/unit_tests/core/rag/datasource/vdb/milvus/test_milvus.py

@@ -11,8 +11,8 @@ def test_default_value():
         config = valid_config.copy()
         del config[key]
         with pytest.raises(ValidationError) as e:
-            MilvusConfig(**config)
+            MilvusConfig.model_validate(config)
         assert e.value.errors()[0]["msg"] == f"Value error, config MILVUS_{key.upper()} is required"
 
-    config = MilvusConfig(**valid_config)
+    config = MilvusConfig.model_validate(valid_config)
     assert config.database == "default"

api/tests/unit_tests/core/workflow/nodes/test_list_operator.py

@@ -35,7 +35,7 @@ def list_operator_node():
         "extract_by": ExtractConfig(enabled=False, serial="1"),
         "title": "Test Title",
     }
-    node_data = ListOperatorNodeData(**config)
+    node_data = ListOperatorNodeData.model_validate(config)
     node_config = {
         "id": "test_node_id",
         "data": node_data.model_dump(),

api/tests/unit_tests/core/workflow/nodes/test_question_classifier_node.py

@@ -17,7 +17,7 @@ def test_init_question_classifier_node_data():
         "vision": {"enabled": True, "configs": {"variable_selector": ["image"], "detail": "low"}},
     }
 
-    node_data = QuestionClassifierNodeData(**data)
+    node_data = QuestionClassifierNodeData.model_validate(data)
 
     assert node_data.query_variable_selector == ["id", "name"]
     assert node_data.model.provider == "openai"
@@ -49,7 +49,7 @@ def test_init_question_classifier_node_data_without_vision_config():
         },
     }
 
-    node_data = QuestionClassifierNodeData(**data)
+    node_data = QuestionClassifierNodeData.model_validate(data)
 
     assert node_data.query_variable_selector == ["id", "name"]
     assert node_data.model.provider == "openai"

api/tests/unit_tests/core/workflow/test_system_variable.py

@@ -46,7 +46,7 @@ class TestSystemVariableSerialization:
     def test_basic_deserialization(self):
         """Test successful deserialization from JSON structure with all fields correctly mapped."""
         # Test with complete data
-        system_var = SystemVariable(**COMPLETE_VALID_DATA)
+        system_var = SystemVariable.model_validate(COMPLETE_VALID_DATA)
 
         # Verify all fields are correctly mapped
         assert system_var.user_id == COMPLETE_VALID_DATA["user_id"]
@@ -59,7 +59,7 @@ class TestSystemVariableSerialization:
         assert system_var.files == []
 
         # Test with minimal data (only required fields)
-        minimal_var = SystemVariable(**VALID_BASE_DATA)
+        minimal_var = SystemVariable.model_validate(VALID_BASE_DATA)
         assert minimal_var.user_id == VALID_BASE_DATA["user_id"]
         assert minimal_var.app_id == VALID_BASE_DATA["app_id"]
         assert minimal_var.workflow_id == VALID_BASE_DATA["workflow_id"]
@@ -75,12 +75,12 @@ class TestSystemVariableSerialization:
 
         # Test workflow_run_id only (preferred alias)
         data_run_id = {**VALID_BASE_DATA, "workflow_run_id": workflow_id}
-        system_var1 = SystemVariable(**data_run_id)
+        system_var1 = SystemVariable.model_validate(data_run_id)
         assert system_var1.workflow_execution_id == workflow_id
 
         # Test workflow_execution_id only (direct field name)
         data_execution_id = {**VALID_BASE_DATA, "workflow_execution_id": workflow_id}
-        system_var2 = SystemVariable(**data_execution_id)
+        system_var2 = SystemVariable.model_validate(data_execution_id)
         assert system_var2.workflow_execution_id == workflow_id
 
         # Test both present - workflow_run_id should take precedence
@@ -89,17 +89,17 @@ class TestSystemVariableSerialization:
             "workflow_execution_id": "should-be-ignored",
             "workflow_run_id": workflow_id,
         }
-        system_var3 = SystemVariable(**data_both)
+        system_var3 = SystemVariable.model_validate(data_both)
         assert system_var3.workflow_execution_id == workflow_id
 
         # Test neither present - should be None
-        system_var4 = SystemVariable(**VALID_BASE_DATA)
+        system_var4 = SystemVariable.model_validate(VALID_BASE_DATA)
         assert system_var4.workflow_execution_id is None
 
     def test_serialization_round_trip(self):
         """Test that serialize → deserialize produces the same result with alias handling."""
         # Create original SystemVariable
-        original = SystemVariable(**COMPLETE_VALID_DATA)
+        original = SystemVariable.model_validate(COMPLETE_VALID_DATA)
 
         # Serialize to dict
         serialized = original.model_dump(mode="json")
@@ -110,7 +110,7 @@ class TestSystemVariableSerialization:
         assert serialized["workflow_run_id"] == COMPLETE_VALID_DATA["workflow_run_id"]
 
         # Deserialize back
-        deserialized = SystemVariable(**serialized)
+        deserialized = SystemVariable.model_validate(serialized)
 
         # Verify all fields match after round-trip
         assert deserialized.user_id == original.user_id
@@ -125,7 +125,7 @@ class TestSystemVariableSerialization:
     def test_json_round_trip(self):
         """Test JSON serialization/deserialization consistency with proper structure."""
         # Create original SystemVariable
-        original = SystemVariable(**COMPLETE_VALID_DATA)
+        original = SystemVariable.model_validate(COMPLETE_VALID_DATA)
 
         # Serialize to JSON string
         json_str = original.model_dump_json()
@@ -137,7 +137,7 @@ class TestSystemVariableSerialization:
         assert json_data["workflow_run_id"] == COMPLETE_VALID_DATA["workflow_run_id"]
 
         # Deserialize from JSON data
-        deserialized = SystemVariable(**json_data)
+        deserialized = SystemVariable.model_validate(json_data)
 
         # Verify key fields match after JSON round-trip
         assert deserialized.workflow_execution_id == original.workflow_execution_id
@@ -149,13 +149,13 @@ class TestSystemVariableSerialization:
         """Test deserialization with File objects in the files field - SystemVariable specific logic."""
         # Test with empty files list
         data_empty = {**VALID_BASE_DATA, "files": []}
-        system_var_empty = SystemVariable(**data_empty)
+        system_var_empty = SystemVariable.model_validate(data_empty)
         assert system_var_empty.files == []
 
         # Test with single File object
         test_file = create_test_file()
         data_single = {**VALID_BASE_DATA, "files": [test_file]}
-        system_var_single = SystemVariable(**data_single)
+        system_var_single = SystemVariable.model_validate(data_single)
         assert len(system_var_single.files) == 1
         assert system_var_single.files[0].filename == "test.txt"
         assert system_var_single.files[0].tenant_id == "test-tenant-id"
@@ -179,14 +179,14 @@ class TestSystemVariableSerialization:
         )
 
         data_multiple = {**VALID_BASE_DATA, "files": [file1, file2]}
-        system_var_multiple = SystemVariable(**data_multiple)
+        system_var_multiple = SystemVariable.model_validate(data_multiple)
         assert len(system_var_multiple.files) == 2
         assert system_var_multiple.files[0].filename == "doc1.txt"
         assert system_var_multiple.files[1].filename == "image.jpg"
 
         # Verify files field serialization/deserialization
         serialized = system_var_multiple.model_dump(mode="json")
-        deserialized = SystemVariable(**serialized)
+        deserialized = SystemVariable.model_validate(serialized)
         assert len(deserialized.files) == 2
         assert deserialized.files[0].filename == "doc1.txt"
         assert deserialized.files[1].filename == "image.jpg"
@@ -197,7 +197,7 @@ class TestSystemVariableSerialization:
 
         # Create with workflow_run_id (alias)
         data_with_alias = {**VALID_BASE_DATA, "workflow_run_id": workflow_id}
-        system_var = SystemVariable(**data_with_alias)
+        system_var = SystemVariable.model_validate(data_with_alias)
 
         # Serialize and verify alias is used
         serialized = system_var.model_dump()
@@ -205,7 +205,7 @@ class TestSystemVariableSerialization:
         assert "workflow_execution_id" not in serialized
 
         # Deserialize and verify field mapping
-        deserialized = SystemVariable(**serialized)
+        deserialized = SystemVariable.model_validate(serialized)
         assert deserialized.workflow_execution_id == workflow_id
 
         # Test JSON serialization path
@@ -213,7 +213,7 @@ class TestSystemVariableSerialization:
         assert json_serialized["workflow_run_id"] == workflow_id
         assert "workflow_execution_id" not in json_serialized
 
-        json_deserialized = SystemVariable(**json_serialized)
+        json_deserialized = SystemVariable.model_validate(json_serialized)
         assert json_deserialized.workflow_execution_id == workflow_id
 
     def test_model_validator_serialization_logic(self):
@@ -222,7 +222,7 @@ class TestSystemVariableSerialization:
 
         # Test direct instantiation with workflow_execution_id (should work)
         data1 = {**VALID_BASE_DATA, "workflow_execution_id": workflow_id}
-        system_var1 = SystemVariable(**data1)
+        system_var1 = SystemVariable.model_validate(data1)
         assert system_var1.workflow_execution_id == workflow_id
 
         # Test serialization of the above (should use alias)
@@ -236,7 +236,7 @@ class TestSystemVariableSerialization:
             "workflow_execution_id": "should-be-removed",
             "workflow_run_id": workflow_id,
         }
-        system_var2 = SystemVariable(**data2)
+        system_var2 = SystemVariable.model_validate(data2)
         assert system_var2.workflow_execution_id == workflow_id
 
         # Verify serialization consistency

api/tests/unit_tests/libs/test_helper.py

@@ -11,7 +11,7 @@ class TestExtractTenantId:
     def test_extract_tenant_id_from_account_with_tenant(self):
         """Test extracting tenant_id from Account with current_tenant_id."""
         # Create a mock Account object
-        account = Account()
+        account = Account(name="test", email="test@example.com")
         # Mock the current_tenant_id property
         account._current_tenant = type("MockTenant", (), {"id": "account-tenant-123"})()
 
@@ -21,7 +21,7 @@ class TestExtractTenantId:
     def test_extract_tenant_id_from_account_without_tenant(self):
         """Test extracting tenant_id from Account without current_tenant_id."""
         # Create a mock Account object
-        account = Account()
+        account = Account(name="test", email="test@example.com")
         account._current_tenant = None
 
         tenant_id = extract_tenant_id(account)

api/tests/unit_tests/repositories/workflow_node_execution/test_sqlalchemy_repository.py

@@ -59,12 +59,11 @@ def session():
 @pytest.fixture
 def mock_user():
     """Create a user instance for testing."""
-    user = Account()
+    user = Account(name="test", email="test@example.com")
     user.id = "test-user-id"
 
-    tenant = Tenant()
+    tenant = Tenant(name="Test Workspace")
     tenant.id = "test-tenant"
-    tenant.name = "Test Workspace"
     user._current_tenant = MagicMock()
     user._current_tenant.id = "test-tenant"
 

api/tests/unit_tests/services/test_metadata_bug_complete.py

@@ -118,7 +118,7 @@ class TestMetadataBugCompleteValidation:
 
             # But would crash when trying to create MetadataArgs
             with pytest.raises((ValueError, TypeError)):
-                MetadataArgs(**args)
+                MetadataArgs.model_validate(args)
 
     def test_7_end_to_end_validation_layers(self):
         """Test all validation layers work together correctly."""
@@ -131,7 +131,7 @@ class TestMetadataBugCompleteValidation:
         valid_data = {"type": "string", "name": "test_metadata"}
 
         # Should create valid Pydantic object
-        metadata_args = MetadataArgs(**valid_data)
+        metadata_args = MetadataArgs.model_validate(valid_data)
         assert metadata_args.type == "string"
         assert metadata_args.name == "test_metadata"
 

api/tests/unit_tests/services/test_metadata_nullable_bug.py

@@ -76,7 +76,7 @@ class TestMetadataNullableBug:
             # Step 2: Try to create MetadataArgs with None values
             # This should fail at Pydantic validation level
             with pytest.raises((ValueError, TypeError)):
-                metadata_args = MetadataArgs(**args)
+                metadata_args = MetadataArgs.model_validate(args)
 
         # Step 3: If we bypass Pydantic (simulating the bug scenario)
         # Move this outside the request context to avoid Flask-Login issues

api/tests/unit_tests/services/workflow/test_workflow_draft_variable_service.py

@@ -47,7 +47,8 @@ class TestDraftVariableSaver:
 
     def test__should_variable_be_visible(self):
         mock_session = MagicMock(spec=Session)
-        mock_user = Account(id=str(uuid.uuid4()))
+        mock_user = Account(name="test", email="test@example.com")
+        mock_user.id = str(uuid.uuid4())
         test_app_id = self._get_test_app_id()
         saver = DraftVariableSaver(
             session=mock_session,