scriptoriumadmin/litellm
1
0
Fork 0
mirror of https://github.com/BerriAI/litellm.git synced 2026-09-16 23:41:43 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 102

add type for AllCallbacks

Browse source
This commit is contained in:
Ishaan Jaff 2024-06-04 16:33:42 -07:00
parent 3b823c7587
commit b8c9326df4
1 changed files with 48 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

48
litellm/proxy/_types.py
View file

@ -1346,3 +1346,51 @@ class InvitationModel(LiteLLMBase):
class ConfigFieldInfo(LiteLLMBase):
field_name: str
field_value: Any
class CallbackOnUI(LiteLLMBase):
litellm_callback_name: str
litellm_callback_params: Optional[list]
ui_callback_name: str
class AllCallbacks(LiteLLMBase):
langfuse: CallbackOnUI = CallbackOnUI(
litellm_callback_name="langfuse",
ui_callback_name="Langfuse",
litellm_callback_params=[
"LANGFUSE_PUBLIC_KEY",
"LANGFUSE_SECRET_KEY",
],
)
opentelemetry: CallbackOnUI = CallbackOnUI(
litellm_callback_name="otel",
ui_callback_name="OpenTelemetry",
litellm_callback_params=[
"OTEL_EXPORTER",
"OTEL_ENDPOINT",
"OTEL_HEADERS",
],
)
openmeter: CallbackOnUI = CallbackOnUI(
litellm_callback_name="openmeter",
ui_callback_name="openmeter",
litellm_callback_params=[
"OPENMETER_API_ENDPOINT",
"OPENMETER_API_KEY",
],
)
custom_callback_api: CallbackOnUI = CallbackOnUI(
litellm_callback_name="custom_callback_api",
litellm_callback_params=["GENERIC_LOGGER_ENDPOINT"],
ui_callback_name="Custom Callback API",
)
datadog: CallbackOnUI = CallbackOnUI(
litellm_callback_name="datadog",
litellm_callback_params=["DD_API_KEY", "DD_SITE"],
ui_callback_name="datadog",
)