diff --git a/litellm/llms/openai/embeddings/guardrail_translation/__init__.py b/litellm/llms/openai/embeddings/guardrail_translation/__init__.py new file mode 100644 index 00000000000..a60662282ca --- /dev/null +++ b/litellm/llms/openai/embeddings/guardrail_translation/__init__.py @@ -0,0 +1,13 @@ +"""OpenAI Embeddings handler for Unified Guardrails.""" + +from litellm.llms.openai.embeddings.guardrail_translation.handler import ( + OpenAIEmbeddingsHandler, +) +from litellm.types.utils import CallTypes + +guardrail_translation_mappings = { + CallTypes.embedding: OpenAIEmbeddingsHandler, + CallTypes.aembedding: OpenAIEmbeddingsHandler, +} + +__all__ = ["guardrail_translation_mappings", "OpenAIEmbeddingsHandler"]