diff --git a/api/contexts/__init__.py b/api/contexts/__init__.py
index 3c5d76d14a..2126a06f75 100644
--- a/api/contexts/__init__.py
+++ b/api/contexts/__init__.py
@@ -3,9 +3,9 @@ from threading import Lock
 from typing import TYPE_CHECKING
 
 from contexts.wrapper import RecyclableContextVar
-from core.datasource.__base.datasource_provider import DatasourcePluginProviderController
 
 if TYPE_CHECKING:
+    from core.datasource.__base.datasource_provider import DatasourcePluginProviderController
     from core.model_runtime.entities.model_entities import AIModelEntity
     from core.plugin.entities.plugin_daemon import PluginModelProviderEntity
     from core.tools.plugin_tool.provider import PluginToolProviderController