diff --git a/api/core/datasource/__base/datasource_runtime.py b/api/core/datasource/__base/datasource_runtime.py
index b5e9d086e0..c6dc130196 100644
--- a/api/core/datasource/__base/datasource_runtime.py
+++ b/api/core/datasource/__base/datasource_runtime.py
@@ -3,6 +3,8 @@ from typing import TYPE_CHECKING, Any, Optional
 from openai import BaseModel
 from pydantic import Field
 
+# Import InvokeFrom locally to avoid circular import
+from core.app.entities.app_invoke_entities import InvokeFrom
 from core.datasource.entities.datasource_entities import DatasourceInvokeFrom
 
 if TYPE_CHECKING:
@@ -28,8 +30,7 @@ class FakeDatasourceRuntime(DatasourceRuntime):
     """
 
     def __init__(self):
-        # Import InvokeFrom locally to avoid circular import
-        from core.app.entities.app_invoke_entities import InvokeFrom
+
 
         super().__init__(
             tenant_id="fake_tenant_id",