scriptoriumadmin/litellm
1
0
Fork 0
mirror of https://github.com/BerriAI/litellm.git synced 2026-09-15 23:31:29 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 165

(sap) fix after bot review

Browse source
This commit is contained in:
Vasilisa Parshikova 2026-03-11 16:19:12 +04:00 • committed by Sameer Kankute
parent 889861f388
commit 8f7018d83d
No known key found for this signature in database
2 changed files with 11 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

2
litellm/llms/sap/embed/transformation.py
View file

@ -48,7 +48,7 @@ class EmbeddingsResponse(BaseModel):
class EmbeddingModel(BaseModel):
name: str
version: str = "latest"
params: dict = Field(default_factory=dict, validation_alias="parameters")
params: dict = Field(default_factory=dict)
timeout: int = Field(default=600, ge=1, le=600)
max_retries: int = Field(default=2, ge=0, le=5)

11
tests/test_litellm/llms/sap/embed/test_sap_embed_transformation.py
View file

@ -47,4 +47,13 @@ def test_basic_config_transform(fake_token_creator, fake_deployment_url):
optional_params={},
headers={}
)
assert body == expected_dict
assert body == expected_dict
def test_model_params(fake_token_creator, fake_deployment_url):
body = GenAIHubEmbeddingConfig().transform_embedding_request(
model="text-embedding-3-small",
input="Hi",
optional_params={"parameters": {"truncate": "END"}},
headers={}
)
assert body["config"]["modules"]["embeddings"]["model"]["params"] == {"truncate": "END"}