scriptoriumadmin/litellm
1
0
Fork 0
mirror of https://github.com/BerriAI/litellm.git synced 2026-10-07 02:59:05 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 170

Add optional pipeline field to Policy model

Browse source
This commit is contained in:
Ishaan Jaffer 2026-02-13 18:07:00 -08:00
parent 396064a6cb
commit 968b4cf90e
1 changed files with 7 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

8
litellm/types/proxy/policy_engine/policy_types.py
View file

@ -29,10 +29,12 @@ Key concepts:
- `condition`: Optional model condition for when guardrails apply
"""
from typing import Any, Dict, List, Optional, Union
from typing import Dict, List, Optional, Union
from pydantic import BaseModel, ConfigDict, Field
from litellm.types.proxy.policy_engine.pipeline_types import GuardrailPipeline
# ─────────────────────────────────────────────────────────────────────────────
# Policy Condition
# ─────────────────────────────────────────────────────────────────────────────
@ -231,6 +233,10 @@ class Policy(BaseModel):
default=None,
description="Optional condition for when this policy's guardrails apply.",
)
pipeline: Optional[GuardrailPipeline] = Field(
default=None,
description="Optional pipeline for ordered, conditional guardrail execution.",
)
model_config = ConfigDict(extra="forbid")