llama-stack-client 0.3.4 → 0.4.0-rc1

package/src/resources/shared.ts

@@ -3,303 +3,49 @@
 //
 // This source code is licensed under the terms described in the LICENSE file in
 // the root directory of this source tree.
-
+//
 // File generated from our OpenAPI spec by Stainless. See CONTRIBUTING.md for details.
 
-import * as Shared from './shared';
-import * as InferenceAPI from './inference';
-import * as ToolRuntimeAPI from './tool-runtime/tool-runtime';
-
-/**
- * Configuration for an agent.
- */
-export interface AgentConfig {
-  /**
-   * The system instructions for the agent
-   */
-  instructions: string;
-
-  /**
-   * The model identifier to use for the agent
-   */
-  model: string;
-
-  client_tools?: Array<ToolRuntimeAPI.ToolDef>;
-
-  /**
-   * Optional flag indicating whether session data has to be persisted
-   */
-  enable_session_persistence?: boolean;
-
-  input_shields?: Array<string>;
-
-  max_infer_iters?: number;
-
-  /**
-   * Optional name for the agent, used in telemetry and identification
-   */
-  name?: string;
-
-  output_shields?: Array<string>;
-
-  /**
-   * Optional response format configuration
-   */
-  response_format?: ResponseFormat;
-
-  /**
-   * Sampling parameters.
-   */
-  sampling_params?: SamplingParams;
-
-  /**
-   * @deprecated Whether tool use is required or automatic. This is a hint to the
-   * model which may not be followed. It depends on the Instruction Following
-   * capabilities of the model.
-   */
-  tool_choice?: 'auto' | 'required' | 'none';
-
-  /**
-   * Configuration for tool use.
-   */
-  tool_config?: AgentConfig.ToolConfig;
-
-  /**
-   * @deprecated Prompt format for calling custom / zero shot tools.
-   */
-  tool_prompt_format?: 'json' | 'function_tag' | 'python_list';
-
-  toolgroups?: Array<string | AgentConfig.AgentToolGroupWithArgs>;
-}
-
-export namespace AgentConfig {
-  /**
-   * Configuration for tool use.
-   */
-  export interface ToolConfig {
-    /**
-     * (Optional) Config for how to override the default system prompt. -
-     * `SystemMessageBehavior.append`: Appends the provided system message to the
-     * default system prompt. - `SystemMessageBehavior.replace`: Replaces the default
-     * system prompt with the provided system message. The system message can include
-     * the string '{{function_definitions}}' to indicate where the function definitions
-     * should be inserted.
-     */
-    system_message_behavior?: 'append' | 'replace';
-
-    /**
-     * (Optional) Whether tool use is automatic, required, or none. Can also specify a
-     * tool name to use a specific tool. Defaults to ToolChoice.auto.
-     */
-    tool_choice?: 'auto' | 'required' | 'none' | (string & {});
-
-    /**
-     * (Optional) Instructs the model how to format tool calls. By default, Llama Stack
-     * will attempt to use a format that is best adapted to the model. -
-     * `ToolPromptFormat.json`: The tool calls are formatted as a JSON object. -
-     * `ToolPromptFormat.function_tag`: The tool calls are enclosed in a
-     * <function=function_name> tag. - `ToolPromptFormat.python_list`: The tool calls
-     * are output as Python syntax -- a list of function calls.
-     */
-    tool_prompt_format?: 'json' | 'function_tag' | 'python_list';
-  }
-
-  export interface AgentToolGroupWithArgs {
-    args: { [key: string]: boolean | number | string | Array<unknown> | unknown | null };
-
-    name: string;
-  }
-}
-
-/**
- * Response from a batch completion request.
- */
-export interface BatchCompletion {
-  /**
-   * List of completion responses, one for each input in the batch
-   */
-  batch: Array<InferenceAPI.CompletionResponse>;
-}
-
-/**
- * Response from a chat completion request.
- */
-export interface ChatCompletionResponse {
-  /**
-   * The complete response message
-   */
-  completion_message: CompletionMessage;
-
-  /**
-   * Optional log probabilities for generated tokens
-   */
-  logprobs?: Array<InferenceAPI.TokenLogProbs>;
-
-  /**
-   * (Optional) List of metrics associated with the API response
-   */
-  metrics?: Array<Metric>;
-}
-
-/**
- * A message containing the model's (assistant) response in a chat conversation.
- */
-export interface CompletionMessage {
-  /**
-   * The content of the model's response
-   */
-  content: InterleavedContent;
-
-  /**
-   * Must be "assistant" to identify this as the model's response
-   */
-  role: 'assistant';
-
-  /**
-   * Reason why the model stopped generating. Options are: -
-   * `StopReason.end_of_turn`: The model finished generating the entire response. -
-   * `StopReason.end_of_message`: The model finished generating but generated a
-   * partial response -- usually, a tool call. The user may call the tool and
-   * continue the conversation with the tool's response. -
-   * `StopReason.out_of_tokens`: The model ran out of token budget.
-   */
-  stop_reason: 'end_of_turn' | 'end_of_message' | 'out_of_tokens';
-
-  /**
-   * List of tool calls. Each tool call is a ToolCall object.
-   */
-  tool_calls?: Array<ToolCall>;
-}
-
-/**
- * A text content delta for streaming responses.
- */
-export type ContentDelta = ContentDelta.TextDelta | ContentDelta.ImageDelta | ContentDelta.ToolCallDelta;
-
-export namespace ContentDelta {
-  /**
-   * A text content delta for streaming responses.
-   */
-  export interface TextDelta {
-    /**
-     * The incremental text content
-     */
-    text: string;
-
-    /**
-     * Discriminator type of the delta. Always "text"
-     */
-    type: 'text';
-  }
-
-  /**
-   * An image content delta for streaming responses.
-   */
-  export interface ImageDelta {
-    /**
-     * The incremental image data as bytes
-     */
-    image: string;
-
-    /**
-     * Discriminator type of the delta. Always "image"
-     */
-    type: 'image';
-  }
-
-  /**
-   * A tool call content delta for streaming responses.
-   */
-  export interface ToolCallDelta {
-    /**
-     * Current parsing status of the tool call
-     */
-    parse_status: 'started' | 'in_progress' | 'failed' | 'succeeded';
-
-    /**
-     * Either an in-progress tool call string or the final parsed tool call
-     */
-    tool_call: string | Shared.ToolCall;
-
-    /**
-     * Discriminator type of the delta. Always "tool_call"
-     */
-    type: 'tool_call';
-  }
-}
-
 /**
- * A document to be used for document ingestion in the RAG Tool.
+ * A image content item
  */
-export interface Document {
-  /**
-   * The content of the document.
-   */
-  content:
-    | string
-    | Document.ImageContentItem
-    | Document.TextContentItem
-    | Array<InterleavedContentItem>
-    | Document.URL;
-
-  /**
-   * The unique identifier for the document.
-   */
-  document_id: string;
-
-  /**
-   * Additional metadata for the document.
-   */
-  metadata: { [key: string]: boolean | number | string | Array<unknown> | unknown | null };
-
-  /**
-   * The MIME type of the document.
-   */
-  mime_type?: string;
-}
+export type InterleavedContent =
+  | string
+  | InterleavedContent.ImageContentItem
+  | InterleavedContent.TextContentItem
+  | Array<InterleavedContent.ImageContentItem | InterleavedContent.TextContentItem>;
 
-export namespace Document {
+export namespace InterleavedContent {
   /**
    * A image content item
    */
   export interface ImageContentItem {
     /**
-     * Image as a base64 encoded string or an URL
+     * A URL or a base64 encoded string
      */
     image: ImageContentItem.Image;
 
-    /**
-     * Discriminator type of the content item. Always "image"
-     */
-    type: 'image';
+    type?: 'image';
   }
 
   export namespace ImageContentItem {
     /**
-     * Image as a base64 encoded string or an URL
+     * A URL or a base64 encoded string
      */
     export interface Image {
-      /**
-       * base64 encoded image data as string
-       */
-      data?: string;
+      data?: string | null;
 
       /**
-       * A URL of the image or data URL in the format of data:image/{type};base64,{data}.
-       * Note that URL could have length limits.
+       * A URL reference to external content.
        */
-      url?: Image.URL;
+      url?: Image.URL | null;
     }
 
     export namespace Image {
       /**
-       * A URL of the image or data URL in the format of data:image/{type};base64,{data}.
-       * Note that URL could have length limits.
+       * A URL reference to external content.
        */
       export interface URL {
-        /**
-         * The URL string pointing to the resource
-         */
         uri: string;
       }
     }
@@ -309,79 +55,41 @@ export namespace Document {
    * A text content item
    */
   export interface TextContentItem {
-    /**
-     * Text content
-     */
     text: string;
 
-    /**
-     * Discriminator type of the content item. Always "text"
-     */
-    type: 'text';
+    type?: 'text';
   }
 
-  /**
-   * A URL reference to external content.
-   */
-  export interface URL {
-    /**
-     * The URL string pointing to the resource
-     */
-    uri: string;
-  }
-}
-
-/**
- * A image content item
- */
-export type InterleavedContent =
-  | string
-  | InterleavedContent.ImageContentItem
-  | InterleavedContent.TextContentItem
-  | Array<InterleavedContentItem>;
-
-export namespace InterleavedContent {
   /**
    * A image content item
    */
   export interface ImageContentItem {
     /**
-     * Image as a base64 encoded string or an URL
+     * A URL or a base64 encoded string
      */
     image: ImageContentItem.Image;
 
-    /**
-     * Discriminator type of the content item. Always "image"
-     */
-    type: 'image';
+    type?: 'image';
   }
 
   export namespace ImageContentItem {
     /**
-     * Image as a base64 encoded string or an URL
+     * A URL or a base64 encoded string
      */
     export interface Image {
-      /**
-       * base64 encoded image data as string
-       */
-      data?: string;
+      data?: string | null;
 
       /**
-       * A URL of the image or data URL in the format of data:image/{type};base64,{data}.
-       * Note that URL could have length limits.
+       * A URL reference to external content.
        */
-      url?: Image.URL;
+      url?: Image.URL | null;
     }
 
     export namespace Image {
       /**
-       * A URL of the image or data URL in the format of data:image/{type};base64,{data}.
-       * Note that URL could have length limits.
+       * A URL reference to external content.
        */
       export interface URL {
-        /**
-         * The URL string pointing to the resource
-         */
         uri: string;
       }
     }
@@ -391,15 +99,9 @@ export namespace InterleavedContent {
    * A text content item
    */
   export interface TextContentItem {
-    /**
-     * Text content
-     */
     text: string;
 
-    /**
-     * Discriminator type of the content item. Always "text"
-     */
-    type: 'text';
+    type?: 'text';
   }
 }
 
@@ -416,42 +118,31 @@ export namespace InterleavedContentItem {
    */
   export interface ImageContentItem {
     /**
-     * Image as a base64 encoded string or an URL
+     * A URL or a base64 encoded string
      */
     image: ImageContentItem.Image;
 
-    /**
-     * Discriminator type of the content item. Always "image"
-     */
-    type: 'image';
+    type?: 'image';
   }
 
   export namespace ImageContentItem {
     /**
-     * Image as a base64 encoded string or an URL
+     * A URL or a base64 encoded string
      */
     export interface Image {
-      /**
-       * base64 encoded image data as string
-       */
-      data?: string;
+      data?: string | null;
 
       /**
-       * A URL of the image or data URL in the format of data:image/{type};base64,{data}.
-       * Note that URL could have length limits.
+       * A URL reference to external content.
        */
-      url?: Image.URL;
+      url?: Image.URL | null;
     }
 
     export namespace Image {
       /**
-       * A URL of the image or data URL in the format of data:image/{type};base64,{data}.
-       * Note that URL could have length limits.
+       * A URL reference to external content.
        */
       export interface URL {
-        /**
-         * The URL string pointing to the resource
-         */
         uri: string;
       }
     }
@@ -461,43 +152,12 @@ export namespace InterleavedContentItem {
    * A text content item
    */
   export interface TextContentItem {
-    /**
-     * Text content
-     */
     text: string;
 
-    /**
-     * Discriminator type of the content item. Always "text"
-     */
-    type: 'text';
+    type?: 'text';
   }
 }
 
-/**
- * A message from the user in a chat conversation.
- */
-export type Message = UserMessage | SystemMessage | ToolResponseMessage | CompletionMessage;
-
-/**
- * A metric value included in API responses.
- */
-export interface Metric {
-  /**
-   * The name of the metric
-   */
-  metric: string;
-
-  /**
-   * The numeric value of the metric
-   */
-  value: number;
-
-  /**
-   * (Optional) The unit of measurement for the metric value
-   */
-  unit?: string;
-}
-
 /**
  * Parameter type for string values.
  */
@@ -510,278 +170,70 @@ export type ParamType =
   | ParamType.JsonType
   | ParamType.UnionType
   | ParamType.ChatCompletionInputType
-  | ParamType.CompletionInputType
-  | ParamType.AgentTurnInputType;
+  | ParamType.CompletionInputType;
 
 export namespace ParamType {
   /**
    * Parameter type for string values.
    */
   export interface StringType {
-    /**
-     * Discriminator type. Always "string"
-     */
-    type: 'string';
+    type?: 'string';
   }
 
   /**
    * Parameter type for numeric values.
    */
   export interface NumberType {
-    /**
-     * Discriminator type. Always "number"
-     */
-    type: 'number';
+    type?: 'number';
   }
 
   /**
    * Parameter type for boolean values.
    */
   export interface BooleanType {
-    /**
-     * Discriminator type. Always "boolean"
-     */
-    type: 'boolean';
+    type?: 'boolean';
   }
 
   /**
    * Parameter type for array values.
    */
   export interface ArrayType {
-    /**
-     * Discriminator type. Always "array"
-     */
-    type: 'array';
+    type?: 'array';
   }
 
   /**
    * Parameter type for object values.
    */
   export interface ObjectType {
-    /**
-     * Discriminator type. Always "object"
-     */
-    type: 'object';
+    type?: 'object';
   }
 
   /**
    * Parameter type for JSON values.
    */
   export interface JsonType {
-    /**
-     * Discriminator type. Always "json"
-     */
-    type: 'json';
+    type?: 'json';
   }
 
   /**
    * Parameter type for union values.
    */
   export interface UnionType {
-    /**
-     * Discriminator type. Always "union"
-     */
-    type: 'union';
+    type?: 'union';
   }
 
   /**
    * Parameter type for chat completion input.
    */
   export interface ChatCompletionInputType {
-    /**
-     * Discriminator type. Always "chat_completion_input"
-     */
-    type: 'chat_completion_input';
+    type?: 'chat_completion_input';
   }
 
   /**
    * Parameter type for completion input.
    */
   export interface CompletionInputType {
-    /**
-     * Discriminator type. Always "completion_input"
-     */
-    type: 'completion_input';
-  }
-
-  /**
-   * Parameter type for agent turn input.
-   */
-  export interface AgentTurnInputType {
-    /**
-     * Discriminator type. Always "agent_turn_input"
-     */
-    type: 'agent_turn_input';
-  }
-}
-
-/**
- * Configuration for the RAG query generation.
- */
-export interface QueryConfig {
-  /**
-   * Template for formatting each retrieved chunk in the context. Available
-   * placeholders: {index} (1-based chunk ordinal), {chunk.content} (chunk content
-   * string), {metadata} (chunk metadata dict). Default: "Result {index}\nContent:
-   * {chunk.content}\nMetadata: {metadata}\n"
-   */
-  chunk_template: string;
-
-  /**
-   * Maximum number of chunks to retrieve.
-   */
-  max_chunks: number;
-
-  /**
-   * Maximum number of tokens in the context.
-   */
-  max_tokens_in_context: number;
-
-  /**
-   * Configuration for the query generator.
-   */
-  query_generator_config: QueryGeneratorConfig;
-
-  /**
-   * Search mode for retrieval—either "vector", "keyword", or "hybrid". Default
-   * "vector".
-   */
-  mode?: 'vector' | 'keyword' | 'hybrid';
-
-  /**
-   * Configuration for the ranker to use in hybrid search. Defaults to RRF ranker.
-   */
-  ranker?: QueryConfig.RrfRanker | QueryConfig.WeightedRanker;
-}
-
-export namespace QueryConfig {
-  /**
-   * Reciprocal Rank Fusion (RRF) ranker configuration.
-   */
-  export interface RrfRanker {
-    /**
-     * The impact factor for RRF scoring. Higher values give more weight to
-     * higher-ranked results. Must be greater than 0
-     */
-    impact_factor: number;
-
-    /**
-     * The type of ranker, always "rrf"
-     */
-    type: 'rrf';
-  }
-
-  /**
-   * Weighted ranker configuration that combines vector and keyword scores.
-   */
-  export interface WeightedRanker {
-    /**
-     * Weight factor between 0 and 1. 0 means only use keyword scores, 1 means only use
-     * vector scores, values in between blend both scores.
-     */
-    alpha: number;
-
-    /**
-     * The type of ranker, always "weighted"
-     */
-    type: 'weighted';
-  }
-}
-
-/**
- * Configuration for the default RAG query generator.
- */
-export type QueryGeneratorConfig =
-  | QueryGeneratorConfig.DefaultRagQueryGeneratorConfig
-  | QueryGeneratorConfig.LlmragQueryGeneratorConfig;
-
-export namespace QueryGeneratorConfig {
-  /**
-   * Configuration for the default RAG query generator.
-   */
-  export interface DefaultRagQueryGeneratorConfig {
-    /**
-     * String separator used to join query terms
-     */
-    separator: string;
-
-    /**
-     * Type of query generator, always 'default'
-     */
-    type: 'default';
-  }
-
-  /**
-   * Configuration for the LLM-based RAG query generator.
-   */
-  export interface LlmragQueryGeneratorConfig {
-    /**
-     * Name of the language model to use for query generation
-     */
-    model: string;
-
-    /**
-     * Template string for formatting the query generation prompt
-     */
-    template: string;
-
-    /**
-     * Type of query generator, always 'llm'
-     */
-    type: 'llm';
-  }
-}
-
-/**
- * Result of a RAG query containing retrieved content and metadata.
- */
-export interface QueryResult {
-  /**
-   * Additional metadata about the query result
-   */
-  metadata: { [key: string]: boolean | number | string | Array<unknown> | unknown | null };
-
-  /**
-   * (Optional) The retrieved content from the query
-   */
-  content?: InterleavedContent;
-}
-
-/**
- * Configuration for JSON schema-guided response generation.
- */
-export type ResponseFormat = ResponseFormat.JsonSchemaResponseFormat | ResponseFormat.GrammarResponseFormat;
-
-export namespace ResponseFormat {
-  /**
-   * Configuration for JSON schema-guided response generation.
-   */
-  export interface JsonSchemaResponseFormat {
-    /**
-     * The JSON schema the response should conform to. In a Python SDK, this is often a
-     * `pydantic` model.
-     */
-    json_schema: { [key: string]: boolean | number | string | Array<unknown> | unknown | null };
-
-    /**
-     * Must be "json_schema" to identify this format type
-     */
-    type: 'json_schema';
-  }
-
-  /**
-   * Configuration for grammar-guided response generation.
-   */
-  export interface GrammarResponseFormat {
-    /**
-     * The BNF grammar specification the response should conform to
-     */
-    bnf: { [key: string]: boolean | number | string | Array<unknown> | unknown | null };
-
-    /**
-     * Must be "grammar" to identify this format type
-     */
-    type: 'grammar';
+    type?: 'completion_input';
   }
 }
 
@@ -790,52 +242,33 @@ export namespace ResponseFormat {
  */
 export interface SafetyViolation {
   /**
-   * Additional metadata including specific violation codes for debugging and
-   * telemetry
-   */
-  metadata: { [key: string]: boolean | number | string | Array<unknown> | unknown | null };
-
-  /**
-   * Severity level of the violation
+   * Severity level of a safety violation.
    */
   violation_level: 'info' | 'warn' | 'error';
 
-  /**
-   * (Optional) Message to convey to the user about the violation
-   */
-  user_message?: string;
+  metadata?: { [key: string]: unknown };
+
+  user_message?: string | null;
 }
 
 /**
  * Sampling parameters.
  */
 export interface SamplingParams {
-  /**
-   * The sampling strategy.
-   */
-  strategy:
-    | SamplingParams.GreedySamplingStrategy
-    | SamplingParams.TopPSamplingStrategy
-    | SamplingParams.TopKSamplingStrategy;
+  max_tokens?: number | null;
 
-  /**
-   * The maximum number of tokens that can be generated in the completion. The token
-   * count of your prompt plus max_tokens cannot exceed the model's context length.
-   */
-  max_tokens?: number;
+  repetition_penalty?: number | null;
 
-  /**
-   * Number between -2.0 and 2.0. Positive values penalize new tokens based on
-   * whether they appear in the text so far, increasing the model's likelihood to
-   * talk about new topics.
-   */
-  repetition_penalty?: number;
+  stop?: Array<string> | null;
 
   /**
-   * Up to 4 sequences where the API will stop generating further tokens. The
-   * returned text will not contain the stop sequence.
+   * Greedy sampling strategy that selects the highest probability token at each
+   * step.
    */
-  stop?: Array<string>;
+  strategy?:
+    | SamplingParams.GreedySamplingStrategy
+    | SamplingParams.TopPSamplingStrategy
+    | SamplingParams.TopKSamplingStrategy;
 }
 
 export namespace SamplingParams {
@@ -844,10 +277,7 @@ export namespace SamplingParams {
    * step.
    */
   export interface GreedySamplingStrategy {
-    /**
-     * Must be "greedy" to identify this sampling strategy
-     */
-    type: 'greedy';
+    type?: 'greedy';
   }
 
   /**
@@ -855,35 +285,20 @@ export namespace SamplingParams {
    * with cumulative probability >= p.
    */
   export interface TopPSamplingStrategy {
-    /**
-     * Must be "top_p" to identify this sampling strategy
-     */
-    type: 'top_p';
+    temperature: number | null;
 
-    /**
-     * Controls randomness in sampling. Higher values increase randomness
-     */
-    temperature?: number;
+    top_p?: number | null;
 
-    /**
-     * Cumulative probability threshold for nucleus sampling. Defaults to 0.95
-     */
-    top_p?: number;
+    type?: 'top_p';
   }
 
   /**
    * Top-k sampling strategy that restricts sampling to the k most likely tokens.
    */
   export interface TopKSamplingStrategy {
-    /**
-     * Number of top tokens to consider for sampling. Must be at least 1
-     */
     top_k: number;
 
-    /**
-     * Must be "top_k" to identify this sampling strategy
-     */
-    type: 'top_k';
+    type?: 'top_k';
   }
 }
 
@@ -891,15 +306,9 @@ export namespace SamplingParams {
  * A scoring result for a single row.
  */
 export interface ScoringResult {
-  /**
-   * Map of metric name to aggregated value
-   */
-  aggregated_results: { [key: string]: boolean | number | string | Array<unknown> | unknown | null };
+  aggregated_results: { [key: string]: unknown };
 
-  /**
-   * The scoring result for each row. Each row is a map of column name to value.
-   */
-  score_rows: Array<{ [key: string]: boolean | number | string | Array<unknown> | unknown | null }>;
+  score_rows: Array<{ [key: string]: unknown }>;
 }
 
 /**
@@ -907,85 +316,103 @@ export interface ScoringResult {
  */
 export interface SystemMessage {
   /**
-   * The content of the "system prompt". If multiple system messages are provided,
-   * they are concatenated. The underlying Llama Stack code may also add other system
-   * messages (for example, for formatting tool definitions).
-   */
-  content: InterleavedContent;
-
-  /**
-   * Must be "system" to identify this as a system message
+   * A image content item
    */
-  role: 'system';
-}
-
-export interface ToolCall {
-  arguments:
+  content:
     | string
-    | {
-        [key: string]:
-          | string
-          | number
-          | boolean
-          | Array<string | number | boolean | null>
-          | { [key: string]: string | number | boolean | null }
-          | null;
-      };
-
-  call_id: string;
-
-  tool_name: 'brave_search' | 'wolfram_alpha' | 'photogen' | 'code_interpreter' | (string & {});
+    | SystemMessage.ImageContentItemInput
+    | SystemMessage.TextContentItem
+    | Array<SystemMessage.ImageContentItemInput | SystemMessage.TextContentItem>;
 
-  arguments_json?: string;
+  role?: 'system';
 }
 
-export interface ToolParamDefinition {
-  param_type: string;
+export namespace SystemMessage {
+  /**
+   * A image content item
+   */
+  export interface ImageContentItemInput {
+    /**
+     * A URL or a base64 encoded string
+     */
+    image: ImageContentItemInput.Image;
+
+    type?: 'image';
+  }
 
-  default?: boolean | number | string | Array<unknown> | unknown | null;
+  export namespace ImageContentItemInput {
+    /**
+     * A URL or a base64 encoded string
+     */
+    export interface Image {
+      data?: string | null;
 
-  description?: string;
+      /**
+       * A URL reference to external content.
+       */
+      url?: Image.URL | null;
+    }
 
-  required?: boolean;
-}
+    export namespace Image {
+      /**
+       * A URL reference to external content.
+       */
+      export interface URL {
+        uri: string;
+      }
+    }
+  }
 
-/**
- * A message representing the result of a tool invocation.
- */
-export interface ToolResponseMessage {
   /**
-   * Unique identifier for the tool call this response is for
+   * A text content item
    */
-  call_id: string;
+  export interface TextContentItem {
+    text: string;
 
-  /**
-   * The response content from the tool
-   */
-  content: InterleavedContent;
+    type?: 'text';
+  }
 
   /**
-   * Must be "tool" to identify this as a tool response
+   * A image content item
    */
-  role: 'tool';
-}
+  export interface ImageContentItemInput {
+    /**
+     * A URL or a base64 encoded string
+     */
+    image: ImageContentItemInput.Image;
 
-/**
- * A message from the user in a chat conversation.
- */
-export interface UserMessage {
-  /**
-   * The content of the message, which can include text and other media
-   */
-  content: InterleavedContent;
+    type?: 'image';
+  }
 
-  /**
-   * Must be "user" to identify this as a user message
-   */
-  role: 'user';
+  export namespace ImageContentItemInput {
+    /**
+     * A URL or a base64 encoded string
+     */
+    export interface Image {
+      data?: string | null;
+
+      /**
+       * A URL reference to external content.
+       */
+      url?: Image.URL | null;
+    }
+
+    export namespace Image {
+      /**
+       * A URL reference to external content.
+       */
+      export interface URL {
+        uri: string;
+      }
+    }
+  }
 
   /**
-   * (Optional) This field is used internally by Llama Stack to pass RAG context.
-   * This field may be removed in the API in the future.
+   * A text content item
    */
-  context?: InterleavedContent;
+  export interface TextContentItem {
+    text: string;
+
+    type?: 'text';
+  }
 }