Fix : enterprise tests

This commit is contained in:
Sameer Kankute 2026-02-26 12:18:01 +05:30
parent 24424abec8
commit b09bfbfd85

View file

@ -3,7 +3,7 @@ from dataclasses import dataclass
from enum import Enum
from typing import Any, Dict, List, Literal, Optional, Tuple
from pydantic import BaseModel, Field
from pydantic import BaseModel, Field, field_validator
from typing_extensions import Annotated
import litellm
@ -721,6 +721,13 @@ class UserAPIKeyLabelValues(BaseModel):
Optional[str], Field(..., alias=UserAPIKeyLabelNames.STREAM.value)
] = None
@field_validator("stream", mode="before")
@classmethod
def coerce_stream_to_str(cls, v: Any) -> Optional[str]:
if v is None:
return None
return str(v)
class PrometheusMetricsConfig(BaseModel):
"""Configuration for filtering Prometheus metrics"""