llama-stack-client 0.3.4 → 0.4.0-rc1

package/src/resources/shields.ts

@@ -3,7 +3,7 @@
 //
 // 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 { APIResource } from '../resource';
@@ -28,6 +28,8 @@ export class Shields extends APIResource {
 
   /**
    * Unregister a shield.
+   *
+   * @deprecated
    */
   delete(identifier: string, options?: Core.RequestOptions): Core.APIPromise<void> {
     return this._client.delete(`/v1/shields/${identifier}`, {
@@ -38,6 +40,8 @@ export class Shields extends APIResource {
 
   /**
    * Register a shield.
+   *
+   * @deprecated
    */
   register(body: ShieldRegisterParams, options?: Core.RequestOptions): Core.APIPromise<Shield> {
     return this._client.post('/v1/shields', { body, ...options });
@@ -52,45 +56,36 @@ export interface ListShieldsResponse {
  * A safety shield resource that can be used to check content.
  */
 export interface Shield {
+  /**
+   * Unique identifier for this resource in llama stack
+   */
   identifier: string;
 
-  provider_id: string;
-
   /**
-   * The resource type, always shield
+   * ID of the provider that owns this resource
    */
-  type: 'shield';
+  provider_id: string;
+
+  params?: { [key: string]: unknown } | null;
 
   /**
-   * (Optional) Configuration parameters for the shield
+   * Unique identifier for this resource in the provider
    */
-  params?: { [key: string]: boolean | number | string | Array<unknown> | unknown | null };
+  provider_resource_id?: string | null;
 
-  provider_resource_id?: string;
+  type?: 'shield';
 }
 
 export type ShieldListResponse = Array<Shield>;
 
 export interface ShieldRegisterParams {
-  /**
-   * The identifier of the shield to register.
-   */
   shield_id: string;
 
-  /**
-   * The parameters of the shield.
-   */
-  params?: { [key: string]: boolean | number | string | Array<unknown> | unknown | null };
+  params?: { [key: string]: unknown } | null;
 
-  /**
-   * The identifier of the provider.
-   */
-  provider_id?: string;
+  provider_id?: string | null;
 
-  /**
-   * The identifier of the shield in the provider.
-   */
-  provider_shield_id?: string;
+  provider_shield_id?: string | null;
 }
 
 export declare namespace Shields {

package/src/resources/tool-runtime.ts

@@ -3,7 +3,215 @@
 //
 // 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.
 
-export * from './tool-runtime/index';
+import { APIResource } from '../resource';
+import { isRequestOptions } from '../core';
+import * as Core from '../core';
+
+export class ToolRuntime extends APIResource {
+  /**
+   * Run a tool with the given arguments.
+   *
+   * @deprecated
+   */
+  invokeTool(
+    body: ToolRuntimeInvokeToolParams,
+    options?: Core.RequestOptions,
+  ): Core.APIPromise<ToolInvocationResult> {
+    return this._client.post('/v1/tool-runtime/invoke', { body, ...options });
+  }
+
+  /**
+   * List all tools in the runtime.
+   *
+   * @deprecated
+   */
+  listTools(
+    query?: ToolRuntimeListToolsParams,
+    options?: Core.RequestOptions,
+  ): Core.APIPromise<ToolRuntimeListToolsResponse>;
+  listTools(options?: Core.RequestOptions): Core.APIPromise<ToolRuntimeListToolsResponse>;
+  listTools(
+    query: ToolRuntimeListToolsParams | Core.RequestOptions = {},
+    options?: Core.RequestOptions,
+  ): Core.APIPromise<ToolRuntimeListToolsResponse> {
+    if (isRequestOptions(query)) {
+      return this.listTools({}, query);
+    }
+    return (
+      this._client.get('/v1/tool-runtime/list-tools', { query, ...options }) as Core.APIPromise<{
+        data: ToolRuntimeListToolsResponse;
+      }>
+    )._thenUnwrap((obj) => obj.data);
+  }
+}
+
+/**
+ * Tool definition used in runtime contexts.
+ */
+export interface ToolDef {
+  name: string;
+
+  description?: string | null;
+
+  input_schema?: { [key: string]: unknown } | null;
+
+  metadata?: { [key: string]: unknown } | null;
+
+  output_schema?: { [key: string]: unknown } | null;
+
+  toolgroup_id?: string | null;
+}
+
+/**
+ * Result of a tool invocation.
+ */
+export interface ToolInvocationResult {
+  /**
+   * A image content item
+   */
+  content?:
+    | string
+    | ToolInvocationResult.ImageContentItemOutput
+    | ToolInvocationResult.TextContentItem
+    | Array<ToolInvocationResult.ImageContentItemOutput | ToolInvocationResult.TextContentItem>
+    | null;
+
+  error_code?: number | null;
+
+  error_message?: string | null;
+
+  metadata?: { [key: string]: unknown } | null;
+}
+
+export namespace ToolInvocationResult {
+  /**
+   * A image content item
+   */
+  export interface ImageContentItemOutput {
+    /**
+     * A URL or a base64 encoded string
+     */
+    image: ImageContentItemOutput.Image;
+
+    type?: 'image';
+  }
+
+  export namespace ImageContentItemOutput {
+    /**
+     * 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;
+      }
+    }
+  }
+
+  /**
+   * A text content item
+   */
+  export interface TextContentItem {
+    text: string;
+
+    type?: 'text';
+  }
+
+  /**
+   * A image content item
+   */
+  export interface ImageContentItemOutput {
+    /**
+     * A URL or a base64 encoded string
+     */
+    image: ImageContentItemOutput.Image;
+
+    type?: 'image';
+  }
+
+  export namespace ImageContentItemOutput {
+    /**
+     * 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;
+      }
+    }
+  }
+
+  /**
+   * A text content item
+   */
+  export interface TextContentItem {
+    text: string;
+
+    type?: 'text';
+  }
+}
+
+export type ToolRuntimeListToolsResponse = Array<ToolDef>;
+
+export interface ToolRuntimeInvokeToolParams {
+  kwargs: { [key: string]: unknown };
+
+  tool_name: string;
+
+  authorization?: string | null;
+}
+
+export interface ToolRuntimeListToolsParams {
+  authorization?: string | null;
+
+  /**
+   * A URL reference to external content.
+   */
+  mcp_endpoint?: ToolRuntimeListToolsParams.McpEndpoint | null;
+
+  tool_group_id?: string | null;
+}
+
+export namespace ToolRuntimeListToolsParams {
+  /**
+   * A URL reference to external content.
+   */
+  export interface McpEndpoint {
+    uri: string;
+  }
+}
+
+export declare namespace ToolRuntime {
+  export {
+    type ToolDef as ToolDef,
+    type ToolInvocationResult as ToolInvocationResult,
+    type ToolRuntimeListToolsResponse as ToolRuntimeListToolsResponse,
+    type ToolRuntimeInvokeToolParams as ToolRuntimeInvokeToolParams,
+    type ToolRuntimeListToolsParams as ToolRuntimeListToolsParams,
+  };
+}

package/src/resources/toolgroups.ts

@@ -3,7 +3,7 @@
 //
 // 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 { APIResource } from '../resource';
@@ -12,6 +12,8 @@ import * as Core from '../core';
 export class Toolgroups extends APIResource {
   /**
    * List tool groups with optional provider.
+   *
+   * @deprecated
    */
   list(options?: Core.RequestOptions): Core.APIPromise<ToolgroupListResponse> {
     return (
@@ -21,6 +23,8 @@ export class Toolgroups extends APIResource {
 
   /**
    * Get a tool group by its ID.
+   *
+   * @deprecated
    */
   get(toolgroupId: string, options?: Core.RequestOptions): Core.APIPromise<ToolGroup> {
     return this._client.get(`/v1/toolgroups/${toolgroupId}`, options);
@@ -28,6 +32,8 @@ export class Toolgroups extends APIResource {
 
   /**
    * Register a tool group.
+   *
+   * @deprecated
    */
   register(body: ToolgroupRegisterParams, options?: Core.RequestOptions): Core.APIPromise<void> {
     return this._client.post('/v1/toolgroups', {
@@ -39,6 +45,8 @@ export class Toolgroups extends APIResource {
 
   /**
    * Unregister a tool group.
+   *
+   * @deprecated
    */
   unregister(toolgroupId: string, options?: Core.RequestOptions): Core.APIPromise<void> {
     return this._client.delete(`/v1/toolgroups/${toolgroupId}`, {
@@ -52,9 +60,6 @@ export class Toolgroups extends APIResource {
  * Response containing a list of tool groups.
  */
 export interface ListToolGroupsResponse {
-  /**
-   * List of tool groups
-   */
   data: ToolgroupListResponse;
 }
 
@@ -62,75 +67,60 @@ export interface ListToolGroupsResponse {
  * A group of related tools managed together.
  */
 export interface ToolGroup {
+  /**
+   * Unique identifier for this resource in llama stack
+   */
   identifier: string;
 
-  provider_id: string;
-
   /**
-   * Type of resource, always 'tool_group'
+   * ID of the provider that owns this resource
    */
-  type: 'tool_group';
+  provider_id: string;
+
+  args?: { [key: string]: unknown } | null;
 
   /**
-   * (Optional) Additional arguments for the tool group
+   * A URL reference to external content.
    */
-  args?: { [key: string]: boolean | number | string | Array<unknown> | unknown | null };
+  mcp_endpoint?: ToolGroup.McpEndpoint | null;
 
   /**
-   * (Optional) Model Context Protocol endpoint for remote tools
+   * Unique identifier for this resource in the provider
    */
-  mcp_endpoint?: ToolGroup.McpEndpoint;
+  provider_resource_id?: string | null;
 
-  provider_resource_id?: string;
+  type?: 'tool_group';
 }
 
 export namespace ToolGroup {
   /**
-   * (Optional) Model Context Protocol endpoint for remote tools
+   * A URL reference to external content.
    */
   export interface McpEndpoint {
-    /**
-     * The URL string pointing to the resource
-     */
     uri: string;
   }
 }
 
-/**
- * List of tool groups
- */
 export type ToolgroupListResponse = Array<ToolGroup>;
 
 export interface ToolgroupRegisterParams {
-  /**
-   * The ID of the provider to use for the tool group.
-   */
   provider_id: string;
 
-  /**
-   * The ID of the tool group to register.
-   */
   toolgroup_id: string;
 
-  /**
-   * A dictionary of arguments to pass to the tool group.
-   */
-  args?: { [key: string]: boolean | number | string | Array<unknown> | unknown | null };
+  args?: { [key: string]: unknown } | null;
 
   /**
-   * The MCP endpoint to use for the tool group.
+   * A URL reference to external content.
    */
-  mcp_endpoint?: ToolgroupRegisterParams.McpEndpoint;
+  mcp_endpoint?: ToolgroupRegisterParams.McpEndpoint | null;
 }
 
 export namespace ToolgroupRegisterParams {
   /**
-   * The MCP endpoint to use for the tool group.
+   * A URL reference to external content.
    */
   export interface McpEndpoint {
-    /**
-     * The URL string pointing to the resource
-     */
     uri: string;
   }
 }

package/src/resources/tools.ts

@@ -3,16 +3,19 @@
 //
 // 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 { APIResource } from '../resource';
 import { isRequestOptions } from '../core';
 import * as Core from '../core';
+import * as ToolRuntimeAPI from './tool-runtime';
 
 export class Tools extends APIResource {
   /**
    * List tools with optional tool group.
+   *
+   * @deprecated
    */
   list(query?: ToolListParams, options?: Core.RequestOptions): Core.APIPromise<ToolListResponse>;
   list(options?: Core.RequestOptions): Core.APIPromise<ToolListResponse>;
@@ -30,107 +33,20 @@ export class Tools extends APIResource {
 
   /**
    * Get a tool by its name.
+   *
+   * @deprecated
    */
-  get(toolName: string, options?: Core.RequestOptions): Core.APIPromise<Tool> {
+  get(toolName: string, options?: Core.RequestOptions): Core.APIPromise<ToolRuntimeAPI.ToolDef> {
     return this._client.get(`/v1/tools/${toolName}`, options);
   }
 }
 
-/**
- * Response containing a list of tools.
- */
-export interface ListToolsResponse {
-  /**
-   * List of tools
-   */
-  data: ToolListResponse;
-}
-
-/**
- * A tool that can be invoked by agents.
- */
-export interface Tool {
-  /**
-   * Human-readable description of what the tool does
-   */
-  description: string;
-
-  identifier: string;
-
-  /**
-   * List of parameters this tool accepts
-   */
-  parameters: Array<Tool.Parameter>;
-
-  provider_id: string;
-
-  /**
-   * ID of the tool group this tool belongs to
-   */
-  toolgroup_id: string;
-
-  /**
-   * Type of resource, always 'tool'
-   */
-  type: 'tool';
-
-  /**
-   * (Optional) Additional metadata about the tool
-   */
-  metadata?: { [key: string]: boolean | number | string | Array<unknown> | unknown | null };
-
-  provider_resource_id?: string;
-}
-
-export namespace Tool {
-  /**
-   * 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;
-  }
-}
-
-/**
- * List of tools
- */
-export type ToolListResponse = Array<Tool>;
+export type ToolListResponse = Array<ToolRuntimeAPI.ToolDef>;
 
 export interface ToolListParams {
-  /**
-   * The ID of the tool group to list tools for.
-   */
-  toolgroup_id?: string;
+  toolgroup_id?: string | null;
 }
 
 export declare namespace Tools {
-  export {
-    type ListToolsResponse as ListToolsResponse,
-    type Tool as Tool,
-    type ToolListResponse as ToolListResponse,
-    type ToolListParams as ToolListParams,
-  };
+  export { type ToolListResponse as ToolListResponse, type ToolListParams as ToolListParams };
 }