| ] | ] | ||||
| "tests/*" = [ | "tests/*" = [ | ||||
| "F811", # redefined-while-unused | "F811", # redefined-while-unused | ||||
| "F401", # unused-import | |||||
| ] | ] | ||||
| [lint.pyflakes] | [lint.pyflakes] | ||||
| extend-generics = [ | |||||
| allowed-unused-imports = [ | |||||
| "_pytest.monkeypatch", | "_pytest.monkeypatch", | ||||
| "tests.integration_tests", | "tests.integration_tests", | ||||
| "tests.unit_tests", | |||||
| ] | ] |
| Single-database configuration for Flask. | Single-database configuration for Flask. | ||||
| from collections.abc import Generator | |||||
| from unittest.mock import MagicMock | from unittest.mock import MagicMock | ||||
| import google.generativeai.types.generation_types as generation_config_types # type: ignore | import google.generativeai.types.generation_types as generation_config_types # type: ignore |
| from unittest.mock import MagicMock | |||||
| from core.rag.datasource.vdb.baidu.baidu_vector import BaiduConfig, BaiduVector | from core.rag.datasource.vdb.baidu.baidu_vector import BaiduConfig, BaiduVector | ||||
| from tests.integration_tests.vdb.__mock.baiduvectordb import setup_baiduvectordb_mock | from tests.integration_tests.vdb.__mock.baiduvectordb import setup_baiduvectordb_mock | ||||
| from tests.integration_tests.vdb.test_vector_store import AbstractVectorTest, get_example_text, setup_mock_redis | from tests.integration_tests.vdb.test_vector_store import AbstractVectorTest, get_example_text, setup_mock_redis |
| from unittest.mock import MagicMock, patch | |||||
| import pytest | import pytest | ||||
| from core.rag.datasource.vdb.tidb_vector.tidb_vector import TiDBVector, TiDBVectorConfig | from core.rag.datasource.vdb.tidb_vector.tidb_vector import TiDBVector, TiDBVectorConfig |
| from configs import dify_config | from configs import dify_config | ||||
| from core.app.app_config.entities import ModelConfigEntity | from core.app.app_config.entities import ModelConfigEntity | ||||
| from core.file import File, FileTransferMethod, FileType, FileUploadConfig, ImageConfig | |||||
| from core.file import File, FileTransferMethod, FileType | |||||
| from core.memory.token_buffer_memory import TokenBufferMemory | from core.memory.token_buffer_memory import TokenBufferMemory | ||||
| from core.model_runtime.entities.message_entities import ( | from core.model_runtime.entities.message_entities import ( | ||||
| AssistantPromptMessage, | AssistantPromptMessage, |
| import uuid | import uuid | ||||
| from collections.abc import Generator | from collections.abc import Generator | ||||
| from datetime import UTC, datetime, timezone | |||||
| from datetime import UTC, datetime | |||||
| from core.workflow.entities.variable_pool import VariablePool | from core.workflow.entities.variable_pool import VariablePool | ||||
| from core.workflow.enums import SystemVariableKey | from core.workflow.enums import SystemVariableKey |
| from core.model_runtime.entities.model_entities import AIModelEntity, FetchFrom, ModelFeature, ModelType | from core.model_runtime.entities.model_entities import AIModelEntity, FetchFrom, ModelFeature, ModelType | ||||
| from core.model_runtime.model_providers.model_provider_factory import ModelProviderFactory | from core.model_runtime.model_providers.model_provider_factory import ModelProviderFactory | ||||
| from core.prompt.entities.advanced_prompt_entities import MemoryConfig | from core.prompt.entities.advanced_prompt_entities import MemoryConfig | ||||
| from core.variables import ArrayAnySegment, ArrayFileSegment, NoneSegment, StringSegment | |||||
| from core.workflow.entities.variable_entities import VariableSelector | |||||
| from core.variables import ArrayAnySegment, ArrayFileSegment, NoneSegment | |||||
| from core.workflow.entities.variable_pool import VariablePool | from core.workflow.entities.variable_pool import VariablePool | ||||
| from core.workflow.graph_engine import Graph, GraphInitParams, GraphRuntimeState | from core.workflow.graph_engine import Graph, GraphInitParams, GraphRuntimeState | ||||
| from core.workflow.nodes.answer import AnswerStreamGenerateRoute | from core.workflow.nodes.answer import AnswerStreamGenerateRoute |
| from core.workflow.graph_engine.entities.event import ( | from core.workflow.graph_engine.entities.event import ( | ||||
| GraphRunFailedEvent, | GraphRunFailedEvent, | ||||
| GraphRunPartialSucceededEvent, | GraphRunPartialSucceededEvent, | ||||
| GraphRunSucceededEvent, | |||||
| NodeRunRetryEvent, | NodeRunRetryEvent, | ||||
| ) | ) | ||||
| from tests.unit_tests.core.workflow.nodes.test_continue_on_error import ContinueOnErrorTestHelper | from tests.unit_tests.core.workflow.nodes.test_continue_on_error import ContinueOnErrorTestHelper |
| import pytest | |||||
| from core.variables import SegmentType | from core.variables import SegmentType | ||||
| from core.workflow.nodes.variable_assigner.v2.enums import Operation | from core.workflow.nodes.variable_assigner.v2.enums import Operation | ||||
| from core.workflow.nodes.variable_assigner.v2.helpers import is_input_value_valid | from core.workflow.nodes.variable_assigner.v2.helpers import is_input_value_valid |
| from unittest.mock import MagicMock, patch | |||||
| from unittest.mock import patch | |||||
| import pytest | import pytest | ||||
| from oss2 import Auth # type: ignore | from oss2 import Auth # type: ignore |
| from textwrap import dedent | |||||
| import pytest | import pytest | ||||
| from core.tools.utils.text_processing_utils import remove_leading_symbols | from core.tools.utils.text_processing_utils import remove_leading_symbols |