From 851ff30019b8d4f8692785cf1a0a9699953335b1 Mon Sep 17 00:00:00 2001 From: Sameer Kankute Date: Wed, 24 Jun 2026 17:33:16 +0530 Subject: [PATCH] fix: remove unused Optional imports after UP045 migration Co-Authored-By: Claude Sonnet 4.6 --- litellm/llms/anthropic/chat/handler.py | 1 - litellm/llms/base_llm/rerank/transformation.py | 2 +- litellm/llms/cohere/rerank/transformation.py | 2 +- litellm/llms/cohere/rerank_v2/transformation.py | 2 +- litellm/llms/dashscope/rerank/transformation.py | 2 +- litellm/llms/deepinfra/rerank/transformation.py | 2 +- litellm/llms/fireworks_ai/rerank/transformation.py | 2 +- litellm/llms/github_copilot/chat/transformation.py | 2 +- litellm/llms/hosted_vllm/rerank/transformation.py | 2 +- litellm/llms/huggingface/rerank/transformation.py | 2 +- litellm/llms/jina_ai/rerank/transformation.py | 2 +- litellm/llms/nvidia_nim/rerank/transformation.py | 2 +- litellm/llms/vertex_ai/rerank/transformation.py | 2 +- litellm/llms/voyage/rerank/transformation.py | 2 +- litellm/llms/watsonx/rerank/transformation.py | 2 +- litellm/proxy/spend_tracking/cold_storage_handler.py | 1 - litellm/proxy/spend_tracking/spend_management_endpoints.py | 1 - litellm/rerank_api/main.py | 2 +- litellm/rerank_api/rerank_utils.py | 2 +- 19 files changed, 16 insertions(+), 19 deletions(-) diff --git a/litellm/llms/anthropic/chat/handler.py b/litellm/llms/anthropic/chat/handler.py index 59751ae158a..18741e2cd0f 100644 --- a/litellm/llms/anthropic/chat/handler.py +++ b/litellm/llms/anthropic/chat/handler.py @@ -11,7 +11,6 @@ from typing import ( Dict, List, Literal, - Optional, Tuple, Union, cast, diff --git a/litellm/llms/base_llm/rerank/transformation.py b/litellm/llms/base_llm/rerank/transformation.py index 6fec006d52b..eac44ba85c5 100644 --- a/litellm/llms/base_llm/rerank/transformation.py +++ b/litellm/llms/base_llm/rerank/transformation.py @@ -1,5 +1,5 @@ from abc import ABC, abstractmethod -from typing import TYPE_CHECKING, Any, Dict, List, Optional, Tuple, Union +from typing import TYPE_CHECKING, Any, Dict, List, Tuple, Union import httpx diff --git a/litellm/llms/cohere/rerank/transformation.py b/litellm/llms/cohere/rerank/transformation.py index 4a72e843924..dd3f0f1a446 100644 --- a/litellm/llms/cohere/rerank/transformation.py +++ b/litellm/llms/cohere/rerank/transformation.py @@ -1,4 +1,4 @@ -from typing import Any, Dict, List, Optional, Union +from typing import Any, Dict, List, Union import httpx diff --git a/litellm/llms/cohere/rerank_v2/transformation.py b/litellm/llms/cohere/rerank_v2/transformation.py index b311c151955..7c68a431a90 100644 --- a/litellm/llms/cohere/rerank_v2/transformation.py +++ b/litellm/llms/cohere/rerank_v2/transformation.py @@ -1,4 +1,4 @@ -from typing import Any, Dict, List, Optional, Union +from typing import Any, Dict, List, Union from litellm.llms.cohere.rerank.transformation import CohereRerankConfig from litellm.types.rerank import OptionalRerankParams, RerankRequest diff --git a/litellm/llms/dashscope/rerank/transformation.py b/litellm/llms/dashscope/rerank/transformation.py index fda0c9d70f3..365e15fdd7a 100644 --- a/litellm/llms/dashscope/rerank/transformation.py +++ b/litellm/llms/dashscope/rerank/transformation.py @@ -22,7 +22,7 @@ as supported only for gte-rerank-v2 / qwen3-vl-rerank. Docs - https://help.aliyun.com/zh/model-studio/text-rerank-api """ -from typing import Any, Dict, List, Optional, Union +from typing import Any, Dict, List, Union import httpx diff --git a/litellm/llms/deepinfra/rerank/transformation.py b/litellm/llms/deepinfra/rerank/transformation.py index cbd98d604c4..385aa051d00 100644 --- a/litellm/llms/deepinfra/rerank/transformation.py +++ b/litellm/llms/deepinfra/rerank/transformation.py @@ -2,7 +2,7 @@ Translate between Cohere's `/rerank` format and Deepinfra's `/rerank` format. """ -from typing import Any, Dict, List, Optional, Union +from typing import Any, Dict, List, Union import httpx diff --git a/litellm/llms/fireworks_ai/rerank/transformation.py b/litellm/llms/fireworks_ai/rerank/transformation.py index 174d91c4a1e..400d511a02f 100644 --- a/litellm/llms/fireworks_ai/rerank/transformation.py +++ b/litellm/llms/fireworks_ai/rerank/transformation.py @@ -4,7 +4,7 @@ Fireworks AI Rerank API transformation Reference: https://docs.fireworks.ai/inference-api-reference/rerank """ -from typing import Any, Dict, List, Optional, Union +from typing import Any, Dict, List, Union import httpx diff --git a/litellm/llms/github_copilot/chat/transformation.py b/litellm/llms/github_copilot/chat/transformation.py index 06c0ad7c8f8..54cbf69a4ac 100644 --- a/litellm/llms/github_copilot/chat/transformation.py +++ b/litellm/llms/github_copilot/chat/transformation.py @@ -1,5 +1,5 @@ import json -from typing import Any, List, Optional, Tuple +from typing import Any, List, Tuple import os diff --git a/litellm/llms/hosted_vllm/rerank/transformation.py b/litellm/llms/hosted_vllm/rerank/transformation.py index 8a96f9363d0..47495350460 100644 --- a/litellm/llms/hosted_vllm/rerank/transformation.py +++ b/litellm/llms/hosted_vllm/rerank/transformation.py @@ -2,7 +2,7 @@ Transformation logic for Hosted VLLM rerank """ -from typing import Any, Dict, List, Optional, Union +from typing import Any, Dict, List, Union import httpx diff --git a/litellm/llms/huggingface/rerank/transformation.py b/litellm/llms/huggingface/rerank/transformation.py index 75b66b890c8..c94fc65acbd 100644 --- a/litellm/llms/huggingface/rerank/transformation.py +++ b/litellm/llms/huggingface/rerank/transformation.py @@ -1,5 +1,5 @@ import os -from typing import TYPE_CHECKING, Any, Dict, List, Optional, Tuple, Union +from typing import TYPE_CHECKING, Any, Dict, List, Tuple, Union import httpx from typing_extensions import TypedDict diff --git a/litellm/llms/jina_ai/rerank/transformation.py b/litellm/llms/jina_ai/rerank/transformation.py index a23f31d6de7..cabaf079edc 100644 --- a/litellm/llms/jina_ai/rerank/transformation.py +++ b/litellm/llms/jina_ai/rerank/transformation.py @@ -6,7 +6,7 @@ Why separate file? Make it easy to see how transformation works Docs - https://jina.ai/reranker """ -from typing import Any, Dict, List, Optional, Tuple, Union +from typing import Any, Dict, List, Tuple, Union from httpx import URL, Response diff --git a/litellm/llms/nvidia_nim/rerank/transformation.py b/litellm/llms/nvidia_nim/rerank/transformation.py index 9eccb54fced..25e75d7876e 100644 --- a/litellm/llms/nvidia_nim/rerank/transformation.py +++ b/litellm/llms/nvidia_nim/rerank/transformation.py @@ -1,4 +1,4 @@ -from typing import Any, Dict, List, Literal, Optional, Union +from typing import Any, Dict, List, Literal, Union import httpx from typing_extensions import Required, TypedDict diff --git a/litellm/llms/vertex_ai/rerank/transformation.py b/litellm/llms/vertex_ai/rerank/transformation.py index ce576956cab..382a8498d40 100644 --- a/litellm/llms/vertex_ai/rerank/transformation.py +++ b/litellm/llms/vertex_ai/rerank/transformation.py @@ -4,7 +4,7 @@ Translates from Cohere's `/v1/rerank` input format to Vertex AI Discovery Engine Why separate file? Make it easy to see how transformation works """ -from typing import Any, Dict, List, Optional, Union +from typing import Any, Dict, List, Union import httpx diff --git a/litellm/llms/voyage/rerank/transformation.py b/litellm/llms/voyage/rerank/transformation.py index 7d2ead6876a..05991d4cc8c 100644 --- a/litellm/llms/voyage/rerank/transformation.py +++ b/litellm/llms/voyage/rerank/transformation.py @@ -4,7 +4,7 @@ Transformation logic for Voyage AI's /v1/rerank endpoint. Docs - https://docs.voyageai.com/docs/reranker """ -from typing import Any, Dict, List, Optional, Tuple, Union +from typing import Any, Dict, List, Tuple, Union import httpx diff --git a/litellm/llms/watsonx/rerank/transformation.py b/litellm/llms/watsonx/rerank/transformation.py index a438b16d7dc..790606c7e6d 100644 --- a/litellm/llms/watsonx/rerank/transformation.py +++ b/litellm/llms/watsonx/rerank/transformation.py @@ -5,7 +5,7 @@ Docs - https://cloud.ibm.com/apidocs/watsonx-ai#text-rerank """ import uuid -from typing import Any, Dict, List, Optional, Union, cast +from typing import Any, Dict, List, Union, cast import httpx diff --git a/litellm/proxy/spend_tracking/cold_storage_handler.py b/litellm/proxy/spend_tracking/cold_storage_handler.py index aff4c29dc06..85bbd5d0792 100644 --- a/litellm/proxy/spend_tracking/cold_storage_handler.py +++ b/litellm/proxy/spend_tracking/cold_storage_handler.py @@ -4,7 +4,6 @@ This module is responsible for handling Getting/Setting the proxy server request It allows fetching a dict of the proxy server request from s3 or GCS bucket. """ -from typing import Optional import litellm from litellm import _custom_logger_compatible_callbacks_literal diff --git a/litellm/proxy/spend_tracking/spend_management_endpoints.py b/litellm/proxy/spend_tracking/spend_management_endpoints.py index 792d5b7d3c6..23430a53440 100644 --- a/litellm/proxy/spend_tracking/spend_management_endpoints.py +++ b/litellm/proxy/spend_tracking/spend_management_endpoints.py @@ -11,7 +11,6 @@ from typing import ( Literal, Mapping, NamedTuple, - Optional, Union, ) diff --git a/litellm/rerank_api/main.py b/litellm/rerank_api/main.py index d19f7c14615..35d40423bca 100644 --- a/litellm/rerank_api/main.py +++ b/litellm/rerank_api/main.py @@ -1,7 +1,7 @@ import asyncio import contextvars from functools import partial -from typing import Any, Coroutine, Dict, List, Literal, Optional, Union +from typing import Any, Coroutine, Dict, List, Literal, Union import litellm from litellm._logging import verbose_logger diff --git a/litellm/rerank_api/rerank_utils.py b/litellm/rerank_api/rerank_utils.py index 1b418a54505..856029e45ea 100644 --- a/litellm/rerank_api/rerank_utils.py +++ b/litellm/rerank_api/rerank_utils.py @@ -1,4 +1,4 @@ -from typing import Any, Dict, List, Optional, Union +from typing import Any, Dict, List, Union from litellm.llms.base_llm.rerank.transformation import BaseRerankConfig