mirror of
https://github.com/BerriAI/litellm.git
synced 2026-09-22 00:31:44 +00:00
60 lines
1.7 KiB
Python
60 lines
1.7 KiB
Python
"""
|
|
Proxy model table model.
|
|
|
|
Canonical definition for ``litellm_proxymodeltable``. Re-exported from
|
|
``litellm.proxy._types`` for backwards compatibility.
|
|
"""
|
|
|
|
import json
|
|
from datetime import datetime
|
|
|
|
from pydantic import ConfigDict, model_validator
|
|
|
|
from litellm.types.llms.base import LiteLLMPydanticObjectBase
|
|
|
|
|
|
class LiteLLM_ProxyModelTable(LiteLLMPydanticObjectBase):
|
|
model_id: str
|
|
model_name: str
|
|
litellm_params: dict
|
|
model_info: dict | None = None
|
|
blocked: bool = False
|
|
created_at: datetime | None = None
|
|
created_by: str | None = None
|
|
updated_at: datetime | None = None
|
|
updated_by: str | None = None
|
|
|
|
model_config = ConfigDict(protected_namespaces=())
|
|
|
|
@model_validator(mode="before")
|
|
@classmethod
|
|
def check_potential_json_str(cls, values):
|
|
if not isinstance(values, dict):
|
|
return values
|
|
if isinstance(values.get("litellm_params"), str):
|
|
try:
|
|
values["litellm_params"] = json.loads(values["litellm_params"])
|
|
except json.JSONDecodeError:
|
|
pass
|
|
if isinstance(values.get("model_info"), str):
|
|
try:
|
|
values["model_info"] = json.loads(values["model_info"])
|
|
except json.JSONDecodeError:
|
|
pass
|
|
return values
|
|
|
|
@property
|
|
def is_blocked(self) -> bool:
|
|
return self.blocked
|
|
|
|
@property
|
|
def team_id(self) -> str | None:
|
|
if self.model_info:
|
|
return self.model_info.get("team_id")
|
|
return None
|
|
|
|
@property
|
|
def team_public_model_name(self) -> str | None:
|
|
if self.model_info:
|
|
return self.model_info.get("team_public_model_name")
|
|
return None
|