Added LiteLLM to the stack
This commit is contained in:
@@ -0,0 +1,34 @@
|
||||
import os
|
||||
from typing import Optional
|
||||
|
||||
from litellm_enterprise.types.proxy.proxy_server import CustomAuthSettings
|
||||
|
||||
custom_auth_settings: Optional[CustomAuthSettings] = None
|
||||
|
||||
|
||||
class EnterpriseProxyConfig:
|
||||
async def load_custom_auth_settings(
|
||||
self, general_settings: dict
|
||||
) -> CustomAuthSettings:
|
||||
custom_auth_settings = general_settings.get("custom_auth_settings", None)
|
||||
if custom_auth_settings is not None:
|
||||
custom_auth_settings = CustomAuthSettings(
|
||||
mode=custom_auth_settings.get("mode"),
|
||||
)
|
||||
return custom_auth_settings
|
||||
|
||||
async def load_enterprise_config(self, general_settings: dict) -> None:
|
||||
global custom_auth_settings
|
||||
custom_auth_settings = await self.load_custom_auth_settings(general_settings)
|
||||
return None
|
||||
|
||||
@staticmethod
|
||||
def get_custom_docs_description() -> Optional[str]:
|
||||
from litellm.proxy.proxy_server import premium_user
|
||||
|
||||
docs_description: Optional[str] = None
|
||||
if premium_user:
|
||||
# check if premium_user has custom_docs_description
|
||||
docs_description = os.getenv("DOCS_DESCRIPTION")
|
||||
|
||||
return docs_description
|
Reference in New Issue
Block a user