llama-stack-client 0.0.1-alpha.0 → 0.1.0

package/resources/synthetic-data-generation.js

@@ -1,24 +1,12 @@
 "use strict";
 // File generated from our OpenAPI spec by Stainless. See CONTRIBUTING.md for details.
 Object.defineProperty(exports, "__esModule", { value: true });
-exports.SyntheticDataGenerationResource = void 0;
+exports.SyntheticDataGeneration = void 0;
 const resource_1 = require("../resource.js");
-class SyntheticDataGenerationResource extends resource_1.APIResource {
-    generate(params, options) {
-        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,
-            },
-        });
+class SyntheticDataGeneration extends resource_1.APIResource {
+    generate(body, options) {
+        return this._client.post('/v1/synthetic-data-generation/generate', { body, ...options });
     }
 }
-exports.SyntheticDataGenerationResource = SyntheticDataGenerationResource;
-(function (SyntheticDataGenerationResource) {
-})(SyntheticDataGenerationResource = exports.SyntheticDataGenerationResource || (exports.SyntheticDataGenerationResource = {}));
+exports.SyntheticDataGeneration = SyntheticDataGeneration;
 //# sourceMappingURL=synthetic-data-generation.js.map

package/resources/synthetic-data-generation.js.map

@@ -1 +1 @@
-{"version":3,"file":"synthetic-data-generation.js","sourceRoot":"","sources":["../src/resources/synthetic-data-generation.ts"],"names":[],"mappings":";AAAA,sFAAsF;;;AAEtF,6CAA0C;AAM1C,MAAa,+BAAgC,SAAQ,sBAAW;IAC9D,QAAQ,CACN,MAA6C,EAC7C,OAA6B;QAE7B,MAAM,EAAE,2BAA2B,EAAE,uBAAuB,EAAE,GAAG,IAAI,EAAE,GAAG,MAAM,CAAC;QACjF,OAAO,IAAI,CAAC,OAAO,CAAC,IAAI,CAAC,qCAAqC,EAAE;YAC9D,IAAI;YACJ,GAAG,OAAO;YACV,OAAO,EAAE;gBACP,GAAG,CAAC,uBAAuB,IAAI,IAAI,CAAC,CAAC;oBACnC,EAAE,2BAA2B,EAAE,uBAAuB,EAAE;oBAC1D,CAAC,CAAC,SAAS,CAAC;gBACZ,GAAG,OAAO,EAAE,OAAO;aACpB;SACF,CAAC,CAAC;IACL,CAAC;CACF;AAjBD,0EAiBC;AAiCD,WAAiB,+BAA+B;AAGhD,CAAC,EAHgB,+BAA+B,GAA/B,uCAA+B,KAA/B,uCAA+B,QAG/C"}
+{"version":3,"file":"synthetic-data-generation.js","sourceRoot":"","sources":["../src/resources/synthetic-data-generation.ts"],"names":[],"mappings":";AAAA,sFAAsF;;;AAEtF,6CAA0C;AAI1C,MAAa,uBAAwB,SAAQ,sBAAW;IACtD,QAAQ,CACN,IAA2C,EAC3C,OAA6B;QAE7B,OAAO,IAAI,CAAC,OAAO,CAAC,IAAI,CAAC,wCAAwC,EAAE,EAAE,IAAI,EAAE,GAAG,OAAO,EAAE,CAAC,CAAC;IAC3F,CAAC;CACF;AAPD,0DAOC"}

package/resources/synthetic-data-generation.mjs

@@ -1,20 +1,8 @@
 // File generated from our OpenAPI spec by Stainless. See CONTRIBUTING.md for details.
 import { APIResource } from "../resource.mjs";
-export class SyntheticDataGenerationResource extends APIResource {
-    generate(params, options) {
-        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 class SyntheticDataGeneration extends APIResource {
+    generate(body, options) {
+        return this._client.post('/v1/synthetic-data-generation/generate', { body, ...options });
     }
 }
-(function (SyntheticDataGenerationResource) {
-})(SyntheticDataGenerationResource || (SyntheticDataGenerationResource = {}));
 //# sourceMappingURL=synthetic-data-generation.mjs.map

package/resources/synthetic-data-generation.mjs.map

@@ -1 +1 @@
-{"version":3,"file":"synthetic-data-generation.mjs","sourceRoot":"","sources":["../src/resources/synthetic-data-generation.ts"],"names":[],"mappings":"AAAA,sFAAsF;OAE/E,EAAE,WAAW,EAAE;AAMtB,MAAM,OAAO,+BAAgC,SAAQ,WAAW;IAC9D,QAAQ,CACN,MAA6C,EAC7C,OAA6B;QAE7B,MAAM,EAAE,2BAA2B,EAAE,uBAAuB,EAAE,GAAG,IAAI,EAAE,GAAG,MAAM,CAAC;QACjF,OAAO,IAAI,CAAC,OAAO,CAAC,IAAI,CAAC,qCAAqC,EAAE;YAC9D,IAAI;YACJ,GAAG,OAAO;YACV,OAAO,EAAE;gBACP,GAAG,CAAC,uBAAuB,IAAI,IAAI,CAAC,CAAC;oBACnC,EAAE,2BAA2B,EAAE,uBAAuB,EAAE;oBAC1D,CAAC,CAAC,SAAS,CAAC;gBACZ,GAAG,OAAO,EAAE,OAAO;aACpB;SACF,CAAC,CAAC;IACL,CAAC;CACF;AAiCD,WAAiB,+BAA+B;AAGhD,CAAC,EAHgB,+BAA+B,KAA/B,+BAA+B,QAG/C"}
+{"version":3,"file":"synthetic-data-generation.mjs","sourceRoot":"","sources":["../src/resources/synthetic-data-generation.ts"],"names":[],"mappings":"AAAA,sFAAsF;OAE/E,EAAE,WAAW,EAAE;AAItB,MAAM,OAAO,uBAAwB,SAAQ,WAAW;IACtD,QAAQ,CACN,IAA2C,EAC3C,OAA6B;QAE7B,OAAO,IAAI,CAAC,OAAO,CAAC,IAAI,CAAC,wCAAwC,EAAE,EAAE,IAAI,EAAE,GAAG,OAAO,EAAE,CAAC,CAAC;IAC3F,CAAC;CACF"}

package/resources/telemetry.d.ts

@@ -1,39 +1,18 @@
 import { APIResource } from "../resource.js";
 import * as Core from "../core.js";
-import * as TelemetryAPI from "./telemetry.js";
 export declare class Telemetry extends APIResource {
-    getTrace(params: TelemetryGetTraceParams, options?: Core.RequestOptions): Core.APIPromise<TelemetryGetTraceResponse>;
-    log(params: TelemetryLogParams, options?: Core.RequestOptions): Core.APIPromise<void>;
+    getSpan(traceId: string, spanId: string, options?: Core.RequestOptions): Core.APIPromise<TelemetryGetSpanResponse>;
+    getSpanTree(spanId: string, query?: TelemetryGetSpanTreeParams, options?: Core.RequestOptions): Core.APIPromise<TelemetryGetSpanTreeResponse>;
+    getSpanTree(spanId: string, options?: Core.RequestOptions): Core.APIPromise<TelemetryGetSpanTreeResponse>;
+    getTrace(traceId: string, options?: Core.RequestOptions): Core.APIPromise<Trace>;
+    logEvent(body: TelemetryLogEventParams, options?: Core.RequestOptions): Core.APIPromise<void>;
+    querySpans(query: TelemetryQuerySpansParams, options?: Core.RequestOptions): Core.APIPromise<TelemetryQuerySpansResponse>;
+    queryTraces(query?: TelemetryQueryTracesParams, options?: Core.RequestOptions): Core.APIPromise<TelemetryQueryTracesResponse>;
+    queryTraces(options?: Core.RequestOptions): Core.APIPromise<TelemetryQueryTracesResponse>;
+    saveSpansToDataset(body: TelemetrySaveSpansToDatasetParams, options?: Core.RequestOptions): Core.APIPromise<void>;
 }
-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 declare namespace TelemetryLogParams {
+export type Event = Event.UnstructuredLogEvent | Event.MetricEvent | Event.StructuredLogEvent;
+export declare namespace Event {
     interface UnstructuredLogEvent {
         message: string;
         severity: 'verbose' | 'debug' | 'info' | 'warn' | 'error' | 'critical';
@@ -73,9 +52,90 @@ export declare namespace TelemetryLogParams {
         }
     }
 }
+export interface QueryCondition {
+    key: string;
+    op: 'eq' | 'ne' | 'gt' | 'lt';
+    value: boolean | number | string | Array<unknown> | unknown | null;
+}
+export interface QuerySpansResponse {
+    data: Array<QuerySpansResponse.Data>;
+}
+export declare namespace QuerySpansResponse {
+    interface Data {
+        name: string;
+        span_id: string;
+        start_time: string;
+        trace_id: string;
+        attributes?: Record<string, boolean | number | string | Array<unknown> | unknown | null>;
+        end_time?: string;
+        parent_span_id?: string;
+    }
+}
+export interface SpanWithStatus {
+    name: string;
+    span_id: string;
+    start_time: string;
+    trace_id: string;
+    attributes?: Record<string, boolean | number | string | Array<unknown> | unknown | null>;
+    end_time?: string;
+    parent_span_id?: string;
+    status?: 'ok' | 'error';
+}
+export interface Trace {
+    root_span_id: string;
+    start_time: string;
+    trace_id: string;
+    end_time?: string;
+}
+export interface TelemetryGetSpanResponse {
+    name: string;
+    span_id: string;
+    start_time: string;
+    trace_id: string;
+    attributes?: Record<string, boolean | number | string | Array<unknown> | unknown | null>;
+    end_time?: string;
+    parent_span_id?: string;
+}
+export type TelemetryGetSpanTreeResponse = Record<string, SpanWithStatus>;
+export type TelemetryQuerySpansResponse = Array<TelemetryQuerySpansResponse.TelemetryQuerySpansResponseItem>;
+export declare namespace TelemetryQuerySpansResponse {
+    interface TelemetryQuerySpansResponseItem {
+        name: string;
+        span_id: string;
+        start_time: string;
+        trace_id: string;
+        attributes?: Record<string, boolean | number | string | Array<unknown> | unknown | null>;
+        end_time?: string;
+        parent_span_id?: string;
+    }
+}
+export type TelemetryQueryTracesResponse = Array<Trace>;
+export interface TelemetryGetSpanTreeParams {
+    attributes_to_return?: Array<string>;
+    max_depth?: number;
+}
+export interface TelemetryLogEventParams {
+    event: Event;
+    ttl_seconds: number;
+}
+export interface TelemetryQuerySpansParams {
+    attribute_filters: Array<QueryCondition>;
+    attributes_to_return: Array<string>;
+    max_depth?: number;
+}
+export interface TelemetryQueryTracesParams {
+    attribute_filters?: Array<QueryCondition>;
+    limit?: number;
+    offset?: number;
+    order_by?: Array<string>;
+}
+export interface TelemetrySaveSpansToDatasetParams {
+    attribute_filters: Array<QueryCondition>;
+    attributes_to_save: Array<string>;
+    dataset_id: string;
+    max_depth?: number;
+}
 export declare namespace Telemetry {
-    export import TelemetryGetTraceResponse = TelemetryAPI.TelemetryGetTraceResponse;
-    export import TelemetryGetTraceParams = TelemetryAPI.TelemetryGetTraceParams;
-    export import TelemetryLogParams = TelemetryAPI.TelemetryLogParams;
+    export { type Event as Event, type QueryCondition as QueryCondition, type QuerySpansResponse as QuerySpansResponse, type SpanWithStatus as SpanWithStatus, type Trace as Trace, type TelemetryGetSpanResponse as TelemetryGetSpanResponse, type TelemetryGetSpanTreeResponse as TelemetryGetSpanTreeResponse, type TelemetryQuerySpansResponse as TelemetryQuerySpansResponse, type TelemetryQueryTracesResponse as TelemetryQueryTracesResponse, type TelemetryGetSpanTreeParams as TelemetryGetSpanTreeParams, type TelemetryLogEventParams as TelemetryLogEventParams, type TelemetryQuerySpansParams as TelemetryQuerySpansParams, type TelemetryQueryTracesParams as TelemetryQueryTracesParams, type TelemetrySaveSpansToDatasetParams as TelemetrySaveSpansToDatasetParams, };
 }
 //# sourceMappingURL=telemetry.d.ts.map

package/resources/telemetry.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"telemetry.d.ts","sourceRoot":"","sources":["../src/resources/telemetry.ts"],"names":[],"mappings":"AAEA,OAAO,EAAE,WAAW,EAAE,MAAM,aAAa,CAAC;AAC1C,OAAO,KAAK,IAAI,MAAM,SAAS,CAAC;AAChC,OAAO,KAAK,YAAY,MAAM,aAAa,CAAC;AAE5C,qBAAa,SAAU,SAAQ,WAAW;IACxC,QAAQ,CACN,MAAM,EAAE,uBAAuB,EAC/B,OAAO,CAAC,EAAE,IAAI,CAAC,cAAc,GAC5B,IAAI,CAAC,UAAU,CAAC,yBAAyB,CAAC;IAc7C,GAAG,CAAC,MAAM,EAAE,kBAAkB,EAAE,OAAO,CAAC,EAAE,IAAI,CAAC,cAAc,GAAG,IAAI,CAAC,UAAU,CAAC,IAAI,CAAC;CActF;AAED,MAAM,WAAW,yBAAyB;IACxC,YAAY,EAAE,MAAM,CAAC;IAErB,UAAU,EAAE,MAAM,CAAC;IAEnB,QAAQ,EAAE,MAAM,CAAC;IAEjB,QAAQ,CAAC,EAAE,MAAM,CAAC;CACnB;AAED,MAAM,WAAW,uBAAuB;IACtC;;OAEG;IACH,QAAQ,EAAE,MAAM,CAAC;IAEjB;;;OAGG;IACH,2BAA2B,CAAC,EAAE,MAAM,CAAC;CACtC;AAED,MAAM,WAAW,kBAAkB;IACjC;;OAEG;IACH,KAAK,EACD,kBAAkB,CAAC,oBAAoB,GACvC,kBAAkB,CAAC,WAAW,GAC9B,kBAAkB,CAAC,kBAAkB,CAAC;IAE1C;;;OAGG;IACH,2BAA2B,CAAC,EAAE,MAAM,CAAC;CACtC;AAED,yBAAiB,kBAAkB,CAAC;IAClC,UAAiB,oBAAoB;QACnC,OAAO,EAAE,MAAM,CAAC;QAEhB,QAAQ,EAAE,SAAS,GAAG,OAAO,GAAG,MAAM,GAAG,MAAM,GAAG,OAAO,GAAG,UAAU,CAAC;QAEvE,OAAO,EAAE,MAAM,CAAC;QAEhB,SAAS,EAAE,MAAM,CAAC;QAElB,QAAQ,EAAE,MAAM,CAAC;QAEjB,IAAI,EAAE,kBAAkB,CAAC;QAEzB,UAAU,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,GAAG,MAAM,GAAG,MAAM,GAAG,KAAK,CAAC,OAAO,CAAC,GAAG,OAAO,GAAG,IAAI,CAAC,CAAC;KAC1F;IAED,UAAiB,WAAW;QAC1B,MAAM,EAAE,MAAM,CAAC;QAEf,OAAO,EAAE,MAAM,CAAC;QAEhB,SAAS,EAAE,MAAM,CAAC;QAElB,QAAQ,EAAE,MAAM,CAAC;QAEjB,IAAI,EAAE,QAAQ,CAAC;QAEf,IAAI,EAAE,MAAM,CAAC;QAEb,KAAK,EAAE,MAAM,CAAC;QAEd,UAAU,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,GAAG,MAAM,GAAG,MAAM,GAAG,KAAK,CAAC,OAAO,CAAC,GAAG,OAAO,GAAG,IAAI,CAAC,CAAC;KAC1F;IAED,UAAiB,kBAAkB;QACjC,OAAO,EAAE,kBAAkB,CAAC,gBAAgB,GAAG,kBAAkB,CAAC,cAAc,CAAC;QAEjF,OAAO,EAAE,MAAM,CAAC;QAEhB,SAAS,EAAE,MAAM,CAAC;QAElB,QAAQ,EAAE,MAAM,CAAC;QAEjB,IAAI,EAAE,gBAAgB,CAAC;QAEvB,UAAU,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,GAAG,MAAM,GAAG,MAAM,GAAG,KAAK,CAAC,OAAO,CAAC,GAAG,OAAO,GAAG,IAAI,CAAC,CAAC;KAC1F;IAED,UAAiB,kBAAkB,CAAC;QAClC,UAAiB,gBAAgB;YAC/B,IAAI,EAAE,MAAM,CAAC;YAEb,IAAI,EAAE,YAAY,CAAC;YAEnB,cAAc,CAAC,EAAE,MAAM,CAAC;SACzB;QAED,UAAiB,cAAc;YAC7B,MAAM,EAAE,IAAI,GAAG,OAAO,CAAC;YAEvB,IAAI,EAAE,UAAU,CAAC;SAClB;KACF;CACF;AAED,yBAAiB,SAAS,CAAC;IACzB,MAAM,QAAQ,yBAAyB,GAAG,YAAY,CAAC,yBAAyB,CAAC;IACjF,MAAM,QAAQ,uBAAuB,GAAG,YAAY,CAAC,uBAAuB,CAAC;IAC7E,MAAM,QAAQ,kBAAkB,GAAG,YAAY,CAAC,kBAAkB,CAAC;CACpE"}
+{"version":3,"file":"telemetry.d.ts","sourceRoot":"","sources":["../src/resources/telemetry.ts"],"names":[],"mappings":"AAEA,OAAO,EAAE,WAAW,EAAE,MAAM,aAAa,CAAC;AAE1C,OAAO,KAAK,IAAI,MAAM,SAAS,CAAC;AAEhC,qBAAa,SAAU,SAAQ,WAAW;IACxC,OAAO,CACL,OAAO,EAAE,MAAM,EACf,MAAM,EAAE,MAAM,EACd,OAAO,CAAC,EAAE,IAAI,CAAC,cAAc,GAC5B,IAAI,CAAC,UAAU,CAAC,wBAAwB,CAAC;IAI5C,WAAW,CACT,MAAM,EAAE,MAAM,EACd,KAAK,CAAC,EAAE,0BAA0B,EAClC,OAAO,CAAC,EAAE,IAAI,CAAC,cAAc,GAC5B,IAAI,CAAC,UAAU,CAAC,4BAA4B,CAAC;IAChD,WAAW,CAAC,MAAM,EAAE,MAAM,EAAE,OAAO,CAAC,EAAE,IAAI,CAAC,cAAc,GAAG,IAAI,CAAC,UAAU,CAAC,4BAA4B,CAAC;IAgBzG,QAAQ,CAAC,OAAO,EAAE,MAAM,EAAE,OAAO,CAAC,EAAE,IAAI,CAAC,cAAc,GAAG,IAAI,CAAC,UAAU,CAAC,KAAK,CAAC;IAIhF,QAAQ,CAAC,IAAI,EAAE,uBAAuB,EAAE,OAAO,CAAC,EAAE,IAAI,CAAC,cAAc,GAAG,IAAI,CAAC,UAAU,CAAC,IAAI,CAAC;IAQ7F,UAAU,CACR,KAAK,EAAE,yBAAyB,EAChC,OAAO,CAAC,EAAE,IAAI,CAAC,cAAc,GAC5B,IAAI,CAAC,UAAU,CAAC,2BAA2B,CAAC;IAQ/C,WAAW,CACT,KAAK,CAAC,EAAE,0BAA0B,EAClC,OAAO,CAAC,EAAE,IAAI,CAAC,cAAc,GAC5B,IAAI,CAAC,UAAU,CAAC,4BAA4B,CAAC;IAChD,WAAW,CAAC,OAAO,CAAC,EAAE,IAAI,CAAC,cAAc,GAAG,IAAI,CAAC,UAAU,CAAC,4BAA4B,CAAC;IAezF,kBAAkB,CAChB,IAAI,EAAE,iCAAiC,EACvC,OAAO,CAAC,EAAE,IAAI,CAAC,cAAc,GAC5B,IAAI,CAAC,UAAU,CAAC,IAAI,CAAC;CAOzB;AAED,MAAM,MAAM,KAAK,GAAG,KAAK,CAAC,oBAAoB,GAAG,KAAK,CAAC,WAAW,GAAG,KAAK,CAAC,kBAAkB,CAAC;AAE9F,yBAAiB,KAAK,CAAC;IACrB,UAAiB,oBAAoB;QACnC,OAAO,EAAE,MAAM,CAAC;QAEhB,QAAQ,EAAE,SAAS,GAAG,OAAO,GAAG,MAAM,GAAG,MAAM,GAAG,OAAO,GAAG,UAAU,CAAC;QAEvE,OAAO,EAAE,MAAM,CAAC;QAEhB,SAAS,EAAE,MAAM,CAAC;QAElB,QAAQ,EAAE,MAAM,CAAC;QAEjB,IAAI,EAAE,kBAAkB,CAAC;QAEzB,UAAU,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,GAAG,MAAM,GAAG,MAAM,GAAG,KAAK,CAAC,OAAO,CAAC,GAAG,OAAO,GAAG,IAAI,CAAC,CAAC;KAC1F;IAED,UAAiB,WAAW;QAC1B,MAAM,EAAE,MAAM,CAAC;QAEf,OAAO,EAAE,MAAM,CAAC;QAEhB,SAAS,EAAE,MAAM,CAAC;QAElB,QAAQ,EAAE,MAAM,CAAC;QAEjB,IAAI,EAAE,QAAQ,CAAC;QAEf,IAAI,EAAE,MAAM,CAAC;QAEb,KAAK,EAAE,MAAM,CAAC;QAEd,UAAU,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,GAAG,MAAM,GAAG,MAAM,GAAG,KAAK,CAAC,OAAO,CAAC,GAAG,OAAO,GAAG,IAAI,CAAC,CAAC;KAC1F;IAED,UAAiB,kBAAkB;QACjC,OAAO,EAAE,kBAAkB,CAAC,gBAAgB,GAAG,kBAAkB,CAAC,cAAc,CAAC;QAEjF,OAAO,EAAE,MAAM,CAAC;QAEhB,SAAS,EAAE,MAAM,CAAC;QAElB,QAAQ,EAAE,MAAM,CAAC;QAEjB,IAAI,EAAE,gBAAgB,CAAC;QAEvB,UAAU,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,GAAG,MAAM,GAAG,MAAM,GAAG,KAAK,CAAC,OAAO,CAAC,GAAG,OAAO,GAAG,IAAI,CAAC,CAAC;KAC1F;IAED,UAAiB,kBAAkB,CAAC;QAClC,UAAiB,gBAAgB;YAC/B,IAAI,EAAE,MAAM,CAAC;YAEb,IAAI,EAAE,YAAY,CAAC;YAEnB,cAAc,CAAC,EAAE,MAAM,CAAC;SACzB;QAED,UAAiB,cAAc;YAC7B,MAAM,EAAE,IAAI,GAAG,OAAO,CAAC;YAEvB,IAAI,EAAE,UAAU,CAAC;SAClB;KACF;CACF;AAED,MAAM,WAAW,cAAc;IAC7B,GAAG,EAAE,MAAM,CAAC;IAEZ,EAAE,EAAE,IAAI,GAAG,IAAI,GAAG,IAAI,GAAG,IAAI,CAAC;IAE9B,KAAK,EAAE,OAAO,GAAG,MAAM,GAAG,MAAM,GAAG,KAAK,CAAC,OAAO,CAAC,GAAG,OAAO,GAAG,IAAI,CAAC;CACpE;AAED,MAAM,WAAW,kBAAkB;IACjC,IAAI,EAAE,KAAK,CAAC,kBAAkB,CAAC,IAAI,CAAC,CAAC;CACtC;AAED,yBAAiB,kBAAkB,CAAC;IAClC,UAAiB,IAAI;QACnB,IAAI,EAAE,MAAM,CAAC;QAEb,OAAO,EAAE,MAAM,CAAC;QAEhB,UAAU,EAAE,MAAM,CAAC;QAEnB,QAAQ,EAAE,MAAM,CAAC;QAEjB,UAAU,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,GAAG,MAAM,GAAG,MAAM,GAAG,KAAK,CAAC,OAAO,CAAC,GAAG,OAAO,GAAG,IAAI,CAAC,CAAC;QAEzF,QAAQ,CAAC,EAAE,MAAM,CAAC;QAElB,cAAc,CAAC,EAAE,MAAM,CAAC;KACzB;CACF;AAED,MAAM,WAAW,cAAc;IAC7B,IAAI,EAAE,MAAM,CAAC;IAEb,OAAO,EAAE,MAAM,CAAC;IAEhB,UAAU,EAAE,MAAM,CAAC;IAEnB,QAAQ,EAAE,MAAM,CAAC;IAEjB,UAAU,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,GAAG,MAAM,GAAG,MAAM,GAAG,KAAK,CAAC,OAAO,CAAC,GAAG,OAAO,GAAG,IAAI,CAAC,CAAC;IAEzF,QAAQ,CAAC,EAAE,MAAM,CAAC;IAElB,cAAc,CAAC,EAAE,MAAM,CAAC;IAExB,MAAM,CAAC,EAAE,IAAI,GAAG,OAAO,CAAC;CACzB;AAED,MAAM,WAAW,KAAK;IACpB,YAAY,EAAE,MAAM,CAAC;IAErB,UAAU,EAAE,MAAM,CAAC;IAEnB,QAAQ,EAAE,MAAM,CAAC;IAEjB,QAAQ,CAAC,EAAE,MAAM,CAAC;CACnB;AAED,MAAM,WAAW,wBAAwB;IACvC,IAAI,EAAE,MAAM,CAAC;IAEb,OAAO,EAAE,MAAM,CAAC;IAEhB,UAAU,EAAE,MAAM,CAAC;IAEnB,QAAQ,EAAE,MAAM,CAAC;IAEjB,UAAU,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,GAAG,MAAM,GAAG,MAAM,GAAG,KAAK,CAAC,OAAO,CAAC,GAAG,OAAO,GAAG,IAAI,CAAC,CAAC;IAEzF,QAAQ,CAAC,EAAE,MAAM,CAAC;IAElB,cAAc,CAAC,EAAE,MAAM,CAAC;CACzB;AAED,MAAM,MAAM,4BAA4B,GAAG,MAAM,CAAC,MAAM,EAAE,cAAc,CAAC,CAAC;AAE1E,MAAM,MAAM,2BAA2B,GAAG,KAAK,CAAC,2BAA2B,CAAC,+BAA+B,CAAC,CAAC;AAE7G,yBAAiB,2BAA2B,CAAC;IAC3C,UAAiB,+BAA+B;QAC9C,IAAI,EAAE,MAAM,CAAC;QAEb,OAAO,EAAE,MAAM,CAAC;QAEhB,UAAU,EAAE,MAAM,CAAC;QAEnB,QAAQ,EAAE,MAAM,CAAC;QAEjB,UAAU,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,GAAG,MAAM,GAAG,MAAM,GAAG,KAAK,CAAC,OAAO,CAAC,GAAG,OAAO,GAAG,IAAI,CAAC,CAAC;QAEzF,QAAQ,CAAC,EAAE,MAAM,CAAC;QAElB,cAAc,CAAC,EAAE,MAAM,CAAC;KACzB;CACF;AAED,MAAM,MAAM,4BAA4B,GAAG,KAAK,CAAC,KAAK,CAAC,CAAC;AAExD,MAAM,WAAW,0BAA0B;IACzC,oBAAoB,CAAC,EAAE,KAAK,CAAC,MAAM,CAAC,CAAC;IAErC,SAAS,CAAC,EAAE,MAAM,CAAC;CACpB;AAED,MAAM,WAAW,uBAAuB;IACtC,KAAK,EAAE,KAAK,CAAC;IAEb,WAAW,EAAE,MAAM,CAAC;CACrB;AAED,MAAM,WAAW,yBAAyB;IACxC,iBAAiB,EAAE,KAAK,CAAC,cAAc,CAAC,CAAC;IAEzC,oBAAoB,EAAE,KAAK,CAAC,MAAM,CAAC,CAAC;IAEpC,SAAS,CAAC,EAAE,MAAM,CAAC;CACpB;AAED,MAAM,WAAW,0BAA0B;IACzC,iBAAiB,CAAC,EAAE,KAAK,CAAC,cAAc,CAAC,CAAC;IAE1C,KAAK,CAAC,EAAE,MAAM,CAAC;IAEf,MAAM,CAAC,EAAE,MAAM,CAAC;IAEhB,QAAQ,CAAC,EAAE,KAAK,CAAC,MAAM,CAAC,CAAC;CAC1B;AAED,MAAM,WAAW,iCAAiC;IAChD,iBAAiB,EAAE,KAAK,CAAC,cAAc,CAAC,CAAC;IAEzC,kBAAkB,EAAE,KAAK,CAAC,MAAM,CAAC,CAAC;IAElC,UAAU,EAAE,MAAM,CAAC;IAEnB,SAAS,CAAC,EAAE,MAAM,CAAC;CACpB;AAED,MAAM,CAAC,OAAO,WAAW,SAAS,CAAC;IACjC,OAAO,EACL,KAAK,KAAK,IAAI,KAAK,EACnB,KAAK,cAAc,IAAI,cAAc,EACrC,KAAK,kBAAkB,IAAI,kBAAkB,EAC7C,KAAK,cAAc,IAAI,cAAc,EACrC,KAAK,KAAK,IAAI,KAAK,EACnB,KAAK,wBAAwB,IAAI,wBAAwB,EACzD,KAAK,4BAA4B,IAAI,4BAA4B,EACjE,KAAK,2BAA2B,IAAI,2BAA2B,EAC/D,KAAK,4BAA4B,IAAI,4BAA4B,EACjE,KAAK,0BAA0B,IAAI,0BAA0B,EAC7D,KAAK,uBAAuB,IAAI,uBAAuB,EACvD,KAAK,yBAAyB,IAAI,yBAAyB,EAC3D,KAAK,0BAA0B,IAAI,0BAA0B,EAC7D,KAAK,iCAAiC,IAAI,iCAAiC,GAC5E,CAAC;CACH"}

package/resources/telemetry.js

@@ -3,36 +3,43 @@
 Object.defineProperty(exports, "__esModule", { value: true });
 exports.Telemetry = void 0;
 const resource_1 = require("../resource.js");
+const core_1 = require("../core.js");
 class Telemetry extends resource_1.APIResource {
-    getTrace(params, options) {
-        const { 'X-LlamaStack-ProviderData': xLlamaStackProviderData, ...query } = params;
-        return this._client.get('/telemetry/get_trace', {
-            query,
+    getSpan(traceId, spanId, options) {
+        return this._client.get(`/v1/telemetry/traces/${traceId}/spans/${spanId}`, options);
+    }
+    getSpanTree(spanId, query = {}, options) {
+        if ((0, core_1.isRequestOptions)(query)) {
+            return this.getSpanTree(spanId, {}, query);
+        }
+        return this._client.get(`/v1/telemetry/spans/${spanId}/tree`, { query, ...options })._thenUnwrap((obj) => obj.data);
+    }
+    getTrace(traceId, options) {
+        return this._client.get(`/v1/telemetry/traces/${traceId}`, options);
+    }
+    logEvent(body, options) {
+        return this._client.post('/v1/telemetry/events', {
+            body,
             ...options,
-            headers: {
-                ...(xLlamaStackProviderData != null ?
-                    { 'X-LlamaStack-ProviderData': xLlamaStackProviderData }
-                    : undefined),
-                ...options?.headers,
-            },
+            headers: { Accept: '*/*', ...options?.headers },
         });
     }
-    log(params, options) {
-        const { 'X-LlamaStack-ProviderData': xLlamaStackProviderData, ...body } = params;
-        return this._client.post('/telemetry/log_event', {
+    querySpans(query, options) {
+        return this._client.get('/v1/telemetry/spans', { query, ...options })._thenUnwrap((obj) => obj.data);
+    }
+    queryTraces(query = {}, options) {
+        if ((0, core_1.isRequestOptions)(query)) {
+            return this.queryTraces({}, query);
+        }
+        return this._client.get('/v1/telemetry/traces', { query, ...options })._thenUnwrap((obj) => obj.data);
+    }
+    saveSpansToDataset(body, options) {
+        return this._client.post('/v1/telemetry/spans/export', {
             body,
             ...options,
-            headers: {
-                Accept: '*/*',
-                ...(xLlamaStackProviderData != null ?
-                    { 'X-LlamaStack-ProviderData': xLlamaStackProviderData }
-                    : undefined),
-                ...options?.headers,
-            },
+            headers: { Accept: '*/*', ...options?.headers },
         });
     }
 }
 exports.Telemetry = Telemetry;
-(function (Telemetry) {
-})(Telemetry = exports.Telemetry || (exports.Telemetry = {}));
 //# sourceMappingURL=telemetry.js.map

package/resources/telemetry.js.map

@@ -1 +1 @@
-{"version":3,"file":"telemetry.js","sourceRoot":"","sources":["../src/resources/telemetry.ts"],"names":[],"mappings":";AAAA,sFAAsF;;;AAEtF,6CAA0C;AAI1C,MAAa,SAAU,SAAQ,sBAAW;IACxC,QAAQ,CACN,MAA+B,EAC/B,OAA6B;QAE7B,MAAM,EAAE,2BAA2B,EAAE,uBAAuB,EAAE,GAAG,KAAK,EAAE,GAAG,MAAM,CAAC;QAClF,OAAO,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,sBAAsB,EAAE;YAC9C,KAAK;YACL,GAAG,OAAO;YACV,OAAO,EAAE;gBACP,GAAG,CAAC,uBAAuB,IAAI,IAAI,CAAC,CAAC;oBACnC,EAAE,2BAA2B,EAAE,uBAAuB,EAAE;oBAC1D,CAAC,CAAC,SAAS,CAAC;gBACZ,GAAG,OAAO,EAAE,OAAO;aACpB;SACF,CAAC,CAAC;IACL,CAAC;IAED,GAAG,CAAC,MAA0B,EAAE,OAA6B;QAC3D,MAAM,EAAE,2BAA2B,EAAE,uBAAuB,EAAE,GAAG,IAAI,EAAE,GAAG,MAAM,CAAC;QACjF,OAAO,IAAI,CAAC,OAAO,CAAC,IAAI,CAAC,sBAAsB,EAAE;YAC/C,IAAI;YACJ,GAAG,OAAO;YACV,OAAO,EAAE;gBACP,MAAM,EAAE,KAAK;gBACb,GAAG,CAAC,uBAAuB,IAAI,IAAI,CAAC,CAAC;oBACnC,EAAE,2BAA2B,EAAE,uBAAuB,EAAE;oBAC1D,CAAC,CAAC,SAAS,CAAC;gBACZ,GAAG,OAAO,EAAE,OAAO;aACpB;SACF,CAAC,CAAC;IACL,CAAC;CACF;AAhCD,8BAgCC;AA2GD,WAAiB,SAAS;AAI1B,CAAC,EAJgB,SAAS,GAAT,iBAAS,KAAT,iBAAS,QAIzB"}
+{"version":3,"file":"telemetry.js","sourceRoot":"","sources":["../src/resources/telemetry.ts"],"names":[],"mappings":";AAAA,sFAAsF;;;AAEtF,6CAA0C;AAC1C,qCAA2C;AAG3C,MAAa,SAAU,SAAQ,sBAAW;IACxC,OAAO,CACL,OAAe,EACf,MAAc,EACd,OAA6B;QAE7B,OAAO,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,wBAAwB,OAAO,UAAU,MAAM,EAAE,EAAE,OAAO,CAAC,CAAC;IACtF,CAAC;IAQD,WAAW,CACT,MAAc,EACd,QAA0D,EAAE,EAC5D,OAA6B;QAE7B,IAAI,IAAA,uBAAgB,EAAC,KAAK,CAAC,EAAE;YAC3B,OAAO,IAAI,CAAC,WAAW,CAAC,MAAM,EAAE,EAAE,EAAE,KAAK,CAAC,CAAC;SAC5C;QACD,OACE,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,uBAAuB,MAAM,OAAO,EAAE,EAAE,KAAK,EAAE,GAAG,OAAO,EAAE,CAG7E,CAAC,WAAW,CAAC,CAAC,GAAG,EAAE,EAAE,CAAC,GAAG,CAAC,IAAI,CAAC,CAAC;IACnC,CAAC;IAED,QAAQ,CAAC,OAAe,EAAE,OAA6B;QACrD,OAAO,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,wBAAwB,OAAO,EAAE,EAAE,OAAO,CAAC,CAAC;IACtE,CAAC;IAED,QAAQ,CAAC,IAA6B,EAAE,OAA6B;QACnE,OAAO,IAAI,CAAC,OAAO,CAAC,IAAI,CAAC,sBAAsB,EAAE;YAC/C,IAAI;YACJ,GAAG,OAAO;YACV,OAAO,EAAE,EAAE,MAAM,EAAE,KAAK,EAAE,GAAG,OAAO,EAAE,OAAO,EAAE;SAChD,CAAC,CAAC;IACL,CAAC;IAED,UAAU,CACR,KAAgC,EAChC,OAA6B;QAE7B,OACE,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,qBAAqB,EAAE,EAAE,KAAK,EAAE,GAAG,OAAO,EAAE,CAG9D,CAAC,WAAW,CAAC,CAAC,GAAG,EAAE,EAAE,CAAC,GAAG,CAAC,IAAI,CAAC,CAAC;IACnC,CAAC;IAOD,WAAW,CACT,QAA0D,EAAE,EAC5D,OAA6B;QAE7B,IAAI,IAAA,uBAAgB,EAAC,KAAK,CAAC,EAAE;YAC3B,OAAO,IAAI,CAAC,WAAW,CAAC,EAAE,EAAE,KAAK,CAAC,CAAC;SACpC;QACD,OACE,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,sBAAsB,EAAE,EAAE,KAAK,EAAE,GAAG,OAAO,EAAE,CAG/D,CAAC,WAAW,CAAC,CAAC,GAAG,EAAE,EAAE,CAAC,GAAG,CAAC,IAAI,CAAC,CAAC;IACnC,CAAC;IAED,kBAAkB,CAChB,IAAuC,EACvC,OAA6B;QAE7B,OAAO,IAAI,CAAC,OAAO,CAAC,IAAI,CAAC,4BAA4B,EAAE;YACrD,IAAI;YACJ,GAAG,OAAO;YACV,OAAO,EAAE,EAAE,MAAM,EAAE,KAAK,EAAE,GAAG,OAAO,EAAE,OAAO,EAAE;SAChD,CAAC,CAAC;IACL,CAAC;CACF;AAlFD,8BAkFC"}

package/resources/telemetry.mjs

@@ -1,34 +1,41 @@
 // File generated from our OpenAPI spec by Stainless. See CONTRIBUTING.md for details.
 import { APIResource } from "../resource.mjs";
+import { isRequestOptions } from "../core.mjs";
 export class Telemetry extends APIResource {
-    getTrace(params, options) {
-        const { 'X-LlamaStack-ProviderData': xLlamaStackProviderData, ...query } = params;
-        return this._client.get('/telemetry/get_trace', {
-            query,
+    getSpan(traceId, spanId, options) {
+        return this._client.get(`/v1/telemetry/traces/${traceId}/spans/${spanId}`, options);
+    }
+    getSpanTree(spanId, query = {}, options) {
+        if (isRequestOptions(query)) {
+            return this.getSpanTree(spanId, {}, query);
+        }
+        return this._client.get(`/v1/telemetry/spans/${spanId}/tree`, { query, ...options })._thenUnwrap((obj) => obj.data);
+    }
+    getTrace(traceId, options) {
+        return this._client.get(`/v1/telemetry/traces/${traceId}`, options);
+    }
+    logEvent(body, options) {
+        return this._client.post('/v1/telemetry/events', {
+            body,
             ...options,
-            headers: {
-                ...(xLlamaStackProviderData != null ?
-                    { 'X-LlamaStack-ProviderData': xLlamaStackProviderData }
-                    : undefined),
-                ...options?.headers,
-            },
+            headers: { Accept: '*/*', ...options?.headers },
         });
     }
-    log(params, options) {
-        const { 'X-LlamaStack-ProviderData': xLlamaStackProviderData, ...body } = params;
-        return this._client.post('/telemetry/log_event', {
+    querySpans(query, options) {
+        return this._client.get('/v1/telemetry/spans', { query, ...options })._thenUnwrap((obj) => obj.data);
+    }
+    queryTraces(query = {}, options) {
+        if (isRequestOptions(query)) {
+            return this.queryTraces({}, query);
+        }
+        return this._client.get('/v1/telemetry/traces', { query, ...options })._thenUnwrap((obj) => obj.data);
+    }
+    saveSpansToDataset(body, options) {
+        return this._client.post('/v1/telemetry/spans/export', {
             body,
             ...options,
-            headers: {
-                Accept: '*/*',
-                ...(xLlamaStackProviderData != null ?
-                    { 'X-LlamaStack-ProviderData': xLlamaStackProviderData }
-                    : undefined),
-                ...options?.headers,
-            },
+            headers: { Accept: '*/*', ...options?.headers },
         });
     }
 }
-(function (Telemetry) {
-})(Telemetry || (Telemetry = {}));
 //# sourceMappingURL=telemetry.mjs.map

package/resources/telemetry.mjs.map

@@ -1 +1 @@
-{"version":3,"file":"telemetry.mjs","sourceRoot":"","sources":["../src/resources/telemetry.ts"],"names":[],"mappings":"AAAA,sFAAsF;OAE/E,EAAE,WAAW,EAAE;AAItB,MAAM,OAAO,SAAU,SAAQ,WAAW;IACxC,QAAQ,CACN,MAA+B,EAC/B,OAA6B;QAE7B,MAAM,EAAE,2BAA2B,EAAE,uBAAuB,EAAE,GAAG,KAAK,EAAE,GAAG,MAAM,CAAC;QAClF,OAAO,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,sBAAsB,EAAE;YAC9C,KAAK;YACL,GAAG,OAAO;YACV,OAAO,EAAE;gBACP,GAAG,CAAC,uBAAuB,IAAI,IAAI,CAAC,CAAC;oBACnC,EAAE,2BAA2B,EAAE,uBAAuB,EAAE;oBAC1D,CAAC,CAAC,SAAS,CAAC;gBACZ,GAAG,OAAO,EAAE,OAAO;aACpB;SACF,CAAC,CAAC;IACL,CAAC;IAED,GAAG,CAAC,MAA0B,EAAE,OAA6B;QAC3D,MAAM,EAAE,2BAA2B,EAAE,uBAAuB,EAAE,GAAG,IAAI,EAAE,GAAG,MAAM,CAAC;QACjF,OAAO,IAAI,CAAC,OAAO,CAAC,IAAI,CAAC,sBAAsB,EAAE;YAC/C,IAAI;YACJ,GAAG,OAAO;YACV,OAAO,EAAE;gBACP,MAAM,EAAE,KAAK;gBACb,GAAG,CAAC,uBAAuB,IAAI,IAAI,CAAC,CAAC;oBACnC,EAAE,2BAA2B,EAAE,uBAAuB,EAAE;oBAC1D,CAAC,CAAC,SAAS,CAAC;gBACZ,GAAG,OAAO,EAAE,OAAO;aACpB;SACF,CAAC,CAAC;IACL,CAAC;CACF;AA2GD,WAAiB,SAAS;AAI1B,CAAC,EAJgB,SAAS,KAAT,SAAS,QAIzB"}
+{"version":3,"file":"telemetry.mjs","sourceRoot":"","sources":["../src/resources/telemetry.ts"],"names":[],"mappings":"AAAA,sFAAsF;OAE/E,EAAE,WAAW,EAAE;OACf,EAAE,gBAAgB,EAAE;AAG3B,MAAM,OAAO,SAAU,SAAQ,WAAW;IACxC,OAAO,CACL,OAAe,EACf,MAAc,EACd,OAA6B;QAE7B,OAAO,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,wBAAwB,OAAO,UAAU,MAAM,EAAE,EAAE,OAAO,CAAC,CAAC;IACtF,CAAC;IAQD,WAAW,CACT,MAAc,EACd,QAA0D,EAAE,EAC5D,OAA6B;QAE7B,IAAI,gBAAgB,CAAC,KAAK,CAAC,EAAE;YAC3B,OAAO,IAAI,CAAC,WAAW,CAAC,MAAM,EAAE,EAAE,EAAE,KAAK,CAAC,CAAC;SAC5C;QACD,OACE,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,uBAAuB,MAAM,OAAO,EAAE,EAAE,KAAK,EAAE,GAAG,OAAO,EAAE,CAG7E,CAAC,WAAW,CAAC,CAAC,GAAG,EAAE,EAAE,CAAC,GAAG,CAAC,IAAI,CAAC,CAAC;IACnC,CAAC;IAED,QAAQ,CAAC,OAAe,EAAE,OAA6B;QACrD,OAAO,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,wBAAwB,OAAO,EAAE,EAAE,OAAO,CAAC,CAAC;IACtE,CAAC;IAED,QAAQ,CAAC,IAA6B,EAAE,OAA6B;QACnE,OAAO,IAAI,CAAC,OAAO,CAAC,IAAI,CAAC,sBAAsB,EAAE;YAC/C,IAAI;YACJ,GAAG,OAAO;YACV,OAAO,EAAE,EAAE,MAAM,EAAE,KAAK,EAAE,GAAG,OAAO,EAAE,OAAO,EAAE;SAChD,CAAC,CAAC;IACL,CAAC;IAED,UAAU,CACR,KAAgC,EAChC,OAA6B;QAE7B,OACE,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,qBAAqB,EAAE,EAAE,KAAK,EAAE,GAAG,OAAO,EAAE,CAG9D,CAAC,WAAW,CAAC,CAAC,GAAG,EAAE,EAAE,CAAC,GAAG,CAAC,IAAI,CAAC,CAAC;IACnC,CAAC;IAOD,WAAW,CACT,QAA0D,EAAE,EAC5D,OAA6B;QAE7B,IAAI,gBAAgB,CAAC,KAAK,CAAC,EAAE;YAC3B,OAAO,IAAI,CAAC,WAAW,CAAC,EAAE,EAAE,KAAK,CAAC,CAAC;SACpC;QACD,OACE,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,sBAAsB,EAAE,EAAE,KAAK,EAAE,GAAG,OAAO,EAAE,CAG/D,CAAC,WAAW,CAAC,CAAC,GAAG,EAAE,EAAE,CAAC,GAAG,CAAC,IAAI,CAAC,CAAC;IACnC,CAAC;IAED,kBAAkB,CAChB,IAAuC,EACvC,OAA6B;QAE7B,OAAO,IAAI,CAAC,OAAO,CAAC,IAAI,CAAC,4BAA4B,EAAE;YACrD,IAAI;YACJ,GAAG,OAAO;YACV,OAAO,EAAE,EAAE,MAAM,EAAE,KAAK,EAAE,GAAG,OAAO,EAAE,OAAO,EAAE;SAChD,CAAC,CAAC;IACL,CAAC;CACF"}

package/resources/tool-runtime/index.d.ts

@@ -0,0 +1,3 @@
+export { RagTool, type RagToolInsertParams, type RagToolQueryParams } from "./rag-tool.js";
+export { ToolRuntime, type ToolDef, type ToolInvocationResult, type ToolRuntimeInvokeToolParams, type ToolRuntimeListToolsParams, } from "./tool-runtime.js";
+//# sourceMappingURL=index.d.ts.map

package/resources/tool-runtime/index.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../../src/resources/tool-runtime/index.ts"],"names":[],"mappings":"AAEA,OAAO,EAAE,OAAO,EAAE,KAAK,mBAAmB,EAAE,KAAK,kBAAkB,EAAE,MAAM,YAAY,CAAC;AACxF,OAAO,EACL,WAAW,EACX,KAAK,OAAO,EACZ,KAAK,oBAAoB,EACzB,KAAK,2BAA2B,EAChC,KAAK,0BAA0B,GAChC,MAAM,gBAAgB,CAAC"}

package/resources/tool-runtime/index.js

@@ -0,0 +1,9 @@
+"use strict";
+// File generated from our OpenAPI spec by Stainless. See CONTRIBUTING.md for details.
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.ToolRuntime = exports.RagTool = void 0;
+var rag_tool_1 = require("./rag-tool.js");
+Object.defineProperty(exports, "RagTool", { enumerable: true, get: function () { return rag_tool_1.RagTool; } });
+var tool_runtime_1 = require("./tool-runtime.js");
+Object.defineProperty(exports, "ToolRuntime", { enumerable: true, get: function () { return tool_runtime_1.ToolRuntime; } });
+//# sourceMappingURL=index.js.map

package/resources/tool-runtime/index.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"index.js","sourceRoot":"","sources":["../../src/resources/tool-runtime/index.ts"],"names":[],"mappings":";AAAA,sFAAsF;;;AAEtF,0CAAwF;AAA/E,mGAAA,OAAO,OAAA;AAChB,kDAMwB;AALtB,2GAAA,WAAW,OAAA"}

package/resources/tool-runtime/index.mjs

@@ -0,0 +1,4 @@
+// File generated from our OpenAPI spec by Stainless. See CONTRIBUTING.md for details.
+export { RagTool } from "./rag-tool.mjs";
+export { ToolRuntime, } from "./tool-runtime.mjs";
+//# sourceMappingURL=index.mjs.map

package/resources/tool-runtime/index.mjs.map

@@ -0,0 +1 @@
+{"version":3,"file":"index.mjs","sourceRoot":"","sources":["../../src/resources/tool-runtime/index.ts"],"names":[],"mappings":"AAAA,sFAAsF;OAE/E,EAAE,OAAO,EAAqD;OAC9D,EACL,WAAW,GAKZ"}

package/resources/tool-runtime/rag-tool.d.ts

@@ -0,0 +1,27 @@
+import { APIResource } from "../../resource.js";
+import * as Core from "../../core.js";
+import * as Shared from "../shared.js";
+export declare class RagTool extends APIResource {
+    /**
+     * Index documents so they can be used by the RAG system
+     */
+    insert(body: RagToolInsertParams, options?: Core.RequestOptions): Core.APIPromise<void>;
+    /**
+     * Query the RAG system for context; typically invoked by the agent
+     */
+    query(body: RagToolQueryParams, options?: Core.RequestOptions): Core.APIPromise<Shared.QueryResult>;
+}
+export interface RagToolInsertParams {
+    chunk_size_in_tokens: number;
+    documents: Array<Shared.Document>;
+    vector_db_id: string;
+}
+export interface RagToolQueryParams {
+    content: Shared.InterleavedContent;
+    vector_db_ids: Array<string>;
+    query_config?: Shared.QueryConfig;
+}
+export declare namespace RagTool {
+    export { type RagToolInsertParams as RagToolInsertParams, type RagToolQueryParams as RagToolQueryParams };
+}
+//# sourceMappingURL=rag-tool.d.ts.map

package/resources/tool-runtime/rag-tool.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"rag-tool.d.ts","sourceRoot":"","sources":["../../src/resources/tool-runtime/rag-tool.ts"],"names":[],"mappings":"AAEA,OAAO,EAAE,WAAW,EAAE,MAAM,gBAAgB,CAAC;AAC7C,OAAO,KAAK,IAAI,MAAM,YAAY,CAAC;AACnC,OAAO,KAAK,MAAM,MAAM,WAAW,CAAC;AAEpC,qBAAa,OAAQ,SAAQ,WAAW;IACtC;;OAEG;IACH,MAAM,CAAC,IAAI,EAAE,mBAAmB,EAAE,OAAO,CAAC,EAAE,IAAI,CAAC,cAAc,GAAG,IAAI,CAAC,UAAU,CAAC,IAAI,CAAC;IAQvF;;OAEG;IACH,KAAK,CAAC,IAAI,EAAE,kBAAkB,EAAE,OAAO,CAAC,EAAE,IAAI,CAAC,cAAc,GAAG,IAAI,CAAC,UAAU,CAAC,MAAM,CAAC,WAAW,CAAC;CAGpG;AAED,MAAM,WAAW,mBAAmB;IAClC,oBAAoB,EAAE,MAAM,CAAC;IAE7B,SAAS,EAAE,KAAK,CAAC,MAAM,CAAC,QAAQ,CAAC,CAAC;IAElC,YAAY,EAAE,MAAM,CAAC;CACtB;AAED,MAAM,WAAW,kBAAkB;IACjC,OAAO,EAAE,MAAM,CAAC,kBAAkB,CAAC;IAEnC,aAAa,EAAE,KAAK,CAAC,MAAM,CAAC,CAAC;IAE7B,YAAY,CAAC,EAAE,MAAM,CAAC,WAAW,CAAC;CACnC;AAED,MAAM,CAAC,OAAO,WAAW,OAAO,CAAC;IAC/B,OAAO,EAAE,KAAK,mBAAmB,IAAI,mBAAmB,EAAE,KAAK,kBAAkB,IAAI,kBAAkB,EAAE,CAAC;CAC3G"}

package/resources/tool-runtime/rag-tool.js

@@ -0,0 +1,25 @@
+"use strict";
+// File generated from our OpenAPI spec by Stainless. See CONTRIBUTING.md for details.
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.RagTool = void 0;
+const resource_1 = require("../../resource.js");
+class RagTool extends resource_1.APIResource {
+    /**
+     * Index documents so they can be used by the RAG system
+     */
+    insert(body, options) {
+        return this._client.post('/v1/tool-runtime/rag-tool/insert', {
+            body,
+            ...options,
+            headers: { Accept: '*/*', ...options?.headers },
+        });
+    }
+    /**
+     * Query the RAG system for context; typically invoked by the agent
+     */
+    query(body, options) {
+        return this._client.post('/v1/tool-runtime/rag-tool/query', { body, ...options });
+    }
+}
+exports.RagTool = RagTool;
+//# sourceMappingURL=rag-tool.js.map

package/resources/tool-runtime/rag-tool.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"rag-tool.js","sourceRoot":"","sources":["../../src/resources/tool-runtime/rag-tool.ts"],"names":[],"mappings":";AAAA,sFAAsF;;;AAEtF,gDAA6C;AAI7C,MAAa,OAAQ,SAAQ,sBAAW;IACtC;;OAEG;IACH,MAAM,CAAC,IAAyB,EAAE,OAA6B;QAC7D,OAAO,IAAI,CAAC,OAAO,CAAC,IAAI,CAAC,kCAAkC,EAAE;YAC3D,IAAI;YACJ,GAAG,OAAO;YACV,OAAO,EAAE,EAAE,MAAM,EAAE,KAAK,EAAE,GAAG,OAAO,EAAE,OAAO,EAAE;SAChD,CAAC,CAAC;IACL,CAAC;IAED;;OAEG;IACH,KAAK,CAAC,IAAwB,EAAE,OAA6B;QAC3D,OAAO,IAAI,CAAC,OAAO,CAAC,IAAI,CAAC,iCAAiC,EAAE,EAAE,IAAI,EAAE,GAAG,OAAO,EAAE,CAAC,CAAC;IACpF,CAAC;CACF;AAlBD,0BAkBC"}

package/resources/tool-runtime/rag-tool.mjs

@@ -0,0 +1,21 @@
+// File generated from our OpenAPI spec by Stainless. See CONTRIBUTING.md for details.
+import { APIResource } from "../../resource.mjs";
+export class RagTool extends APIResource {
+    /**
+     * Index documents so they can be used by the RAG system
+     */
+    insert(body, options) {
+        return this._client.post('/v1/tool-runtime/rag-tool/insert', {
+            body,
+            ...options,
+            headers: { Accept: '*/*', ...options?.headers },
+        });
+    }
+    /**
+     * Query the RAG system for context; typically invoked by the agent
+     */
+    query(body, options) {
+        return this._client.post('/v1/tool-runtime/rag-tool/query', { body, ...options });
+    }
+}
+//# sourceMappingURL=rag-tool.mjs.map

package/resources/tool-runtime/rag-tool.mjs.map

@@ -0,0 +1 @@
+{"version":3,"file":"rag-tool.mjs","sourceRoot":"","sources":["../../src/resources/tool-runtime/rag-tool.ts"],"names":[],"mappings":"AAAA,sFAAsF;OAE/E,EAAE,WAAW,EAAE;AAItB,MAAM,OAAO,OAAQ,SAAQ,WAAW;IACtC;;OAEG;IACH,MAAM,CAAC,IAAyB,EAAE,OAA6B;QAC7D,OAAO,IAAI,CAAC,OAAO,CAAC,IAAI,CAAC,kCAAkC,EAAE;YAC3D,IAAI;YACJ,GAAG,OAAO;YACV,OAAO,EAAE,EAAE,MAAM,EAAE,KAAK,EAAE,GAAG,OAAO,EAAE,OAAO,EAAE;SAChD,CAAC,CAAC;IACL,CAAC;IAED;;OAEG;IACH,KAAK,CAAC,IAAwB,EAAE,OAA6B;QAC3D,OAAO,IAAI,CAAC,OAAO,CAAC,IAAI,CAAC,iCAAiC,EAAE,EAAE,IAAI,EAAE,GAAG,OAAO,EAAE,CAAC,CAAC;IACpF,CAAC;CACF"}

package/resources/tool-runtime/tool-runtime.d.ts

@@ -0,0 +1,48 @@
+import { APIResource } from "../../resource.js";
+import * as Core from "../../core.js";
+import * as Shared from "../shared.js";
+import * as RagToolAPI from "./rag-tool.js";
+import { RagTool, RagToolInsertParams, RagToolQueryParams } from "./rag-tool.js";
+import { JSONLDecoder } from "../../internal/decoders/jsonl.js";
+export declare class ToolRuntime extends APIResource {
+    ragTool: RagToolAPI.RagTool;
+    /**
+     * Run a tool with the given arguments
+     */
+    invokeTool(body: ToolRuntimeInvokeToolParams, options?: Core.RequestOptions): Core.APIPromise<ToolInvocationResult>;
+    listTools(query?: ToolRuntimeListToolsParams, options?: Core.RequestOptions): Core.APIPromise<JSONLDecoder<ToolDef>>;
+    listTools(options?: Core.RequestOptions): Core.APIPromise<JSONLDecoder<ToolDef>>;
+}
+export interface ToolDef {
+    name: string;
+    description?: string;
+    metadata?: Record<string, boolean | number | string | Array<unknown> | unknown | null>;
+    parameters?: Array<ToolDef.Parameter>;
+}
+export declare namespace ToolDef {
+    interface Parameter {
+        description: string;
+        name: string;
+        parameter_type: string;
+        required: boolean;
+        default?: boolean | number | string | Array<unknown> | unknown | null;
+    }
+}
+export interface ToolInvocationResult {
+    content: Shared.InterleavedContent;
+    error_code?: number;
+    error_message?: string;
+}
+export interface ToolRuntimeInvokeToolParams {
+    kwargs: Record<string, boolean | number | string | Array<unknown> | unknown | null>;
+    tool_name: string;
+}
+export interface ToolRuntimeListToolsParams {
+    mcp_endpoint?: Shared.URL;
+    tool_group_id?: string;
+}
+export declare namespace ToolRuntime {
+    export { type ToolDef as ToolDef, type ToolInvocationResult as ToolInvocationResult, type ToolRuntimeInvokeToolParams as ToolRuntimeInvokeToolParams, type ToolRuntimeListToolsParams as ToolRuntimeListToolsParams, };
+    export { RagTool as RagTool, type RagToolInsertParams as RagToolInsertParams, type RagToolQueryParams as RagToolQueryParams, };
+}
+//# sourceMappingURL=tool-runtime.d.ts.map

package/resources/tool-runtime/tool-runtime.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"tool-runtime.d.ts","sourceRoot":"","sources":["../../src/resources/tool-runtime/tool-runtime.ts"],"names":[],"mappings":"AAEA,OAAO,EAAE,WAAW,EAAE,MAAM,gBAAgB,CAAC;AAE7C,OAAO,KAAK,IAAI,MAAM,YAAY,CAAC;AACnC,OAAO,KAAK,MAAM,MAAM,WAAW,CAAC;AACpC,OAAO,KAAK,UAAU,MAAM,YAAY,CAAC;AACzC,OAAO,EAAE,OAAO,EAAE,mBAAmB,EAAE,kBAAkB,EAAE,MAAM,YAAY,CAAC;AAC9E,OAAO,EAAE,YAAY,EAAE,MAAM,+BAA+B,CAAC;AAE7D,qBAAa,WAAY,SAAQ,WAAW;IAC1C,OAAO,EAAE,UAAU,CAAC,OAAO,CAAwC;IAEnE;;OAEG;IACH,UAAU,CACR,IAAI,EAAE,2BAA2B,EACjC,OAAO,CAAC,EAAE,IAAI,CAAC,cAAc,GAC5B,IAAI,CAAC,UAAU,CAAC,oBAAoB,CAAC;IAIxC,SAAS,CACP,KAAK,CAAC,EAAE,0BAA0B,EAClC,OAAO,CAAC,EAAE,IAAI,CAAC,cAAc,GAC5B,IAAI,CAAC,UAAU,CAAC,YAAY,CAAC,OAAO,CAAC,CAAC;IACzC,SAAS,CAAC,OAAO,CAAC,EAAE,IAAI,CAAC,cAAc,GAAG,IAAI,CAAC,UAAU,CAAC,YAAY,CAAC,OAAO,CAAC,CAAC;CAiBjF;AAED,MAAM,WAAW,OAAO;IACtB,IAAI,EAAE,MAAM,CAAC;IAEb,WAAW,CAAC,EAAE,MAAM,CAAC;IAErB,QAAQ,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,GAAG,MAAM,GAAG,MAAM,GAAG,KAAK,CAAC,OAAO,CAAC,GAAG,OAAO,GAAG,IAAI,CAAC,CAAC;IAEvF,UAAU,CAAC,EAAE,KAAK,CAAC,OAAO,CAAC,SAAS,CAAC,CAAC;CACvC;AAED,yBAAiB,OAAO,CAAC;IACvB,UAAiB,SAAS;QACxB,WAAW,EAAE,MAAM,CAAC;QAEpB,IAAI,EAAE,MAAM,CAAC;QAEb,cAAc,EAAE,MAAM,CAAC;QAEvB,QAAQ,EAAE,OAAO,CAAC;QAElB,OAAO,CAAC,EAAE,OAAO,GAAG,MAAM,GAAG,MAAM,GAAG,KAAK,CAAC,OAAO,CAAC,GAAG,OAAO,GAAG,IAAI,CAAC;KACvE;CACF;AAED,MAAM,WAAW,oBAAoB;IACnC,OAAO,EAAE,MAAM,CAAC,kBAAkB,CAAC;IAEnC,UAAU,CAAC,EAAE,MAAM,CAAC;IAEpB,aAAa,CAAC,EAAE,MAAM,CAAC;CACxB;AAED,MAAM,WAAW,2BAA2B;IAC1C,MAAM,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,GAAG,MAAM,GAAG,MAAM,GAAG,KAAK,CAAC,OAAO,CAAC,GAAG,OAAO,GAAG,IAAI,CAAC,CAAC;IAEpF,SAAS,EAAE,MAAM,CAAC;CACnB;AAED,MAAM,WAAW,0BAA0B;IACzC,YAAY,CAAC,EAAE,MAAM,CAAC,GAAG,CAAC;IAE1B,aAAa,CAAC,EAAE,MAAM,CAAC;CACxB;AAID,MAAM,CAAC,OAAO,WAAW,WAAW,CAAC;IACnC,OAAO,EACL,KAAK,OAAO,IAAI,OAAO,EACvB,KAAK,oBAAoB,IAAI,oBAAoB,EACjD,KAAK,2BAA2B,IAAI,2BAA2B,EAC/D,KAAK,0BAA0B,IAAI,0BAA0B,GAC9D,CAAC;IAEF,OAAO,EACL,OAAO,IAAI,OAAO,EAClB,KAAK,mBAAmB,IAAI,mBAAmB,EAC/C,KAAK,kBAAkB,IAAI,kBAAkB,GAC9C,CAAC;CACH"}

package/resources/{evaluate/evaluate.js → tool-runtime/tool-runtime.js}

@@ -24,26 +24,37 @@ var __importStar = (this && this.__importStar) || function (mod) {
     return result;
 };
 Object.defineProperty(exports, "__esModule", { value: true });
-exports.Evaluate = void 0;
+exports.ToolRuntime = void 0;
 const resource_1 = require("../../resource.js");
-const QuestionAnsweringAPI = __importStar(require("./question-answering.js"));
-const SummarizationAPI = __importStar(require("./summarization.js"));
-const TextGenerationAPI = __importStar(require("./text-generation.js"));
-const JobsAPI = __importStar(require("./jobs/jobs.js"));
-class Evaluate extends resource_1.APIResource {
+const core_1 = require("../../core.js");
+const RagToolAPI = __importStar(require("./rag-tool.js"));
+const rag_tool_1 = require("./rag-tool.js");
+const jsonl_1 = require("../../internal/decoders/jsonl.js");
+class ToolRuntime extends resource_1.APIResource {
     constructor() {
         super(...arguments);
-        this.jobs = new JobsAPI.Jobs(this._client);
-        this.questionAnswering = new QuestionAnsweringAPI.QuestionAnswering(this._client);
-        this.summarization = new SummarizationAPI.Summarization(this._client);
-        this.textGeneration = new TextGenerationAPI.TextGeneration(this._client);
+        this.ragTool = new RagToolAPI.RagTool(this._client);
+    }
+    /**
+     * Run a tool with the given arguments
+     */
+    invokeTool(body, options) {
+        return this._client.post('/v1/tool-runtime/invoke', { body, ...options });
+    }
+    listTools(query = {}, options) {
+        if ((0, core_1.isRequestOptions)(query)) {
+            return this.listTools({}, query);
+        }
+        return this._client
+            .get('/v1/tool-runtime/list-tools', {
+            query,
+            ...options,
+            headers: { Accept: 'application/jsonl', ...options?.headers },
+            __binaryResponse: true,
+        })
+            ._thenUnwrap((_, props) => jsonl_1.JSONLDecoder.fromResponse(props.response, props.controller));
     }
 }
-exports.Evaluate = Evaluate;
-(function (Evaluate) {
-    Evaluate.Jobs = JobsAPI.Jobs;
-    Evaluate.QuestionAnswering = QuestionAnsweringAPI.QuestionAnswering;
-    Evaluate.Summarization = SummarizationAPI.Summarization;
-    Evaluate.TextGeneration = TextGenerationAPI.TextGeneration;
-})(Evaluate = exports.Evaluate || (exports.Evaluate = {}));
-//# sourceMappingURL=evaluate.js.map
+exports.ToolRuntime = ToolRuntime;
+ToolRuntime.RagTool = rag_tool_1.RagTool;
+//# sourceMappingURL=tool-runtime.js.map