diff --git a/docs/my-website/docs/providers/sap.md b/docs/my-website/docs/providers/sap.md index 11f240adbdd..0b29f66ea57 100644 --- a/docs/my-website/docs/providers/sap.md +++ b/docs/my-website/docs/providers/sap.md @@ -698,9 +698,9 @@ Required parameters: - `messages` Optional parameters: -- `filterings` -- `groundings` -- `translations` +- `filtering` +- `grounding` +- `translation` - `masking` - `tools` diff --git a/tests/test_litellm/llms/sap/embed/test_sap_embed_transformation.py b/tests/test_litellm/llms/sap/embed/test_sap_embed_transformation.py index 3c6236d5503..7bf194b6ad9 100644 --- a/tests/test_litellm/llms/sap/embed/test_sap_embed_transformation.py +++ b/tests/test_litellm/llms/sap/embed/test_sap_embed_transformation.py @@ -62,3 +62,36 @@ def test_model_params(fake_token_creator, fake_deployment_url): headers={} ) assert body["config"]["modules"]["embeddings"]["model"]["params"] == {"truncate": "END"} + +def test_embed_with_masking(fake_token_creator, fake_deployment_url): + masking_config = { + 'providers': + [ + { + 'type': 'sap_data_privacy_integration', + 'method': 'anonymization', + 'entities': [ + {'type': 'profile-address'}, + {'type': 'profile-phone'}, + {'type': 'profile-person'}, + {'type': 'profile-location'} + ] + } + ] + } + with patch( + "litellm.llms.sap.embed.transformation.GenAIHubEmbeddingConfig.deployment_url", + new_callable=PropertyMock, + return_value=fake_deployment_url, + ), patch( + "litellm.llms.sap.embed.transformation.get_token_creator", + return_value=fake_token_creator, + ): + body = GenAIHubEmbeddingConfig().transform_embedding_request( + model="text-embedding-3-small", + input="Hi", + optional_params={"parameters": {"truncate": "END"}, + "masking": masking_config}, + headers={} + ) + assert body["config"]["modules"]["masking"] == masking_config