use correct location for types

This commit is contained in:
Ishaan Jaff 2024-11-22 17:15:47 -08:00
parent b2b3e40d13
commit 7b28c0f3fb
2 changed files with 21 additions and 2 deletions

View file

@ -8,8 +8,7 @@ from httpx import USE_CLIENT_DEFAULT, AsyncHTTPTransport, HTTPTransport
import litellm import litellm
from litellm.caching import InMemoryCache from litellm.caching import InMemoryCache
from litellm.types.llms.custom_http import *
from .types import httpxSpecialProvider
if TYPE_CHECKING: if TYPE_CHECKING:
from litellm import LlmProviders from litellm import LlmProviders

View file

@ -0,0 +1,20 @@
from enum import Enum
import litellm
class httpxSpecialProvider(str, Enum):
"""
Httpx Clients can be created for these litellm internal providers
Example:
- langsmith logging would need a custom async httpx client
- pass through endpoint would need a custom async httpx client
"""
LoggingCallback = "logging_callback"
GuardrailCallback = "guardrail_callback"
Caching = "caching"
Oauth2Check = "oauth2_check"
SecretManager = "secret_manager"
PassThroughEndpoint = "pass_through_endpoint"