scriptoriumadmin/litellm
1
0
Fork 0
mirror of https://github.com/BerriAI/litellm.git synced 2026-09-17 23:51:30 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 67

fix(sap): accept file content parts on user messages

Browse source
This commit is contained in:
Devin AI 2026-07-29 09:07:49 +00:00
parent c274cf321c
commit 7b94aed093
2 changed files with 77 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

18
litellm/llms/sap/chat/models.py
View file

@ -39,6 +39,16 @@ class ImageContent(BaseModel):
image_url: ImageURLContent
class FileContentData(BaseModel):
file_data: str
filename: str | None = None
class FileContent(BaseModel):
type_: Literal["file"] = Field(default="file", alias="type")
file: FileContentData
class FunctionObj(BaseModel):
name: str
arguments: str
@ -95,7 +105,13 @@ class SAPMessage(BaseModel):
class SAPUserMessage(BaseModel):
role: Literal["user"] = "user"
content: Union[str, TextContent, ImageContent, list[Union[TextContent, ImageContent]]]
content: Union[
str,
TextContent,
ImageContent,
FileContent,
list[Union[TextContent, ImageContent, FileContent]],
]
class SAPAssistantMessage(BaseModel):

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

@ -639,3 +639,63 @@ class TestSAPTransformationIntegration:
config["config"]["modules"][1]["translation"]["input"]["type"]
== "sap_document_translation"
)
def test_transform_request_with_file_content_part(self, mock_config):
pdf_data = "data:application/pdf;base64,JVBERi0xLjQK"
result = mock_config.transform_request(
model="anthropic--claude-4-sonnet",
messages=[
{
"role": "user",
"content": [
{"type": "text", "text": "Summarize this document."},
{
"type": "file",
"file": {"file_data": pdf_data, "filename": "report.pdf"},
},
],
}
],
optional_params={},
litellm_params={},
headers={},
)
content = result["config"]["modules"]["prompt_templating"]["prompt"]["template"][0]["content"]
assert content == [
{"type": "text", "text": "Summarize this document."},
{"type": "file", "file": {"file_data": pdf_data, "filename": "report.pdf"}},
]
def test_transform_request_with_file_content_part_without_filename(self, mock_config):
pdf_data = "data:application/pdf;base64,JVBERi0xLjQK"
result = mock_config.transform_request(
model="gemini-2.5-pro",
messages=[
{
"role": "user",
"content": [{"type": "file", "file": {"file_data": pdf_data}}],
}
],
optional_params={},
litellm_params={},
headers={},
)
content = result["config"]["modules"]["prompt_templating"]["prompt"]["template"][0]["content"]
assert content == [{"type": "file", "file": {"file_data": pdf_data}}]
def test_file_content_part_requires_file_data(self, mock_config):
with pytest.raises(ValidationError):
mock_config.transform_request(
model="anthropic--claude-4-sonnet",
messages=[
{
"role": "user",
"content": [{"type": "file", "file": {"filename": "report.pdf"}}],
}
],
optional_params={},
litellm_params={},
headers={},
)