fix - track litellm_model_name in LiteLLM_ErrorLogs

This commit is contained in:
Ishaan Jaff 2024-04-30 17:31:40 -07:00
parent 8a1a043801
commit f2849d0641
3 changed files with 3 additions and 0 deletions

View file

@ -916,6 +916,7 @@ class LiteLLM_ErrorLogs(LiteLLMBase):
request_id: Optional[str] = str(uuid.uuid4())
api_base: Optional[str] = ""
model_group: Optional[str] = ""
litellm_model_name: Optional[str] = ""
model_id: Optional[str] = ""
request_kwargs: Optional[dict] = {}
exception_type: Optional[str] = ""

View file

@ -1258,6 +1258,7 @@ async def _PROXY_failure_handler(
request_id=str(uuid.uuid4()),
model_group=_model_group,
model_id=_model_id,
litellm_model_name=kwargs.get("model"),
request_kwargs=_optional_params,
api_base=api_base,
exception_type=_exception_type,

View file

@ -190,6 +190,7 @@ model LiteLLM_ErrorLogs {
endTime DateTime // Assuming end_time is a DateTime field
api_base String @default("")
model_group String @default("") // public model_name / model_group
litellm_model_name String @default("") // model passed to litellm
model_id String @default("") // ID of model in ProxyModelTable
request_kwargs Json @default("{}")
exception_type String @default("")