forked from phoenix/litellm-mirror
fix(litellm_logging.py): fix linting erros
This commit is contained in:
parent
10b571ca42
commit
69efc4dcdd
1 changed files with 12 additions and 2 deletions
|
@ -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.
|
||||||
|
|
||||||
|
|
Loading…
Add table
Add a link
Reference in a new issue