fix(litellm_logging.py): fix linting erros

This commit is contained in:
Krrish Dholakia 2024-08-01 17:32:22 -07:00
parent 10b571ca42
commit 69efc4dcdd

View file

@ -10,7 +10,7 @@ import sys
import time import time
import traceback import traceback
import uuid import uuid
from typing import Any, Callable, Dict, List, Literal, Optional from typing import Any, Callable, Dict, List, Literal, Optional, Union
from pydantic import BaseModel from pydantic import BaseModel
@ -505,7 +505,17 @@ class Logging:
) )
) )
def _response_cost_calculator(self, result: BaseModel): def _response_cost_calculator(
self,
result: Union[
ModelResponse,
EmbeddingResponse,
ImageResponse,
TranscriptionResponse,
TextCompletionResponse,
HttpxBinaryResponseContent,
],
):
""" """
Calculate response cost using result + logging object variables. Calculate response cost using result + logging object variables.