llama-stack-client 0.6.1 → 0.7.1

package/src/resources/toolgroups.ts

@@ -1,135 +0,0 @@
-// Copyright (c) Meta Platforms, Inc. and affiliates.
-// All rights reserved.
-//
-// 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 * as Core from '../core';
-
-export class Toolgroups extends APIResource {
-  /**
-   * List tool groups with optional provider.
-   *
-   * @deprecated
-   */
-  list(options?: Core.RequestOptions): Core.APIPromise<ToolgroupListResponse> {
-    return (
-      this._client.get('/v1/toolgroups', options) as Core.APIPromise<{ data: ToolgroupListResponse }>
-    )._thenUnwrap((obj) => obj.data);
-  }
-
-  /**
-   * 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);
-  }
-
-  /**
-   * Register a tool group.
-   *
-   * @deprecated
-   */
-  register(body: ToolgroupRegisterParams, options?: Core.RequestOptions): Core.APIPromise<void> {
-    return this._client.post('/v1/toolgroups', {
-      body,
-      ...options,
-      headers: { Accept: '*/*', ...options?.headers },
-    });
-  }
-
-  /**
-   * Unregister a tool group.
-   *
-   * @deprecated
-   */
-  unregister(toolgroupId: string, options?: Core.RequestOptions): Core.APIPromise<void> {
-    return this._client.delete(`/v1/toolgroups/${toolgroupId}`, {
-      ...options,
-      headers: { Accept: '*/*', ...options?.headers },
-    });
-  }
-}
-
-/**
- * Response containing a list of tool groups.
- */
-export interface ListToolGroupsResponse {
-  data: ToolgroupListResponse;
-}
-
-/**
- * A group of related tools managed together.
- */
-export interface ToolGroup {
-  /**
-   * Unique identifier for this resource in llama stack
-   */
-  identifier: string;
-
-  /**
-   * ID of the provider that owns this resource
-   */
-  provider_id: string;
-
-  args?: { [key: string]: unknown } | null;
-
-  /**
-   * A URL reference to external content.
-   */
-  mcp_endpoint?: ToolGroup.McpEndpoint | null;
-
-  /**
-   * Unique identifier for this resource in the provider
-   */
-  provider_resource_id?: string | null;
-
-  type?: 'tool_group';
-}
-
-export namespace ToolGroup {
-  /**
-   * A URL reference to external content.
-   */
-  export interface McpEndpoint {
-    uri: string;
-  }
-}
-
-export type ToolgroupListResponse = Array<ToolGroup>;
-
-export interface ToolgroupRegisterParams {
-  provider_id: string;
-
-  toolgroup_id: string;
-
-  args?: { [key: string]: unknown } | null;
-
-  /**
-   * A URL reference to external content.
-   */
-  mcp_endpoint?: ToolgroupRegisterParams.McpEndpoint | null;
-}
-
-export namespace ToolgroupRegisterParams {
-  /**
-   * A URL reference to external content.
-   */
-  export interface McpEndpoint {
-    uri: string;
-  }
-}
-
-export declare namespace Toolgroups {
-  export {
-    type ListToolGroupsResponse as ListToolGroupsResponse,
-    type ToolGroup as ToolGroup,
-    type ToolgroupListResponse as ToolgroupListResponse,
-    type ToolgroupRegisterParams as ToolgroupRegisterParams,
-  };
-}

package/src/resources/tools.ts

@@ -1,52 +0,0 @@
-// Copyright (c) Meta Platforms, Inc. and affiliates.
-// All rights reserved.
-//
-// 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>;
-  list(
-    query: ToolListParams | Core.RequestOptions = {},
-    options?: Core.RequestOptions,
-  ): Core.APIPromise<ToolListResponse> {
-    if (isRequestOptions(query)) {
-      return this.list({}, query);
-    }
-    return (
-      this._client.get('/v1/tools', { query, ...options }) as Core.APIPromise<{ data: ToolListResponse }>
-    )._thenUnwrap((obj) => obj.data);
-  }
-
-  /**
-   * Get a tool by its name.
-   *
-   * @deprecated
-   */
-  get(toolName: string, options?: Core.RequestOptions): Core.APIPromise<ToolRuntimeAPI.ToolDef> {
-    return this._client.get(`/v1/tools/${toolName}`, options);
-  }
-}
-
-export type ToolListResponse = Array<ToolRuntimeAPI.ToolDef>;
-
-export interface ToolListParams {
-  toolgroup_id?: string | null;
-}
-
-export declare namespace Tools {
-  export { type ToolListResponse as ToolListResponse, type ToolListParams as ToolListParams };
-}

package/tests/api-resources/alpha/post-training/job.test.ts

@@ -1,86 +0,0 @@
-// Copyright (c) Meta Platforms, Inc. and affiliates.
-// All rights reserved.
-//
-// 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 LlamaStackClient from 'llama-stack-client';
-import { Response } from 'node-fetch';
-
-const client = new LlamaStackClient({ baseURL: process.env['TEST_API_BASE_URL'] ?? 'http://127.0.0.1:4010' });
-
-describe('resource job', () => {
-  test('list', async () => {
-    const responsePromise = client.alpha.postTraining.job.list();
-    const rawResponse = await responsePromise.asResponse();
-    expect(rawResponse).toBeInstanceOf(Response);
-    const response = await responsePromise;
-    expect(response).not.toBeInstanceOf(Response);
-    const dataAndResponse = await responsePromise.withResponse();
-    expect(dataAndResponse.data).toBe(response);
-    expect(dataAndResponse.response).toBe(rawResponse);
-  });
-
-  test('list: request options instead of params are passed correctly', async () => {
-    // ensure the request options are being passed correctly by passing an invalid HTTP method in order to cause an error
-    await expect(client.alpha.postTraining.job.list({ path: '/_stainless_unknown_path' })).rejects.toThrow(
-      LlamaStackClient.NotFoundError,
-    );
-  });
-
-  test('artifacts', async () => {
-    const responsePromise = client.alpha.postTraining.job.artifacts();
-    const rawResponse = await responsePromise.asResponse();
-    expect(rawResponse).toBeInstanceOf(Response);
-    const response = await responsePromise;
-    expect(response).not.toBeInstanceOf(Response);
-    const dataAndResponse = await responsePromise.withResponse();
-    expect(dataAndResponse.data).toBe(response);
-    expect(dataAndResponse.response).toBe(rawResponse);
-  });
-
-  test('artifacts: request options instead of params are passed correctly', async () => {
-    // ensure the request options are being passed correctly by passing an invalid HTTP method in order to cause an error
-    await expect(
-      client.alpha.postTraining.job.artifacts({ path: '/_stainless_unknown_path' }),
-    ).rejects.toThrow(LlamaStackClient.NotFoundError);
-  });
-
-  test('cancel', async () => {
-    const responsePromise = client.alpha.postTraining.job.cancel();
-    const rawResponse = await responsePromise.asResponse();
-    expect(rawResponse).toBeInstanceOf(Response);
-    const response = await responsePromise;
-    expect(response).not.toBeInstanceOf(Response);
-    const dataAndResponse = await responsePromise.withResponse();
-    expect(dataAndResponse.data).toBe(response);
-    expect(dataAndResponse.response).toBe(rawResponse);
-  });
-
-  test('cancel: request options instead of params are passed correctly', async () => {
-    // ensure the request options are being passed correctly by passing an invalid HTTP method in order to cause an error
-    await expect(client.alpha.postTraining.job.cancel({ path: '/_stainless_unknown_path' })).rejects.toThrow(
-      LlamaStackClient.NotFoundError,
-    );
-  });
-
-  test('status', async () => {
-    const responsePromise = client.alpha.postTraining.job.status();
-    const rawResponse = await responsePromise.asResponse();
-    expect(rawResponse).toBeInstanceOf(Response);
-    const response = await responsePromise;
-    expect(response).not.toBeInstanceOf(Response);
-    const dataAndResponse = await responsePromise.withResponse();
-    expect(dataAndResponse.data).toBe(response);
-    expect(dataAndResponse.response).toBe(rawResponse);
-  });
-
-  test('status: request options instead of params are passed correctly', async () => {
-    // ensure the request options are being passed correctly by passing an invalid HTTP method in order to cause an error
-    await expect(client.alpha.postTraining.job.status({ path: '/_stainless_unknown_path' })).rejects.toThrow(
-      LlamaStackClient.NotFoundError,
-    );
-  });
-});

package/tests/api-resources/alpha/post-training/post-training.test.ts

@@ -1,134 +0,0 @@
-// Copyright (c) Meta Platforms, Inc. and affiliates.
-// All rights reserved.
-//
-// 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 LlamaStackClient from 'llama-stack-client';
-import { Response } from 'node-fetch';
-
-const client = new LlamaStackClient({ baseURL: process.env['TEST_API_BASE_URL'] ?? 'http://127.0.0.1:4010' });
-
-describe('resource postTraining', () => {
-  test('preferenceOptimize: only required params', async () => {
-    const responsePromise = client.alpha.postTraining.preferenceOptimize({
-      algorithm_config: { beta: 0 },
-      finetuned_model: 'finetuned_model',
-      hyperparam_search_config: { foo: 'bar' },
-      job_uuid: 'job_uuid',
-      logger_config: { foo: 'bar' },
-      training_config: { n_epochs: 0 },
-    });
-    const rawResponse = await responsePromise.asResponse();
-    expect(rawResponse).toBeInstanceOf(Response);
-    const response = await responsePromise;
-    expect(response).not.toBeInstanceOf(Response);
-    const dataAndResponse = await responsePromise.withResponse();
-    expect(dataAndResponse.data).toBe(response);
-    expect(dataAndResponse.response).toBe(rawResponse);
-  });
-
-  test('preferenceOptimize: required and optional params', async () => {
-    const response = await client.alpha.postTraining.preferenceOptimize({
-      algorithm_config: { beta: 0, loss_type: 'sigmoid' },
-      finetuned_model: 'finetuned_model',
-      hyperparam_search_config: { foo: 'bar' },
-      job_uuid: 'job_uuid',
-      logger_config: { foo: 'bar' },
-      training_config: {
-        n_epochs: 0,
-        data_config: {
-          batch_size: 0,
-          data_format: 'instruct',
-          dataset_id: 'dataset_id',
-          shuffle: true,
-          packed: true,
-          train_on_input: true,
-          validation_dataset_id: 'validation_dataset_id',
-        },
-        dtype: 'dtype',
-        efficiency_config: {
-          enable_activation_checkpointing: true,
-          enable_activation_offloading: true,
-          fsdp_cpu_offload: true,
-          memory_efficient_fsdp_wrap: true,
-        },
-        gradient_accumulation_steps: 0,
-        max_steps_per_epoch: 0,
-        max_validation_steps: 0,
-        optimizer_config: {
-          lr: 0,
-          num_warmup_steps: 0,
-          optimizer_type: 'adam',
-          weight_decay: 0,
-        },
-      },
-    });
-  });
-
-  test('supervisedFineTune: only required params', async () => {
-    const responsePromise = client.alpha.postTraining.supervisedFineTune({
-      hyperparam_search_config: { foo: 'bar' },
-      job_uuid: 'job_uuid',
-      logger_config: { foo: 'bar' },
-      training_config: { n_epochs: 0 },
-    });
-    const rawResponse = await responsePromise.asResponse();
-    expect(rawResponse).toBeInstanceOf(Response);
-    const response = await responsePromise;
-    expect(response).not.toBeInstanceOf(Response);
-    const dataAndResponse = await responsePromise.withResponse();
-    expect(dataAndResponse.data).toBe(response);
-    expect(dataAndResponse.response).toBe(rawResponse);
-  });
-
-  test('supervisedFineTune: required and optional params', async () => {
-    const response = await client.alpha.postTraining.supervisedFineTune({
-      hyperparam_search_config: { foo: 'bar' },
-      job_uuid: 'job_uuid',
-      logger_config: { foo: 'bar' },
-      training_config: {
-        n_epochs: 0,
-        data_config: {
-          batch_size: 0,
-          data_format: 'instruct',
-          dataset_id: 'dataset_id',
-          shuffle: true,
-          packed: true,
-          train_on_input: true,
-          validation_dataset_id: 'validation_dataset_id',
-        },
-        dtype: 'dtype',
-        efficiency_config: {
-          enable_activation_checkpointing: true,
-          enable_activation_offloading: true,
-          fsdp_cpu_offload: true,
-          memory_efficient_fsdp_wrap: true,
-        },
-        gradient_accumulation_steps: 0,
-        max_steps_per_epoch: 0,
-        max_validation_steps: 0,
-        optimizer_config: {
-          lr: 0,
-          num_warmup_steps: 0,
-          optimizer_type: 'adam',
-          weight_decay: 0,
-        },
-      },
-      algorithm_config: {
-        alpha: 0,
-        apply_lora_to_mlp: true,
-        apply_lora_to_output: true,
-        lora_attn_modules: ['string'],
-        rank: 0,
-        quantize_base: true,
-        type: 'LoRA',
-        use_dora: true,
-      },
-      checkpoint_dir: 'checkpoint_dir',
-      model: 'model',
-    });
-  });
-});

package/tests/api-resources/tool-runtime.test.ts

@@ -1,68 +0,0 @@
-// Copyright (c) Meta Platforms, Inc. and affiliates.
-// All rights reserved.
-//
-// 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 LlamaStackClient from 'llama-stack-client';
-import { Response } from 'node-fetch';
-
-const client = new LlamaStackClient({ baseURL: process.env['TEST_API_BASE_URL'] ?? 'http://127.0.0.1:4010' });
-
-describe('resource toolRuntime', () => {
-  test('invokeTool: only required params', async () => {
-    const responsePromise = client.toolRuntime.invokeTool({
-      kwargs: { foo: 'bar' },
-      tool_name: 'tool_name',
-    });
-    const rawResponse = await responsePromise.asResponse();
-    expect(rawResponse).toBeInstanceOf(Response);
-    const response = await responsePromise;
-    expect(response).not.toBeInstanceOf(Response);
-    const dataAndResponse = await responsePromise.withResponse();
-    expect(dataAndResponse.data).toBe(response);
-    expect(dataAndResponse.response).toBe(rawResponse);
-  });
-
-  test('invokeTool: required and optional params', async () => {
-    const response = await client.toolRuntime.invokeTool({
-      kwargs: { foo: 'bar' },
-      tool_name: 'tool_name',
-      authorization: 'authorization',
-    });
-  });
-
-  test.skip('listTools', async () => {
-    const responsePromise = client.toolRuntime.listTools();
-    const rawResponse = await responsePromise.asResponse();
-    expect(rawResponse).toBeInstanceOf(Response);
-    const response = await responsePromise;
-    expect(response).not.toBeInstanceOf(Response);
-    const dataAndResponse = await responsePromise.withResponse();
-    expect(dataAndResponse.data).toBe(response);
-    expect(dataAndResponse.response).toBe(rawResponse);
-  });
-
-  test('listTools: request options instead of params are passed correctly', async () => {
-    // ensure the request options are being passed correctly by passing an invalid HTTP method in order to cause an error
-    await expect(client.toolRuntime.listTools({ path: '/_stainless_unknown_path' })).rejects.toThrow(
-      LlamaStackClient.NotFoundError,
-    );
-  });
-
-  test('listTools: request options and params are passed correctly', async () => {
-    // ensure the request options are being passed correctly by passing an invalid HTTP method in order to cause an error
-    await expect(
-      client.toolRuntime.listTools(
-        {
-          authorization: 'authorization',
-          mcp_endpoint: { uri: 'uri' },
-          tool_group_id: 'tool_group_id',
-        },
-        { path: '/_stainless_unknown_path' },
-      ),
-    ).rejects.toThrow(LlamaStackClient.NotFoundError);
-  });
-});

package/tests/api-resources/toolgroups.test.ts

@@ -1,91 +0,0 @@
-// Copyright (c) Meta Platforms, Inc. and affiliates.
-// All rights reserved.
-//
-// 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 LlamaStackClient from 'llama-stack-client';
-import { Response } from 'node-fetch';
-
-const client = new LlamaStackClient({ baseURL: process.env['TEST_API_BASE_URL'] ?? 'http://127.0.0.1:4010' });
-
-describe('resource toolgroups', () => {
-  test('list', async () => {
-    const responsePromise = client.toolgroups.list();
-    const rawResponse = await responsePromise.asResponse();
-    expect(rawResponse).toBeInstanceOf(Response);
-    const response = await responsePromise;
-    expect(response).not.toBeInstanceOf(Response);
-    const dataAndResponse = await responsePromise.withResponse();
-    expect(dataAndResponse.data).toBe(response);
-    expect(dataAndResponse.response).toBe(rawResponse);
-  });
-
-  test('list: request options instead of params are passed correctly', async () => {
-    // ensure the request options are being passed correctly by passing an invalid HTTP method in order to cause an error
-    await expect(client.toolgroups.list({ path: '/_stainless_unknown_path' })).rejects.toThrow(
-      LlamaStackClient.NotFoundError,
-    );
-  });
-
-  test('get', async () => {
-    const responsePromise = client.toolgroups.get('toolgroup_id');
-    const rawResponse = await responsePromise.asResponse();
-    expect(rawResponse).toBeInstanceOf(Response);
-    const response = await responsePromise;
-    expect(response).not.toBeInstanceOf(Response);
-    const dataAndResponse = await responsePromise.withResponse();
-    expect(dataAndResponse.data).toBe(response);
-    expect(dataAndResponse.response).toBe(rawResponse);
-  });
-
-  test('get: request options instead of params are passed correctly', async () => {
-    // ensure the request options are being passed correctly by passing an invalid HTTP method in order to cause an error
-    await expect(client.toolgroups.get('toolgroup_id', { path: '/_stainless_unknown_path' })).rejects.toThrow(
-      LlamaStackClient.NotFoundError,
-    );
-  });
-
-  test('register: only required params', async () => {
-    const responsePromise = client.toolgroups.register({
-      provider_id: 'provider_id',
-      toolgroup_id: 'toolgroup_id',
-    });
-    const rawResponse = await responsePromise.asResponse();
-    expect(rawResponse).toBeInstanceOf(Response);
-    const response = await responsePromise;
-    expect(response).not.toBeInstanceOf(Response);
-    const dataAndResponse = await responsePromise.withResponse();
-    expect(dataAndResponse.data).toBe(response);
-    expect(dataAndResponse.response).toBe(rawResponse);
-  });
-
-  test('register: required and optional params', async () => {
-    const response = await client.toolgroups.register({
-      provider_id: 'provider_id',
-      toolgroup_id: 'toolgroup_id',
-      args: { foo: 'bar' },
-      mcp_endpoint: { uri: 'uri' },
-    });
-  });
-
-  test('unregister', async () => {
-    const responsePromise = client.toolgroups.unregister('toolgroup_id');
-    const rawResponse = await responsePromise.asResponse();
-    expect(rawResponse).toBeInstanceOf(Response);
-    const response = await responsePromise;
-    expect(response).not.toBeInstanceOf(Response);
-    const dataAndResponse = await responsePromise.withResponse();
-    expect(dataAndResponse.data).toBe(response);
-    expect(dataAndResponse.response).toBe(rawResponse);
-  });
-
-  test('unregister: request options instead of params are passed correctly', async () => {
-    // ensure the request options are being passed correctly by passing an invalid HTTP method in order to cause an error
-    await expect(
-      client.toolgroups.unregister('toolgroup_id', { path: '/_stainless_unknown_path' }),
-    ).rejects.toThrow(LlamaStackClient.NotFoundError);
-  });
-});

package/tests/api-resources/tools.test.ts

@@ -1,57 +0,0 @@
-// Copyright (c) Meta Platforms, Inc. and affiliates.
-// All rights reserved.
-//
-// 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 LlamaStackClient from 'llama-stack-client';
-import { Response } from 'node-fetch';
-
-const client = new LlamaStackClient({ baseURL: process.env['TEST_API_BASE_URL'] ?? 'http://127.0.0.1:4010' });
-
-describe('resource tools', () => {
-  test('list', async () => {
-    const responsePromise = client.tools.list();
-    const rawResponse = await responsePromise.asResponse();
-    expect(rawResponse).toBeInstanceOf(Response);
-    const response = await responsePromise;
-    expect(response).not.toBeInstanceOf(Response);
-    const dataAndResponse = await responsePromise.withResponse();
-    expect(dataAndResponse.data).toBe(response);
-    expect(dataAndResponse.response).toBe(rawResponse);
-  });
-
-  test('list: request options instead of params are passed correctly', async () => {
-    // ensure the request options are being passed correctly by passing an invalid HTTP method in order to cause an error
-    await expect(client.tools.list({ path: '/_stainless_unknown_path' })).rejects.toThrow(
-      LlamaStackClient.NotFoundError,
-    );
-  });
-
-  test('list: request options and params are passed correctly', async () => {
-    // ensure the request options are being passed correctly by passing an invalid HTTP method in order to cause an error
-    await expect(
-      client.tools.list({ toolgroup_id: 'toolgroup_id' }, { path: '/_stainless_unknown_path' }),
-    ).rejects.toThrow(LlamaStackClient.NotFoundError);
-  });
-
-  test('get', async () => {
-    const responsePromise = client.tools.get('tool_name');
-    const rawResponse = await responsePromise.asResponse();
-    expect(rawResponse).toBeInstanceOf(Response);
-    const response = await responsePromise;
-    expect(response).not.toBeInstanceOf(Response);
-    const dataAndResponse = await responsePromise.withResponse();
-    expect(dataAndResponse.data).toBe(response);
-    expect(dataAndResponse.response).toBe(rawResponse);
-  });
-
-  test('get: request options instead of params are passed correctly', async () => {
-    // ensure the request options are being passed correctly by passing an invalid HTTP method in order to cause an error
-    await expect(client.tools.get('tool_name', { path: '/_stainless_unknown_path' })).rejects.toThrow(
-      LlamaStackClient.NotFoundError,
-    );
-  });
-});