llama-stack-client 0.2.13 → 0.2.14-rc2

package/src/resources/shared.ts

@@ -99,7 +99,7 @@ export namespace AgentConfig {
   }
 
   export interface AgentToolGroupWithArgs {
-    args: Record<string, boolean | number | string | Array<unknown> | unknown | null>;
+    args: { [key: string]: boolean | number | string | Array<unknown> | unknown | null };
 
     name: string;
   }
@@ -212,7 +212,7 @@ export interface Document {
   /**
    * Additional metadata for the document.
    */
-  metadata: Record<string, boolean | number | string | Array<unknown> | unknown | null>;
+  metadata: { [key: string]: boolean | number | string | Array<unknown> | unknown | null };
 
   /**
    * The MIME type of the document.
@@ -574,7 +574,7 @@ export namespace QueryGeneratorConfig {
 }
 
 export interface QueryResult {
-  metadata: Record<string, boolean | number | string | Array<unknown> | unknown | null>;
+  metadata: { [key: string]: boolean | number | string | Array<unknown> | unknown | null };
 
   /**
    * A image content item
@@ -596,7 +596,7 @@ export namespace ResponseFormat {
      * The JSON schema the response should conform to. In a Python SDK, this is often a
      * `pydantic` model.
      */
-    json_schema: Record<string, boolean | number | string | Array<unknown> | unknown | null>;
+    json_schema: { [key: string]: boolean | number | string | Array<unknown> | unknown | null };
 
     /**
      * Must be "json_schema" to identify this format type
@@ -611,7 +611,7 @@ export namespace ResponseFormat {
     /**
      * The BNF grammar specification the response should conform to
      */
-    bnf: Record<string, boolean | number | string | Array<unknown> | unknown | null>;
+    bnf: { [key: string]: boolean | number | string | Array<unknown> | unknown | null };
 
     /**
      * Must be "grammar" to identify this format type
@@ -635,7 +635,7 @@ export interface ReturnType {
 }
 
 export interface SafetyViolation {
-  metadata: Record<string, boolean | number | string | Array<unknown> | unknown | null>;
+  metadata: { [key: string]: boolean | number | string | Array<unknown> | unknown | null };
 
   violation_level: 'info' | 'warn' | 'error';
 
@@ -701,12 +701,12 @@ export interface ScoringResult {
   /**
    * Map of metric name to aggregated value
    */
-  aggregated_results: Record<string, boolean | number | string | Array<unknown> | unknown | null>;
+  aggregated_results: { [key: string]: boolean | number | string | Array<unknown> | unknown | null };
 
   /**
    * The scoring result for each row. Each row is a map of column name to value.
    */
-  score_rows: Array<Record<string, boolean | number | string | Array<unknown> | unknown | null>>;
+  score_rows: Array<{ [key: string]: boolean | number | string | Array<unknown> | unknown | null }>;
 }
 
 /**
@@ -729,15 +729,15 @@ export interface SystemMessage {
 export interface ToolCall {
   arguments:
     | string
-    | Record<
-        string,
-        | string
-        | number
-        | boolean
-        | Array<string | number | boolean | null>
-        | Record<string, string | number | boolean | null>
-        | null
-      >;
+    | {
+        [key: string]:
+          | string
+          | number
+          | boolean
+          | Array<string | number | boolean | null>
+          | { [key: string]: string | number | boolean | null }
+          | null;
+      };
 
   call_id: string;
 

package/src/resources/shields.ts

@@ -42,7 +42,7 @@ export interface Shield {
 
   type: 'shield';
 
-  params?: Record<string, boolean | number | string | Array<unknown> | unknown | null>;
+  params?: { [key: string]: boolean | number | string | Array<unknown> | unknown | null };
 
   provider_resource_id?: string;
 }
@@ -58,7 +58,7 @@ export interface ShieldRegisterParams {
   /**
    * The parameters of the shield.
    */
-  params?: Record<string, boolean | number | string | Array<unknown> | unknown | null>;
+  params?: { [key: string]: boolean | number | string | Array<unknown> | unknown | null };
 
   /**
    * The identifier of the provider.

package/src/resources/synthetic-data-generation.ts

@@ -18,9 +18,9 @@ export class SyntheticDataGeneration extends APIResource {
  * tuples that pass the threshold.
  */
 export interface SyntheticDataGenerationResponse {
-  synthetic_data: Array<Record<string, boolean | number | string | Array<unknown> | unknown | null>>;
+  synthetic_data: Array<{ [key: string]: boolean | number | string | Array<unknown> | unknown | null }>;
 
-  statistics?: Record<string, boolean | number | string | Array<unknown> | unknown | null>;
+  statistics?: { [key: string]: boolean | number | string | Array<unknown> | unknown | null };
 }
 
 export interface SyntheticDataGenerationGenerateParams {

package/src/resources/telemetry.ts

@@ -107,7 +107,7 @@ export namespace Event {
 
     type: 'unstructured_log';
 
-    attributes?: Record<string, string | number | boolean | null>;
+    attributes?: { [key: string]: string | number | boolean | null };
   }
 
   export interface MetricEvent {
@@ -125,7 +125,7 @@ export namespace Event {
 
     value: number;
 
-    attributes?: Record<string, string | number | boolean | null>;
+    attributes?: { [key: string]: string | number | boolean | null };
   }
 
   export interface StructuredLogEvent {
@@ -139,7 +139,7 @@ export namespace Event {
 
     type: 'structured_log';
 
-    attributes?: Record<string, string | number | boolean | null>;
+    attributes?: { [key: string]: string | number | boolean | null };
   }
 
   export namespace StructuredLogEvent {
@@ -180,7 +180,7 @@ export interface SpanWithStatus {
 
   trace_id: string;
 
-  attributes?: Record<string, boolean | number | string | Array<unknown> | unknown | null>;
+  attributes?: { [key: string]: boolean | number | string | Array<unknown> | unknown | null };
 
   end_time?: string;
 
@@ -208,14 +208,14 @@ export interface TelemetryGetSpanResponse {
 
   trace_id: string;
 
-  attributes?: Record<string, boolean | number | string | Array<unknown> | unknown | null>;
+  attributes?: { [key: string]: boolean | number | string | Array<unknown> | unknown | null };
 
   end_time?: string;
 
   parent_span_id?: string;
 }
 
-export type TelemetryGetSpanTreeResponse = Record<string, SpanWithStatus>;
+export type TelemetryGetSpanTreeResponse = { [key: string]: SpanWithStatus };
 
 export type TelemetryQuerySpansResponse = Array<TelemetryQuerySpansResponse.TelemetryQuerySpansResponseItem>;
 
@@ -229,7 +229,7 @@ export namespace TelemetryQuerySpansResponse {
 
     trace_id: string;
 
-    attributes?: Record<string, boolean | number | string | Array<unknown> | unknown | null>;
+    attributes?: { [key: string]: boolean | number | string | Array<unknown> | unknown | null };
 
     end_time?: string;
 

package/src/resources/tool-runtime/tool-runtime.ts

@@ -48,7 +48,7 @@ export interface ToolDef {
 
   description?: string;
 
-  metadata?: Record<string, boolean | number | string | Array<unknown> | unknown | null>;
+  metadata?: { [key: string]: boolean | number | string | Array<unknown> | unknown | null };
 
   parameters?: Array<ToolDef.Parameter>;
 }
@@ -77,7 +77,7 @@ export interface ToolInvocationResult {
 
   error_message?: string;
 
-  metadata?: Record<string, boolean | number | string | Array<unknown> | unknown | null>;
+  metadata?: { [key: string]: boolean | number | string | Array<unknown> | unknown | null };
 }
 
 export type ToolRuntimeListToolsResponse = Array<ToolDef>;
@@ -86,7 +86,7 @@ export interface ToolRuntimeInvokeToolParams {
   /**
    * A dictionary of arguments to pass to the tool.
    */
-  kwargs: Record<string, boolean | number | string | Array<unknown> | unknown | null>;
+  kwargs: { [key: string]: boolean | number | string | Array<unknown> | unknown | null };
 
   /**
    * The name of the tool to invoke.

package/src/resources/toolgroups.ts

@@ -53,7 +53,7 @@ export interface ToolGroup {
 
   type: 'tool_group';
 
-  args?: Record<string, boolean | number | string | Array<unknown> | unknown | null>;
+  args?: { [key: string]: boolean | number | string | Array<unknown> | unknown | null };
 
   mcp_endpoint?: ToolGroup.McpEndpoint;
 
@@ -82,7 +82,7 @@ export interface ToolgroupRegisterParams {
   /**
    * A dictionary of arguments to pass to the tool group.
    */
-  args?: Record<string, boolean | number | string | Array<unknown> | unknown | null>;
+  args?: { [key: string]: boolean | number | string | Array<unknown> | unknown | null };
 
   /**
    * The MCP endpoint to use for the tool group.

package/src/resources/tools.ts

@@ -47,7 +47,7 @@ export interface Tool {
 
   type: 'tool';
 
-  metadata?: Record<string, boolean | number | string | Array<unknown> | unknown | null>;
+  metadata?: { [key: string]: boolean | number | string | Array<unknown> | unknown | null };
 
   provider_resource_id?: string;
 }

package/src/resources/vector-io.ts

@@ -41,15 +41,91 @@ export namespace QueryChunksResponse {
     content: Shared.InterleavedContent;
 
     /**
-     * Metadata associated with the chunk, such as document ID, source, or other
-     * relevant information.
+     * Metadata associated with the chunk that will be used in the model context during
+     * inference.
      */
-    metadata: Record<string, boolean | number | string | Array<unknown> | unknown | null>;
+    metadata: { [key: string]: boolean | number | string | Array<unknown> | unknown | null };
+
+    /**
+     * Metadata for the chunk that will NOT be used in the context during inference.
+     * The `chunk_metadata` is required backend functionality.
+     */
+    chunk_metadata?: Chunk.ChunkMetadata;
 
     /**
      * Optional embedding for the chunk. If not provided, it will be computed later.
      */
     embedding?: Array<number>;
+
+    /**
+     * The chunk ID that is stored in the vector database. Used for backend
+     * functionality.
+     */
+    stored_chunk_id?: string;
+  }
+
+  export namespace Chunk {
+    /**
+     * Metadata for the chunk that will NOT be used in the context during inference.
+     * The `chunk_metadata` is required backend functionality.
+     */
+    export interface ChunkMetadata {
+      /**
+       * The dimension of the embedding vector for the chunk.
+       */
+      chunk_embedding_dimension?: number;
+
+      /**
+       * The embedding model used to create the chunk's embedding.
+       */
+      chunk_embedding_model?: string;
+
+      /**
+       * The ID of the chunk. If not set, it will be generated based on the document ID
+       * and content.
+       */
+      chunk_id?: string;
+
+      /**
+       * The tokenizer used to create the chunk. Default is Tiktoken.
+       */
+      chunk_tokenizer?: string;
+
+      /**
+       * The window of the chunk, which can be used to group related chunks together.
+       */
+      chunk_window?: string;
+
+      /**
+       * The number of tokens in the content of the chunk.
+       */
+      content_token_count?: number;
+
+      /**
+       * An optional timestamp indicating when the chunk was created.
+       */
+      created_timestamp?: number;
+
+      /**
+       * The ID of the document this chunk belongs to.
+       */
+      document_id?: string;
+
+      /**
+       * The number of tokens in the metadata of the chunk.
+       */
+      metadata_token_count?: number;
+
+      /**
+       * The source of the content, such as a URL, file path, or other identifier.
+       */
+      source?: string;
+
+      /**
+       * An optional timestamp indicating when the chunk was last updated.
+       */
+      updated_timestamp?: number;
+    }
   }
 }
 
@@ -85,15 +161,91 @@ export namespace VectorIoInsertParams {
     content: Shared.InterleavedContent;
 
     /**
-     * Metadata associated with the chunk, such as document ID, source, or other
-     * relevant information.
+     * Metadata associated with the chunk that will be used in the model context during
+     * inference.
      */
-    metadata: Record<string, boolean | number | string | Array<unknown> | unknown | null>;
+    metadata: { [key: string]: boolean | number | string | Array<unknown> | unknown | null };
+
+    /**
+     * Metadata for the chunk that will NOT be used in the context during inference.
+     * The `chunk_metadata` is required backend functionality.
+     */
+    chunk_metadata?: Chunk.ChunkMetadata;
 
     /**
      * Optional embedding for the chunk. If not provided, it will be computed later.
      */
     embedding?: Array<number>;
+
+    /**
+     * The chunk ID that is stored in the vector database. Used for backend
+     * functionality.
+     */
+    stored_chunk_id?: string;
+  }
+
+  export namespace Chunk {
+    /**
+     * Metadata for the chunk that will NOT be used in the context during inference.
+     * The `chunk_metadata` is required backend functionality.
+     */
+    export interface ChunkMetadata {
+      /**
+       * The dimension of the embedding vector for the chunk.
+       */
+      chunk_embedding_dimension?: number;
+
+      /**
+       * The embedding model used to create the chunk's embedding.
+       */
+      chunk_embedding_model?: string;
+
+      /**
+       * The ID of the chunk. If not set, it will be generated based on the document ID
+       * and content.
+       */
+      chunk_id?: string;
+
+      /**
+       * The tokenizer used to create the chunk. Default is Tiktoken.
+       */
+      chunk_tokenizer?: string;
+
+      /**
+       * The window of the chunk, which can be used to group related chunks together.
+       */
+      chunk_window?: string;
+
+      /**
+       * The number of tokens in the content of the chunk.
+       */
+      content_token_count?: number;
+
+      /**
+       * An optional timestamp indicating when the chunk was created.
+       */
+      created_timestamp?: number;
+
+      /**
+       * The ID of the document this chunk belongs to.
+       */
+      document_id?: string;
+
+      /**
+       * The number of tokens in the metadata of the chunk.
+       */
+      metadata_token_count?: number;
+
+      /**
+       * The source of the content, such as a URL, file path, or other identifier.
+       */
+      source?: string;
+
+      /**
+       * An optional timestamp indicating when the chunk was last updated.
+       */
+      updated_timestamp?: number;
+    }
   }
 }
 
@@ -111,7 +263,7 @@ export interface VectorIoQueryParams {
   /**
    * The parameters of the query.
    */
-  params?: Record<string, boolean | number | string | Array<unknown> | unknown | null>;
+  params?: { [key: string]: boolean | number | string | Array<unknown> | unknown | null };
 }
 
 export declare namespace VectorIo {

package/src/resources/vector-stores/files.ts

@@ -1,6 +1,7 @@
 // File generated from our OpenAPI spec by Stainless. See CONTRIBUTING.md for details.
 
 import { APIResource } from '../../resource';
+import { isRequestOptions } from '../../core';
 import * as Core from '../../core';
 
 export class Files extends APIResource {
@@ -14,6 +15,74 @@ export class Files extends APIResource {
   ): Core.APIPromise<VectorStoreFile> {
     return this._client.post(`/v1/openai/v1/vector_stores/${vectorStoreId}/files`, { body, ...options });
   }
+
+  /**
+   * Retrieves a vector store file.
+   */
+  retrieve(
+    vectorStoreId: string,
+    fileId: string,
+    options?: Core.RequestOptions,
+  ): Core.APIPromise<VectorStoreFile> {
+    return this._client.get(`/v1/openai/v1/vector_stores/${vectorStoreId}/files/${fileId}`, options);
+  }
+
+  /**
+   * Updates a vector store file.
+   */
+  update(
+    vectorStoreId: string,
+    fileId: string,
+    body: FileUpdateParams,
+    options?: Core.RequestOptions,
+  ): Core.APIPromise<VectorStoreFile> {
+    return this._client.post(`/v1/openai/v1/vector_stores/${vectorStoreId}/files/${fileId}`, {
+      body,
+      ...options,
+    });
+  }
+
+  /**
+   * List files in a vector store.
+   */
+  list(
+    vectorStoreId: string,
+    query?: FileListParams,
+    options?: Core.RequestOptions,
+  ): Core.APIPromise<FileListResponse>;
+  list(vectorStoreId: string, options?: Core.RequestOptions): Core.APIPromise<FileListResponse>;
+  list(
+    vectorStoreId: string,
+    query: FileListParams | Core.RequestOptions = {},
+    options?: Core.RequestOptions,
+  ): Core.APIPromise<FileListResponse> {
+    if (isRequestOptions(query)) {
+      return this.list(vectorStoreId, {}, query);
+    }
+    return this._client.get(`/v1/openai/v1/vector_stores/${vectorStoreId}/files`, { query, ...options });
+  }
+
+  /**
+   * Delete a vector store file.
+   */
+  delete(
+    vectorStoreId: string,
+    fileId: string,
+    options?: Core.RequestOptions,
+  ): Core.APIPromise<FileDeleteResponse> {
+    return this._client.delete(`/v1/openai/v1/vector_stores/${vectorStoreId}/files/${fileId}`, options);
+  }
+
+  /**
+   * Retrieves the contents of a vector store file.
+   */
+  content(
+    vectorStoreId: string,
+    fileId: string,
+    options?: Core.RequestOptions,
+  ): Core.APIPromise<FileContentResponse> {
+    return this._client.get(`/v1/openai/v1/vector_stores/${vectorStoreId}/files/${fileId}/content`, options);
+  }
 }
 
 /**
@@ -22,7 +91,7 @@ export class Files extends APIResource {
 export interface VectorStoreFile {
   id: string;
 
-  attributes: Record<string, boolean | number | string | Array<unknown> | unknown | null>;
+  attributes: { [key: string]: boolean | number | string | Array<unknown> | unknown | null };
 
   chunking_strategy:
     | VectorStoreFile.VectorStoreChunkingStrategyAuto
@@ -67,6 +136,53 @@ export namespace VectorStoreFile {
   }
 }
 
+/**
+ * Response from listing vector stores.
+ */
+export interface FileListResponse {
+  data: Array<VectorStoreFile>;
+
+  has_more: boolean;
+
+  object: string;
+
+  first_id?: string;
+
+  last_id?: string;
+}
+
+/**
+ * Response from deleting a vector store file.
+ */
+export interface FileDeleteResponse {
+  id: string;
+
+  deleted: boolean;
+
+  object: string;
+}
+
+/**
+ * Response from retrieving the contents of a vector store file.
+ */
+export interface FileContentResponse {
+  attributes: { [key: string]: boolean | number | string | Array<unknown> | unknown | null };
+
+  content: Array<FileContentResponse.Content>;
+
+  file_id: string;
+
+  filename: string;
+}
+
+export namespace FileContentResponse {
+  export interface Content {
+    text: string;
+
+    type: 'text';
+  }
+}
+
 export interface FileCreateParams {
   /**
    * The ID of the file to attach to the vector store.
@@ -76,7 +192,7 @@ export interface FileCreateParams {
   /**
    * The key-value attributes stored with the file, which can be used for filtering.
    */
-  attributes?: Record<string, boolean | number | string | Array<unknown> | unknown | null>;
+  attributes?: { [key: string]: boolean | number | string | Array<unknown> | unknown | null };
 
   /**
    * The chunking strategy to use for the file.
@@ -106,6 +222,33 @@ export namespace FileCreateParams {
   }
 }
 
+export interface FileUpdateParams {
+  /**
+   * The updated key-value attributes to store with the file.
+   */
+  attributes: { [key: string]: boolean | number | string | Array<unknown> | unknown | null };
+}
+
+export interface FileListParams {
+  after?: string;
+
+  before?: string;
+
+  filter?: 'completed' | 'in_progress' | 'cancelled' | 'failed';
+
+  limit?: number;
+
+  order?: string;
+}
+
 export declare namespace Files {
-  export { type VectorStoreFile as VectorStoreFile, type FileCreateParams as FileCreateParams };
+  export {
+    type VectorStoreFile as VectorStoreFile,
+    type FileListResponse as FileListResponse,
+    type FileDeleteResponse as FileDeleteResponse,
+    type FileContentResponse as FileContentResponse,
+    type FileCreateParams as FileCreateParams,
+    type FileUpdateParams as FileUpdateParams,
+    type FileListParams as FileListParams,
+  };
 }

package/src/resources/vector-stores/index.ts

@@ -1,6 +1,15 @@
 // File generated from our OpenAPI spec by Stainless. See CONTRIBUTING.md for details.
 
-export { Files, type VectorStoreFile, type FileCreateParams } from './files';
+export {
+  Files,
+  type VectorStoreFile,
+  type FileListResponse,
+  type FileDeleteResponse,
+  type FileContentResponse,
+  type FileCreateParams,
+  type FileUpdateParams,
+  type FileListParams,
+} from './files';
 export {
   VectorStores,
   type ListVectorStoresResponse,