(sap) fix after bot review

This commit is contained in:
Vasilisa Parshikova 2026-03-11 20:00:20 +04:00 committed by Sameer Kankute
parent 1bd5b6339f
commit 9bdf6e087f
No known key found for this signature in database
2 changed files with 36 additions and 3 deletions

View file

@ -698,9 +698,9 @@ Required parameters:
- `messages`
Optional parameters:
- `filterings`
- `groundings`
- `translations`
- `filtering`
- `grounding`
- `translation`
- `masking`
- `tools`

View file

@ -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