mirror of
https://github.com/BerriAI/litellm.git
synced 2026-10-07 02:59:05 +00:00
Add PipelineTestRequest type for test-pipeline endpoint
This commit is contained in:
parent
ba2d697737
commit
d960fec551
1 changed files with 11 additions and 0 deletions
|
|
@ -302,6 +302,17 @@ class PolicyAttachmentListResponse(BaseModel):
|
|||
# ─────────────────────────────────────────────────────────────────────────────
|
||||
|
||||
|
||||
class PipelineTestRequest(BaseModel):
|
||||
"""Request body for testing a guardrail pipeline with sample messages."""
|
||||
|
||||
pipeline: Dict[str, Any] = Field(
|
||||
description="Pipeline definition with 'mode' and 'steps'.",
|
||||
)
|
||||
test_messages: List[Dict[str, str]] = Field(
|
||||
description="Test messages to run through the pipeline, e.g. [{'role': 'user', 'content': '...'}].",
|
||||
)
|
||||
|
||||
|
||||
class PolicyResolveRequest(BaseModel):
|
||||
"""Request body for resolving effective policies/guardrails for a context."""
|
||||
|
||||
|
|
|
|||
Loading…
Add table
Reference in a new issue