llama-stack-client 0.0.1-alpha.0

package/src/resources/shields.ts

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

package/src/resources/synthetic-data-generation.ts

@@ -0,0 +1,62 @@
+// File generated from our OpenAPI spec by Stainless. See CONTRIBUTING.md for details.
+
+import { APIResource } from '../resource';
+import * as Core from '../core';
+import * as SyntheticDataGenerationAPI from './synthetic-data-generation';
+import * as RewardScoringAPI from './reward-scoring';
+import * as Shared from './shared';
+
+export class SyntheticDataGenerationResource extends APIResource {
+  generate(
+    params: SyntheticDataGenerationGenerateParams,
+    options?: Core.RequestOptions,
+  ): Core.APIPromise<SyntheticDataGeneration> {
+    const { 'X-LlamaStack-ProviderData': xLlamaStackProviderData, ...body } = params;
+    return this._client.post('/synthetic_data_generation/generate', {
+      body,
+      ...options,
+      headers: {
+        ...(xLlamaStackProviderData != null ?
+          { 'X-LlamaStack-ProviderData': xLlamaStackProviderData }
+        : undefined),
+        ...options?.headers,
+      },
+    });
+  }
+}
+
+export interface SyntheticDataGeneration {
+  synthetic_data: Array<RewardScoringAPI.ScoredDialogGenerations>;
+
+  statistics?: Record<string, boolean | number | string | Array<unknown> | unknown | null>;
+}
+
+export interface SyntheticDataGenerationGenerateParams {
+  /**
+   * Body param:
+   */
+  dialogs: Array<
+    Shared.UserMessage | Shared.SystemMessage | Shared.ToolResponseMessage | Shared.CompletionMessage
+  >;
+
+  /**
+   * Body param:
+   */
+  filtering_function: 'none' | 'random' | 'top_k' | 'top_p' | 'top_k_top_p' | 'sigmoid';
+
+  /**
+   * 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 SyntheticDataGenerationResource {
+  export import SyntheticDataGeneration = SyntheticDataGenerationAPI.SyntheticDataGeneration;
+  export import SyntheticDataGenerationGenerateParams = SyntheticDataGenerationAPI.SyntheticDataGenerationGenerateParams;
+}

package/src/resources/telemetry.ts

@@ -0,0 +1,150 @@
+// File generated from our OpenAPI spec by Stainless. See CONTRIBUTING.md for details.
+
+import { APIResource } from '../resource';
+import * as Core from '../core';
+import * as TelemetryAPI from './telemetry';
+
+export class Telemetry extends APIResource {
+  getTrace(
+    params: TelemetryGetTraceParams,
+    options?: Core.RequestOptions,
+  ): Core.APIPromise<TelemetryGetTraceResponse> {
+    const { 'X-LlamaStack-ProviderData': xLlamaStackProviderData, ...query } = params;
+    return this._client.get('/telemetry/get_trace', {
+      query,
+      ...options,
+      headers: {
+        ...(xLlamaStackProviderData != null ?
+          { 'X-LlamaStack-ProviderData': xLlamaStackProviderData }
+        : undefined),
+        ...options?.headers,
+      },
+    });
+  }
+
+  log(params: TelemetryLogParams, options?: Core.RequestOptions): Core.APIPromise<void> {
+    const { 'X-LlamaStack-ProviderData': xLlamaStackProviderData, ...body } = params;
+    return this._client.post('/telemetry/log_event', {
+      body,
+      ...options,
+      headers: {
+        Accept: '*/*',
+        ...(xLlamaStackProviderData != null ?
+          { 'X-LlamaStack-ProviderData': xLlamaStackProviderData }
+        : undefined),
+        ...options?.headers,
+      },
+    });
+  }
+}
+
+export interface TelemetryGetTraceResponse {
+  root_span_id: string;
+
+  start_time: string;
+
+  trace_id: string;
+
+  end_time?: string;
+}
+
+export interface TelemetryGetTraceParams {
+  /**
+   * Query param:
+   */
+  trace_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 TelemetryLogParams {
+  /**
+   * Body param:
+   */
+  event:
+    | TelemetryLogParams.UnstructuredLogEvent
+    | TelemetryLogParams.MetricEvent
+    | TelemetryLogParams.StructuredLogEvent;
+
+  /**
+   * Header param: JSON-encoded provider data which will be made available to the
+   * adapter servicing the API
+   */
+  'X-LlamaStack-ProviderData'?: string;
+}
+
+export namespace TelemetryLogParams {
+  export interface UnstructuredLogEvent {
+    message: string;
+
+    severity: 'verbose' | 'debug' | 'info' | 'warn' | 'error' | 'critical';
+
+    span_id: string;
+
+    timestamp: string;
+
+    trace_id: string;
+
+    type: 'unstructured_log';
+
+    attributes?: Record<string, boolean | number | string | Array<unknown> | unknown | null>;
+  }
+
+  export interface MetricEvent {
+    metric: string;
+
+    span_id: string;
+
+    timestamp: string;
+
+    trace_id: string;
+
+    type: 'metric';
+
+    unit: string;
+
+    value: number;
+
+    attributes?: Record<string, boolean | number | string | Array<unknown> | unknown | null>;
+  }
+
+  export interface StructuredLogEvent {
+    payload: StructuredLogEvent.SpanStartPayload | StructuredLogEvent.SpanEndPayload;
+
+    span_id: string;
+
+    timestamp: string;
+
+    trace_id: string;
+
+    type: 'structured_log';
+
+    attributes?: Record<string, boolean | number | string | Array<unknown> | unknown | null>;
+  }
+
+  export namespace StructuredLogEvent {
+    export interface SpanStartPayload {
+      name: string;
+
+      type: 'span_start';
+
+      parent_span_id?: string;
+    }
+
+    export interface SpanEndPayload {
+      status: 'ok' | 'error';
+
+      type: 'span_end';
+    }
+  }
+}
+
+export namespace Telemetry {
+  export import TelemetryGetTraceResponse = TelemetryAPI.TelemetryGetTraceResponse;
+  export import TelemetryGetTraceParams = TelemetryAPI.TelemetryGetTraceParams;
+  export import TelemetryLogParams = TelemetryAPI.TelemetryLogParams;
+}

package/src/shims/node.ts

@@ -0,0 +1,50 @@
+// @ts-ignore
+import * as types from '../_shims/node-types';
+import { setShims } from '../_shims/registry';
+import { getRuntime } from '../_shims/node-runtime';
+setShims(getRuntime());
+
+declare module '../_shims/manual-types' {
+  export namespace manual {
+    // @ts-ignore
+    export type Agent = types.Agent;
+    // @ts-ignore
+    export import fetch = types.fetch;
+    // @ts-ignore
+    export type Request = types.Request;
+    // @ts-ignore
+    export type RequestInfo = types.RequestInfo;
+    // @ts-ignore
+    export type RequestInit = types.RequestInit;
+    // @ts-ignore
+    export type Response = types.Response;
+    // @ts-ignore
+    export type ResponseInit = types.ResponseInit;
+    // @ts-ignore
+    export type ResponseType = types.ResponseType;
+    // @ts-ignore
+    export type BodyInit = types.BodyInit;
+    // @ts-ignore
+    export type Headers = types.Headers;
+    // @ts-ignore
+    export type HeadersInit = types.HeadersInit;
+    // @ts-ignore
+    export type BlobPropertyBag = types.BlobPropertyBag;
+    // @ts-ignore
+    export type FilePropertyBag = types.FilePropertyBag;
+    // @ts-ignore
+    export type FileFromPathOptions = types.FileFromPathOptions;
+    // @ts-ignore
+    export import FormData = types.FormData;
+    // @ts-ignore
+    export import File = types.File;
+    // @ts-ignore
+    export import Blob = types.Blob;
+    // @ts-ignore
+    export type Readable = types.Readable;
+    // @ts-ignore
+    export type FsReadStream = types.FsReadStream;
+    // @ts-ignore
+    export import ReadableStream = types.ReadableStream;
+  }
+}

package/src/shims/web.ts

@@ -0,0 +1,50 @@
+// @ts-ignore
+import * as types from '../_shims/web-types';
+import { setShims } from '../_shims/registry';
+import { getRuntime } from '../_shims/web-runtime';
+setShims(getRuntime({ manuallyImported: true }));
+
+declare module '../_shims/manual-types' {
+  export namespace manual {
+    // @ts-ignore
+    export type Agent = types.Agent;
+    // @ts-ignore
+    export import fetch = types.fetch;
+    // @ts-ignore
+    export type Request = types.Request;
+    // @ts-ignore
+    export type RequestInfo = types.RequestInfo;
+    // @ts-ignore
+    export type RequestInit = types.RequestInit;
+    // @ts-ignore
+    export type Response = types.Response;
+    // @ts-ignore
+    export type ResponseInit = types.ResponseInit;
+    // @ts-ignore
+    export type ResponseType = types.ResponseType;
+    // @ts-ignore
+    export type BodyInit = types.BodyInit;
+    // @ts-ignore
+    export type Headers = types.Headers;
+    // @ts-ignore
+    export type HeadersInit = types.HeadersInit;
+    // @ts-ignore
+    export type BlobPropertyBag = types.BlobPropertyBag;
+    // @ts-ignore
+    export type FilePropertyBag = types.FilePropertyBag;
+    // @ts-ignore
+    export type FileFromPathOptions = types.FileFromPathOptions;
+    // @ts-ignore
+    export import FormData = types.FormData;
+    // @ts-ignore
+    export import File = types.File;
+    // @ts-ignore
+    export import Blob = types.Blob;
+    // @ts-ignore
+    export type Readable = types.Readable;
+    // @ts-ignore
+    export type FsReadStream = types.FsReadStream;
+    // @ts-ignore
+    export import ReadableStream = types.ReadableStream;
+  }
+}