fix(inference): use flat response message model for chat/completions

Author: stainless-app[bot]

.stats.yml

@@ -1,4 +1,4 @@
 configured_endpoints: 108
-openapi_spec_url: https://storage.googleapis.com/stainless-sdk-openapi-specs/llamastack%2Fllama-stack-client-d6858fead41d2db69218aca5b3b7bc8fe300a1025484c486c3cb304ed39c48bc.yml
-openapi_spec_hash: bb1cc7aff177fad17663182b20e964b6
+openapi_spec_url: https://storage.googleapis.com/stainless-sdk-openapi-specs/llamastack%2Fllama-stack-client-5c711749dbcc9575d8997ac3e0b2a2e45e20ef8de212cdb0fcceb7009b34cc48.yml
+openapi_spec_hash: 8107eabfac6b422964ac2a6688844181
 config_hash: 07e70c7f1980785685ea4f2618dfde62

src/llama_stack_client/types/chat/completion_create_response.py

@@ -6,33 +6,18 @@
 
 # File generated from our OpenAPI spec by Stainless. See CONTRIBUTING.md for details.
 
-from typing import List, Union, Optional
-from typing_extensions import Literal, Annotated, TypeAlias
+from typing import Dict, List, Optional
+from typing_extensions import Literal
 
-from ..._utils import PropertyInfo
 from ..._models import BaseModel
 
 __all__ = [
     "CompletionCreateResponse",
     "Choice",
     "ChoiceMessage",
-    "ChoiceMessageOpenAIUserMessageParamOutput",
-    "ChoiceMessageOpenAIUserMessageParamOutputContentListOpenAIChatCompletionContentPartTextParamOpenAIChatCompletionContentPartImageParamOpenAIFile",
-    "ChoiceMessageOpenAIUserMessageParamOutputContentListOpenAIChatCompletionContentPartTextParamOpenAIChatCompletionContentPartImageParamOpenAIFileOpenAIChatCompletionContentPartTextParam",
-    "ChoiceMessageOpenAIUserMessageParamOutputContentListOpenAIChatCompletionContentPartTextParamOpenAIChatCompletionContentPartImageParamOpenAIFileOpenAIChatCompletionContentPartImageParam",
-    "ChoiceMessageOpenAIUserMessageParamOutputContentListOpenAIChatCompletionContentPartTextParamOpenAIChatCompletionContentPartImageParamOpenAIFileOpenAIChatCompletionContentPartImageParamImageURL",
-    "ChoiceMessageOpenAIUserMessageParamOutputContentListOpenAIChatCompletionContentPartTextParamOpenAIChatCompletionContentPartImageParamOpenAIFileOpenAIFile",
-    "ChoiceMessageOpenAIUserMessageParamOutputContentListOpenAIChatCompletionContentPartTextParamOpenAIChatCompletionContentPartImageParamOpenAIFileOpenAIFileFile",
-    "ChoiceMessageOpenAISystemMessageParam",
-    "ChoiceMessageOpenAISystemMessageParamContentListOpenAIChatCompletionContentPartTextParam",
-    "ChoiceMessageOpenAIAssistantMessageParamOutput",
-    "ChoiceMessageOpenAIAssistantMessageParamOutputContentListOpenAIChatCompletionContentPartTextParam",
-    "ChoiceMessageOpenAIAssistantMessageParamOutputToolCall",
-    "ChoiceMessageOpenAIAssistantMessageParamOutputToolCallFunction",
-    "ChoiceMessageOpenAIToolMessageParam",
-    "ChoiceMessageOpenAIToolMessageParamContentListOpenAIChatCompletionContentPartTextParam",
-    "ChoiceMessageOpenAIDeveloperMessageParam",
-    "ChoiceMessageOpenAIDeveloperMessageParamContentListOpenAIChatCompletionContentPartTextParam",
+    "ChoiceMessageFunctionCall",
+    "ChoiceMessageToolCall",
+    "ChoiceMessageToolCallFunction",
     "ChoiceLogprobs",
     "ChoiceLogprobsContent",
     "ChoiceLogprobsContentTopLogprob",
@@ -44,132 +29,17 @@
 ]
 
 
-class ChoiceMessageOpenAIUserMessageParamOutputContentListOpenAIChatCompletionContentPartTextParamOpenAIChatCompletionContentPartImageParamOpenAIFileOpenAIChatCompletionContentPartTextParam(
-    BaseModel
-):
-    """Text content part for OpenAI-compatible chat completion messages."""
+class ChoiceMessageFunctionCall(BaseModel):
+    """Deprecated: the name and arguments of a function that should be called."""
 
-    text: str
-    """The text content of the message."""
-
-    type: Optional[Literal["text"]] = None
-    """Must be 'text' to identify this as text content."""
-
-
-class ChoiceMessageOpenAIUserMessageParamOutputContentListOpenAIChatCompletionContentPartTextParamOpenAIChatCompletionContentPartImageParamOpenAIFileOpenAIChatCompletionContentPartImageParamImageURL(
-    BaseModel
-):
-    """Image URL specification and processing details."""
-
-    url: str
-    """URL of the image to include in the message."""
-
-    detail: Optional[Literal["low", "high", "auto"]] = None
-    """Level of detail for image processing. Can be 'low', 'high', or 'auto'."""
-
-
-class ChoiceMessageOpenAIUserMessageParamOutputContentListOpenAIChatCompletionContentPartTextParamOpenAIChatCompletionContentPartImageParamOpenAIFileOpenAIChatCompletionContentPartImageParam(
-    BaseModel
-):
-    """Image content part for OpenAI-compatible chat completion messages."""
-
-    image_url: ChoiceMessageOpenAIUserMessageParamOutputContentListOpenAIChatCompletionContentPartTextParamOpenAIChatCompletionContentPartImageParamOpenAIFileOpenAIChatCompletionContentPartImageParamImageURL
-    """Image URL specification and processing details."""
-
-    type: Optional[Literal["image_url"]] = None
-    """Must be 'image_url' to identify this as image content."""
-
-
-class ChoiceMessageOpenAIUserMessageParamOutputContentListOpenAIChatCompletionContentPartTextParamOpenAIChatCompletionContentPartImageParamOpenAIFileOpenAIFileFile(
-    BaseModel
-):
-    """File specification."""
-
-    file_data: Optional[str] = None
-    """Base64-encoded file data."""
-
-    file_id: Optional[str] = None
-    """ID of an uploaded file."""
-
-    filename: Optional[str] = None
-    """Name of the file."""
-
-
-class ChoiceMessageOpenAIUserMessageParamOutputContentListOpenAIChatCompletionContentPartTextParamOpenAIChatCompletionContentPartImageParamOpenAIFileOpenAIFile(
-    BaseModel
-):
-    file: ChoiceMessageOpenAIUserMessageParamOutputContentListOpenAIChatCompletionContentPartTextParamOpenAIChatCompletionContentPartImageParamOpenAIFileOpenAIFileFile
-    """File specification."""
-
-    type: Optional[Literal["file"]] = None
-    """Must be 'file' to identify this as file content."""
-
-
-ChoiceMessageOpenAIUserMessageParamOutputContentListOpenAIChatCompletionContentPartTextParamOpenAIChatCompletionContentPartImageParamOpenAIFile: TypeAlias = Annotated[
-    Union[
-        ChoiceMessageOpenAIUserMessageParamOutputContentListOpenAIChatCompletionContentPartTextParamOpenAIChatCompletionContentPartImageParamOpenAIFileOpenAIChatCompletionContentPartTextParam,
-        ChoiceMessageOpenAIUserMessageParamOutputContentListOpenAIChatCompletionContentPartTextParamOpenAIChatCompletionContentPartImageParamOpenAIFileOpenAIChatCompletionContentPartImageParam,
-        ChoiceMessageOpenAIUserMessageParamOutputContentListOpenAIChatCompletionContentPartTextParamOpenAIChatCompletionContentPartImageParamOpenAIFileOpenAIFile,
-    ],
-    PropertyInfo(discriminator="type"),
-]
-
-
-class ChoiceMessageOpenAIUserMessageParamOutput(BaseModel):
-    """A message from the user in an OpenAI-compatible chat completion request."""
-
-    content: Union[
-        str,
-        List[
-            ChoiceMessageOpenAIUserMessageParamOutputContentListOpenAIChatCompletionContentPartTextParamOpenAIChatCompletionContentPartImageParamOpenAIFile
-        ],
-    ]
-    """The content of the message, which can include text and other media."""
-
-    name: Optional[str] = None
-    """The name of the user message participant."""
-
-    role: Optional[Literal["user"]] = None
-    """Must be 'user' to identify this as a user message."""
-
-
-class ChoiceMessageOpenAISystemMessageParamContentListOpenAIChatCompletionContentPartTextParam(BaseModel):
-    """Text content part for OpenAI-compatible chat completion messages."""
-
-    text: str
-    """The text content of the message."""
-
-    type: Optional[Literal["text"]] = None
-    """Must be 'text' to identify this as text content."""
-
-
-class ChoiceMessageOpenAISystemMessageParam(BaseModel):
-    """A system message providing instructions or context to the model."""
-
-    content: Union[str, List[ChoiceMessageOpenAISystemMessageParamContentListOpenAIChatCompletionContentPartTextParam]]
-    """The content of the 'system prompt'.
-
-    If multiple system messages are provided, they are concatenated.
-    """
+    arguments: Optional[str] = None
+    """Arguments to pass to the function as a JSON string."""
 
     name: Optional[str] = None
-    """The name of the system message participant."""
-
-    role: Optional[Literal["system"]] = None
-    """Must be 'system' to identify this as a system message."""
-
-
-class ChoiceMessageOpenAIAssistantMessageParamOutputContentListOpenAIChatCompletionContentPartTextParam(BaseModel):
-    """Text content part for OpenAI-compatible chat completion messages."""
-
-    text: str
-    """The text content of the message."""
-
-    type: Optional[Literal["text"]] = None
-    """Must be 'text' to identify this as text content."""
+    """Name of the function to call."""
 
 
-class ChoiceMessageOpenAIAssistantMessageParamOutputToolCallFunction(BaseModel):
+class ChoiceMessageToolCallFunction(BaseModel):
     """Function call details for OpenAI-compatible tool calls."""
 
     arguments: Optional[str] = None
@@ -179,13 +49,13 @@ class ChoiceMessageOpenAIAssistantMessageParamOutputToolCallFunction(BaseModel):
     """Name of the function to call."""
 
 
-class ChoiceMessageOpenAIAssistantMessageParamOutputToolCall(BaseModel):
+class ChoiceMessageToolCall(BaseModel):
     """Tool call specification for OpenAI-compatible chat completion responses."""
 
     id: Optional[str] = None
     """Unique identifier for the tool call."""
 
-    function: Optional[ChoiceMessageOpenAIAssistantMessageParamOutputToolCallFunction] = None
+    function: Optional[ChoiceMessageToolCallFunction] = None
     """Function call details for OpenAI-compatible tool calls."""
 
     index: Optional[int] = None
@@ -195,88 +65,29 @@ class ChoiceMessageOpenAIAssistantMessageParamOutputToolCall(BaseModel):
     """Must be 'function' to identify this as a function call."""
 
 
-class ChoiceMessageOpenAIAssistantMessageParamOutput(BaseModel):
-    """
-    A message containing the model's (assistant) response in an OpenAI-compatible chat completion request.
-    """
-
-    content: Union[
-        str,
-        List[ChoiceMessageOpenAIAssistantMessageParamOutputContentListOpenAIChatCompletionContentPartTextParam],
-        None,
-    ] = None
-    """The content of the model's response."""
-
-    name: Optional[str] = None
-    """The name of the assistant message participant."""
-
-    role: Optional[Literal["assistant"]] = None
-    """Must be 'assistant' to identify this as the model's response."""
-
-    tool_calls: Optional[List[ChoiceMessageOpenAIAssistantMessageParamOutputToolCall]] = None
-    """List of tool calls. Each tool call is an OpenAIChatCompletionToolCall object."""
-
-
-class ChoiceMessageOpenAIToolMessageParamContentListOpenAIChatCompletionContentPartTextParam(BaseModel):
-    """Text content part for OpenAI-compatible chat completion messages."""
-
-    text: str
-    """The text content of the message."""
-
-    type: Optional[Literal["text"]] = None
-    """Must be 'text' to identify this as text content."""
-
-
-class ChoiceMessageOpenAIToolMessageParam(BaseModel):
-    """
-    A message representing the result of a tool invocation in an OpenAI-compatible chat completion request.
-    """
-
-    content: Union[str, List[ChoiceMessageOpenAIToolMessageParamContentListOpenAIChatCompletionContentPartTextParam]]
-    """The response content from the tool."""
-
-    tool_call_id: str
-    """Unique identifier for the tool call this response is for."""
-
-    role: Optional[Literal["tool"]] = None
-    """Must be 'tool' to identify this as a tool response."""
-
-
-class ChoiceMessageOpenAIDeveloperMessageParamContentListOpenAIChatCompletionContentPartTextParam(BaseModel):
-    """Text content part for OpenAI-compatible chat completion messages."""
-
-    text: str
-    """The text content of the message."""
-
-    type: Optional[Literal["text"]] = None
-    """Must be 'text' to identify this as text content."""
+class ChoiceMessage(BaseModel):
+    """The message from the model."""
 
+    annotations: Optional[List[Dict[str, object]]] = None
+    """Annotations for the message, when applicable."""
 
-class ChoiceMessageOpenAIDeveloperMessageParam(BaseModel):
-    """A message from the developer in an OpenAI-compatible chat completion request."""
+    audio: Optional[Dict[str, object]] = None
+    """Audio response data when using audio output modality."""
 
-    content: Union[
-        str, List[ChoiceMessageOpenAIDeveloperMessageParamContentListOpenAIChatCompletionContentPartTextParam]
-    ]
-    """The content of the developer message."""
+    content: Optional[str] = None
+    """The content of the message."""
 
-    name: Optional[str] = None
-    """The name of the developer message participant."""
+    function_call: Optional[ChoiceMessageFunctionCall] = None
+    """Deprecated: the name and arguments of a function that should be called."""
 
-    role: Optional[Literal["developer"]] = None
-    """Must be 'developer' to identify this as a developer message."""
+    refusal: Optional[str] = None
+    """The refusal message generated by the model."""
 
+    role: Optional[Literal["assistant"]] = None
+    """The role of the message author, always 'assistant' in responses."""
 
-ChoiceMessage: TypeAlias = Annotated[
-    Union[
-        ChoiceMessageOpenAIUserMessageParamOutput,
-        ChoiceMessageOpenAISystemMessageParam,
-        ChoiceMessageOpenAIAssistantMessageParamOutput,
-        ChoiceMessageOpenAIToolMessageParam,
-        ChoiceMessageOpenAIDeveloperMessageParam,
-    ],
-    PropertyInfo(discriminator="role"),
-]
+    tool_calls: Optional[List[ChoiceMessageToolCall]] = None
+    """The tool calls generated by the model."""
 
 
 class ChoiceLogprobsContentTopLogprob(BaseModel):