From 86c9ae96fc114af87171b923681c4aeb85822fcc Mon Sep 17 00:00:00 2001 From: hx <1367557521@qq.com> Date: Mon, 5 Oct 2026 19:20:58 +0800 Subject: [PATCH] style: sort imports --- tests/unit/llms/xai/test_xai_oauth.py | 8 ++++---- tests/unit/proxy/auth/test_auth_utils.py | 4 ++-- 2 files changed, 6 insertions(+), 6 deletions(-) diff --git a/tests/unit/llms/xai/test_xai_oauth.py b/tests/unit/llms/xai/test_xai_oauth.py index 0456c993832..0ad9cedc00e 100644 --- a/tests/unit/llms/xai/test_xai_oauth.py +++ b/tests/unit/llms/xai/test_xai_oauth.py @@ -4,24 +4,24 @@ import json import os import threading import time -from urllib.parse import parse_qs, urlparse from unittest.mock import MagicMock +from urllib.parse import parse_qs, urlparse import httpx -import litellm import pytest from click.testing import CliRunner +import litellm import litellm.llms.xai.oauth as xai_oauth_module from litellm.litellm_core_utils.get_llm_provider_logic import get_llm_provider +from litellm.llms.xai.chat.transformation import XAIChatConfig from litellm.llms.xai.oauth import ( XAI_OAUTH_CLIENT_ID, XAI_OAUTH_SCOPE, - XAIOAuthError, XAIOAuthAuthenticator, + XAIOAuthError, XAIOAuthLoginRequiredError, ) -from litellm.llms.xai.chat.transformation import XAIChatConfig from litellm.llms.xai.responses.transformation import XAIResponsesAPIConfig from litellm.types.router import GenericLiteLLMParams from litellm.utils import get_optional_params, validate_environment diff --git a/tests/unit/proxy/auth/test_auth_utils.py b/tests/unit/proxy/auth/test_auth_utils.py index f4e865b5218..4f1d0e18db4 100644 --- a/tests/unit/proxy/auth/test_auth_utils.py +++ b/tests/unit/proxy/auth/test_auth_utils.py @@ -17,8 +17,6 @@ from litellm.proxy.auth.auth_utils import ( abbreviate_api_key, check_complete_credentials, custom_auth_common_checks_warning, - log_once_if_budget_reservation_disabled, - warn_once_if_custom_auth_skips_common_checks, get_end_user_id_from_request_body, get_key_mcp_rpm_limit, get_key_model_rpm_limit, @@ -30,6 +28,8 @@ from litellm.proxy.auth.auth_utils import ( get_project_model_tpm_limit, get_request_route_template, is_request_body_safe, + log_once_if_budget_reservation_disabled, + warn_once_if_custom_auth_skips_common_checks, ) from litellm.types.workload_identity import ANTHROPIC_WIF_KWARGS_KEYS, OPENAI_WIF_KWARGS_KEYS