diff --git a/litellm/integrations/gcs_bucket/gcs_bucket_mock_client.py b/litellm/integrations/gcs_bucket/gcs_bucket_mock_client.py index de3e1739482..2d14f5eb962 100644 --- a/litellm/integrations/gcs_bucket/gcs_bucket_mock_client.py +++ b/litellm/integrations/gcs_bucket/gcs_bucket_mock_client.py @@ -8,11 +8,7 @@ Usage: Set GCS_MOCK=true in environment variables or config to enable mock mode. """ -import httpx -import json import asyncio -from datetime import timedelta -from typing import Dict, Optional from litellm._logging import verbose_logger from litellm.integrations.mock_client_factory import MockClientConfig, create_mock_client_factory, MockResponse diff --git a/litellm/integrations/mock_client_factory.py b/litellm/integrations/mock_client_factory.py index c45707c263d..2f04fae9f76 100644 --- a/litellm/integrations/mock_client_factory.py +++ b/litellm/integrations/mock_client_factory.py @@ -10,7 +10,7 @@ import httpx import json import asyncio from datetime import timedelta -from typing import Dict, Optional, Callable, List, cast +from typing import Dict, Optional, List, cast from dataclasses import dataclass from litellm._logging import verbose_logger @@ -96,7 +96,7 @@ def _is_url_match(url, matchers: List[str]) -> bool: return False -def create_mock_client_factory(config: MockClientConfig): +def create_mock_client_factory(config: MockClientConfig): # noqa: PLR0915 """ Factory function that creates mock client functions based on configuration.