phoenix/litellm
1
0
Fork 0
forked from phoenix/litellm-mirror
Code Pull requests Activity

fix: fix linting issue

Browse source
This commit is contained in:
Krrish Dholakia 2024-03-21 08:05:47 -07:00
parent d91f9a9f50
commit 2ce5de903f
3 changed files with 3 additions and 9 deletions
Download patch file Download diff file Expand all files Collapse all files

4
enterprise/enterprise_hooks/google_text_moderation.py
View file

@ -96,9 +96,7 @@ class _ENTERPRISE_GoogleTextModeration(CustomLogger):
async def async_moderation_hook( async def async_moderation_hook(
self, self,
data: dict, data: dict,
call_type: ( call_type: Literal["completion", "embeddings", "image_generation"],
Literal["completion"] | Literal["embeddings"] | Literal["image_generation"]
),
): ):
""" """
- Calls Google's Text Moderation API - Calls Google's Text Moderation API

4
enterprise/enterprise_hooks/llama_guard.py
View file

@ -99,9 +99,7 @@ class _ENTERPRISE_LlamaGuard(CustomLogger):
async def async_moderation_hook( async def async_moderation_hook(
self, self,
data: dict, data: dict,
call_type: ( call_type: Literal["completion", "embeddings", "image_generation"],
Literal["completion"] | Literal["embeddings"] | Literal["image_generation"]
),
): ):
""" """
- Calls the Llama Guard Endpoint - Calls the Llama Guard Endpoint

4
enterprise/enterprise_hooks/llm_guard.py
View file

@ -95,9 +95,7 @@ class _ENTERPRISE_LLMGuard(CustomLogger):
async def async_moderation_hook( async def async_moderation_hook(
self, self,
data: dict, data: dict,
call_type: ( call_type: Literal["completion", "embeddings", "image_generation"],
Literal["completion"] | Literal["embeddings"] | Literal["image_generation"]
),
): ):
""" """
- Calls the LLM Guard Endpoint - Calls the LLM Guard Endpoint