llama-stack-client 0.2.13 → 0.2.14-rc1

package/src/resources/agents/agents.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';
 import * as Shared from '../shared';
 import * as SessionAPI from './session';
@@ -8,6 +9,8 @@ import {
   Session,
   SessionCreateParams,
   SessionCreateResponse,
+  SessionListParams,
+  SessionListResponse,
   SessionResource,
   SessionRetrieveParams,
 } from './session';
@@ -40,6 +43,28 @@ export class Agents extends APIResource {
     return this._client.post('/v1/agents', { body, ...options });
   }
 
+  /**
+   * Describe an agent by its ID.
+   */
+  retrieve(agentId: string, options?: Core.RequestOptions): Core.APIPromise<AgentRetrieveResponse> {
+    return this._client.get(`/v1/agents/${agentId}`, options);
+  }
+
+  /**
+   * List all agents.
+   */
+  list(query?: AgentListParams, options?: Core.RequestOptions): Core.APIPromise<AgentListResponse>;
+  list(options?: Core.RequestOptions): Core.APIPromise<AgentListResponse>;
+  list(
+    query: AgentListParams | Core.RequestOptions = {},
+    options?: Core.RequestOptions,
+  ): Core.APIPromise<AgentListResponse> {
+    if (isRequestOptions(query)) {
+      return this.list({}, query);
+    }
+    return this._client.get('/v1/agents', { query, ...options });
+  }
+
   /**
    * Delete an agent by its ID and its associated sessions and turns.
    */
@@ -211,13 +236,44 @@ export interface ToolResponse {
 
   tool_name: 'brave_search' | 'wolfram_alpha' | 'photogen' | 'code_interpreter' | (string & {});
 
-  metadata?: Record<string, boolean | number | string | Array<unknown> | unknown | null>;
+  metadata?: { [key: string]: boolean | number | string | Array<unknown> | unknown | null };
 }
 
 export interface AgentCreateResponse {
   agent_id: string;
 }
 
+export interface AgentRetrieveResponse {
+  /**
+   * Configuration for an agent.
+   */
+  agent_config: Shared.AgentConfig;
+
+  agent_id: string;
+
+  created_at: string;
+}
+
+/**
+ * A generic paginated response that follows a simple format.
+ */
+export interface AgentListResponse {
+  /**
+   * The list of items for the current page
+   */
+  data: Array<{ [key: string]: boolean | number | string | Array<unknown> | unknown | null }>;
+
+  /**
+   * Whether there are more items available after this set
+   */
+  has_more: boolean;
+
+  /**
+   * The URL for accessing this list
+   */
+  url?: string;
+}
+
 export interface AgentCreateParams {
   /**
    * The configuration for the agent.
@@ -225,6 +281,18 @@ export interface AgentCreateParams {
   agent_config: Shared.AgentConfig;
 }
 
+export interface AgentListParams {
+  /**
+   * The number of agents to return.
+   */
+  limit?: number;
+
+  /**
+   * The index to start the pagination from.
+   */
+  start_index?: number;
+}
+
 Agents.SessionResource = SessionResource;
 Agents.Steps = Steps;
 Agents.TurnResource = TurnResource;
@@ -237,15 +305,20 @@ export declare namespace Agents {
     type ToolExecutionStep as ToolExecutionStep,
     type ToolResponse as ToolResponse,
     type AgentCreateResponse as AgentCreateResponse,
+    type AgentRetrieveResponse as AgentRetrieveResponse,
+    type AgentListResponse as AgentListResponse,
     type AgentCreateParams as AgentCreateParams,
+    type AgentListParams as AgentListParams,
   };
 
   export {
     SessionResource as SessionResource,
     type Session as Session,
     type SessionCreateResponse as SessionCreateResponse,
+    type SessionListResponse as SessionListResponse,
     type SessionCreateParams as SessionCreateParams,
     type SessionRetrieveParams as SessionRetrieveParams,
+    type SessionListParams as SessionListParams,
   };
 
   export { Steps as Steps, type StepRetrieveResponse as StepRetrieveResponse };

package/src/resources/agents/index.ts

@@ -8,14 +8,19 @@ export {
   type ToolExecutionStep,
   type ToolResponse,
   type AgentCreateResponse,
+  type AgentRetrieveResponse,
+  type AgentListResponse,
   type AgentCreateParams,
+  type AgentListParams,
 } from './agents';
 export {
   SessionResource,
   type Session,
   type SessionCreateResponse,
+  type SessionListResponse,
   type SessionCreateParams,
   type SessionRetrieveParams,
+  type SessionListParams,
 } from './session';
 export { Steps, type StepRetrieveResponse } from './steps';
 export {

package/src/resources/agents/session.ts

@@ -39,6 +39,26 @@ export class SessionResource extends APIResource {
     return this._client.get(`/v1/agents/${agentId}/session/${sessionId}`, { query, ...options });
   }
 
+  /**
+   * List all session(s) of a given agent.
+   */
+  list(
+    agentId: string,
+    query?: SessionListParams,
+    options?: Core.RequestOptions,
+  ): Core.APIPromise<SessionListResponse>;
+  list(agentId: string, options?: Core.RequestOptions): Core.APIPromise<SessionListResponse>;
+  list(
+    agentId: string,
+    query: SessionListParams | Core.RequestOptions = {},
+    options?: Core.RequestOptions,
+  ): Core.APIPromise<SessionListResponse> {
+    if (isRequestOptions(query)) {
+      return this.list(agentId, {}, query);
+    }
+    return this._client.get(`/v1/agents/${agentId}/sessions`, { query, ...options });
+  }
+
   /**
    * Delete an agent session by its ID and its associated turns.
    */
@@ -67,6 +87,26 @@ export interface SessionCreateResponse {
   session_id: string;
 }
 
+/**
+ * A generic paginated response that follows a simple format.
+ */
+export interface SessionListResponse {
+  /**
+   * The list of items for the current page
+   */
+  data: Array<{ [key: string]: boolean | number | string | Array<unknown> | unknown | null }>;
+
+  /**
+   * Whether there are more items available after this set
+   */
+  has_more: boolean;
+
+  /**
+   * The URL for accessing this list
+   */
+  url?: string;
+}
+
 export interface SessionCreateParams {
   /**
    * The name of the session to create.
@@ -81,11 +121,25 @@ export interface SessionRetrieveParams {
   turn_ids?: Array<string>;
 }
 
+export interface SessionListParams {
+  /**
+   * The number of sessions to return.
+   */
+  limit?: number;
+
+  /**
+   * The index to start the pagination from.
+   */
+  start_index?: number;
+}
+
 export declare namespace SessionResource {
   export {
     type Session as Session,
     type SessionCreateResponse as SessionCreateResponse,
+    type SessionListResponse as SessionListResponse,
     type SessionCreateParams as SessionCreateParams,
     type SessionRetrieveParams as SessionRetrieveParams,
+    type SessionListParams as SessionListParams,
   };
 }

package/src/resources/agents/turn.ts

@@ -242,7 +242,7 @@ export namespace TurnResponseEventPayload {
      */
     step_type: 'inference' | 'tool_execution' | 'shield_call' | 'memory_retrieval';
 
-    metadata?: Record<string, boolean | number | string | Array<unknown> | unknown | null>;
+    metadata?: { [key: string]: boolean | number | string | Array<unknown> | unknown | null };
   }
 
   export interface AgentTurnResponseStepProgressPayload {
@@ -453,7 +453,7 @@ export namespace TurnCreateParams {
   }
 
   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;
   }

package/src/resources/benchmarks.ts

@@ -37,7 +37,7 @@ export interface Benchmark {
 
   identifier: string;
 
-  metadata: Record<string, boolean | number | string | Array<unknown> | unknown | null>;
+  metadata: { [key: string]: boolean | number | string | Array<unknown> | unknown | null };
 
   provider_id: string;
 
@@ -73,7 +73,7 @@ export interface BenchmarkRegisterParams {
   /**
    * The metadata to use for the benchmark.
    */
-  metadata?: Record<string, boolean | number | string | Array<unknown> | unknown | null>;
+  metadata?: { [key: string]: boolean | number | string | Array<unknown> | unknown | null };
 
   /**
    * The ID of the provider benchmark to use for the benchmark.

package/src/resources/chat/completions.ts

@@ -1841,17 +1841,17 @@ export interface CompletionCreateParamsBase {
   /**
    * (Optional) The function call to use.
    */
-  function_call?: string | Record<string, boolean | number | string | Array<unknown> | unknown | null>;
+  function_call?: string | { [key: string]: boolean | number | string | Array<unknown> | unknown | null };
 
   /**
    * (Optional) List of functions to use.
    */
-  functions?: Array<Record<string, boolean | number | string | Array<unknown> | unknown | null>>;
+  functions?: Array<{ [key: string]: boolean | number | string | Array<unknown> | unknown | null }>;
 
   /**
    * (Optional) The logit bias to use.
    */
-  logit_bias?: Record<string, number>;
+  logit_bias?: { [key: string]: number };
 
   /**
    * (Optional) The log probabilities to use.
@@ -1909,7 +1909,7 @@ export interface CompletionCreateParamsBase {
   /**
    * (Optional) The stream options to use.
    */
-  stream_options?: Record<string, boolean | number | string | Array<unknown> | unknown | null>;
+  stream_options?: { [key: string]: boolean | number | string | Array<unknown> | unknown | null };
 
   /**
    * (Optional) The temperature to use.
@@ -1919,12 +1919,12 @@ export interface CompletionCreateParamsBase {
   /**
    * (Optional) The tool choice to use.
    */
-  tool_choice?: string | Record<string, boolean | number | string | Array<unknown> | unknown | null>;
+  tool_choice?: string | { [key: string]: boolean | number | string | Array<unknown> | unknown | null };
 
   /**
    * (Optional) The tools to use.
    */
-  tools?: Array<Record<string, boolean | number | string | Array<unknown> | unknown | null>>;
+  tools?: Array<{ [key: string]: boolean | number | string | Array<unknown> | unknown | null }>;
 
   /**
    * (Optional) The top log probabilities to use.
@@ -2221,7 +2221,7 @@ export namespace CompletionCreateParams {
 
       description?: string;
 
-      schema?: Record<string, boolean | number | string | Array<unknown> | unknown | null>;
+      schema?: { [key: string]: boolean | number | string | Array<unknown> | unknown | null };
 
       strict?: boolean;
     }

package/src/resources/completions.ts

@@ -179,7 +179,7 @@ export interface CompletionCreateParamsBase {
   /**
    * (Optional) The logit bias to use.
    */
-  logit_bias?: Record<string, number>;
+  logit_bias?: { [key: string]: number };
 
   /**
    * (Optional) The log probabilities to use.
@@ -221,7 +221,7 @@ export interface CompletionCreateParamsBase {
   /**
    * (Optional) The stream options to use.
    */
-  stream_options?: Record<string, boolean | number | string | Array<unknown> | unknown | null>;
+  stream_options?: { [key: string]: boolean | number | string | Array<unknown> | unknown | null };
 
   /**
    * (Optional) The suffix that should be appended to the completion.

package/src/resources/datasets.ts

@@ -21,6 +21,21 @@ export class Datasets extends APIResource {
     )._thenUnwrap((obj) => obj.data);
   }
 
+  /**
+   * Append rows to a dataset.
+   */
+  appendrows(
+    datasetId: string,
+    body: DatasetAppendrowsParams,
+    options?: Core.RequestOptions,
+  ): Core.APIPromise<void> {
+    return this._client.post(`/v1/datasetio/append-rows/${datasetId}`, {
+      body,
+      ...options,
+      headers: { Accept: '*/*', ...options?.headers },
+    });
+  }
+
   /**
    * Get a paginated list of rows from a dataset. Uses offset-based pagination where:
    *
@@ -77,7 +92,7 @@ export interface ListDatasetsResponse {
 export interface DatasetRetrieveResponse {
   identifier: string;
 
-  metadata: Record<string, boolean | number | string | Array<unknown> | unknown | null>;
+  metadata: { [key: string]: boolean | number | string | Array<unknown> | unknown | null };
 
   provider_id: string;
 
@@ -120,7 +135,7 @@ export namespace DatasetRetrieveResponse {
      * "content": "Hello, world!"}, {"role": "assistant", "content": "Hello, world!"}]}
      * ]
      */
-    rows: Array<Record<string, boolean | number | string | Array<unknown> | unknown | null>>;
+    rows: Array<{ [key: string]: boolean | number | string | Array<unknown> | unknown | null }>;
 
     type: 'rows';
   }
@@ -132,7 +147,7 @@ export namespace DatasetListResponse {
   export interface DatasetListResponseItem {
     identifier: string;
 
-    metadata: Record<string, boolean | number | string | Array<unknown> | unknown | null>;
+    metadata: { [key: string]: boolean | number | string | Array<unknown> | unknown | null };
 
     provider_id: string;
 
@@ -175,7 +190,7 @@ export namespace DatasetListResponse {
        * "content": "Hello, world!"}, {"role": "assistant", "content": "Hello, world!"}]}
        * ]
        */
-      rows: Array<Record<string, boolean | number | string | Array<unknown> | unknown | null>>;
+      rows: Array<{ [key: string]: boolean | number | string | Array<unknown> | unknown | null }>;
 
       type: 'rows';
     }
@@ -189,7 +204,7 @@ export interface DatasetIterrowsResponse {
   /**
    * The list of items for the current page
    */
-  data: Array<Record<string, boolean | number | string | Array<unknown> | unknown | null>>;
+  data: Array<{ [key: string]: boolean | number | string | Array<unknown> | unknown | null }>;
 
   /**
    * Whether there are more items available after this set
@@ -205,7 +220,7 @@ export interface DatasetIterrowsResponse {
 export interface DatasetRegisterResponse {
   identifier: string;
 
-  metadata: Record<string, boolean | number | string | Array<unknown> | unknown | null>;
+  metadata: { [key: string]: boolean | number | string | Array<unknown> | unknown | null };
 
   provider_id: string;
 
@@ -248,12 +263,19 @@ export namespace DatasetRegisterResponse {
      * "content": "Hello, world!"}, {"role": "assistant", "content": "Hello, world!"}]}
      * ]
      */
-    rows: Array<Record<string, boolean | number | string | Array<unknown> | unknown | null>>;
+    rows: Array<{ [key: string]: boolean | number | string | Array<unknown> | unknown | null }>;
 
     type: 'rows';
   }
 }
 
+export interface DatasetAppendrowsParams {
+  /**
+   * The rows to append to the dataset.
+   */
+  rows: Array<{ [key: string]: boolean | number | string | Array<unknown> | unknown | null }>;
+}
+
 export interface DatasetIterrowsParams {
   /**
    * The number of rows to get.
@@ -302,7 +324,7 @@ export interface DatasetRegisterParams {
   /**
    * The metadata for the dataset. - E.g. {"description": "My dataset"}.
    */
-  metadata?: Record<string, boolean | number | string | Array<unknown> | unknown | null>;
+  metadata?: { [key: string]: boolean | number | string | Array<unknown> | unknown | null };
 }
 
 export namespace DatasetRegisterParams {
@@ -329,7 +351,7 @@ export namespace DatasetRegisterParams {
      * "content": "Hello, world!"}, {"role": "assistant", "content": "Hello, world!"}]}
      * ]
      */
-    rows: Array<Record<string, boolean | number | string | Array<unknown> | unknown | null>>;
+    rows: Array<{ [key: string]: boolean | number | string | Array<unknown> | unknown | null }>;
 
     type: 'rows';
   }
@@ -342,6 +364,7 @@ export declare namespace Datasets {
     type DatasetListResponse as DatasetListResponse,
     type DatasetIterrowsResponse as DatasetIterrowsResponse,
     type DatasetRegisterResponse as DatasetRegisterResponse,
+    type DatasetAppendrowsParams as DatasetAppendrowsParams,
     type DatasetIterrowsParams as DatasetIterrowsParams,
     type DatasetRegisterParams as DatasetRegisterParams,
   };

package/src/resources/eval/eval.ts

@@ -64,7 +64,7 @@ export interface BenchmarkConfig {
    * Map between scoring function id and parameters for each scoring function you
    * want to run
    */
-  scoring_params: Record<string, ScoringFunctionsAPI.ScoringFnParams>;
+  scoring_params: { [key: string]: ScoringFunctionsAPI.ScoringFnParams };
 
   /**
    * (Optional) The number of examples to evaluate. If not provided, all examples in
@@ -121,12 +121,12 @@ export interface EvaluateResponse {
   /**
    * The generations from the evaluation.
    */
-  generations: Array<Record<string, boolean | number | string | Array<unknown> | unknown | null>>;
+  generations: Array<{ [key: string]: boolean | number | string | Array<unknown> | unknown | null }>;
 
   /**
    * The scores from the evaluation.
    */
-  scores: Record<string, Shared.ScoringResult>;
+  scores: { [key: string]: Shared.ScoringResult };
 }
 
 export interface Job {
@@ -144,7 +144,7 @@ export interface EvalEvaluateRowsParams {
   /**
    * The rows to evaluate.
    */
-  input_rows: Array<Record<string, boolean | number | string | Array<unknown> | unknown | null>>;
+  input_rows: Array<{ [key: string]: boolean | number | string | Array<unknown> | unknown | null }>;
 
   /**
    * The scoring functions to use for the evaluation.
@@ -161,7 +161,7 @@ export interface EvalEvaluateRowsAlphaParams {
   /**
    * The rows to evaluate.
    */
-  input_rows: Array<Record<string, boolean | number | string | Array<unknown> | unknown | null>>;
+  input_rows: Array<{ [key: string]: boolean | number | string | Array<unknown> | unknown | null }>;
 
   /**
    * The scoring functions to use for the evaluation.

package/src/resources/index.ts

@@ -9,7 +9,10 @@ export {
   type ToolExecutionStep,
   type ToolResponse,
   type AgentCreateResponse,
+  type AgentRetrieveResponse,
+  type AgentListResponse,
   type AgentCreateParams,
+  type AgentListParams,
 } from './agents/agents';
 export {
   Benchmarks,
@@ -33,6 +36,7 @@ export {
   type DatasetListResponse,
   type DatasetIterrowsResponse,
   type DatasetRegisterResponse,
+  type DatasetAppendrowsParams,
   type DatasetIterrowsParams,
   type DatasetRegisterParams,
 } from './datasets';

package/src/resources/inference.ts

@@ -190,7 +190,7 @@ export interface TokenLogProbs {
   /**
    * Dictionary mapping tokens to their log probabilities
    */
-  logprobs_by_token: Record<string, number>;
+  logprobs_by_token: { [key: string]: number };
 }
 
 export interface InferenceBatchChatCompletionResponse {
@@ -284,7 +284,7 @@ export namespace InferenceBatchChatCompletionParams {
 
     description?: string;
 
-    parameters?: Record<string, Shared.ToolParamDefinition>;
+    parameters?: { [key: string]: Shared.ToolParamDefinition };
   }
 }
 
@@ -447,7 +447,7 @@ export namespace InferenceChatCompletionParams {
 
     description?: string;
 
-    parameters?: Record<string, Shared.ToolParamDefinition>;
+    parameters?: { [key: string]: Shared.ToolParamDefinition };
   }
 
   export type InferenceChatCompletionParamsNonStreaming =

package/src/resources/inspect.ts

@@ -26,9 +26,9 @@ export interface HealthInfo {
 export interface ProviderInfo {
   api: string;
 
-  config: Record<string, boolean | number | string | Array<unknown> | unknown | null>;
+  config: { [key: string]: boolean | number | string | Array<unknown> | unknown | null };
 
-  health: Record<string, boolean | number | string | Array<unknown> | unknown | null>;
+  health: { [key: string]: boolean | number | string | Array<unknown> | unknown | null };
 
   provider_id: string;
 

package/src/resources/models.ts

@@ -45,7 +45,7 @@ export interface ListModelsResponse {
 export interface Model {
   identifier: string;
 
-  metadata: Record<string, boolean | number | string | Array<unknown> | unknown | null>;
+  metadata: { [key: string]: boolean | number | string | Array<unknown> | unknown | null };
 
   model_type: 'llm' | 'embedding';
 
@@ -67,7 +67,7 @@ export interface ModelRegisterParams {
   /**
    * Any additional metadata for this model.
    */
-  metadata?: Record<string, boolean | number | string | Array<unknown> | unknown | null>;
+  metadata?: { [key: string]: boolean | number | string | Array<unknown> | unknown | null };
 
   /**
    * The type of model to register.

package/src/resources/post-training/job.ts

@@ -2,7 +2,6 @@
 
 import { APIResource } from '../../resource';
 import * as Core from '../../core';
-import { ListPostTrainingJobsResponse } from './post-training';
 import * as PostTrainingAPI from './post-training';
 
 export class Job extends APIResource {
@@ -74,7 +73,7 @@ export interface JobStatusResponse {
 
   completed_at?: string;
 
-  resources_allocated?: Record<string, boolean | number | string | Array<unknown> | unknown | null>;
+  resources_allocated?: { [key: string]: boolean | number | string | Array<unknown> | unknown | null };
 
   scheduled_at?: string;
 

package/src/resources/post-training/post-training.ts

@@ -95,7 +95,7 @@ export interface PostTrainingPreferenceOptimizeParams {
   /**
    * The hyperparam search configuration.
    */
-  hyperparam_search_config: Record<string, boolean | number | string | Array<unknown> | unknown | null>;
+  hyperparam_search_config: { [key: string]: boolean | number | string | Array<unknown> | unknown | null };
 
   /**
    * The UUID of the job to create.
@@ -105,7 +105,7 @@ export interface PostTrainingPreferenceOptimizeParams {
   /**
    * The logger configuration.
    */
-  logger_config: Record<string, boolean | number | string | Array<unknown> | unknown | null>;
+  logger_config: { [key: string]: boolean | number | string | Array<unknown> | unknown | null };
 
   /**
    * The training configuration.
@@ -191,7 +191,7 @@ export interface PostTrainingSupervisedFineTuneParams {
   /**
    * The hyperparam search configuration.
    */
-  hyperparam_search_config: Record<string, boolean | number | string | Array<unknown> | unknown | null>;
+  hyperparam_search_config: { [key: string]: boolean | number | string | Array<unknown> | unknown | null };
 
   /**
    * The UUID of the job to create.
@@ -201,7 +201,7 @@ export interface PostTrainingSupervisedFineTuneParams {
   /**
    * The logger configuration.
    */
-  logger_config: Record<string, boolean | number | string | Array<unknown> | unknown | null>;
+  logger_config: { [key: string]: boolean | number | string | Array<unknown> | unknown | null };
 
   /**
    * The training configuration.