Skip to content

chore: move Protect models to galileo.protect to avoid conflicts #268

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Draft
ajaynayak wants to merge 1 commit into
base: main
Choose a base branch
from
Draft

chore: move Protect models to galileo.protect to avoid conflicts #268

Show file tree
Hide file tree
Changes from all commits
Commits
Show all changes
1 commit
Select commit
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
6 changes: 0 additions & 6 deletions src/galileo/__init__.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -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
Copy link
Contributor

@setu4993 setu4993 Aug 8, 2025

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

We should probably export all of these from here as ProtectPayload, ProtectRequest, etc.


__version__ = "1.14.0"
21 changes: 21 additions & 0 deletions src/galileo/protect/__init__.py
View file
Open in desktop
Original file line number Diff line number Diff line change
@@ -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
Loading