diff --git a/src/galileo/__init__.py b/src/galileo/__init__.py
index 1a789df63..8cbf739c4 100644
--- a/src/galileo/__init__.py
+++ b/src/galileo/__init__.py
@@ -21,11 +21,5 @@
 from galileo_core.schemas.logging.span import LlmSpan, RetrieverSpan, Span, StepWithChildSpans, ToolSpan, WorkflowSpan
 from galileo_core.schemas.logging.step import StepType
 from galileo_core.schemas.logging.trace import Trace
-from galileo_core.schemas.protect.execution_status import ExecutionStatus
-from galileo_core.schemas.protect.payload import Payload
-from galileo_core.schemas.protect.request import Request
-from galileo_core.schemas.protect.response import Response
-from galileo_core.schemas.protect.ruleset import Ruleset
-from galileo_core.schemas.protect.stage import StageType
 
 __version__ = "1.14.0"
diff --git a/src/galileo/protect/__init__.py b/src/galileo/protect/__init__.py
new file mode 100644
index 000000000..654710768
--- /dev/null
+++ b/src/galileo/protect/__init__.py
@@ -0,0 +1,21 @@
+# flake8: noqa: F401
+# ruff: noqa: F401
+
+from galileo.protect import ainvoke_protect, invoke_protect
+from galileo.stages import (
+    create_protect_stage,
+    get_protect_stage,
+    pause_protect_stage,
+    resume_protect_stage,
+    update_protect_stage,
+)
+from galileo_core.schemas.protect.action import ActionResult, ActionType, OverrideAction, PassthroughAction
+from galileo_core.schemas.protect.execution_status import ExecutionStatus
+from galileo_core.schemas.protect.metric import MetricComputation, MetricComputationStatus, MetricValueType
+from galileo_core.schemas.protect.payload import Payload
+from galileo_core.schemas.protect.request import Request
+from galileo_core.schemas.protect.response import Response, TraceMetadata
+from galileo_core.schemas.protect.rule import Rule, RuleOperator
+from galileo_core.schemas.protect.ruleset import Ruleset
+from galileo_core.schemas.protect.stage import Stage, StageType, StageWithRulesets
+from galileo_core.schemas.protect.subscription_config import SubscriptionConfig