aidial-sdk 0.36.0.dev2__tar.gz → 0.36.0.dev4__tar.gz

{aidial_sdk-0.36.0.dev2 → aidial_sdk-0.36.0.dev4}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: aidial-sdk
-Version: 0.36.0.dev2
+Version: 0.36.0.dev4
 Summary: Framework to create applications and model adapters for AI DIAL
 License-Expression: Apache-2.0
 License-File: LICENSE

{aidial_sdk-0.36.0.dev2 → aidial_sdk-0.36.0.dev4}/aidial_sdk/chat_completion/__init__.py

@@ -3,14 +3,17 @@ from aidial_sdk.chat_completion.choice import Choice
 from aidial_sdk.chat_completion.enums import FinishReason, Status
 from aidial_sdk.chat_completion.form import Button, FormMetaclass
 from aidial_sdk.chat_completion.request import (
-    Addon,
     Attachment,
     CacheBreakpoint,
     CustomContent,
     Function,
     FunctionCall,
     FunctionChoice,
+    InputAudio,
+    InputFile,
     Message,
+    MessageContentAudioPart,
+    MessageContentFilePart,
     MessageContentImagePart,
     MessageContentPart,
     MessageContentRefusalPart,
@@ -49,45 +52,48 @@ from aidial_sdk.deployment.truncate_prompt import (
 )
 
 __all__ = [
+    "Attachment",
+    "Button",
+    "CacheBreakpoint",
     "ChatCompletion",
     "Choice",
+    "ConfigurationRequest",
+    "ConfigurationResponse",
+    "CustomContent",
     "FinishReason",
-    "Status",
-    "Button",
     "FormMetaclass",
-    "Addon",
-    "Attachment",
-    "CacheBreakpoint",
-    "CustomContent",
     "Function",
     "FunctionCall",
     "FunctionChoice",
+    "InputAudio",
+    "InputFile",
     "Message",
+    "MessageContentAudioPart",
+    "MessageContentFilePart",
     "MessageContentImagePart",
     "MessageContentPart",
     "MessageContentRefusalPart",
     "MessageContentTextPart",
     "MessageCustomFields",
     "Request",
+    "RequestStage",
+    "Response",
     "ResponseFormat",
     "ResponseFormatJsonObject",
     "ResponseFormatJsonSchema",
     "ResponseFormatJsonSchemaObject",
     "ResponseFormatText",
     "Role",
-    "Tool",
-    "ToolCall",
-    "ToolChoice",
-    "ToolCustomFields",
-    "RequestStage",
-    "Response",
     "Stage",
-    "ConfigurationRequest",
-    "ConfigurationResponse",
+    "Status",
     "TokenizeError",
     "TokenizeRequest",
     "TokenizeResponse",
     "TokenizeSuccess",
+    "Tool",
+    "ToolCall",
+    "ToolChoice",
+    "ToolCustomFields",
     "TruncatePromptError",
     "TruncatePromptRequest",
     "TruncatePromptResponse",

{aidial_sdk-0.36.0.dev2 → aidial_sdk-0.36.0.dev4}/aidial_sdk/chat_completion/request.py

@@ -91,13 +91,38 @@ class MessageContentTextPart(ExtraAllowModel):
     text: StrictStr
 
 
+class InputFile(ExtraAllowModel):
+    file_data: StrictStr | None = None
+    file_id: StrictStr | None = None
+    filename: StrictStr | None = None
+
+
+class MessageContentFilePart(ExtraAllowModel):
+    type: Literal["file"]
+    file: InputFile
+
+
+class InputAudio(ExtraAllowModel):
+    data: StrictStr
+    format: Literal["wav", "mp3"] | StrictStr
+
+
+class MessageContentAudioPart(ExtraAllowModel):
+    type: Literal["input_audio"]
+    input_audio: InputAudio
+
+
 class MessageContentRefusalPart(ExtraAllowModel):
     type: Literal["refusal"]
     refusal: StrictStr
 
 
 MessageContentPart = (
-    MessageContentTextPart | MessageContentImagePart | MessageContentRefusalPart
+    MessageContentTextPart
+    | MessageContentImagePart
+    | MessageContentFilePart
+    | MessageContentAudioPart
+    | MessageContentRefusalPart
 )
 
 
@@ -139,11 +164,6 @@ class Message(ExtraAllowModel):
             assert_never(self.content)
 
 
-class Addon(ExtraAllowModel):
-    name: StrictStr | None = None
-    url: StrictStr | None = None
-
-
 class Function(ExtraAllowModel):
     name: StrictStr
     strict: bool = False
@@ -279,7 +299,6 @@ class ChatCompletionRequestCustomFields(ExtraAllowModel):
 
 
 class ChatCompletionRequest(AzureChatCompletionRequest):
-    addons: list[Addon] | None = None
     max_prompt_tokens: PositiveInt | None = None
     custom_fields: ChatCompletionRequestCustomFields | None = None
 

{aidial_sdk-0.36.0.dev2 → aidial_sdk-0.36.0.dev4}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "aidial-sdk"
-version = "0.36.0.dev2"
+version = "0.36.0.dev4"
 description = "Framework to create applications and model adapters for AI DIAL"
 authors = [{ name = "EPAM RAIL", email = "SpecialEPM-DIALDevTeam@epam.com" }]
 license = "Apache-2.0"