Skip to content

test: migrate dataset service dataset mock tests to testcontainers#35194

Merged
asukaminato0721 merged 4 commits intolanggenius:mainfrom
jamesrayammons:test/migrate-dataset-service-dataset-to-testcontainers
Apr 14, 2026
Merged

test: migrate dataset service dataset mock tests to testcontainers#35194
asukaminato0721 merged 4 commits intolanggenius:mainfrom
jamesrayammons:test/migrate-dataset-service-dataset-to-testcontainers

Conversation

@jamesrayammons
Copy link
Copy Markdown
Contributor

@jamesrayammons jamesrayammons commented Apr 14, 2026

Summary

  • Add Testcontainers integration coverage for SQL-backed dataset permission, lifecycle, collection-binding, and partial-member behaviors
  • Remove overlapping mock-based unit coverage from api/tests/unit_tests/services/test_dataset_service_dataset.py
  • Rely on existing Testcontainers suites for retrieval and document indexing coverage already migrated elsewhere

Part of #32454

Test plan

  • Verify migrated SQL cases now use real Testcontainers database fixtures
  • Keep remaining non-SQL unit coverage in place
  • Confirm the migrated files parse cleanly after the split

@dosubot dosubot Bot added the size:XS This PR changes 0-9 lines, ignoring generated files. label Apr 14, 2026
@dosubot dosubot Bot added the lgtm This PR has been approved by a maintainer label Apr 14, 2026
@github-actions
Copy link
Copy Markdown
Contributor

github-actions Bot commented Apr 14, 2026

Pyrefly Diff

base → PR 
--- /tmp/pyrefly_base.txt	2026-04-14 17:57:06.021808334 +0000
+++ /tmp/pyrefly_pr.txt	2026-04-14 17:56:55.351718816 +0000
@@ -1165,6 +1165,12 @@
 ERROR Argument `Literal['normal']` is not assignable to parameter `status` with type `SQLCoreOperations[TenantStatus] | TenantStatus` in function `models.account.Tenant.__init__` [bad-argument-type]
   --> tests/test_containers_integration_tests/services/test_dataset_service_get_segments.py:44:62
 ERROR Argument `Literal['active']` is not assignable to parameter `status` with type `AccountStatus | SQLCoreOperations[AccountStatus]` in function `models.account.Account.__init__` [bad-argument-type]
+  --> tests/test_containers_integration_tests/services/test_dataset_service_permissions.py:36:20
+ERROR Argument `Literal['normal']` is not assignable to parameter `status` with type `SQLCoreOperations[TenantStatus] | TenantStatus` in function `models.account.Tenant.__init__` [bad-argument-type]
+  --> tests/test_containers_integration_tests/services/test_dataset_service_permissions.py:38:58
+ERROR Argument `Literal['active']` is not assignable to parameter `status` with type `AccountStatus | SQLCoreOperations[AccountStatus]` in function `models.account.Account.__init__` [bad-argument-type]
+  --> tests/test_containers_integration_tests/services/test_dataset_service_permissions.py:65:20
+ERROR Argument `Literal['active']` is not assignable to parameter `status` with type `AccountStatus | SQLCoreOperations[AccountStatus]` in function `models.account.Account.__init__` [bad-argument-type]
   --> tests/test_containers_integration_tests/services/test_dataset_service_retrieval.py:45:20
 ERROR Argument `Literal['normal']` is not assignable to parameter `status` with type `SQLCoreOperations[TenantStatus] | TenantStatus` in function `models.account.Tenant.__init__` [bad-argument-type]
   --> tests/test_containers_integration_tests/services/test_dataset_service_retrieval.py:49:20
@@ -6114,43 +6120,35 @@
 ERROR Class member `FixedDateTime.now` overrides parent class `datetime` in an inconsistent manner [bad-override]
    --> tests/unit_tests/services/test_clear_free_plan_tenant_expired_logs.py:460:13
 ERROR Argument `SimpleNamespace` is not assignable to parameter `account` with type `Account` in function `services.dataset_service.DatasetService.create_empty_dataset` [bad-argument-type]
-   --> tests/unit_tests/services/test_dataset_service_dataset.py:310:93
+   --> tests/unit_tests/services/test_dataset_service_dataset.py:184:93
 ERROR Argument `SimpleNamespace` is not assignable to parameter `account` with type `Account` in function `services.dataset_service.DatasetService.create_empty_dataset` [bad-argument-type]
-   --> tests/unit_tests/services/test_dataset_service_dataset.py:334:25
+   --> tests/unit_tests/services/test_dataset_service_dataset.py:208:25
 ERROR Argument `SimpleNamespace` is not assignable to parameter `account` with type `Account` in function `services.dataset_service.DatasetService.create_empty_dataset` [bad-argument-type]
-   --> tests/unit_tests/services/test_dataset_service_dataset.py:377:25
+   --> tests/unit_tests/services/test_dataset_service_dataset.py:251:25
 ERROR Argument `SimpleNamespace` is not assignable to parameter `dataset` with type `Dataset` in function `services.dataset_service.DatasetService._update_pipeline_knowledge_base_node_data` [bad-argument-type]
-   --> tests/unit_tests/services/test_dataset_service_dataset.py:694:70
+   --> tests/unit_tests/services/test_dataset_service_dataset.py:568:70
 ERROR Argument `SimpleNamespace` is not assignable to parameter `dataset` with type `Dataset` in function `services.dataset_service.DatasetService._update_pipeline_knowledge_base_node_data` [bad-argument-type]
-   --> tests/unit_tests/services/test_dataset_service_dataset.py:704:70
+   --> tests/unit_tests/services/test_dataset_service_dataset.py:578:70
 ERROR Argument `SimpleNamespace` is not assignable to parameter `dataset` with type `Dataset` in function `services.dataset_service.DatasetService._update_pipeline_knowledge_base_node_data` [bad-argument-type]
-   --> tests/unit_tests/services/test_dataset_service_dataset.py:743:70
+   --> tests/unit_tests/services/test_dataset_service_dataset.py:617:70
 ERROR Argument `SimpleNamespace` is not assignable to parameter `dataset` with type `Dataset` in function `services.dataset_service.DatasetService._update_pipeline_knowledge_base_node_data` [bad-argument-type]
-   --> tests/unit_tests/services/test_dataset_service_dataset.py:766:74
+   --> tests/unit_tests/services/test_dataset_service_dataset.py:640:74
 ERROR Object of class `FakeAccount` has no attribute `current_tenant_id` [missing-attribute]
-   --> tests/unit_tests/services/test_dataset_service_dataset.py:833:9
+   --> tests/unit_tests/services/test_dataset_service_dataset.py:707:9
 ERROR Object of class `FakeAccount` has no attribute `current_tenant_id` [missing-attribute]
-   --> tests/unit_tests/services/test_dataset_service_dataset.py:871:9
+   --> tests/unit_tests/services/test_dataset_service_dataset.py:745:9
 ERROR Argument `None` is not assignable to parameter `embedding_model_provider` with type `str` in function `tests.unit_tests.services.dataset_service_test_helpers.DatasetServiceUnitDataFactory.create_dataset_mock` [bad-argument-type]
-   --> tests/unit_tests/services/test_dataset_service_dataset.py:937:38
+   --> tests/unit_tests/services/test_dataset_service_dataset.py:811:38
 ERROR Argument `None` is not assignable to parameter `embedding_model` with type `str` in function `tests.unit_tests.services.dataset_service_test_helpers.DatasetServiceUnitDataFactory.create_dataset_mock` [bad-argument-type]
-   --> tests/unit_tests/services/test_dataset_service_dataset.py:938:29
+   --> tests/unit_tests/services/test_dataset_service_dataset.py:812:29
 ERROR Object of class `FakeAccount` has no attribute `current_tenant_id` [missing-attribute]
-   --> tests/unit_tests/services/test_dataset_service_dataset.py:996:9
+   --> tests/unit_tests/services/test_dataset_service_dataset.py:870:9
 ERROR Object of class `FakeAccount` has no attribute `current_tenant_id` [missing-attribute]
-    --> tests/unit_tests/services/test_dataset_service_dataset.py:1031:9
+   --> tests/unit_tests/services/test_dataset_service_dataset.py:905:9
 ERROR Argument `SimpleNamespace` is not assignable to parameter `user` with type `Account | None` in function `services.dataset_service.DatasetService.check_dataset_operator_permission` [bad-argument-type]
-    --> tests/unit_tests/services/test_dataset_service_dataset.py:1439:67
+    --> tests/unit_tests/services/test_dataset_service_dataset.py:1216:67
 ERROR Argument `SimpleNamespace` is not assignable to parameter `dataset` with type `Dataset | None` in function `services.dataset_service.DatasetService.check_dataset_operator_permission` [bad-argument-type]
-    --> tests/unit_tests/services/test_dataset_service_dataset.py:1442:81
-ERROR Argument `SimpleNamespace` is not assignable to parameter `user` with type `Account | None` in function `services.dataset_service.DatasetService.check_dataset_operator_permission` [bad-argument-type]
-    --> tests/unit_tests/services/test_dataset_service_dataset.py:1455:67
-ERROR Argument `SimpleNamespace` is not assignable to parameter `user` with type `Account | None` in function `services.dataset_service.DatasetService.check_dataset_operator_permission` [bad-argument-type]
-    --> tests/unit_tests/services/test_dataset_service_dataset.py:1468:71
-ERROR Object of class `FakeAccount` has no attribute `current_tenant_id` [missing-attribute]
-    --> tests/unit_tests/services/test_dataset_service_dataset.py:1527:9
-ERROR Object of class `FakeAccount` has no attribute `current_tenant_id` [missing-attribute]
-    --> tests/unit_tests/services/test_dataset_service_dataset.py:1549:9
+    --> tests/unit_tests/services/test_dataset_service_dataset.py:1219:81
 ERROR Argument `SimpleNamespace` is not assignable to parameter `current_user` with type `Account` in function `services.dataset_service.DocumentService.prepare_document_batch_download_zip` [bad-argument-type]
    --> tests/unit_tests/services/test_dataset_service_document.py:340:34
 ERROR Argument `SimpleNamespace` is not assignable to parameter `current_user` with type `Account` in function `services.dataset_service.DocumentService.prepare_document_batch_download_zip` [bad-argument-type]

@asukaminato0721 asukaminato0721 self-requested a review April 14, 2026 18:25
@github-actions
Copy link
Copy Markdown
Contributor

github-actions Bot commented Apr 14, 2026

Pyrefly Diff

base → PR 
--- /tmp/pyrefly_base.txt	2026-04-14 18:26:10.924144614 +0000
+++ /tmp/pyrefly_pr.txt	2026-04-14 18:25:59.637152560 +0000
@@ -714,8 +714,6 @@
    --> tests/test_containers_integration_tests/controllers/service_api/dataset/test_dataset.py:183:29
 ERROR Missing argument `tag_id` in function `controllers.service_api.dataset.dataset.TagDeletePayload.__init__` [missing-argument]
    --> tests/test_containers_integration_tests/controllers/service_api/dataset/test_dataset.py:195:29
-ERROR `in` is not supported between `Literal['webapp-logo']` and `None` [not-iterable]
-   --> tests/test_containers_integration_tests/controllers/web/test_site.py:133:16
 ERROR Argument `SimpleNamespace` is not assignable to parameter `webapp_settings` with type `WebAppSettings | None` in function `controllers.web.wraps._validate_user_accessibility` [bad-argument-type]
   --> tests/test_containers_integration_tests/controllers/web/test_wraps.py:74:33
 ERROR Argument `SimpleNamespace` is not assignable to parameter `webapp_settings` with type `WebAppSettings | None` in function `controllers.web.wraps._validate_user_accessibility` [bad-argument-type]
@@ -1167,6 +1165,12 @@
 ERROR Argument `Literal['normal']` is not assignable to parameter `status` with type `SQLCoreOperations[TenantStatus] | TenantStatus` in function `models.account.Tenant.__init__` [bad-argument-type]
   --> tests/test_containers_integration_tests/services/test_dataset_service_get_segments.py:44:62
 ERROR Argument `Literal['active']` is not assignable to parameter `status` with type `AccountStatus | SQLCoreOperations[AccountStatus]` in function `models.account.Account.__init__` [bad-argument-type]
+  --> tests/test_containers_integration_tests/services/test_dataset_service_permissions.py:36:20
+ERROR Argument `Literal['normal']` is not assignable to parameter `status` with type `SQLCoreOperations[TenantStatus] | TenantStatus` in function `models.account.Tenant.__init__` [bad-argument-type]
+  --> tests/test_containers_integration_tests/services/test_dataset_service_permissions.py:38:58
+ERROR Argument `Literal['active']` is not assignable to parameter `status` with type `AccountStatus | SQLCoreOperations[AccountStatus]` in function `models.account.Account.__init__` [bad-argument-type]
+  --> tests/test_containers_integration_tests/services/test_dataset_service_permissions.py:65:20
+ERROR Argument `Literal['active']` is not assignable to parameter `status` with type `AccountStatus | SQLCoreOperations[AccountStatus]` in function `models.account.Account.__init__` [bad-argument-type]
   --> tests/test_containers_integration_tests/services/test_dataset_service_retrieval.py:45:20
 ERROR Argument `Literal['normal']` is not assignable to parameter `status` with type `SQLCoreOperations[TenantStatus] | TenantStatus` in function `models.account.Tenant.__init__` [bad-argument-type]
   --> tests/test_containers_integration_tests/services/test_dataset_service_retrieval.py:49:20
@@ -1272,7 +1276,7 @@
    --> tests/test_containers_integration_tests/services/test_model_provider_service.py:187:24
 ERROR Object of class `NoneType` has no attribute `id` [missing-attribute]
   --> tests/test_containers_integration_tests/services/test_ops_service.py:59:13
-ERROR Argument `dict[str, Any] | dict[Unknown, Unknown] | object | None` is not assignable to parameter `tracing_config` with type `SQLCoreOperations[dict[str, Any] | None] | dict[str, Any] | None` in function `models.model.TraceAppConfig.__init__` [bad-argument-type]
+ERROR Argument `dict[Unknown, Unknown] | object | None` is not assignable to parameter `tracing_config` with type `SQLCoreOperations[dict[Unknown, Unknown] | None] | dict[Unknown, Unknown] | None` in function `models.model.TraceAppConfig.__init__` [bad-argument-type]
   --> tests/test_containers_integration_tests/services/test_ops_service.py:84:28
 ERROR `None` is not subscriptable [unsupported-operation]
    --> tests/test_containers_integration_tests/services/test_recommended_app_service.py:198:16
@@ -2077,37 +2081,37 @@
 ERROR Could not find name `Import` [unknown-name]
    --> tests/unit_tests/controllers/inner_api/app/test_dsl.py:119:71
 ERROR Missing argument `tenant_model` in function `protected_view` [missing-argument]
-   --> tests/unit_tests/controllers/inner_api/plugin/test_plugin_wraps.py:169:44
+   --> tests/unit_tests/controllers/inner_api/plugin/test_plugin_wraps.py:168:44
 ERROR Missing argument `user_model` in function `protected_view` [missing-argument]
-   --> tests/unit_tests/controllers/inner_api/plugin/test_plugin_wraps.py:169:44
+   --> tests/unit_tests/controllers/inner_api/plugin/test_plugin_wraps.py:168:44
 ERROR Missing argument `tenant_model` in function `protected_view` [missing-argument]
-   --> tests/unit_tests/controllers/inner_api/plugin/test_plugin_wraps.py:186:31
+   --> tests/unit_tests/controllers/inner_api/plugin/test_plugin_wraps.py:185:31
 ERROR Missing argument `user_model` in function `protected_view` [missing-argument]
-   --> tests/unit_tests/controllers/inner_api/plugin/test_plugin_wraps.py:186:31
+   --> tests/unit_tests/controllers/inner_api/plugin/test_plugin_wraps.py:185:31
 ERROR Missing argument `tenant_model` in function `protected_view` [missing-argument]
-   --> tests/unit_tests/controllers/inner_api/plugin/test_plugin_wraps.py:201:35
+   --> tests/unit_tests/controllers/inner_api/plugin/test_plugin_wraps.py:200:35
 ERROR Missing argument `user_model` in function `protected_view` [missing-argument]
-   --> tests/unit_tests/controllers/inner_api/plugin/test_plugin_wraps.py:201:35
+   --> tests/unit_tests/controllers/inner_api/plugin/test_plugin_wraps.py:200:35
 ERROR Missing argument `tenant_model` in function `protected_view` [missing-argument]
-   --> tests/unit_tests/controllers/inner_api/plugin/test_plugin_wraps.py:223:44
+   --> tests/unit_tests/controllers/inner_api/plugin/test_plugin_wraps.py:222:44
 ERROR Missing argument `user_model` in function `protected_view` [missing-argument]
-   --> tests/unit_tests/controllers/inner_api/plugin/test_plugin_wraps.py:223:44
+   --> tests/unit_tests/controllers/inner_api/plugin/test_plugin_wraps.py:222:44
 ERROR Argument `type[PluginTestPayload]` is not assignable to parameter `payload_type` with type `type[BaseModel]` in function `controllers.inner_api.plugin.wraps.plugin_data` [bad-argument-type]
-   --> tests/unit_tests/controllers/inner_api/plugin/test_plugin_wraps.py:251:35
+   --> tests/unit_tests/controllers/inner_api/plugin/test_plugin_wraps.py:250:35
 ERROR Missing argument `payload` in function `protected_view` [missing-argument]
-   --> tests/unit_tests/controllers/inner_api/plugin/test_plugin_wraps.py:257:36
+   --> tests/unit_tests/controllers/inner_api/plugin/test_plugin_wraps.py:256:36
 ERROR Argument `type[PluginTestPayload]` is not assignable to parameter `payload_type` with type `type[BaseModel]` in function `controllers.inner_api.plugin.wraps.plugin_data` [bad-argument-type]
-   --> tests/unit_tests/controllers/inner_api/plugin/test_plugin_wraps.py:266:35
+   --> tests/unit_tests/controllers/inner_api/plugin/test_plugin_wraps.py:265:35
 ERROR Missing argument `payload` in function `protected_view` [missing-argument]
-   --> tests/unit_tests/controllers/inner_api/plugin/test_plugin_wraps.py:273:31
+   --> tests/unit_tests/controllers/inner_api/plugin/test_plugin_wraps.py:272:31
 ERROR Argument `type[TestPluginData.test_should_raise_error_on_invalid_payload.InvalidPayload]` is not assignable to parameter `payload_type` with type `type[BaseModel]` in function `controllers.inner_api.plugin.wraps.plugin_data` [bad-argument-type]
-   --> tests/unit_tests/controllers/inner_api/plugin/test_plugin_wraps.py:284:35
+   --> tests/unit_tests/controllers/inner_api/plugin/test_plugin_wraps.py:283:35
 ERROR Missing argument `payload` in function `protected_view` [missing-argument]
-   --> tests/unit_tests/controllers/inner_api/plugin/test_plugin_wraps.py:291:31
+   --> tests/unit_tests/controllers/inner_api/plugin/test_plugin_wraps.py:290:31
 ERROR Argument `type[PluginTestPayload]` is not assignable to parameter `payload_type` with type `type[BaseModel]` in function `controllers.inner_api.plugin.wraps.plugin_data` [bad-argument-type]
-   --> tests/unit_tests/controllers/inner_api/plugin/test_plugin_wraps.py:297:35
+   --> tests/unit_tests/controllers/inner_api/plugin/test_plugin_wraps.py:296:35
 ERROR Missing argument `payload` in function `protected_view` [missing-argument]
-   --> tests/unit_tests/controllers/inner_api/plugin/test_plugin_wraps.py:303:36
+   --> tests/unit_tests/controllers/inner_api/plugin/test_plugin_wraps.py:302:36
 ERROR Cannot index into `Iterable[bytes]` [bad-index]
    --> tests/unit_tests/controllers/service_api/app/test_audio.py:189:16
 ERROR Cannot index into `Response` [bad-index]
@@ -2250,6 +2254,8 @@
    --> tests/unit_tests/controllers/web/test_pydantic_models.py:308:38
 ERROR Missing argument `inputs` in function `controllers.common.controller_schemas.WorkflowRunPayload.__init__` [missing-argument]
    --> tests/unit_tests/controllers/web/test_pydantic_models.py:329:31
+ERROR `in` is not supported between `Literal['webapp-logo']` and `None` [not-iterable]
+   --> tests/unit_tests/controllers/web/test_site.py:126:16
 ERROR `None` is not subscriptable [unsupported-operation]
   --> tests/unit_tests/controllers/web/test_web_passport.py:36:16
 ERROR Argument `SimpleNamespace` is not assignable to parameter `app_model` with type `App` in function `controllers.web.workflow_events.WorkflowEventsApi.get` [bad-argument-type]
@@ -3948,13 +3954,13 @@
 ERROR Class `Extension` has no class attribute `_Extension__module_extensions` [missing-attribute]
   --> tests/unit_tests/core/extension/test_extension.py:29:30
 ERROR Class member `ConcreteTool.query` overrides parent class `ExternalDataTool` in an inconsistent manner [bad-override]
-  --> tests/unit_tests/core/external_data_tool/test_base.py:20:17
+  --> tests/unit_tests/core/external_data_tool/test_base.py:18:17
 ERROR Class member `ConcreteTool.query` overrides parent class `ExternalDataTool` in an inconsistent manner [bad-override]
-  --> tests/unit_tests/core/external_data_tool/test_base.py:36:17
+  --> tests/unit_tests/core/external_data_tool/test_base.py:34:17
 ERROR Class member `ConcreteTool.query` overrides parent class `ExternalDataTool` in an inconsistent manner [bad-override]
-  --> tests/unit_tests/core/external_data_tool/test_base.py:51:17
+  --> tests/unit_tests/core/external_data_tool/test_base.py:49:17
 ERROR Class member `ConcreteTool.query` overrides parent class `ExternalDataTool` in an inconsistent manner [bad-override]
-  --> tests/unit_tests/core/external_data_tool/test_base.py:63:17
+  --> tests/unit_tests/core/external_data_tool/test_base.py:61:17
 ERROR Argument `None` is not assignable to parameter `schema` with type `dict[str, Any]` in function `core.llm_generator.output_parser.structured_output.remove_additional_properties` [bad-argument-type]
   --> tests/unit_tests/core/llm_generator/output_parser/test_structured_output.py:50:38
 ERROR Argument `list[@_]` is not assignable to parameter `schema` with type `dict[str, Any]` in function `core.llm_generator.output_parser.structured_output.remove_additional_properties` [bad-argument-type]
@@ -4494,85 +4500,85 @@
 ERROR Argument `dict[str, str]` is not assignable to parameter `reranking_model` with type `RerankingModelDict | None` in function `core.rag.data_post_processor.data_post_processor.DataPostProcessor._get_rerank_model_instance` [bad-argument-type]
    --> tests/unit_tests/core/rag/data_post_processor/test_data_post_processor.py:182:33
 ERROR Class member `_Field.__eq__` overrides parent class `object` in an inconsistent manner [bad-override]
-  --> tests/unit_tests/core/rag/datasource/keyword/jieba/test_jieba.py:25:9
+  --> tests/unit_tests/core/rag/datasource/keyword/jieba/test_jieba.py:24:9
 ERROR Argument `SimpleNamespace` is not assignable to parameter `dataset` with type `Dataset` in function `core.rag.datasource.keyword.jieba.jieba.Jieba.__init__` [bad-argument-type]
-  --> tests/unit_tests/core/rag/datasource/keyword/jieba/test_jieba.py:95:21
+  --> tests/unit_tests/core/rag/datasource/keyword/jieba/test_jieba.py:94:21
 ERROR Argument `list[Document | SimpleNamespace]` is not assignable to parameter `texts` with type `list[Document]` in function `core.rag.datasource.keyword.jieba.jieba.Jieba.create` [bad-argument-type]
-   --> tests/unit_tests/core/rag/datasource/keyword/jieba/test_jieba.py:105:9
+   --> tests/unit_tests/core/rag/datasource/keyword/jieba/test_jieba.py:104:9
 ERROR Object of class `FunctionType` has no attribute `assert_called_once` [missing-attribute]
-   --> tests/unit_tests/core/rag/datasource/keyword/jieba/test_jieba.py:112:5
+   --> tests/unit_tests/core/rag/datasource/keyword/jieba/test_jieba.py:111:5
 ERROR Object of class `FunctionType` has no attribute `call_args` [missing-attribute]
-   --> tests/unit_tests/core/rag/datasource/keyword/jieba/test_jieba.py:113:17
+   --> tests/unit_tests/core/rag/datasource/keyword/jieba/test_jieba.py:112:17
 ERROR Object of class `FunctionType` has no attribute `call_args` [missing-attribute]
-   --> tests/unit_tests/core/rag/datasource/keyword/jieba/test_jieba.py:117:19
+   --> tests/unit_tests/core/rag/datasource/keyword/jieba/test_jieba.py:116:19
 ERROR Argument `SimpleNamespace` is not assignable to parameter `dataset` with type `Dataset` in function `core.rag.datasource.keyword.jieba.jieba.Jieba.__init__` [bad-argument-type]
-   --> tests/unit_tests/core/rag/datasource/keyword/jieba/test_jieba.py:124:21
+   --> tests/unit_tests/core/rag/datasource/keyword/jieba/test_jieba.py:123:21
 ERROR Object of class `FunctionType` has no attribute `call_count` [missing-attribute]
-   --> tests/unit_tests/core/rag/datasource/keyword/jieba/test_jieba.py:139:12
+   --> tests/unit_tests/core/rag/datasource/keyword/jieba/test_jieba.py:138:12
 ERROR Object of class `FunctionType` has no attribute `call_args_list` [missing-attribute]
-   --> tests/unit_tests/core/rag/datasource/keyword/jieba/test_jieba.py:140:18
+   --> tests/unit_tests/core/rag/datasource/keyword/jieba/test_jieba.py:139:18
 ERROR Object of class `FunctionType` has no attribute `call_args_list` [missing-attribute]
-   --> tests/unit_tests/core/rag/datasource/keyword/jieba/test_jieba.py:141:19
+   --> tests/unit_tests/core/rag/datasource/keyword/jieba/test_jieba.py:140:19
 ERROR Object of class `FunctionType` has no attribute `assert_called_once` [missing-attribute]
-   --> tests/unit_tests/core/rag/datasource/keyword/jieba/test_jieba.py:144:5
+   --> tests/unit_tests/core/rag/datasource/keyword/jieba/test_jieba.py:143:5
 ERROR Argument `SimpleNamespace` is not assignable to parameter `dataset` with type `Dataset` in function `core.rag.datasource.keyword.jieba.jieba.Jieba.__init__` [bad-argument-type]
-   --> tests/unit_tests/core/rag/datasource/keyword/jieba/test_jieba.py:148:21
+   --> tests/unit_tests/core/rag/datasource/keyword/jieba/test_jieba.py:147:21
 ERROR Object of class `FunctionType` has no attribute `call_args` [missing-attribute]
-   --> tests/unit_tests/core/rag/datasource/keyword/jieba/test_jieba.py:160:16
+   --> tests/unit_tests/core/rag/datasource/keyword/jieba/test_jieba.py:159:16
 ERROR Argument `SimpleNamespace` is not assignable to parameter `dataset` with type `Dataset` in function `core.rag.datasource.keyword.jieba.jieba.Jieba.__init__` [bad-argument-type]
-   --> tests/unit_tests/core/rag/datasource/keyword/jieba/test_jieba.py:164:21
+   --> tests/unit_tests/core/rag/datasource/keyword/jieba/test_jieba.py:163:21
 ERROR Argument `SimpleNamespace` is not assignable to parameter `dataset` with type `Dataset` in function `core.rag.datasource.keyword.jieba.jieba.Jieba.__init__` [bad-argument-type]
-   --> tests/unit_tests/core/rag/datasource/keyword/jieba/test_jieba.py:175:21
+   --> tests/unit_tests/core/rag/datasource/keyword/jieba/test_jieba.py:174:21
 ERROR Object of class `FunctionType` has no attribute `assert_called_once_with` [missing-attribute]
-   --> tests/unit_tests/core/rag/datasource/keyword/jieba/test_jieba.py:182:5
+   --> tests/unit_tests/core/rag/datasource/keyword/jieba/test_jieba.py:181:5
 ERROR Object of class `FunctionType` has no attribute `assert_called_once_with` [missing-attribute]
-   --> tests/unit_tests/core/rag/datasource/keyword/jieba/test_jieba.py:183:5
+   --> tests/unit_tests/core/rag/datasource/keyword/jieba/test_jieba.py:182:5
 ERROR Argument `SimpleNamespace` is not assignable to parameter `dataset` with type `Dataset` in function `core.rag.datasource.keyword.jieba.jieba.Jieba.__init__` [bad-argument-type]
-   --> tests/unit_tests/core/rag/datasource/keyword/jieba/test_jieba.py:187:21
+   --> tests/unit_tests/core/rag/datasource/keyword/jieba/test_jieba.py:186:21
 ERROR Object of class `FunctionType` has no attribute `assert_not_called` [missing-attribute]
-   --> tests/unit_tests/core/rag/datasource/keyword/jieba/test_jieba.py:194:5
+   --> tests/unit_tests/core/rag/datasource/keyword/jieba/test_jieba.py:193:5
 ERROR Object of class `FunctionType` has no attribute `assert_called_once_with` [missing-attribute]
-   --> tests/unit_tests/core/rag/datasource/keyword/jieba/test_jieba.py:195:5
+   --> tests/unit_tests/core/rag/datasource/keyword/jieba/test_jieba.py:194:5
 ERROR Argument `SimpleNamespace` is not assignable to parameter `dataset` with type `Dataset` in function `core.rag.datasource.keyword.jieba.jieba.Jieba.__init__` [bad-argument-type]
-   --> tests/unit_tests/core/rag/datasource/keyword/jieba/test_jieba.py:204:21
+   --> tests/unit_tests/core/rag/datasource/keyword/jieba/test_jieba.py:203:21
 ERROR Argument `SimpleNamespace` is not assignable to parameter `dataset` with type `Dataset` in function `core.rag.datasource.keyword.jieba.jieba.Jieba.__init__` [bad-argument-type]
-   --> tests/unit_tests/core/rag/datasource/keyword/jieba/test_jieba.py:232:24
+   --> tests/unit_tests/core/rag/datasource/keyword/jieba/test_jieba.py:231:24
 ERROR Argument `SimpleNamespace` is not assignable to parameter `dataset` with type `Dataset` in function `core.rag.datasource.keyword.jieba.jieba.Jieba.__init__` [bad-argument-type]
-   --> tests/unit_tests/core/rag/datasource/keyword/jieba/test_jieba.py:236:26
+   --> tests/unit_tests/core/rag/datasource/keyword/jieba/test_jieba.py:235:26
 ERROR Argument `SimpleNamespace` is not assignable to parameter `dataset` with type `Dataset` in function `core.rag.datasource.keyword.jieba.jieba.Jieba.__init__` [bad-argument-type]
-   --> tests/unit_tests/core/rag/datasource/keyword/jieba/test_jieba.py:246:21
+   --> tests/unit_tests/core/rag/datasource/keyword/jieba/test_jieba.py:245:21
 ERROR Argument `SimpleNamespace` is not assignable to parameter `dataset` with type `Dataset` in function `core.rag.datasource.keyword.jieba.jieba.Jieba.__init__` [bad-argument-type]
-   --> tests/unit_tests/core/rag/datasource/keyword/jieba/test_jieba.py:257:21
+   --> tests/unit_tests/core/rag/datasource/keyword/jieba/test_jieba.py:256:21
 ERROR Argument `SimpleNamespace` is not assignable to parameter `dataset` with type `Dataset` in function `core.rag.datasource.keyword.jieba.jieba.Jieba.__init__` [bad-argument-type]
-   --> tests/unit_tests/core/rag/datasource/keyword/jieba/test_jieba.py:273:21
+   --> tests/unit_tests/core/rag/datasource/keyword/jieba/test_jieba.py:272:21
 ERROR Argument `SimpleNamespace` is not assignable to parameter `dataset` with type `Dataset` in function `core.rag.datasource.keyword.jieba.jieba.Jieba.__init__` [bad-argument-type]
-   --> tests/unit_tests/core/rag/datasource/keyword/jieba/test_jieba.py:277:42
+   --> tests/unit_tests/core/rag/datasource/keyword/jieba/test_jieba.py:276:42
 ERROR Argument `SimpleNamespace` is not assignable to parameter `dataset` with type `Dataset` in function `core.rag.datasource.keyword.jieba.jieba.Jieba.__init__` [bad-argument-type]
-   --> tests/unit_tests/core/rag/datasource/keyword/jieba/test_jieba.py:291:21
+   --> tests/unit_tests/core/rag/datasource/keyword/jieba/test_jieba.py:290:21
 ERROR Argument `SimpleNamespace` is not assignable to parameter `dataset` with type `Dataset` in function `core.rag.datasource.keyword.jieba.jieba.Jieba.__init__` [bad-argument-type]
-   --> tests/unit_tests/core/rag/datasource/keyword/jieba/test_jieba.py:307:21
+   --> tests/unit_tests/core/rag/datasource/keyword/jieba/test_jieba.py:306:21
 ERROR Argument `SimpleNamespace` is not assignable to parameter `dataset` with type `Dataset` in function `core.rag.datasource.keyword.jieba.jieba.Jieba.__init__` [bad-argument-type]
-   --> tests/unit_tests/core/rag/datasource/keyword/jieba/test_jieba.py:321:21
+   --> tests/unit_tests/core/rag/datasource/keyword/jieba/test_jieba.py:320:21
 ERROR Argument `SimpleNamespace` is not assignable to parameter `dataset` with type `Dataset` in function `core.rag.datasource.keyword.jieba.jieba.Jieba.__init__` [bad-argument-type]
-   --> tests/unit_tests/core/rag/datasource/keyword/jieba/test_jieba.py:343:21
+   --> tests/unit_tests/core/rag/datasource/keyword/jieba/test_jieba.py:342:21
 ERROR Argument `SimpleNamespace` is not assignable to parameter `dataset` with type `Dataset` in function `core.rag.datasource.keyword.jieba.jieba.Jieba.__init__` [bad-argument-type]
-   --> tests/unit_tests/core/rag/datasource/keyword/jieba/test_jieba.py:363:21
+   --> tests/unit_tests/core/rag/datasource/keyword/jieba/test_jieba.py:362:21
 ERROR Object of class `FunctionType` has no attribute `assert_called_once_with` [missing-attribute]
-   --> tests/unit_tests/core/rag/datasource/keyword/jieba/test_jieba.py:369:5
+   --> tests/unit_tests/core/rag/datasource/keyword/jieba/test_jieba.py:368:5
 ERROR Object of class `FunctionType` has no attribute `assert_called_once` [missing-attribute]
-   --> tests/unit_tests/core/rag/datasource/keyword/jieba/test_jieba.py:370:5
+   --> tests/unit_tests/core/rag/datasource/keyword/jieba/test_jieba.py:369:5
 ERROR Object of class `FunctionType` has no attribute `reset_mock` [missing-attribute]
-   --> tests/unit_tests/core/rag/datasource/keyword/jieba/test_jieba.py:372:5
+   --> tests/unit_tests/core/rag/datasource/keyword/jieba/test_jieba.py:371:5
 ERROR Object of class `FunctionType` has no attribute `assert_called_once` [missing-attribute]
-   --> tests/unit_tests/core/rag/datasource/keyword/jieba/test_jieba.py:374:5
+   --> tests/unit_tests/core/rag/datasource/keyword/jieba/test_jieba.py:373:5
 ERROR Argument `SimpleNamespace` is not assignable to parameter `dataset` with type `Dataset` in function `core.rag.datasource.keyword.jieba.jieba.Jieba.__init__` [bad-argument-type]
-   --> tests/unit_tests/core/rag/datasource/keyword/jieba/test_jieba.py:378:21
+   --> tests/unit_tests/core/rag/datasource/keyword/jieba/test_jieba.py:377:21
 ERROR `SimpleNamespace` is not assignable to TypedDict key `segment` with type `DocumentSegment` [bad-typed-dict-key]
-   --> tests/unit_tests/core/rag/datasource/keyword/jieba/test_jieba.py:390:25
+   --> tests/unit_tests/core/rag/datasource/keyword/jieba/test_jieba.py:389:25
 ERROR `SimpleNamespace` is not assignable to TypedDict key `segment` with type `DocumentSegment` [bad-typed-dict-key]
-   --> tests/unit_tests/core/rag/datasource/keyword/jieba/test_jieba.py:391:25
+   --> tests/unit_tests/core/rag/datasource/keyword/jieba/test_jieba.py:390:25
 ERROR Object of class `FunctionType` has no attribute `call_args` [missing-attribute]
-   --> tests/unit_tests/core/rag/datasource/keyword/jieba/test_jieba.py:397:19
+   --> tests/unit_tests/core/rag/datasource/keyword/jieba/test_jieba.py:396:19
 ERROR Object of class `ModuleType` has no attribute `analyse` [missing-attribute]
   --> tests/unit_tests/core/rag/datasource/keyword/jieba/test_jieba_keyword_table_handler.py:30:5
 ERROR Object of class `ModuleType` has no attribute `default_tfidf` [missing-attribute]
@@ -4672,7 +4678,7 @@
 ERROR `Literal['john@test.com']` is not assignable to attribute `text` with type `Ellipsis` [bad-assignment]
    --> tests/unit_tests/core/rag/extractor/test_word_extractor.py:695:14
 ERROR Argument `Literal['semantic_search']` is not assignable to parameter `retrieval_method` with type `RetrievalMethod` in function `core.rag.index_processor.processor.paragraph_index_processor.ParagraphIndexProcessor.retrieve` [bad-argument-type]
-   --> tests/unit_tests/core/rag/indexing/processor/test_paragraph_index_processor.py:246:39
+   --> tests/unit_tests/core/rag/indexing/processor/test_paragraph_index_processor.py:241:39
 ERROR Argument `Literal['semantic_search']` is not assignable to parameter `retrieval_method` with type `RetrievalMethod` in function `core.rag.index_processor.processor.parent_child_index_processor.ParentChildIndexProcessor.retrieve` [bad-argument-type]
    --> tests/unit_tests/core/rag/indexing/processor/test_parent_child_index_processor.py:305:39
 ERROR Argument `SimpleNamespace` is not assignable to parameter `rules` with type `Rule` in function `core.rag.index_processor.processor.parent_child_index_processor.ParentChildIndexProcessor._split_child_nodes` [bad-argument-type]
@@ -4680,9 +4686,9 @@
 ERROR Argument `SimpleNamespace` is not assignable to parameter `rules` with type `Rule` in function `core.rag.index_processor.processor.parent_child_index_processor.ParentChildIndexProcessor._split_child_nodes` [bad-argument-type]
    --> tests/unit_tests/core/rag/indexing/processor/test_parent_child_index_processor.py:333:63
 ERROR Argument `Literal['semantic_search']` is not assignable to parameter `retrieval_method` with type `RetrievalMethod` in function `core.rag.index_processor.processor.qa_index_processor.QAIndexProcessor.retrieve` [bad-argument-type]
-   --> tests/unit_tests/core/rag/indexing/processor/test_qa_index_processor.py:268:39
+   --> tests/unit_tests/core/rag/indexing/processor/test_qa_index_processor.py:267:39
 ERROR Missing required key `enable` for TypedDict `SummaryIndexSettingDict` [bad-typed-dict-key]
-   --> tests/unit_tests/core/rag/indexing/processor/test_qa_index_processor.py:334:78
+   --> tests/unit_tests/core/rag/indexing/processor/test_qa_index_processor.py:333:78
 ERROR Argument `None` is not assignable to parameter `state` with type `InstanceState[Any]` in function `sqlalchemy.orm.exc.ObjectDeletedError.__init__` [bad-argument-type]
    --> tests/unit_tests/core/rag/indexing/test_indexing_runner.py:924:84
 ERROR Argument `Literal['completed']` is not assignable to parameter `after_indexing_status` with type `IndexingStatus` in function `core.indexing_runner.IndexingRunner._update_document_index_status` [bad-argument-type]
@@ -4698,39 +4704,39 @@
 ERROR Object of class `FunctionType` has no attribute `call_count` [missing-attribute]
     --> tests/unit_tests/core/rag/rerank/test_reranker.py:1630:16
 ERROR Argument `list[float] | None` is not assignable to parameter `obj` with type `Sized` in function `len` [bad-argument-type]
-    --> tests/unit_tests/core/rag/retrieval/test_dataset_retrieval.py:1949:20
+    --> tests/unit_tests/core/rag/retrieval/test_dataset_retrieval.py:1948:20
 ERROR Could not find name `metadata_name` [unknown-name]
-    --> tests/unit_tests/core/rag/retrieval/test_dataset_retrieval.py:2769:29
+    --> tests/unit_tests/core/rag/retrieval/test_dataset_retrieval.py:2768:29
 ERROR Could not find name `metadata_name` [unknown-name]
-    --> tests/unit_tests/core/rag/retrieval/test_dataset_retrieval.py:2770:29
+    --> tests/unit_tests/core/rag/retrieval/test_dataset_retrieval.py:2769:29
 ERROR Argument `Iterator[Any | Unknown] | Iterator[Any]` is not assignable to parameter `invoke_result` with type `Generator[Unknown]` in function `core.rag.retrieval.dataset_retrieval.DatasetRetrieval._handle_invoke_result` [bad-argument-type]
-    --> tests/unit_tests/core/rag/retrieval/test_dataset_retrieval.py:3782:64
+    --> tests/unit_tests/core/rag/retrieval/test_dataset_retrieval.py:3781:64
 ERROR Argument `Iterator[Any | Unknown] | Iterator[Any]` is not assignable to parameter `invoke_result` with type `Generator[Unknown]` in function `core.rag.retrieval.dataset_retrieval.DatasetRetrieval._handle_invoke_result` [bad-argument-type]
-    --> tests/unit_tests/core/rag/retrieval/test_dataset_retrieval.py:3786:67
+    --> tests/unit_tests/core/rag/retrieval/test_dataset_retrieval.py:3785:67
 ERROR `None` is not subscriptable [unsupported-operation]
-    --> tests/unit_tests/core/rag/retrieval/test_dataset_retrieval.py:4025:16
+    --> tests/unit_tests/core/rag/retrieval/test_dataset_retrieval.py:4024:16
 ERROR Argument `list[SimpleNamespace]` is not assignable to parameter `available_datasets` with type `list[Dataset]` in function `core.rag.retrieval.dataset_retrieval.DatasetRetrieval.single_retrieve` [bad-argument-type]
-    --> tests/unit_tests/core/rag/retrieval/test_dataset_retrieval.py:4547:40
+    --> tests/unit_tests/core/rag/retrieval/test_dataset_retrieval.py:4546:40
 ERROR Argument `list[SimpleNamespace]` is not assignable to parameter `available_datasets` with type `list[Dataset]` in function `core.rag.retrieval.dataset_retrieval.DatasetRetrieval.single_retrieve` [bad-argument-type]
-    --> tests/unit_tests/core/rag/retrieval/test_dataset_retrieval.py:4599:40
+    --> tests/unit_tests/core/rag/retrieval/test_dataset_retrieval.py:4598:40
 ERROR Argument `SimpleNamespace` is not assignable to parameter `metadata_condition` with type `MetadataFilteringCondition | None` in function `core.rag.retrieval.dataset_retrieval.DatasetRetrieval.single_retrieve` [bad-argument-type]
-    --> tests/unit_tests/core/rag/retrieval/test_dataset_retrieval.py:4604:40
+    --> tests/unit_tests/core/rag/retrieval/test_dataset_retrieval.py:4603:40
 ERROR Argument `list[SimpleNamespace]` is not assignable to parameter `available_datasets` with type `list[Dataset]` in function `core.rag.retrieval.dataset_retrieval.DatasetRetrieval.single_retrieve` [bad-argument-type]
-    --> tests/unit_tests/core/rag/retrieval/test_dataset_retrieval.py:4620:36
+    --> tests/unit_tests/core/rag/retrieval/test_dataset_retrieval.py:4619:36
 ERROR Argument `list[SimpleNamespace]` is not assignable to parameter `available_datasets` with type `list[Dataset]` in function `core.rag.retrieval.dataset_retrieval.DatasetRetrieval.single_retrieve` [bad-argument-type]
-    --> tests/unit_tests/core/rag/retrieval/test_dataset_retrieval.py:4650:36
+    --> tests/unit_tests/core/rag/retrieval/test_dataset_retrieval.py:4649:36
 ERROR Argument `SimpleNamespace` is not assignable to parameter `metadata_condition` with type `MetadataFilteringCondition | None` in function `core.rag.retrieval.dataset_retrieval.DatasetRetrieval.single_retrieve` [bad-argument-type]
-    --> tests/unit_tests/core/rag/retrieval/test_dataset_retrieval.py:4655:36
+    --> tests/unit_tests/core/rag/retrieval/test_dataset_retrieval.py:4654:36
 ERROR Argument `list[SimpleNamespace]` is not assignable to parameter `available_datasets` with type `list[Dataset]` in function `core.rag.retrieval.dataset_retrieval.DatasetRetrieval.single_retrieve` [bad-argument-type]
-    --> tests/unit_tests/core/rag/retrieval/test_dataset_retrieval.py:4663:36
+    --> tests/unit_tests/core/rag/retrieval/test_dataset_retrieval.py:4662:36
 ERROR Argument `list[SimpleNamespace]` is not assignable to parameter `available_datasets` with type `list[Dataset]` in function `core.rag.retrieval.dataset_retrieval.DatasetRetrieval.multiple_retrieve` [bad-argument-type]
-    --> tests/unit_tests/core/rag/retrieval/test_dataset_retrieval.py:4699:36
+    --> tests/unit_tests/core/rag/retrieval/test_dataset_retrieval.py:4698:36
 ERROR Argument `list[SimpleNamespace]` is not assignable to parameter `available_datasets` with type `list[Dataset]` in function `core.rag.retrieval.dataset_retrieval.DatasetRetrieval.multiple_retrieve` [bad-argument-type]
-    --> tests/unit_tests/core/rag/retrieval/test_dataset_retrieval.py:4727:36
+    --> tests/unit_tests/core/rag/retrieval/test_dataset_retrieval.py:4726:36
 ERROR Argument `list[SimpleNamespace]` is not assignable to parameter `available_datasets` with type `list[Dataset]` in function `core.rag.retrieval.dataset_retrieval.DatasetRetrieval.multiple_retrieve` [bad-argument-type]
-    --> tests/unit_tests/core/rag/retrieval/test_dataset_retrieval.py:4785:40
+    --> tests/unit_tests/core/rag/retrieval/test_dataset_retrieval.py:4784:40
 ERROR Argument `list[SimpleNamespace]` is not assignable to parameter `available_datasets` with type `list[Dataset]` in function `core.rag.retrieval.dataset_retrieval.DatasetRetrieval.multiple_retrieve` [bad-argument-type]
-    --> tests/unit_tests/core/rag/retrieval/test_dataset_retrieval.py:4829:44
+    --> tests/unit_tests/core/rag/retrieval/test_dataset_retrieval.py:4828:44
 ERROR Argument `Iterator[Any | Unknown] | Iterator[Any]` is not assignable to parameter `invoke_result` with type `Generator[Unknown]` in function `core.rag.retrieval.router.multi_dataset_react_route.ReactMultiDatasetRouter._handle_invoke_result` [bad-argument-type]
    --> tests/unit_tests/core/rag/retrieval/test_multi_dataset_react_route.py:199:52
 ERROR Argument `None` is not assignable to parameter `text` with type `str` in function `core.rag.splitter.text_splitter.RecursiveCharacterTextSplitter.split_text` [bad-argument-type]
@@ -5244,11 +5250,11 @@
 ERROR Argument `list[SimpleNamespace]` is not assignable to parameter `providers` with type `list[ToolProviderApiEntity]` in function `core.tools.builtin_tool.providers._positions.BuiltinToolProviderSort.sort` [bad-argument-type]
    --> tests/unit_tests/core/tools/test_builtin_tools_extra.py:328:53
 ERROR `None` is not assignable to attribute `runtime` with type `ToolRuntime` [bad-assignment]
-  --> tests/unit_tests/core/tools/test_custom_tool.py:97:20
+  --> tests/unit_tests/core/tools/test_custom_tool.py:96:20
 ERROR `list[SimpleNamespace]` is not assignable to attribute `parameters` with type `list[ToolParameter] | None` [bad-assignment]
-   --> tests/unit_tests/core/tools/test_custom_tool.py:118:34
+   --> tests/unit_tests/core/tools/test_custom_tool.py:117:34
 ERROR `list[SimpleNamespace]` is not assignable to attribute `parameters` with type `list[ToolParameter] | None` [bad-assignment]
-   --> tests/unit_tests/core/tools/test_custom_tool.py:124:34
+   --> tests/unit_tests/core/tools/test_custom_tool.py:123:34
 ERROR Object of class `BlobChunkMessage` has no attribute `text`
 ERROR Argument `SimpleNamespace` is not assignable to parameter `db_provider` with type `ApiToolProvider` in function `core.tools.custom_tool.provider.ApiToolProviderController.from_db` [bad-argument-type]
   --> tests/unit_tests/core/tools/test_custom_tool_provider.py:36:52
@@ -5305,11 +5311,11 @@
 ERROR Argument `SimpleNamespace` is not assignable to parameter `controller` with type `ToolProviderController` in function `core.tools.utils.encryption.create_tool_provider_encrypter` [bad-argument-type]
    --> tests/unit_tests/core/tools/utils/test_encryption.py:203:75
 ERROR `None` is not subscriptable [unsupported-operation]
-  --> tests/unit_tests/core/tools/utils/test_message_transformer.py:88:12
+  --> tests/unit_tests/core/tools/utils/test_message_transformer.py:86:12
 ERROR `None` is not subscriptable [unsupported-operation]
-  --> tests/unit_tests/core/tools/utils/test_message_transformer.py:89:12
+  --> tests/unit_tests/core/tools/utils/test_message_transformer.py:87:12
 ERROR `None` is not subscriptable [unsupported-operation]
-   --> tests/unit_tests/core/tools/utils/test_message_transformer.py:109:12
+   --> tests/unit_tests/core/tools/utils/test_message_transformer.py:107:12
 ERROR Argument `SimpleNamespace` is not assignable to parameter `dataset` with type `Dataset` in function `core.tools.utils.dataset_retriever.dataset_retriever_tool.DatasetRetrieverTool.from_dataset` [bad-argument-type]
    --> tests/unit_tests/core/tools/utils/test_misc_utils_extra.py:112:17
 ERROR Argument `_FakeFlaskApp` is not assignable to parameter `flask_app` with type `Flask` in function `core.tools.utils.dataset_retriever.dataset_multi_retriever_tool.DatasetMultiRetrieverTool._retriever` [bad-argument-type]
@@ -5317,15 +5323,15 @@
 ERROR Argument `_FakeFlaskApp` is not assignable to parameter `flask_app` with type `Flask` in function `core.tools.utils.dataset_retriever.dataset_multi_retriever_tool.DatasetMultiRetrieverTool._retriever` [bad-argument-type]
    --> tests/unit_tests/core/tools/utils/test_misc_utils_extra.py:376:27
 ERROR Argument `Literal['builtin']` is not assignable to parameter `tool_type` with type `ToolProviderType` in function `core.tools.utils.model_invocation_utils.ModelInvocationUtils.invoke` [bad-argument-type]
-   --> tests/unit_tests/core/tools/utils/test_model_invocation_utils.py:111:31
+   --> tests/unit_tests/core/tools/utils/test_model_invocation_utils.py:110:31
 ERROR Argument `Literal['builtin']` is not assignable to parameter `tool_type` with type `ToolProviderType` in function `core.tools.utils.model_invocation_utils.ModelInvocationUtils.invoke` [bad-argument-type]
-   --> tests/unit_tests/core/tools/utils/test_model_invocation_utils.py:161:35
+   --> tests/unit_tests/core/tools/utils/test_model_invocation_utils.py:160:35
 ERROR `None` is not subscriptable [unsupported-operation]
-   --> tests/unit_tests/core/tools/utils/test_parser.py:115:12
+   --> tests/unit_tests/core/tools/utils/test_parser.py:114:12
 ERROR Argument `list[ToolParameter] | None` is not assignable to parameter `obj` with type `Sized` in function `len` [bad-argument-type]
-   --> tests/unit_tests/core/tools/utils/test_parser.py:175:16
+   --> tests/unit_tests/core/tools/utils/test_parser.py:174:16
 ERROR Type `None` is not iterable [not-iterable]
-   --> tests/unit_tests/core/tools/utils/test_parser.py:178:54
+   --> tests/unit_tests/core/tools/utils/test_parser.py:177:54
 ERROR Argument `Literal['not a dict']` is not assignable to parameter `d` with type `dict[str, Any]` in function `core.tools.utils.message_transformer.safe_json_dict` [bad-argument-type]
    --> tests/unit_tests/core/tools/utils/test_tool_engine_serialization.py:206:28
 ERROR Argument `list[int]` is not assignable to parameter `d` with type `dict[str, Any]` in function `core.tools.utils.message_transformer.safe_json_dict` [bad-argument-type]
@@ -5661,19 +5667,21 @@
 ERROR Missing argument `name` in function `core.workflow.nodes.trigger_webhook.entities.WebhookBodyParameter.__init__` [missing-argument]
   --> tests/unit_tests/core/workflow/nodes/webhook/test_entities.py:77:29
 ERROR Argument `dict[str, dict[str, Any] | str]` is not assignable to parameter `config` with type `NodeConfigDict` in function `graphon.nodes.base.node.Node.__init__` [bad-argument-type]
-  --> tests/unit_tests/core/workflow/nodes/webhook/test_webhook_file_conversion.py:61:16
+  --> tests/unit_tests/core/workflow/nodes/webhook/test_webhook_file_conversion.py:60:16
 ERROR Object of class `GraphRuntimeState` has no attribute `app_config` [missing-attribute]
-  --> tests/unit_tests/core/workflow/nodes/webhook/test_webhook_file_conversion.py:67:5
+  --> tests/unit_tests/core/workflow/nodes/webhook/test_webhook_file_conversion.py:66:5
 ERROR Object of class `GraphRuntimeState` has no attribute `app_config` [missing-attribute]
-  --> tests/unit_tests/core/workflow/nodes/webhook/test_webhook_file_conversion.py:68:5
+  --> tests/unit_tests/core/workflow/nodes/webhook/test_webhook_file_conversion.py:67:5
 ERROR Object of class `TriggerWebhookNode` has no attribute `node_id` [missing-attribute]
-  --> tests/unit_tests/core/workflow/nodes/webhook/test_webhook_file_conversion.py:72:5
+  --> tests/unit_tests/core/workflow/nodes/webhook/test_webhook_file_conversion.py:71:5
 ERROR Argument `dict[str, dict[str, Any] | str]` is not assignable to parameter `config` with type `NodeConfigDict` in function `graphon.nodes.base.node.Node.__init__` [bad-argument-type]
-  --> tests/unit_tests/core/workflow/nodes/webhook/test_webhook_node.py:52:16
+  --> tests/unit_tests/core/workflow/nodes/webhook/test_webhook_node.py:51:16
 ERROR Object of class `GraphRuntimeState` has no attribute `app_config` [missing-attribute]
-  --> tests/unit_tests/core/workflow/nodes/webhook/test_webhook_node.py:58:5
+  --> tests/unit_tests/core/workflow/nodes/webhook/test_webhook_node.py:57:5
 ERROR Object of class `TriggerWebhookNode` has no attribute `node_id` [missing-attribute]
-  --> tests/unit_tests/core/workflow/nodes/webhook/test_webhook_node.py:61:5
+  --> tests/unit_tests/core/workflow/nodes/webhook/test_webhook_node.py:60:5
+ERROR Cannot index into `object` [bad-index]
+   --> tests/unit_tests/core/workflow/nodes/webhook/test_webhook_node.py:106:12
 ERROR Cannot index into `object` [bad-index]
    --> tests/unit_tests/core/workflow/nodes/webhook/test_webhook_node.py:107:12
 ERROR Cannot index into `object` [bad-index]
@@ -5690,8 +5698,6 @@
    --> tests/unit_tests/core/workflow/nodes/webhook/test_webhook_node.py:113:12
 ERROR Cannot index into `object` [bad-index]
    --> tests/unit_tests/core/workflow/nodes/webhook/test_webhook_node.py:114:12
-ERROR Cannot index into `object` [bad-index]
-   --> tests/unit_tests/core/workflow/nodes/webhook/test_webhook_node.py:115:12
 ERROR Argument `SimpleNamespace` is not assignable to parameter `variable_pool` with type `VariablePool | None` in function `core.workflow.human_input_compat.EmailDeliveryConfig.render_body_template` [bad-argument-type]
   --> tests/unit_tests/core/workflow/test_human_input_compat.py:29:23
 ERROR Object of class `_InteractiveSurfaceDeliveryConfig` has no attribute `recipients` [missing-attribute]
@@ -5889,7 +5895,7 @@
 ERROR Argument `_FakeRedis` is not assignable to parameter `redis_client` with type `_RateLimiterRedisClient` in function `libs.helper.RateLimiter.__init__` [bad-argument-type]
   --> tests/unit_tests/libs/test_rate_limiter.py:42:22
 ERROR Argument `dict[@_, @_]` is not assignable to parameter `headers` with type `Message` in function `python_http_client.exceptions.HTTPError.__init__` [bad-argument-type]
-  --> tests/unit_tests/libs/test_sendgrid_client.py:39:99
+  --> tests/unit_tests/libs/test_sendgrid_client.py:38:99
 ERROR Argument `None` is not assignable to parameter `duration_str` with type `str` in function `libs.time_parser.parse_time_duration` [bad-argument-type]
   --> tests/unit_tests/libs/test_time_parser.py:54:38
 ERROR Argument `MockRequest` is not assignable to parameter `request` with type `Request` in function `libs.token.extract_access_token` [bad-argument-type]
@@ -5992,6 +5998,8 @@
    --> tests/unit_tests/services/document_indexing_task_proxy.py:831:114
 ERROR Argument `None` is not assignable to parameter `plan` with type `CloudPlan` in function `DocumentIndexingTaskProxyTestDataFactory.create_mock_features` [bad-argument-type]
    --> tests/unit_tests/services/document_indexing_task_proxy.py:855:114
+ERROR Argument `Literal['invalid_action']` is not assignable to parameter `action` with type `Literal['archive', 'disable', 'enable', 'un_archive']` in function `services.dataset_service.DocumentService.batch_update_document_status` [bad-argument-type]
+  --> tests/unit_tests/services/document_service_status.py:70:81
 ERROR Unexpected keyword argument `workspace_id` in function `services.enterprise.enterprise_service.DefaultWorkspaceJoinResult.__init__` [unexpected-keyword]
    --> tests/unit_tests/services/enterprise/test_enterprise_service.py:239:40
 ERROR Unexpected keyword argument `workspace_id` in function `services.enterprise.enterprise_service.DefaultWorkspaceJoinResult.__init__` [unexpected-keyword]
@@ -6080,9 +6088,9 @@
 ERROR TypedDict `AnnotationSettingDisabledDict` does not have key `embedding_model` [bad-typed-dict-key]
     --> tests/unit_tests/services/test_annotation_service.py:1251:27
 ERROR Argument `_FakeStreams` is not assignable to parameter `redis_client` with type `Redis[Unknown] | RedisCluster[Unknown]` in function `libs.broadcast_channel.redis.streams_channel.StreamsBroadcastChannel.__init__` [bad-argument-type]
-  --> tests/unit_tests/services/test_app_generate_service_streaming_integration.py:96:36
+  --> tests/unit_tests/services/test_app_generate_service_streaming_integration.py:95:36
 ERROR Argument `_FakeRedisClient` is not assignable to parameter `redis_client` with type `Redis[Unknown] | RedisCluster[Unknown]` in function `libs.broadcast_channel.redis.channel.BroadcastChannel.__init__` [bad-argument-type]
-   --> tests/unit_tests/services/test_app_generate_service_streaming_integration.py:116:34
+   --> tests/unit_tests/services/test_app_generate_service_streaming_integration.py:115:34
 ERROR Argument `TestAsyncWorkflowService.test_should_dispatch_to_matching_celery_task_when_triggering_workflow.DummyAccount` is not assignable to parameter `user` with type `Account | EndUser` in function `services.async_workflow_service.AsyncWorkflowService.trigger_workflow_async` [bad-argument-type]
    --> tests/unit_tests/services/test_async_workflow_service.py:152:88
 ERROR Argument `SimpleNamespace` is not assignable to parameter `user` with type `Account | EndUser` in function `services.async_workflow_service.AsyncWorkflowService.trigger_workflow_async` [bad-argument-type]
@@ -6096,7 +6104,7 @@
 ERROR Argument `SimpleNamespace` is not assignable to parameter `user` with type `Account | EndUser` in function `services.async_workflow_service.AsyncWorkflowService.reinvoke_trigger` [bad-argument-type]
    --> tests/unit_tests/services/test_async_workflow_service.py:318:22
 ERROR Argument `None` is not assignable to parameter `file` with type `FileStorage` in function `services.audio_service.AudioService.transcript_asr` [bad-argument-type]
-   --> tests/unit_tests/services/test_audio_service.py:318:61
+   --> tests/unit_tests/services/test_audio_service.py:317:61
 ERROR Argument `BillingInfo` is not assignable to parameter `result` with type `dict[Unknown, Unknown]` in function `TestBillingServiceSubscriptionInfoDataType._assert_billing_info_types` [bad-argument-type]
     --> tests/unit_tests/services/test_billing_service.py:1613:41
 ERROR Argument `BillingInfo` is not assignable to parameter `result` with type `dict[Unknown, Unknown]` in function `TestBillingServiceSubscriptionInfoDataType._assert_billing_info_types` [bad-argument-type]
@@ -6112,43 +6120,35 @@
 ERROR Class member `FixedDateTime.now` overrides parent class `datetime` in an inconsistent manner [bad-override]
    --> tests/unit_tests/services/test_clear_free_plan_tenant_expired_logs.py:460:13
 ERROR Argument `SimpleNamespace` is not assignable to parameter `account` with type `Account` in function `services.dataset_service.DatasetService.create_empty_dataset` [bad-argument-type]
-   --> tests/unit_tests/services/test_dataset_service_dataset.py:310:93
+   --> tests/unit_tests/services/test_dataset_service_dataset.py:180:93
 ERROR Argument `SimpleNamespace` is not assignable to parameter `account` with type `Account` in function `services.dataset_service.DatasetService.create_empty_dataset` [bad-argument-type]
-   --> tests/unit_tests/services/test_dataset_service_dataset.py:334:25
+   --> tests/unit_tests/services/test_dataset_service_dataset.py:204:25
 ERROR Argument `SimpleNamespace` is not assignable to parameter `account` with type `Account` in function `services.dataset_service.DatasetService.create_empty_dataset` [bad-argument-type]
-   --> tests/unit_tests/services/test_dataset_service_dataset.py:377:25
+   --> tests/unit_tests/services/test_dataset_service_dataset.py:247:25
 ERROR Argument `SimpleNamespace` is not assignable to parameter `dataset` with type `Dataset` in function `services.dataset_service.DatasetService._update_pipeline_knowledge_base_node_data` [bad-argument-type]
-   --> tests/unit_tests/services/test_dataset_service_dataset.py:694:70
+   --> tests/unit_tests/services/test_dataset_service_dataset.py:564:70
 ERROR Argument `SimpleNamespace` is not assignable to parameter `dataset` with type `Dataset` in function `services.dataset_service.DatasetService._update_pipeline_knowledge_base_node_data` [bad-argument-type]
-   --> tests/unit_tests/services/test_dataset_service_dataset.py:704:70
+   --> tests/unit_tests/services/test_dataset_service_dataset.py:574:70
 ERROR Argument `SimpleNamespace` is not assignable to parameter `dataset` with type `Dataset` in function `services.dataset_service.DatasetService._update_pipeline_knowledge_base_node_data` [bad-argument-type]
-   --> tests/unit_tests/services/test_dataset_service_dataset.py:743:70
+   --> tests/unit_tests/services/test_dataset_service_dataset.py:613:70
 ERROR Argument `SimpleNamespace` is not assignable to parameter `dataset` with type `Dataset` in function `services.dataset_service.DatasetService._update_pipeline_knowledge_base_node_data` [bad-argument-type]
-   --> tests/unit_tests/services/test_dataset_service_dataset.py:766:74
+   --> tests/unit_tests/services/test_dataset_service_dataset.py:636:74
 ERROR Object of class `FakeAccount` has no attribute `current_tenant_id` [missing-attribute]
-   --> tests/unit_tests/services/test_dataset_service_dataset.py:833:9
+   --> tests/unit_tests/services/test_dataset_service_dataset.py:703:9
 ERROR Object of class `FakeAccount` has no attribute `current_tenant_id` [missing-attribute]
-   --> tests/unit_tests/services/test_dataset_service_dataset.py:871:9
+   --> tests/unit_tests/services/test_dataset_service_dataset.py:741:9
 ERROR Argument `None` is not assignable to parameter `embedding_model_provider` with type `str` in function `tests.unit_tests.services.dataset_service_test_helpers.DatasetServiceUnitDataFactory.create_dataset_mock` [bad-argument-type]
-   --> tests/unit_tests/services/test_dataset_service_dataset.py:937:38
+   --> tests/unit_tests/services/test_dataset_service_dataset.py:807:38
 ERROR Argument `None` is not assignable to parameter `embedding_model` with type `str` in function `tests.unit_tests.services.dataset_service_test_helpers.DatasetServiceUnitDataFactory.create_dataset_mock` [bad-argument-type]
-   --> tests/unit_tests/services/test_dataset_service_dataset.py:938:29
+   --> tests/unit_tests/services/test_dataset_service_dataset.py:808:29
 ERROR Object of class `FakeAccount` has no attribute `current_tenant_id` [missing-attribute]
-   --> tests/unit_tests/services/test_dataset_service_dataset.py:996:9
+   --> tests/unit_tests/services/test_dataset_service_dataset.py:866:9
 ERROR Object of class `FakeAccount` has no attribute `current_tenant_id` [missing-attribute]
-    --> tests/unit_tests/services/test_dataset_service_dataset.py:1031:9
+   --> tests/unit_tests/services/test_dataset_service_dataset.py:901:9
 ERROR Argument `SimpleNamespace` is not assignable to parameter `user` with type `Account | None` in function `services.dataset_service.DatasetService.check_dataset_operator_permission` [bad-argument-type]
-    --> tests/unit_tests/services/test_dataset_service_dataset.py:1439:67
+    --> tests/unit_tests/services/test_dataset_service_dataset.py:1212:67
 ERROR Argument `SimpleNamespace` is not assignable to parameter `dataset` with type `Dataset | None` in function `services.dataset_service.DatasetService.check_dataset_operator_permission` [bad-argument-type]
-    --> tests/unit_tests/services/test_dataset_service_dataset.py:1442:81
-ERROR Argument `SimpleNamespace` is not assignable to parameter `user` with type `Account | None` in function `services.dataset_service.DatasetService.check_dataset_operator_permission` [bad-argument-type]
-    --> tests/unit_tests/services/test_dataset_service_dataset.py:1455:67
-ERROR Argument `SimpleNamespace` is not assignable to parameter `user` with type `Account | None` in function `services.dataset_service.DatasetService.check_dataset_operator_permission` [bad-argument-type]
-    --> tests/unit_tests/services/test_dataset_service_dataset.py:1468:71
-ERROR Object of class `FakeAccount` has no attribute `current_tenant_id` [missing-attribute]
-    --> tests/unit_tests/services/test_dataset_service_dataset.py:1527:9
-ERROR Object of class `FakeAccount` has no attribute `current_tenant_id` [missing-attribute]
-    --> tests/unit_tests/services/test_dataset_service_dataset.py:1549:9
+    --> tests/unit_tests/services/test_dataset_service_dataset.py:1215:81
 ERROR Argument `SimpleNamespace` is not assignable to parameter `current_user` with type `Account` in function `services.dataset_service.DocumentService.prepare_document_batch_download_zip` [bad-argument-type]
    --> tests/unit_tests/services/test_dataset_service_document.py:340:34
 ERROR Argument `SimpleNamespace` is not assignable to parameter `current_user` with type `Account` in function `services.dataset_service.DocumentService.prepare_document_batch_download_zip` [bad-argument-type]

auto-merge was automatically disabled April 14, 2026 18:45

Head branch was pushed to by a user without write access

@github-actions
Copy link
Copy Markdown
Contributor

github-actions Bot commented Apr 14, 2026

Pyrefly Diff

base → PR 
--- /tmp/pyrefly_base.txt	2026-04-14 18:47:03.944157939 +0000
+++ /tmp/pyrefly_pr.txt	2026-04-14 18:46:52.961127920 +0000
@@ -1155,6 +1155,12 @@
 ERROR Argument `Literal['normal']` is not assignable to parameter `status` with type `SQLCoreOperations[TenantStatus] | TenantStatus` in function `models.account.Tenant.__init__` [bad-argument-type]
   --> tests/test_containers_integration_tests/services/test_dataset_service_get_segments.py:44:62
 ERROR Argument `Literal['active']` is not assignable to parameter `status` with type `AccountStatus | SQLCoreOperations[AccountStatus]` in function `models.account.Account.__init__` [bad-argument-type]
+  --> tests/test_containers_integration_tests/services/test_dataset_service_permissions.py:37:20
+ERROR Argument `Literal['normal']` is not assignable to parameter `status` with type `SQLCoreOperations[TenantStatus] | TenantStatus` in function `models.account.Tenant.__init__` [bad-argument-type]
+  --> tests/test_containers_integration_tests/services/test_dataset_service_permissions.py:39:58
+ERROR Argument `Literal['active']` is not assignable to parameter `status` with type `AccountStatus | SQLCoreOperations[AccountStatus]` in function `models.account.Account.__init__` [bad-argument-type]
+  --> tests/test_containers_integration_tests/services/test_dataset_service_permissions.py:66:20
+ERROR Argument `Literal['active']` is not assignable to parameter `status` with type `AccountStatus | SQLCoreOperations[AccountStatus]` in function `models.account.Account.__init__` [bad-argument-type]
   --> tests/test_containers_integration_tests/services/test_dataset_service_retrieval.py:45:20
 ERROR Argument `Literal['normal']` is not assignable to parameter `status` with type `SQLCoreOperations[TenantStatus] | TenantStatus` in function `models.account.Tenant.__init__` [bad-argument-type]
   --> tests/test_containers_integration_tests/services/test_dataset_service_retrieval.py:49:20
@@ -6100,43 +6106,35 @@
 ERROR Class member `FixedDateTime.now` overrides parent class `datetime` in an inconsistent manner [bad-override]
    --> tests/unit_tests/services/test_clear_free_plan_tenant_expired_logs.py:460:13
 ERROR Argument `SimpleNamespace` is not assignable to parameter `account` with type `Account` in function `services.dataset_service.DatasetService.create_empty_dataset` [bad-argument-type]
-   --> tests/unit_tests/services/test_dataset_service_dataset.py:310:93
+   --> tests/unit_tests/services/test_dataset_service_dataset.py:180:93
 ERROR Argument `SimpleNamespace` is not assignable to parameter `account` with type `Account` in function `services.dataset_service.DatasetService.create_empty_dataset` [bad-argument-type]
-   --> tests/unit_tests/services/test_dataset_service_dataset.py:334:25
+   --> tests/unit_tests/services/test_dataset_service_dataset.py:204:25
 ERROR Argument `SimpleNamespace` is not assignable to parameter `account` with type `Account` in function `services.dataset_service.DatasetService.create_empty_dataset` [bad-argument-type]
-   --> tests/unit_tests/services/test_dataset_service_dataset.py:377:25
+   --> tests/unit_tests/services/test_dataset_service_dataset.py:247:25
 ERROR Argument `SimpleNamespace` is not assignable to parameter `dataset` with type `Dataset` in function `services.dataset_service.DatasetService._update_pipeline_knowledge_base_node_data` [bad-argument-type]
-   --> tests/unit_tests/services/test_dataset_service_dataset.py:694:70
+   --> tests/unit_tests/services/test_dataset_service_dataset.py:564:70
 ERROR Argument `SimpleNamespace` is not assignable to parameter `dataset` with type `Dataset` in function `services.dataset_service.DatasetService._update_pipeline_knowledge_base_node_data` [bad-argument-type]
-   --> tests/unit_tests/services/test_dataset_service_dataset.py:704:70
+   --> tests/unit_tests/services/test_dataset_service_dataset.py:574:70
 ERROR Argument `SimpleNamespace` is not assignable to parameter `dataset` with type `Dataset` in function `services.dataset_service.DatasetService._update_pipeline_knowledge_base_node_data` [bad-argument-type]
-   --> tests/unit_tests/services/test_dataset_service_dataset.py:743:70
+   --> tests/unit_tests/services/test_dataset_service_dataset.py:613:70
 ERROR Argument `SimpleNamespace` is not assignable to parameter `dataset` with type `Dataset` in function `services.dataset_service.DatasetService._update_pipeline_knowledge_base_node_data` [bad-argument-type]
-   --> tests/unit_tests/services/test_dataset_service_dataset.py:766:74
+   --> tests/unit_tests/services/test_dataset_service_dataset.py:636:74
 ERROR Object of class `FakeAccount` has no attribute `current_tenant_id` [missing-attribute]
-   --> tests/unit_tests/services/test_dataset_service_dataset.py:833:9
+   --> tests/unit_tests/services/test_dataset_service_dataset.py:703:9
 ERROR Object of class `FakeAccount` has no attribute `current_tenant_id` [missing-attribute]
-   --> tests/unit_tests/services/test_dataset_service_dataset.py:871:9
+   --> tests/unit_tests/services/test_dataset_service_dataset.py:741:9
 ERROR Argument `None` is not assignable to parameter `embedding_model_provider` with type `str` in function `tests.unit_tests.services.dataset_service_test_helpers.DatasetServiceUnitDataFactory.create_dataset_mock` [bad-argument-type]
-   --> tests/unit_tests/services/test_dataset_service_dataset.py:937:38
+   --> tests/unit_tests/services/test_dataset_service_dataset.py:807:38
 ERROR Argument `None` is not assignable to parameter `embedding_model` with type `str` in function `tests.unit_tests.services.dataset_service_test_helpers.DatasetServiceUnitDataFactory.create_dataset_mock` [bad-argument-type]
-   --> tests/unit_tests/services/test_dataset_service_dataset.py:938:29
+   --> tests/unit_tests/services/test_dataset_service_dataset.py:808:29
 ERROR Object of class `FakeAccount` has no attribute `current_tenant_id` [missing-attribute]
-   --> tests/unit_tests/services/test_dataset_service_dataset.py:996:9
+   --> tests/unit_tests/services/test_dataset_service_dataset.py:866:9
 ERROR Object of class `FakeAccount` has no attribute `current_tenant_id` [missing-attribute]
-    --> tests/unit_tests/services/test_dataset_service_dataset.py:1031:9
+   --> tests/unit_tests/services/test_dataset_service_dataset.py:901:9
 ERROR Argument `SimpleNamespace` is not assignable to parameter `user` with type `Account | None` in function `services.dataset_service.DatasetService.check_dataset_operator_permission` [bad-argument-type]
-    --> tests/unit_tests/services/test_dataset_service_dataset.py:1439:67
+    --> tests/unit_tests/services/test_dataset_service_dataset.py:1212:67
 ERROR Argument `SimpleNamespace` is not assignable to parameter `dataset` with type `Dataset | None` in function `services.dataset_service.DatasetService.check_dataset_operator_permission` [bad-argument-type]
-    --> tests/unit_tests/services/test_dataset_service_dataset.py:1442:81
-ERROR Argument `SimpleNamespace` is not assignable to parameter `user` with type `Account | None` in function `services.dataset_service.DatasetService.check_dataset_operator_permission` [bad-argument-type]
-    --> tests/unit_tests/services/test_dataset_service_dataset.py:1455:67
-ERROR Argument `SimpleNamespace` is not assignable to parameter `user` with type `Account | None` in function `services.dataset_service.DatasetService.check_dataset_operator_permission` [bad-argument-type]
-    --> tests/unit_tests/services/test_dataset_service_dataset.py:1468:71
-ERROR Object of class `FakeAccount` has no attribute `current_tenant_id` [missing-attribute]
-    --> tests/unit_tests/services/test_dataset_service_dataset.py:1527:9
-ERROR Object of class `FakeAccount` has no attribute `current_tenant_id` [missing-attribute]
-    --> tests/unit_tests/services/test_dataset_service_dataset.py:1549:9
+    --> tests/unit_tests/services/test_dataset_service_dataset.py:1215:81
 ERROR Argument `SimpleNamespace` is not assignable to parameter `current_user` with type `Account` in function `services.dataset_service.DocumentService.prepare_document_batch_download_zip` [bad-argument-type]
    --> tests/unit_tests/services/test_dataset_service_document.py:340:34
 ERROR Argument `SimpleNamespace` is not assignable to parameter `current_user` with type `Account` in function `services.dataset_service.DocumentService.prepare_document_batch_download_zip` [bad-argument-type]

@github-actions
Copy link
Copy Markdown
Contributor

github-actions Bot commented Apr 14, 2026

Pyrefly Diff

base → PR 
--- /tmp/pyrefly_base.txt	2026-04-14 19:05:04.080907940 +0000
+++ /tmp/pyrefly_pr.txt	2026-04-14 19:04:53.401746405 +0000
@@ -1155,6 +1155,12 @@
 ERROR Argument `Literal['normal']` is not assignable to parameter `status` with type `SQLCoreOperations[TenantStatus] | TenantStatus` in function `models.account.Tenant.__init__` [bad-argument-type]
   --> tests/test_containers_integration_tests/services/test_dataset_service_get_segments.py:44:62
 ERROR Argument `Literal['active']` is not assignable to parameter `status` with type `AccountStatus | SQLCoreOperations[AccountStatus]` in function `models.account.Account.__init__` [bad-argument-type]
+  --> tests/test_containers_integration_tests/services/test_dataset_service_permissions.py:37:20
+ERROR Argument `Literal['normal']` is not assignable to parameter `status` with type `SQLCoreOperations[TenantStatus] | TenantStatus` in function `models.account.Tenant.__init__` [bad-argument-type]
+  --> tests/test_containers_integration_tests/services/test_dataset_service_permissions.py:39:58
+ERROR Argument `Literal['active']` is not assignable to parameter `status` with type `AccountStatus | SQLCoreOperations[AccountStatus]` in function `models.account.Account.__init__` [bad-argument-type]
+  --> tests/test_containers_integration_tests/services/test_dataset_service_permissions.py:66:20
+ERROR Argument `Literal['active']` is not assignable to parameter `status` with type `AccountStatus | SQLCoreOperations[AccountStatus]` in function `models.account.Account.__init__` [bad-argument-type]
   --> tests/test_containers_integration_tests/services/test_dataset_service_retrieval.py:45:20
 ERROR Argument `Literal['normal']` is not assignable to parameter `status` with type `SQLCoreOperations[TenantStatus] | TenantStatus` in function `models.account.Tenant.__init__` [bad-argument-type]
   --> tests/test_containers_integration_tests/services/test_dataset_service_retrieval.py:49:20
@@ -6100,43 +6106,35 @@
 ERROR Class member `FixedDateTime.now` overrides parent class `datetime` in an inconsistent manner [bad-override]
    --> tests/unit_tests/services/test_clear_free_plan_tenant_expired_logs.py:460:13
 ERROR Argument `SimpleNamespace` is not assignable to parameter `account` with type `Account` in function `services.dataset_service.DatasetService.create_empty_dataset` [bad-argument-type]
-   --> tests/unit_tests/services/test_dataset_service_dataset.py:310:93
+   --> tests/unit_tests/services/test_dataset_service_dataset.py:180:93
 ERROR Argument `SimpleNamespace` is not assignable to parameter `account` with type `Account` in function `services.dataset_service.DatasetService.create_empty_dataset` [bad-argument-type]
-   --> tests/unit_tests/services/test_dataset_service_dataset.py:334:25
+   --> tests/unit_tests/services/test_dataset_service_dataset.py:204:25
 ERROR Argument `SimpleNamespace` is not assignable to parameter `account` with type `Account` in function `services.dataset_service.DatasetService.create_empty_dataset` [bad-argument-type]
-   --> tests/unit_tests/services/test_dataset_service_dataset.py:377:25
+   --> tests/unit_tests/services/test_dataset_service_dataset.py:247:25
 ERROR Argument `SimpleNamespace` is not assignable to parameter `dataset` with type `Dataset` in function `services.dataset_service.DatasetService._update_pipeline_knowledge_base_node_data` [bad-argument-type]
-   --> tests/unit_tests/services/test_dataset_service_dataset.py:694:70
+   --> tests/unit_tests/services/test_dataset_service_dataset.py:564:70
 ERROR Argument `SimpleNamespace` is not assignable to parameter `dataset` with type `Dataset` in function `services.dataset_service.DatasetService._update_pipeline_knowledge_base_node_data` [bad-argument-type]
-   --> tests/unit_tests/services/test_dataset_service_dataset.py:704:70
+   --> tests/unit_tests/services/test_dataset_service_dataset.py:574:70
 ERROR Argument `SimpleNamespace` is not assignable to parameter `dataset` with type `Dataset` in function `services.dataset_service.DatasetService._update_pipeline_knowledge_base_node_data` [bad-argument-type]
-   --> tests/unit_tests/services/test_dataset_service_dataset.py:743:70
+   --> tests/unit_tests/services/test_dataset_service_dataset.py:613:70
 ERROR Argument `SimpleNamespace` is not assignable to parameter `dataset` with type `Dataset` in function `services.dataset_service.DatasetService._update_pipeline_knowledge_base_node_data` [bad-argument-type]
-   --> tests/unit_tests/services/test_dataset_service_dataset.py:766:74
+   --> tests/unit_tests/services/test_dataset_service_dataset.py:636:74
 ERROR Object of class `FakeAccount` has no attribute `current_tenant_id` [missing-attribute]
-   --> tests/unit_tests/services/test_dataset_service_dataset.py:833:9
+   --> tests/unit_tests/services/test_dataset_service_dataset.py:703:9
 ERROR Object of class `FakeAccount` has no attribute `current_tenant_id` [missing-attribute]
-   --> tests/unit_tests/services/test_dataset_service_dataset.py:871:9
+   --> tests/unit_tests/services/test_dataset_service_dataset.py:741:9
 ERROR Argument `None` is not assignable to parameter `embedding_model_provider` with type `str` in function `tests.unit_tests.services.dataset_service_test_helpers.DatasetServiceUnitDataFactory.create_dataset_mock` [bad-argument-type]
-   --> tests/unit_tests/services/test_dataset_service_dataset.py:937:38
+   --> tests/unit_tests/services/test_dataset_service_dataset.py:807:38
 ERROR Argument `None` is not assignable to parameter `embedding_model` with type `str` in function `tests.unit_tests.services.dataset_service_test_helpers.DatasetServiceUnitDataFactory.create_dataset_mock` [bad-argument-type]
-   --> tests/unit_tests/services/test_dataset_service_dataset.py:938:29
+   --> tests/unit_tests/services/test_dataset_service_dataset.py:808:29
 ERROR Object of class `FakeAccount` has no attribute `current_tenant_id` [missing-attribute]
-   --> tests/unit_tests/services/test_dataset_service_dataset.py:996:9
+   --> tests/unit_tests/services/test_dataset_service_dataset.py:866:9
 ERROR Object of class `FakeAccount` has no attribute `current_tenant_id` [missing-attribute]
-    --> tests/unit_tests/services/test_dataset_service_dataset.py:1031:9
+   --> tests/unit_tests/services/test_dataset_service_dataset.py:901:9
 ERROR Argument `SimpleNamespace` is not assignable to parameter `user` with type `Account | None` in function `services.dataset_service.DatasetService.check_dataset_operator_permission` [bad-argument-type]
-    --> tests/unit_tests/services/test_dataset_service_dataset.py:1439:67
+    --> tests/unit_tests/services/test_dataset_service_dataset.py:1212:67
 ERROR Argument `SimpleNamespace` is not assignable to parameter `dataset` with type `Dataset | None` in function `services.dataset_service.DatasetService.check_dataset_operator_permission` [bad-argument-type]
-    --> tests/unit_tests/services/test_dataset_service_dataset.py:1442:81
-ERROR Argument `SimpleNamespace` is not assignable to parameter `user` with type `Account | None` in function `services.dataset_service.DatasetService.check_dataset_operator_permission` [bad-argument-type]
-    --> tests/unit_tests/services/test_dataset_service_dataset.py:1455:67
-ERROR Argument `SimpleNamespace` is not assignable to parameter `user` with type `Account | None` in function `services.dataset_service.DatasetService.check_dataset_operator_permission` [bad-argument-type]
-    --> tests/unit_tests/services/test_dataset_service_dataset.py:1468:71
-ERROR Object of class `FakeAccount` has no attribute `current_tenant_id` [missing-attribute]
-    --> tests/unit_tests/services/test_dataset_service_dataset.py:1527:9
-ERROR Object of class `FakeAccount` has no attribute `current_tenant_id` [missing-attribute]
-    --> tests/unit_tests/services/test_dataset_service_dataset.py:1549:9
+    --> tests/unit_tests/services/test_dataset_service_dataset.py:1215:81
 ERROR Argument `SimpleNamespace` is not assignable to parameter `current_user` with type `Account` in function `services.dataset_service.DocumentService.prepare_document_batch_download_zip` [bad-argument-type]
    --> tests/unit_tests/services/test_dataset_service_document.py:340:34
 ERROR Argument `SimpleNamespace` is not assignable to parameter `current_user` with type `Account` in function `services.dataset_service.DocumentService.prepare_document_batch_download_zip` [bad-argument-type]

@asukaminato0721 asukaminato0721 added this pull request to the merge queue Apr 14, 2026
Merged via the queue into langgenius:main with commit 693080a Apr 14, 2026
27 of 28 checks passed
HanqingZ pushed a commit to HanqingZ/dify that referenced this pull request Apr 23, 2026
@jamesrayammons @autofix-ci @HanqingZ
test: migrate dataset service dataset mock tests to testcontainers (l…
77ce413 
…anggenius#35194)

Co-authored-by: autofix-ci[bot] <114827586+autofix-ci[bot]@users.noreply.github.com>
asukaminato0721 pushed a commit to asukaminato0721/dify that referenced this pull request Apr 24, 2026
@jamesrayammons @autofix-ci @asukaminato0721
test: migrate dataset service dataset mock tests to testcontainers (l…
a5a0baf 
…anggenius#35194)

Co-authored-by: autofix-ci[bot] <114827586+autofix-ci[bot]@users.noreply.github.com>
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Reviewers

@asukaminato0721 asukaminato0721 asukaminato0721 approved these changes

@laipz8200 laipz8200 Awaiting requested review from laipz8200 laipz8200 is a code owner

@QuantumGhost QuantumGhost Awaiting requested review from QuantumGhost QuantumGhost is a code owner

Assignees

No one assigned

Labels

lgtm This PR has been approved by a maintainer size:XS This PR changes 0-9 lines, ignoring generated files.

Projects

None yet

Milestone

No milestone

Development

Successfully merging this pull request may close these issues.

2 participants

@jamesrayammons @asukaminato0721