scriptoriumadmin/litellm
1
0
Fork 0
mirror of https://github.com/BerriAI/litellm.git synced 2026-10-07 02:59:05 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 103

(sap) fix after bot review

Browse source
This commit is contained in:
Vasilisa Parshikova 2026-03-11 21:36:28 +04:00 • committed by Sameer Kankute
parent 315f1755e8
commit 6123651b30
No known key found for this signature in database
3 changed files with 9 additions and 4 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

3
litellm/llms/sap/chat/transformation.py
View file

@ -266,7 +266,8 @@ class GenAIHubOrchestrationConfig(OpenAIGPTConfig):
stream_config = {}
if "stream_options" in optional_params:
stream_options = optional_params.pop("stream_options", {})
stream_config["chunk_size"] = stream_options.get("chunk_size", 100)
if "chunk_size" in stream_options:
stream_config["chunk_size"] = stream_options.get("chunk_size")
if "delimiters" in stream_options:
stream_config["delimiters"] = stream_options.get("delimiters")

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

@ -167,7 +167,8 @@ class GenAIHubEmbeddingConfig(BaseEmbeddingConfig):
input_dict = {"text": input}
if optional_params.get("type"):
input_dict["type"] = optional_params.get("type")
masking = {"masking": optional_params.get("masking")} if optional_params.get("masking") else {}
masking = optional_params.get("masking")
masking = {"masking": masking} if masking is not None else {}
body = {
"config": {
"modules": {

7
tests/test_litellm/llms/sap/chat/test_sap_transformation.py
View file

@ -426,9 +426,12 @@ class TestSAPTransformationIntegration:
headers={}
)
assert "translation" not in config["config"]["modules"][0]
assert config["config"]["modules"][1]["translation"] == translation_config
translation = config["config"]["modules"][1]["translation"]
assert translation["input"]["config"]["source_language"] == "en-US"
assert translation["input"]["config"]["target_language"] == "de-DE"
assert translation["output"]["config"]["target_language"] == "fr-FR"
assert config["config"]["modules"][1]["prompt_templating"]["model"]["name"] == "gpt-5"
assert config["config"]["modules"][0]["prompt_templating"]["model"]["name"] == "gpt-4o"
assert config["config"]["modules"][0]["prompt_templating"]["model"]["params"] == {}
assert config["config"]["modules"][1]["prompt_templating"]["prompt"]["template"][0]["content"] == "Hello world!"
assert config["config"]["modules"][0]["prompt_templating"]["prompt"]["template"][0]["content"] == "Hello."
assert config["config"]["modules"][0]["prompt_templating"]["prompt"]["template"][0]["content"] == "Hello."