diff --git a/apps/api/src/models/__init__.py b/apps/api/src/models/__init__.py
index 0ea612d..49a69a4 100644
--- a/apps/api/src/models/__init__.py
+++ b/apps/api/src/models/__init__.py
@@ -12,6 +12,7 @@ from src.models.tool_instance import ToolInstance
 from src.models.tool_type import ToolType
 from src.models.user import User
 from src.models.user_config import UserConfig
+from src.models.workspace import Workspace
 
 __all__ = [
     "Base",
@@ -29,4 +30,5 @@ __all__ = [
     "ToolType",
     "User",
     "UserConfig",
+    "Workspace",
 ]
diff --git a/apps/api/src/models/tool_type.py b/apps/api/src/models/tool_type.py
index 24b93dc..1d7b6a3 100644
--- a/apps/api/src/models/tool_type.py
+++ b/apps/api/src/models/tool_type.py
@@ -7,8 +7,9 @@ from sqlalchemy.orm import Mapped, mapped_column, relationship
 
 from src.models.base import Base, TimestampMixin, UUIDPrimaryKeyMixin
 
+from src.models.tool_definition_manifest import ToolDefinitionManifest
+
 if TYPE_CHECKING:
-    from src.models.tool_definition_manifest import ToolDefinitionManifest
     from src.models.user import User