From a0bebc34137bbddb79d6168124785e4a1a4263b8 Mon Sep 17 00:00:00 2001 From: Ishaan Jaff Date: Wed, 9 Oct 2024 15:23:39 +0530 Subject: [PATCH] fix literal ai typing errors --- litellm/integrations/literal_ai.py | 8 +++++--- 1 file changed, 5 insertions(+), 3 deletions(-) diff --git a/litellm/integrations/literal_ai.py b/litellm/integrations/literal_ai.py index f83b714e5..5bf9afd7e 100644 --- a/litellm/integrations/literal_ai.py +++ b/litellm/integrations/literal_ai.py @@ -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