phoenix/litellm
1
0
Fork 0
forked from phoenix/litellm-mirror
Code Pull requests Activity

fix literal ai typing errors

Browse source
This commit is contained in:
Ishaan Jaff 2024-10-09 15:23:39 +05:30
parent 1fd437e263
commit a0bebc3413
1 changed files with 5 additions and 3 deletions
Download patch file Download diff file Expand all files Collapse all files

8
litellm/integrations/literal_ai.py
View file

@ -3,7 +3,7 @@
import asyncio
import os
import uuid
from typing import Optional
from typing import List, Optional
import httpx
@ -192,9 +192,11 @@ class LiteralAILogger(CustomBatchLogger):
metadata = kwargs.get("litellm_params", {}).get("metadata", {})
settings = logging_payload["model_parameters"]
messages = logging_payload["messages"]
choices = logging_payload.get("response", {}).get("choices", [])
response = logging_payload["response"]
choices: List = []
if isinstance(response, dict) and "choices" in response:
choices = response["choices"]
message_completion = choices[0]["message"] if choices else None
prompt_id = None
variables = None