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 diff --git a/litellm/router.py b/litellm/router.py index 3760f93b62..8d44882cdf 100644 --- a/litellm/router.py +++ b/litellm/router.py @@ -115,9 +115,6 @@ from litellm.router_utils.handle_error import ( from litellm.router_utils.pre_call_checks.deployment_affinity_check import ( DeploymentAffinityCheck, ) -from litellm.router_utils.pre_call_checks.encrypted_content_affinity_check import ( - EncryptedContentAffinityCheck, -) from litellm.router_utils.pre_call_checks.model_rate_limit_check import ( ModelRateLimitingCheck, )