feat(api): manual updates

Author: stainless-app[bot]

.stats.yml

@@ -1,4 +1,4 @@
 configured_endpoints: 106
 openapi_spec_url: https://storage.googleapis.com/stainless-sdk-openapi-specs/llamastack%2Fllama-stack-client-4f6633567c1a079df49d0cf58f37251a4bb0ee2f2a496ac83c9fee26eb325f9c.yml
 openapi_spec_hash: af5b3d3bbecf48f15c90b982ccac852e
-config_hash: de16e52db65de71ac35adcdb665a74f5
+config_hash: ddcbd66d7ac80290da208232a746e30f

api.md

@@ -20,6 +20,7 @@ Types:
 - <code><a href="./src/resources/shared.ts">SafetyViolation</a></code>
 - <code><a href="./src/resources/shared.ts">SamplingParams</a></code>
 - <code><a href="./src/resources/shared.ts">ScoringResult</a></code>
+- <code><a href="./src/resources/shared.ts">SharedTokenLogProbs</a></code>
 - <code><a href="./src/resources/shared.ts">SystemMessage</a></code>
 - <code><a href="./src/resources/shared.ts">ToolCall</a></code>
 - <code><a href="./src/resources/shared.ts">ToolCallOrString</a></code>
@@ -59,7 +60,6 @@ Methods:
 
 Types:
 
-- <code><a href="./src/resources/tool-runtime/tool-runtime.ts">ToolDef</a></code>
 - <code><a href="./src/resources/tool-runtime/tool-runtime.ts">ToolInvocationResult</a></code>
 - <code><a href="./src/resources/tool-runtime/tool-runtime.ts">ToolRuntimeListToolsResponse</a></code>
 
@@ -221,7 +221,6 @@ Methods:
 Types:
 
 - <code><a href="./src/resources/inference.ts">ChatCompletionResponseStreamChunk</a></code>
-- <code><a href="./src/resources/inference.ts">CompletionResponse</a></code>
 - <code><a href="./src/resources/inference.ts">EmbeddingsResponse</a></code>
 - <code><a href="./src/resources/inference.ts">TokenLogProbs</a></code>
 - <code><a href="./src/resources/inference.ts">InferenceBatchChatCompletionResponse</a></code>

src/index.ts

@@ -45,7 +45,6 @@ import {
 } from './resources/files';
 import {
   ChatCompletionResponseStreamChunk,
-  CompletionResponse,
   EmbeddingsResponse,
   Inference,
   InferenceBatchChatCompletionParams,
@@ -182,7 +181,6 @@ import {
   Responses,
 } from './resources/responses/responses';
 import {
-  ToolDef,
   ToolInvocationResult,
   ToolRuntime,
   ToolRuntimeInvokeToolParams,
@@ -445,7 +443,6 @@ export declare namespace LlamaStackClient {
 
   export {
     ToolRuntime as ToolRuntime,
-    type ToolDef as ToolDef,
     type ToolInvocationResult as ToolInvocationResult,
     type ToolRuntimeListToolsResponse as ToolRuntimeListToolsResponse,
     type ToolRuntimeInvokeToolParams as ToolRuntimeInvokeToolParams,
@@ -512,7 +509,6 @@ export declare namespace LlamaStackClient {
   export {
     Inference as Inference,
     type ChatCompletionResponseStreamChunk as ChatCompletionResponseStreamChunk,
-    type CompletionResponse as CompletionResponse,
     type EmbeddingsResponse as EmbeddingsResponse,
     type TokenLogProbs as TokenLogProbs,
     type InferenceBatchChatCompletionResponse as InferenceBatchChatCompletionResponse,
@@ -697,6 +693,7 @@ export declare namespace LlamaStackClient {
   export type SafetyViolation = API.SafetyViolation;
   export type SamplingParams = API.SamplingParams;
   export type ScoringResult = API.ScoringResult;
+  export type SharedTokenLogProbs = API.SharedTokenLogProbs;
   export type SystemMessage = API.SystemMessage;
   export type ToolCall = API.ToolCall;
   export type ToolCallOrString = API.ToolCallOrString;

src/pagination.ts

@@ -48,15 +48,14 @@ export class DatasetsIterrows<Item> extends AbstractPage<Item> implements Datase
   }
 
   nextPageInfo(): PageInfo | null {
-    const cursor = this.next_index;
-    if (!cursor) {
+    const offset = this.next_index;
+    if (!offset) {
       return null;
     }
 
-    return {
-      params: {
-        start_index: cursor,
-      },
-    };
+    const length = this.getPaginatedItems().length;
+    const currentCount = offset + length;
+
+    return { params: { start_index: currentCount } };
   }
 }

src/resources/index.ts

@@ -64,7 +64,6 @@ export {
 export {
   Inference,
   type ChatCompletionResponseStreamChunk,
-  type CompletionResponse,
   type EmbeddingsResponse,
   type TokenLogProbs,
   type InferenceBatchChatCompletionResponse,
@@ -154,7 +153,6 @@ export {
 } from './telemetry';
 export {
   ToolRuntime,
-  type ToolDef,
   type ToolInvocationResult,
   type ToolRuntimeListToolsResponse,
   type ToolRuntimeInvokeToolParams,

src/resources/inference.ts

@@ -131,7 +131,7 @@ export namespace ChatCompletionResponseStreamChunk {
     /**
      * Optional log probabilities for generated tokens
      */
-    logprobs?: Array<InferenceAPI.TokenLogProbs>;
+    logprobs?: Array<Shared.SharedTokenLogProbs>;
 
     /**
      * Optional reason why generation stopped, if complete
@@ -160,53 +160,6 @@ export namespace ChatCompletionResponseStreamChunk {
   }
 }
 
-/**
- * Response from a completion request.
- */
-export interface CompletionResponse {
-  /**
-   * The generated completion text
-   */
-  content: string;
-
-  /**
-   * Reason why generation stopped
-   */
-  stop_reason: 'end_of_turn' | 'end_of_message' | 'out_of_tokens';
-
-  /**
-   * Optional log probabilities for generated tokens
-   */
-  logprobs?: Array<TokenLogProbs>;
-
-  /**
-   * (Optional) List of metrics associated with the API response
-   */
-  metrics?: Array<CompletionResponse.Metric>;
-}
-
-export namespace CompletionResponse {
-  /**
-   * 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;
-  }
-}
-
 /**
  * Response containing generated embeddings.
  */
@@ -620,7 +573,6 @@ export interface InferenceEmbeddingsParams {
 export declare namespace Inference {
   export {
     type ChatCompletionResponseStreamChunk as ChatCompletionResponseStreamChunk,
-    type CompletionResponse as CompletionResponse,
     type EmbeddingsResponse as EmbeddingsResponse,
     type TokenLogProbs as TokenLogProbs,
     type InferenceBatchChatCompletionResponse as InferenceBatchChatCompletionResponse,

src/resources/shared.ts

@@ -1,8 +1,6 @@
 // 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.
@@ -18,7 +16,7 @@ export interface AgentConfig {
    */
   model: string;
 
-  client_tools?: Array<ToolRuntimeAPI.ToolDef>;
+  client_tools?: Array<AgentConfig.ClientTool>;
 
   /**
    * Optional flag indicating whether session data has to be persisted
@@ -67,6 +65,63 @@ export interface AgentConfig {
 }
 
 export namespace AgentConfig {
+  /**
+   * Tool definition used in runtime contexts.
+   */
+  export interface ClientTool {
+    /**
+     * Name of the tool
+     */
+    name: string;
+
+    /**
+     * (Optional) Human-readable description of what the tool does
+     */
+    description?: string;
+
+    /**
+     * (Optional) Additional metadata about the tool
+     */
+    metadata?: { [key: string]: boolean | number | string | Array<unknown> | unknown | null };
+
+    /**
+     * (Optional) List of parameters this tool accepts
+     */
+    parameters?: Array<ClientTool.Parameter>;
+  }
+
+  export namespace ClientTool {
+    /**
+     * Parameter definition for a tool.
+     */
+    export interface Parameter {
+      /**
+       * Human-readable description of what the parameter does
+       */
+      description: string;
+
+      /**
+       * Name of the parameter
+       */
+      name: string;
+
+      /**
+       * Type of the parameter (e.g., string, integer)
+       */
+      parameter_type: string;
+
+      /**
+       * Whether this parameter is required for tool invocation
+       */
+      required: boolean;
+
+      /**
+       * (Optional) Default value for the parameter if not provided
+       */
+      default?: boolean | number | string | Array<unknown> | unknown | null;
+    }
+  }
+
   /**
    * Configuration for tool use.
    */
@@ -112,7 +167,56 @@ export interface BatchCompletion {
   /**
    * List of completion responses, one for each input in the batch
    */
-  batch: Array<InferenceAPI.CompletionResponse>;
+  batch: Array<BatchCompletion.Batch>;
+}
+
+export namespace BatchCompletion {
+  /**
+   * Response from a completion request.
+   */
+  export interface Batch {
+    /**
+     * The generated completion text
+     */
+    content: string;
+
+    /**
+     * Reason why generation stopped
+     */
+    stop_reason: 'end_of_turn' | 'end_of_message' | 'out_of_tokens';
+
+    /**
+     * Optional log probabilities for generated tokens
+     */
+    logprobs?: Array<Shared.SharedTokenLogProbs>;
+
+    /**
+     * (Optional) List of metrics associated with the API response
+     */
+    metrics?: Array<Batch.Metric>;
+  }
+
+  export namespace Batch {
+    /**
+     * 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;
+    }
+  }
 }
 
 /**
@@ -127,7 +231,7 @@ export interface ChatCompletionResponse {
   /**
    * Optional log probabilities for generated tokens
    */
-  logprobs?: Array<InferenceAPI.TokenLogProbs>;
+  logprobs?: Array<SharedTokenLogProbs>;
 
   /**
    * (Optional) List of metrics associated with the API response
@@ -912,6 +1016,16 @@ export interface ScoringResult {
   score_rows: Array<{ [key: string]: boolean | number | string | Array<unknown> | unknown | null }>;
 }
 
+/**
+ * Log probabilities for generated tokens.
+ */
+export interface SharedTokenLogProbs {
+  /**
+   * Dictionary mapping tokens to their log probabilities
+   */
+  logprobs_by_token: { [key: string]: number };
+}
+
 /**
  * A system message providing instructions or context to the model.
  */

src/resources/tool-runtime/index.ts

@@ -3,7 +3,6 @@
 export { RagTool, type RagToolInsertParams, type RagToolQueryParams } from './rag-tool';
 export {
   ToolRuntime,
-  type ToolDef,
   type ToolInvocationResult,
   type ToolRuntimeListToolsResponse,
   type ToolRuntimeInvokeToolParams,