From e02e58971fda3c3a9cc456eea5d9f78010cdc59c Mon Sep 17 00:00:00 2001 From: Krrish Dholakia Date: Sat, 27 Sep 2025 13:06:05 -0700 Subject: [PATCH] fix: fix linting errors on main --- litellm/__init__.py | 9 ++++++--- litellm/llms/watsonx/chat/handler.py | 2 +- litellm/main.py | 12 ++++++------ 3 files changed, 13 insertions(+), 10 deletions(-) diff --git a/litellm/__init__.py b/litellm/__init__.py index 6df9d0780a..02bb773d26 100644 --- a/litellm/__init__.py +++ b/litellm/__init__.py @@ -377,7 +377,9 @@ public_model_groups: Optional[List[str]] = None public_model_groups_links: Dict[str, str] = {} #### REQUEST PRIORITIZATION ###### priority_reservation: Optional[Dict[str, float]] = None -priority_reservation_settings: "PriorityReservationSettings" = PriorityReservationSettings() +priority_reservation_settings: "PriorityReservationSettings" = ( + PriorityReservationSettings() +) ######## Networking Settings ######## @@ -443,7 +445,7 @@ def identify(event_details): ####### ADDITIONAL PARAMS ################### configurable params if you use proxy models like Helicone, map spend to org id, etc. api_base: Optional[str] = None headers = None -api_version = None +api_version: Optional[str] = None organization = None project = None config_path = None @@ -494,7 +496,7 @@ azure_ai_models: Set = set() jina_ai_models: Set = set() voyage_models: Set = set() infinity_models: Set = set() -heroku_models: Set = set() +heroku_models: Set = set() databricks_models: Set = set() cloudflare_models: Set = set() codestral_models: Set = set() @@ -1357,6 +1359,7 @@ from .passthrough import allm_passthrough_route, llm_passthrough_route ### GLOBAL CONFIG ### global_bitbucket_config: Optional[Dict[str, Any]] = None + def set_global_bitbucket_config(config: Dict[str, Any]) -> None: """Set global BitBucket configuration for prompt management.""" global global_bitbucket_config diff --git a/litellm/llms/watsonx/chat/handler.py b/litellm/llms/watsonx/chat/handler.py index 17d3fa0468..bc0effe4a1 100644 --- a/litellm/llms/watsonx/chat/handler.py +++ b/litellm/llms/watsonx/chat/handler.py @@ -21,7 +21,7 @@ class WatsonXChatHandler(OpenAILikeChatHandler): *, model: str, messages: list, - api_base: str, + api_base: Optional[str], custom_llm_provider: str, custom_prompt_dict: dict, model_response: ModelResponse, diff --git a/litellm/main.py b/litellm/main.py index a7b5f960bc..71d5509448 100644 --- a/litellm/main.py +++ b/litellm/main.py @@ -1547,13 +1547,13 @@ def completion( # type: ignore # noqa: PLR0915 api_key or litellm.api_key or litellm.azure_key - or get_secret("AZURE_OPENAI_API_KEY") - or get_secret("AZURE_API_KEY") + or get_secret_str("AZURE_OPENAI_API_KEY") + or get_secret_str("AZURE_API_KEY") ) azure_ad_token = optional_params.get("extra_body", {}).pop( "azure_ad_token", None - ) or get_secret("AZURE_AD_TOKEN") + ) or get_secret_str("AZURE_AD_TOKEN") azure_ad_token_provider = litellm_params.get( "azure_ad_token_provider", None @@ -1579,7 +1579,7 @@ def completion( # type: ignore # noqa: PLR0915 headers=headers, api_key=api_key, api_base=api_base, - api_version=api_version, + api_version=cast(str, api_version), api_type=api_type, azure_ad_token=azure_ad_token, azure_ad_token_provider=azure_ad_token_provider, @@ -2861,7 +2861,7 @@ def completion( # type: ignore # noqa: PLR0915 logging_obj=logging, acompletion=acompletion, timeout=timeout, - custom_llm_provider=custom_llm_provider, + custom_llm_provider=custom_llm_provider, # type: ignore client=client, api_base=api_base, extra_headers=extra_headers, @@ -2930,7 +2930,7 @@ def completion( # type: ignore # noqa: PLR0915 logging_obj=logging, acompletion=acompletion, timeout=timeout, - custom_llm_provider=custom_llm_provider, + custom_llm_provider=custom_llm_provider, # type: ignore client=client, api_base=api_base, extra_headers=extra_headers,