chore: initial public snapshot for github upload

llm-gateway-competitors/litellm-wheel-src/litellm/llms/sap/chat/models.py

@@ -0,0 +1,130 @@
+from typing import Union, Literal
+
+from pydantic import BaseModel, Field, field_validator
+
+
+def validate_different_content(v: Union[str, dict, list]) -> str:
+    if v in ((), {}, []):
+        return ""
+    elif isinstance(v, dict) and "text" in v:
+        return v["text"]
+    elif isinstance(v, list):
+        new_v = []
+        for item in v:
+            if isinstance(item, dict) and "text" in item:
+                if item["text"]:
+                    new_v.append(item["text"])
+            elif isinstance(item, str):
+                new_v.append(item)
+        return "\n".join(new_v)
+    elif isinstance(v, str):
+        return v
+    raise ValueError("Content must be a string")
+    return v
+
+
+class TextContent(BaseModel):
+    type_: Literal["text"] = Field(default="text", alias="type")
+    text: str
+
+
+class ImageURLContent(BaseModel):
+    url: str
+    detail: str = "auto"
+
+
+class ImageContent(BaseModel):
+    type_: Literal["image_url"] = Field(default="image_url", alias="type")
+    image_url: ImageURLContent
+
+
+class FunctionObj(BaseModel):
+    name: str
+    arguments: str
+
+
+class FunctionTool(BaseModel):
+    description: str = ""
+    name: str
+    parameters: dict = {"type": "object", "properties": {}}
+    strict: bool = False
+
+    @field_validator("parameters", mode="before")
+    @classmethod
+    def ensure_object_type(cls, v: dict) -> dict:
+        """Ensure parameters has type='object' as required by SAP Orchestration Service."""
+        if not v:
+            return {"type": "object", "properties": {}}
+        if "type" not in v:
+            v = {"type": "object", **v}
+        if "properties" not in v:
+            v["properties"] = {}
+        return v
+
+
+class ChatCompletionTool(BaseModel):
+    type_: Literal["function"] = Field(default="function", alias="type")
+    function: FunctionTool
+
+
+class MessageToolCall(BaseModel):
+    id: str
+    type_: Literal["function"] = Field(default="function", alias="type")
+    function: FunctionObj
+
+
+class SAPMessage(BaseModel):
+    """
+    Model for SystemChatMessage and DeveloperChatMessage
+    """
+
+    role: Literal["system", "developer"] = "system"
+    content: str
+
+    _content_validator = field_validator("content", mode="before")(
+        validate_different_content
+    )
+
+
+class SAPUserMessage(BaseModel):
+    role: Literal["user"] = "user"
+    content: Union[
+        str, TextContent, ImageContent, list[Union[TextContent, ImageContent]]
+    ]
+
+
+class SAPAssistantMessage(BaseModel):
+    role: Literal["assistant"] = "assistant"
+    content: str = ""
+    refusal: str = ""
+    tool_calls: list[MessageToolCall] = []
+
+    _content_validator = field_validator("content", mode="before")(
+        validate_different_content
+    )
+
+
+class SAPToolChatMessage(BaseModel):
+    role: Literal["tool"] = "tool"
+    tool_call_id: str
+    content: str
+
+    _content_validator = field_validator("content", mode="before")(
+        validate_different_content
+    )
+
+
+class ResponseFormat(BaseModel):
+    type_: Literal["text", "json_object"] = Field(default="text", alias="type")
+
+
+class JSONResponseSchema(BaseModel):
+    description: str = ""
+    name: str
+    schema_: dict = Field(default_factory=dict, alias="schema")
+    strict: bool = False
+
+
+class ResponseFormatJSONSchema(BaseModel):
+    type_: Literal["json_schema"] = Field(default="json_schema", alias="type")
+    json_schema: JSONResponseSchema