fix test

litellm/types/router.py

@@ -8,10 +8,13 @@ import uuid
 from typing import Any, Dict, List, Literal, Optional, Tuple, Union, get_type_hints
 
 import httpx
+from httpx import AsyncClient, Client
 from openai import AsyncAzureOpenAI, AsyncOpenAI, AzureOpenAI, OpenAI
 from pydantic import BaseModel, ConfigDict, Field
 from typing_extensions import Required, TypedDict
 
+from litellm.llms.custom_httpx.http_handler import AsyncHTTPHandler, HTTPHandler
+
 from ..exceptions import RateLimitError
 from .completion import CompletionRequest
 from .embedding import EmbeddingRequest
@@ -154,7 +157,18 @@ class GenericLiteLLMParams(BaseModel):
     configurable_clientside_auth_params: CONFIGURABLE_CLIENTSIDE_AUTH_PARAMS = None
 
     # for passing in custom OpenAI / Azure OpenAI clients
-    client: Optional[Union[OpenAI, AsyncOpenAI, AzureOpenAI, AsyncAzureOpenAI]] = None
+    client: Optional[
+        Union[
+            OpenAI,
+            AsyncOpenAI,
+            AzureOpenAI,
+            AsyncAzureOpenAI,
+            AsyncClient,
+            Client,
+            AsyncHTTPHandler,
+            HTTPHandler,
+        ]
+    ] = None
     ## LOGGING PARAMS ##
     litellm_trace_id: Optional[str] = None
     ## UNIFIED PROJECT/REGION ##