From f0cd93aeb229258724a09b4547fdcf3eeb270969 Mon Sep 17 00:00:00 2001 From: Julio Quinteros Pro Date: Wed, 4 Mar 2026 10:47:47 -0300 Subject: [PATCH] fix: remove unused imports in tool_management_endpoints and streaming_iterator - Remove unused ToolOutputPolicy import - Remove unused _WsClientConnection TYPE_CHECKING import Co-Authored-By: Claude Opus 4.6 --- .../proxy/management_endpoints/tool_management_endpoints.py | 1 - litellm/responses/streaming_iterator.py | 5 +---- 2 files changed, 1 insertion(+), 5 deletions(-) diff --git a/litellm/proxy/management_endpoints/tool_management_endpoints.py b/litellm/proxy/management_endpoints/tool_management_endpoints.py index 7fdd3475c0..19ca2c9f6b 100644 --- a/litellm/proxy/management_endpoints/tool_management_endpoints.py +++ b/litellm/proxy/management_endpoints/tool_management_endpoints.py @@ -26,7 +26,6 @@ from litellm.types.tool_management import ( ToolDetailResponse, ToolInputPolicy, ToolListResponse, - ToolOutputPolicy, ToolPolicyOption, ToolPolicyOptionsResponse, ToolPolicyUpdateRequest, diff --git a/litellm/responses/streaming_iterator.py b/litellm/responses/streaming_iterator.py index 09d770b0fa..0af5c46712 100644 --- a/litellm/responses/streaming_iterator.py +++ b/litellm/responses/streaming_iterator.py @@ -3,7 +3,7 @@ import json import time import traceback from datetime import datetime -from typing import TYPE_CHECKING, Any, Dict, List, Optional +from typing import Any, Dict, List, Optional import httpx @@ -688,9 +688,6 @@ class MockResponsesAPIStreamingIterator(BaseResponsesAPIStreamingIterator): # WebSocket mode streaming (bidirectional forwarding) # --------------------------------------------------------------------------- -if TYPE_CHECKING: - from websockets.asyncio.client import ClientConnection as _WsClientConnection - from litellm._logging import verbose_logger from litellm.litellm_core_utils.thread_pool_executor import executor as _ws_executor