scriptoriumadmin/litellm
1
0
Fork 0
mirror of https://github.com/BerriAI/litellm.git synced 2026-09-15 23:31:29 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 165

fix(v2 managed agents): enforce idle_timeout_minutes <= timeout_minutes on SandboxSpec

Browse source 
Per contract \xc2\xa76.2, idle_timeout_minutes must be in [1, timeout_minutes].
Previously only ge=1 was enforced, so idle=1440 with timeout=10 would pass.

Add a Pydantic v2 model_validator(mode="after") that raises ValueError when
idle_timeout_minutes > timeout_minutes; FastAPI surfaces it as a 422.
This commit is contained in:
Ishaan Jaffer 2026-05-07 10:43:03 -07:00
parent cf454e38c2
commit 49fd3ed1ba
No known key found for this signature in database
1 changed files with 12 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

14
litellm/managed_agents/types.py
View file

@ -10,7 +10,7 @@ forward-compatibility with proxy-side enrichment.
from datetime import datetime
from typing import Any, Dict, List, Literal, Optional
from pydantic import BaseModel, ConfigDict, Field
from pydantic import BaseModel, ConfigDict, Field, model_validator
# ---------------------------------------------------------------------------
# Status / sandbox enums
@ -73,7 +73,7 @@ class SandboxSpec(BaseModel):
`image` is optional and falls back to a pinned built-in.
`timeout_minutes` is bounded to [1, 1440].
`idle_timeout_minutes` must be in [1, timeout_minutes] (enforced by handler).
`idle_timeout_minutes` must be in [1, timeout_minutes].
"""
model_config = ConfigDict(extra="forbid")
@ -84,6 +84,16 @@ class SandboxSpec(BaseModel):
idle_timeout_minutes: int = Field(default=10, ge=1)
image: Optional[str] = None
@model_validator(mode="after")
def _validate_idle_timeout_within_timeout(self) -> "SandboxSpec":
"""Enforce contract §6.2: `idle_timeout_minutes` ∈ [1, `timeout_minutes`]."""
if self.idle_timeout_minutes > self.timeout_minutes:
raise ValueError(
f"idle_timeout_minutes ({self.idle_timeout_minutes}) must be "
f"<= timeout_minutes ({self.timeout_minutes})"
)
return self
class Repo(BaseModel):
model_config = ConfigDict(extra="forbid")