From 6fc0e640cc73677ac5c14ad5d6b03ef45353f76b Mon Sep 17 00:00:00 2001 From: hgaib Date: Wed, 26 Aug 2026 06:25:52 +0000 Subject: [PATCH] fix(token_kiosk): use public method for provider info to pass reportPrivateUsage lint Signed-off-by: hgaib --- litellm/litellm_core_utils/get_llm_provider_logic.py | 2 +- litellm/llms/token_kiosk/chat/transformation.py | 7 ++++++- tests/test_litellm/test_token_kiosk.py | 2 +- 3 files changed, 8 insertions(+), 3 deletions(-) diff --git a/litellm/litellm_core_utils/get_llm_provider_logic.py b/litellm/litellm_core_utils/get_llm_provider_logic.py index 88c6bb16d6a..31d2c58cba9 100644 --- a/litellm/litellm_core_utils/get_llm_provider_logic.py +++ b/litellm/litellm_core_utils/get_llm_provider_logic.py @@ -830,7 +830,7 @@ def _get_openai_compatible_provider_info( ( api_base, dynamic_api_key, - ) = litellm.TokenKioskConfig()._get_openai_compatible_provider_info(api_base, api_key) + ) = litellm.TokenKioskConfig().get_openai_compatible_provider_info(api_base, api_key) elif custom_llm_provider == "wandb": api_base = api_base or get_secret("WANDB_API_BASE") or "https://api.inference.wandb.ai/v1" dynamic_api_key = api_key or get_secret_str("WANDB_API_KEY") diff --git a/litellm/llms/token_kiosk/chat/transformation.py b/litellm/llms/token_kiosk/chat/transformation.py index 4503d9b2009..2ea42364a75 100644 --- a/litellm/llms/token_kiosk/chat/transformation.py +++ b/litellm/llms/token_kiosk/chat/transformation.py @@ -9,13 +9,18 @@ from litellm.secret_managers.main import get_secret_str class TokenKioskConfig(OpenAIGPTConfig): - def _get_openai_compatible_provider_info( + def get_openai_compatible_provider_info( self, api_base: str | None, api_key: str | None ) -> tuple[str | None, str | None]: final_api_base: Final = api_base or get_secret_str("TOKEN_KIOSK_API_BASE") or "https://agent-router.gaib.ai/v1" dynamic_api_key: Final = api_key or get_secret_str("TOKEN_KIOSK_API_KEY") return final_api_base, dynamic_api_key + def _get_openai_compatible_provider_info( + self, api_base: str | None, api_key: str | None + ) -> tuple[str | None, str | None]: + return self.get_openai_compatible_provider_info(api_base, api_key) + def get_complete_url( self, api_base: str | None, diff --git a/tests/test_litellm/test_token_kiosk.py b/tests/test_litellm/test_token_kiosk.py index aa252016463..d2720742a6d 100644 --- a/tests/test_litellm/test_token_kiosk.py +++ b/tests/test_litellm/test_token_kiosk.py @@ -41,7 +41,7 @@ def test_token_kiosk_llm_provider_enum(): def test_token_kiosk_get_openai_compatible_provider_info(): config = TokenKioskConfig() - api_base, api_key = config._get_openai_compatible_provider_info( + api_base, api_key = config.get_openai_compatible_provider_info( api_base=None, api_key="test-key" ) assert api_base == "https://agent-router.gaib.ai/v1"