From 72678525112464d41a2382d2dc12b34faa62b665 Mon Sep 17 00:00:00 2001 From: Ishaan Jaff Date: Sat, 5 Oct 2024 15:03:39 +0530 Subject: [PATCH] linting error fix --- litellm/types/utils.py | 1 + litellm/utils.py | 1 + 2 files changed, 2 insertions(+) diff --git a/litellm/types/utils.py b/litellm/types/utils.py index 4cabf6834..faae0ff22 100644 --- a/litellm/types/utils.py +++ b/litellm/types/utils.py @@ -67,6 +67,7 @@ class ModelInfo(TypedDict, total=False): input_cost_per_image: Optional[float] # only for vertex ai models input_cost_per_audio_per_second: Optional[float] # only for vertex ai models input_cost_per_video_per_second: Optional[float] # only for vertex ai models + input_cost_per_second: Optional[float] # for OpenAI Speech models output_cost_per_token: Required[float] output_cost_per_character: Optional[float] # only for vertex ai models output_cost_per_token_above_128k_tokens: Optional[ diff --git a/litellm/utils.py b/litellm/utils.py index 57166fd2a..f1597f221 100644 --- a/litellm/utils.py +++ b/litellm/utils.py @@ -5082,6 +5082,7 @@ def get_model_info(model: str, custom_llm_provider: Optional[str] = None) -> Mod "input_cost_per_token_above_128k_tokens", None ), input_cost_per_query=_model_info.get("input_cost_per_query", None), + input_cost_per_second=_model_info.get("input_cost_per_second", None), output_cost_per_token=_output_cost_per_token, output_cost_per_character=_model_info.get( "output_cost_per_character", None