llama-stack-client 0.0.35 → 0.1.0

package/src/resources/memory/memory.ts

@@ -1,297 +0,0 @@
-// 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 MemoryAPI from './memory';
-import * as DocumentsAPI from './documents';
-
-export class Memory extends APIResource {
-  documents: DocumentsAPI.Documents = new DocumentsAPI.Documents(this._client);
-
-  create(params: MemoryCreateParams, options?: Core.RequestOptions): Core.APIPromise<unknown> {
-    const { body, 'X-LlamaStack-ProviderData': xLlamaStackProviderData } = params;
-    return this._client.post('/memory/create', {
-      body: body,
-      ...options,
-      headers: {
-        ...(xLlamaStackProviderData != null ?
-          { 'X-LlamaStack-ProviderData': xLlamaStackProviderData }
-        : undefined),
-        ...options?.headers,
-      },
-    });
-  }
-
-  retrieve(params: MemoryRetrieveParams, options?: Core.RequestOptions): Core.APIPromise<unknown> {
-    const { 'X-LlamaStack-ProviderData': xLlamaStackProviderData, ...query } = params;
-    return this._client.get('/memory/get', {
-      query,
-      ...options,
-      headers: {
-        ...(xLlamaStackProviderData != null ?
-          { 'X-LlamaStack-ProviderData': xLlamaStackProviderData }
-        : undefined),
-        ...options?.headers,
-      },
-    });
-  }
-
-  update(params: MemoryUpdateParams, options?: Core.RequestOptions): Core.APIPromise<void> {
-    const { 'X-LlamaStack-ProviderData': xLlamaStackProviderData, ...body } = params;
-    return this._client.post('/memory/update', {
-      body,
-      ...options,
-      headers: {
-        Accept: '*/*',
-        ...(xLlamaStackProviderData != null ?
-          { 'X-LlamaStack-ProviderData': xLlamaStackProviderData }
-        : undefined),
-        ...options?.headers,
-      },
-    });
-  }
-
-  list(params?: MemoryListParams, options?: Core.RequestOptions): Core.APIPromise<unknown>;
-  list(options?: Core.RequestOptions): Core.APIPromise<unknown>;
-  list(
-    params: MemoryListParams | Core.RequestOptions = {},
-    options?: Core.RequestOptions,
-  ): Core.APIPromise<unknown> {
-    if (isRequestOptions(params)) {
-      return this.list({}, params);
-    }
-    const { 'X-LlamaStack-ProviderData': xLlamaStackProviderData } = params;
-    return this._client.get('/memory/list', {
-      ...options,
-      headers: {
-        ...(xLlamaStackProviderData != null ?
-          { 'X-LlamaStack-ProviderData': xLlamaStackProviderData }
-        : undefined),
-        ...options?.headers,
-      },
-    });
-  }
-
-  drop(params: MemoryDropParams, options?: Core.RequestOptions): Core.APIPromise<string> {
-    const { 'X-LlamaStack-ProviderData': xLlamaStackProviderData, ...body } = params;
-    return this._client.post('/memory/drop', {
-      body,
-      ...options,
-      headers: {
-        Accept: 'application/json',
-        ...(xLlamaStackProviderData != null ?
-          { 'X-LlamaStack-ProviderData': xLlamaStackProviderData }
-        : undefined),
-        ...options?.headers,
-      },
-    });
-  }
-
-  insert(params: MemoryInsertParams, options?: Core.RequestOptions): Core.APIPromise<void> {
-    const { 'X-LlamaStack-ProviderData': xLlamaStackProviderData, ...body } = params;
-    return this._client.post('/memory/insert', {
-      body,
-      ...options,
-      headers: {
-        Accept: '*/*',
-        ...(xLlamaStackProviderData != null ?
-          { 'X-LlamaStack-ProviderData': xLlamaStackProviderData }
-        : undefined),
-        ...options?.headers,
-      },
-    });
-  }
-
-  query(params: MemoryQueryParams, options?: Core.RequestOptions): Core.APIPromise<QueryDocuments> {
-    const { 'X-LlamaStack-ProviderData': xLlamaStackProviderData, ...body } = params;
-    return this._client.post('/memory/query', {
-      body,
-      ...options,
-      headers: {
-        ...(xLlamaStackProviderData != null ?
-          { 'X-LlamaStack-ProviderData': xLlamaStackProviderData }
-        : undefined),
-        ...options?.headers,
-      },
-    });
-  }
-}
-
-export interface QueryDocuments {
-  chunks: Array<QueryDocuments.Chunk>;
-
-  scores: Array<number>;
-}
-
-export namespace QueryDocuments {
-  export interface Chunk {
-    content: string | Array<string>;
-
-    document_id: string;
-
-    token_count: number;
-  }
-}
-
-export type MemoryCreateResponse = unknown;
-
-export type MemoryRetrieveResponse = unknown;
-
-export type MemoryListResponse = unknown;
-
-export type MemoryDropResponse = string;
-
-export interface MemoryCreateParams {
-  /**
-   * Body param:
-   */
-  body: unknown;
-
-  /**
-   * Header param: JSON-encoded provider data which will be made available to the
-   * adapter servicing the API
-   */
-  'X-LlamaStack-ProviderData'?: string;
-}
-
-export interface MemoryRetrieveParams {
-  /**
-   * Query param:
-   */
-  bank_id: string;
-
-  /**
-   * Header param: JSON-encoded provider data which will be made available to the
-   * adapter servicing the API
-   */
-  'X-LlamaStack-ProviderData'?: string;
-}
-
-export interface MemoryUpdateParams {
-  /**
-   * Body param:
-   */
-  bank_id: string;
-
-  /**
-   * Body param:
-   */
-  documents: Array<MemoryUpdateParams.Document>;
-
-  /**
-   * Header param: JSON-encoded provider data which will be made available to the
-   * adapter servicing the API
-   */
-  'X-LlamaStack-ProviderData'?: string;
-}
-
-export namespace MemoryUpdateParams {
-  export interface Document {
-    content: string | Array<string>;
-
-    document_id: string;
-
-    metadata: Record<string, boolean | number | string | Array<unknown> | unknown | null>;
-
-    mime_type?: string;
-  }
-}
-
-export interface MemoryListParams {
-  /**
-   * JSON-encoded provider data which will be made available to the adapter servicing
-   * the API
-   */
-  'X-LlamaStack-ProviderData'?: string;
-}
-
-export interface MemoryDropParams {
-  /**
-   * Body param:
-   */
-  bank_id: string;
-
-  /**
-   * Header param: JSON-encoded provider data which will be made available to the
-   * adapter servicing the API
-   */
-  'X-LlamaStack-ProviderData'?: string;
-}
-
-export interface MemoryInsertParams {
-  /**
-   * Body param:
-   */
-  bank_id: string;
-
-  /**
-   * Body param:
-   */
-  documents: Array<MemoryInsertParams.Document>;
-
-  /**
-   * Body param:
-   */
-  ttl_seconds?: number;
-
-  /**
-   * Header param: JSON-encoded provider data which will be made available to the
-   * adapter servicing the API
-   */
-  'X-LlamaStack-ProviderData'?: string;
-}
-
-export namespace MemoryInsertParams {
-  export interface Document {
-    content: string | Array<string>;
-
-    document_id: string;
-
-    metadata: Record<string, boolean | number | string | Array<unknown> | unknown | null>;
-
-    mime_type?: string;
-  }
-}
-
-export interface MemoryQueryParams {
-  /**
-   * Body param:
-   */
-  bank_id: string;
-
-  /**
-   * Body param:
-   */
-  query: string | Array<string>;
-
-  /**
-   * Body param:
-   */
-  params?: Record<string, boolean | number | string | Array<unknown> | unknown | null>;
-
-  /**
-   * Header param: JSON-encoded provider data which will be made available to the
-   * adapter servicing the API
-   */
-  'X-LlamaStack-ProviderData'?: string;
-}
-
-export namespace Memory {
-  export import QueryDocuments = MemoryAPI.QueryDocuments;
-  export import MemoryCreateResponse = MemoryAPI.MemoryCreateResponse;
-  export import MemoryRetrieveResponse = MemoryAPI.MemoryRetrieveResponse;
-  export import MemoryListResponse = MemoryAPI.MemoryListResponse;
-  export import MemoryDropResponse = MemoryAPI.MemoryDropResponse;
-  export import MemoryCreateParams = MemoryAPI.MemoryCreateParams;
-  export import MemoryRetrieveParams = MemoryAPI.MemoryRetrieveParams;
-  export import MemoryUpdateParams = MemoryAPI.MemoryUpdateParams;
-  export import MemoryListParams = MemoryAPI.MemoryListParams;
-  export import MemoryDropParams = MemoryAPI.MemoryDropParams;
-  export import MemoryInsertParams = MemoryAPI.MemoryInsertParams;
-  export import MemoryQueryParams = MemoryAPI.MemoryQueryParams;
-  export import Documents = DocumentsAPI.Documents;
-  export import DocumentRetrieveResponse = DocumentsAPI.DocumentRetrieveResponse;
-  export import DocumentRetrieveParams = DocumentsAPI.DocumentRetrieveParams;
-  export import DocumentDeleteParams = DocumentsAPI.DocumentDeleteParams;
-}

package/src/resources/memory-banks.ts

@@ -1,84 +0,0 @@
-// 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 MemoryBanksAPI from './memory-banks';
-
-export class MemoryBanks extends APIResource {
-  list(params?: MemoryBankListParams, options?: Core.RequestOptions): Core.APIPromise<MemoryBankSpec>;
-  list(options?: Core.RequestOptions): Core.APIPromise<MemoryBankSpec>;
-  list(
-    params: MemoryBankListParams | Core.RequestOptions = {},
-    options?: Core.RequestOptions,
-  ): Core.APIPromise<MemoryBankSpec> {
-    if (isRequestOptions(params)) {
-      return this.list({}, params);
-    }
-    const { 'X-LlamaStack-ProviderData': xLlamaStackProviderData } = params;
-    return this._client.get('/memory_banks/list', {
-      ...options,
-      headers: {
-        ...(xLlamaStackProviderData != null ?
-          { 'X-LlamaStack-ProviderData': xLlamaStackProviderData }
-        : undefined),
-        ...options?.headers,
-      },
-    });
-  }
-
-  get(params: MemoryBankGetParams, options?: Core.RequestOptions): Core.APIPromise<MemoryBankSpec | null> {
-    const { 'X-LlamaStack-ProviderData': xLlamaStackProviderData, ...query } = params;
-    return this._client.get('/memory_banks/get', {
-      query,
-      ...options,
-      headers: {
-        ...(xLlamaStackProviderData != null ?
-          { 'X-LlamaStack-ProviderData': xLlamaStackProviderData }
-        : undefined),
-        ...options?.headers,
-      },
-    });
-  }
-}
-
-export interface MemoryBankSpec {
-  bank_type: 'vector' | 'keyvalue' | 'keyword' | 'graph';
-
-  provider_config: MemoryBankSpec.ProviderConfig;
-}
-
-export namespace MemoryBankSpec {
-  export interface ProviderConfig {
-    config: Record<string, boolean | number | string | Array<unknown> | unknown | null>;
-
-    provider_id: string;
-  }
-}
-
-export interface MemoryBankListParams {
-  /**
-   * JSON-encoded provider data which will be made available to the adapter servicing
-   * the API
-   */
-  'X-LlamaStack-ProviderData'?: string;
-}
-
-export interface MemoryBankGetParams {
-  /**
-   * Query param:
-   */
-  bank_type: 'vector' | 'keyvalue' | 'keyword' | 'graph';
-
-  /**
-   * Header param: JSON-encoded provider data which will be made available to the
-   * adapter servicing the API
-   */
-  'X-LlamaStack-ProviderData'?: string;
-}
-
-export namespace MemoryBanks {
-  export import MemoryBankSpec = MemoryBanksAPI.MemoryBankSpec;
-  export import MemoryBankListParams = MemoryBanksAPI.MemoryBankListParams;
-  export import MemoryBankGetParams = MemoryBanksAPI.MemoryBankGetParams;
-}

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

@@ -1,192 +0,0 @@
-// 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 JobsAPI from './jobs';
-import * as PostTrainingAPI from './post-training';
-
-export class Jobs extends APIResource {
-  list(
-    params?: JobListParams,
-    options?: Core.RequestOptions,
-  ): Core.APIPromise<PostTrainingAPI.PostTrainingJob>;
-  list(options?: Core.RequestOptions): Core.APIPromise<PostTrainingAPI.PostTrainingJob>;
-  list(
-    params: JobListParams | Core.RequestOptions = {},
-    options?: Core.RequestOptions,
-  ): Core.APIPromise<PostTrainingAPI.PostTrainingJob> {
-    if (isRequestOptions(params)) {
-      return this.list({}, params);
-    }
-    const { 'X-LlamaStack-ProviderData': xLlamaStackProviderData } = params;
-    return this._client.get('/post_training/jobs', {
-      ...options,
-      headers: {
-        ...(xLlamaStackProviderData != null ?
-          { 'X-LlamaStack-ProviderData': xLlamaStackProviderData }
-        : undefined),
-        ...options?.headers,
-      },
-    });
-  }
-
-  artifacts(
-    params: JobArtifactsParams,
-    options?: Core.RequestOptions,
-  ): Core.APIPromise<PostTrainingJobArtifacts> {
-    const { 'X-LlamaStack-ProviderData': xLlamaStackProviderData, ...query } = params;
-    return this._client.get('/post_training/job/artifacts', {
-      query,
-      ...options,
-      headers: {
-        ...(xLlamaStackProviderData != null ?
-          { 'X-LlamaStack-ProviderData': xLlamaStackProviderData }
-        : undefined),
-        ...options?.headers,
-      },
-    });
-  }
-
-  cancel(params: JobCancelParams, options?: Core.RequestOptions): Core.APIPromise<void> {
-    const { 'X-LlamaStack-ProviderData': xLlamaStackProviderData, ...body } = params;
-    return this._client.post('/post_training/job/cancel', {
-      body,
-      ...options,
-      headers: {
-        Accept: '*/*',
-        ...(xLlamaStackProviderData != null ?
-          { 'X-LlamaStack-ProviderData': xLlamaStackProviderData }
-        : undefined),
-        ...options?.headers,
-      },
-    });
-  }
-
-  logs(params: JobLogsParams, options?: Core.RequestOptions): Core.APIPromise<PostTrainingJobLogStream> {
-    const { 'X-LlamaStack-ProviderData': xLlamaStackProviderData, ...query } = params;
-    return this._client.get('/post_training/job/logs', {
-      query,
-      ...options,
-      headers: {
-        ...(xLlamaStackProviderData != null ?
-          { 'X-LlamaStack-ProviderData': xLlamaStackProviderData }
-        : undefined),
-        ...options?.headers,
-      },
-    });
-  }
-
-  status(params: JobStatusParams, options?: Core.RequestOptions): Core.APIPromise<PostTrainingJobStatus> {
-    const { 'X-LlamaStack-ProviderData': xLlamaStackProviderData, ...query } = params;
-    return this._client.get('/post_training/job/status', {
-      query,
-      ...options,
-      headers: {
-        ...(xLlamaStackProviderData != null ?
-          { 'X-LlamaStack-ProviderData': xLlamaStackProviderData }
-        : undefined),
-        ...options?.headers,
-      },
-    });
-  }
-}
-
-export interface PostTrainingJobArtifacts {
-  checkpoints: Array<unknown>;
-
-  job_uuid: string;
-}
-
-export interface PostTrainingJobLogStream {
-  job_uuid: string;
-
-  log_lines: Array<string>;
-}
-
-export interface PostTrainingJobStatus {
-  checkpoints: Array<unknown>;
-
-  job_uuid: string;
-
-  status: 'running' | 'completed' | 'failed' | 'scheduled';
-
-  completed_at?: string;
-
-  resources_allocated?: Record<string, boolean | number | string | Array<unknown> | unknown | null>;
-
-  scheduled_at?: string;
-
-  started_at?: string;
-}
-
-export interface JobListParams {
-  /**
-   * JSON-encoded provider data which will be made available to the adapter servicing
-   * the API
-   */
-  'X-LlamaStack-ProviderData'?: string;
-}
-
-export interface JobArtifactsParams {
-  /**
-   * Query param:
-   */
-  job_uuid: string;
-
-  /**
-   * Header param: JSON-encoded provider data which will be made available to the
-   * adapter servicing the API
-   */
-  'X-LlamaStack-ProviderData'?: string;
-}
-
-export interface JobCancelParams {
-  /**
-   * Body param:
-   */
-  job_uuid: string;
-
-  /**
-   * Header param: JSON-encoded provider data which will be made available to the
-   * adapter servicing the API
-   */
-  'X-LlamaStack-ProviderData'?: string;
-}
-
-export interface JobLogsParams {
-  /**
-   * Query param:
-   */
-  job_uuid: string;
-
-  /**
-   * Header param: JSON-encoded provider data which will be made available to the
-   * adapter servicing the API
-   */
-  'X-LlamaStack-ProviderData'?: string;
-}
-
-export interface JobStatusParams {
-  /**
-   * Query param:
-   */
-  job_uuid: string;
-
-  /**
-   * Header param: JSON-encoded provider data which will be made available to the
-   * adapter servicing the API
-   */
-  'X-LlamaStack-ProviderData'?: string;
-}
-
-export namespace Jobs {
-  export import PostTrainingJobArtifacts = JobsAPI.PostTrainingJobArtifacts;
-  export import PostTrainingJobLogStream = JobsAPI.PostTrainingJobLogStream;
-  export import PostTrainingJobStatus = JobsAPI.PostTrainingJobStatus;
-  export import JobListParams = JobsAPI.JobListParams;
-  export import JobArtifactsParams = JobsAPI.JobArtifactsParams;
-  export import JobCancelParams = JobsAPI.JobCancelParams;
-  export import JobLogsParams = JobsAPI.JobLogsParams;
-  export import JobStatusParams = JobsAPI.JobStatusParams;
-}

package/src/resources/reward-scoring.ts

@@ -1,82 +0,0 @@
-// File generated from our OpenAPI spec by Stainless. See CONTRIBUTING.md for details.
-
-import { APIResource } from '../resource';
-import * as Core from '../core';
-import * as RewardScoringAPI from './reward-scoring';
-import * as Shared from './shared';
-
-export class RewardScoringResource extends APIResource {
-  score(params: RewardScoringScoreParams, options?: Core.RequestOptions): Core.APIPromise<RewardScoring> {
-    const { 'X-LlamaStack-ProviderData': xLlamaStackProviderData, ...body } = params;
-    return this._client.post('/reward_scoring/score', {
-      body,
-      ...options,
-      headers: {
-        ...(xLlamaStackProviderData != null ?
-          { 'X-LlamaStack-ProviderData': xLlamaStackProviderData }
-        : undefined),
-        ...options?.headers,
-      },
-    });
-  }
-}
-
-export interface RewardScoring {
-  scored_generations: Array<ScoredDialogGenerations>;
-}
-
-export interface ScoredDialogGenerations {
-  dialog: Array<
-    Shared.UserMessage | Shared.SystemMessage | Shared.ToolResponseMessage | Shared.CompletionMessage
-  >;
-
-  scored_generations: Array<ScoredDialogGenerations.ScoredGeneration>;
-}
-
-export namespace ScoredDialogGenerations {
-  export interface ScoredGeneration {
-    message:
-      | Shared.UserMessage
-      | Shared.SystemMessage
-      | Shared.ToolResponseMessage
-      | Shared.CompletionMessage;
-
-    score: number;
-  }
-}
-
-export interface RewardScoringScoreParams {
-  /**
-   * Body param:
-   */
-  dialog_generations: Array<RewardScoringScoreParams.DialogGeneration>;
-
-  /**
-   * Body param:
-   */
-  model: string;
-
-  /**
-   * Header param: JSON-encoded provider data which will be made available to the
-   * adapter servicing the API
-   */
-  'X-LlamaStack-ProviderData'?: string;
-}
-
-export namespace RewardScoringScoreParams {
-  export interface DialogGeneration {
-    dialog: Array<
-      Shared.UserMessage | Shared.SystemMessage | Shared.ToolResponseMessage | Shared.CompletionMessage
-    >;
-
-    sampled_generations: Array<
-      Shared.UserMessage | Shared.SystemMessage | Shared.ToolResponseMessage | Shared.CompletionMessage
-    >;
-  }
-}
-
-export namespace RewardScoringResource {
-  export import RewardScoring = RewardScoringAPI.RewardScoring;
-  export import ScoredDialogGenerations = RewardScoringAPI.ScoredDialogGenerations;
-  export import RewardScoringScoreParams = RewardScoringAPI.RewardScoringScoreParams;
-}