openlayer 0.4.0 → 0.7.0

package/lib/tracing/steps.mjs

@@ -0,0 +1,78 @@
+import { v4 as uuidv4 } from 'uuid';
+import { StepType } from "./enums.mjs";
+export class Step {
+    constructor(name, inputs = null, output = null, metadata = {}) {
+        this.stepType = null;
+        this.endTime = null;
+        this.groundTruth = null;
+        this.latency = null;
+        this.steps = [];
+        this.name = name;
+        this.id = uuidv4();
+        this.inputs = inputs;
+        this.output = output;
+        this.metadata = metadata;
+        this.startTime = Date.now();
+    }
+    addNestedStep(nestedStep) {
+        this.steps.push(nestedStep);
+    }
+    log(data) {
+        Object.keys(data).forEach((key) => {
+            if (key in this) {
+                // @ts-ignore
+                this[key] = data[key];
+            }
+        });
+    }
+    toJSON() {
+        return {
+            name: this.name,
+            id: this.id,
+            type: this.stepType,
+            inputs: this.inputs,
+            output: this.output,
+            groundTruth: this.groundTruth,
+            metadata: this.metadata,
+            steps: this.steps.map((nestedStep) => nestedStep.toJSON()),
+            latency: this.latency,
+            startTime: this.startTime,
+            endTime: this.endTime,
+        };
+    }
+}
+export class UserCallStep extends Step {
+    constructor(name, inputs = null, output = null, metadata = {}) {
+        super(name, inputs, output, metadata);
+        this.stepType = StepType.USER_CALL;
+    }
+}
+export class ChatCompletionStep extends Step {
+    constructor(name, inputs = null, output = null, metadata = {}) {
+        super(name, inputs, output, metadata);
+        this.provider = null;
+        this.promptTokens = null;
+        this.completionTokens = null;
+        this.tokens = null;
+        this.cost = null;
+        this.model = null;
+        this.modelParameters = null;
+        this.rawOutput = null;
+        this.stepType = StepType.CHAT_COMPLETION;
+    }
+    toJSON() {
+        const stepData = super.toJSON();
+        return {
+            ...stepData,
+            provider: this.provider,
+            promptTokens: this.promptTokens,
+            completionTokens: this.completionTokens,
+            tokens: this.tokens,
+            cost: this.cost,
+            model: this.model,
+            modelParameters: this.modelParameters,
+            rawOutput: this.rawOutput,
+        };
+    }
+}
+//# sourceMappingURL=steps.mjs.map

package/lib/tracing/steps.mjs.map

@@ -0,0 +1 @@
+{"version":3,"file":"steps.mjs","sourceRoot":"","sources":["../../src/lib/tracing/steps.ts"],"names":[],"mappings":"OAAO,EAAE,EAAE,IAAI,MAAM,EAAE,MAAM,MAAM;OAC5B,EAAE,QAAQ,EAAE;AA2BnB,MAAM,OAAO,IAAI;IAaf,YAAY,IAAY,EAAE,SAAc,IAAI,EAAE,SAAc,IAAI,EAAE,WAAgC,EAAE;QAPpG,aAAQ,GAAoB,IAAI,CAAC;QAEjC,YAAO,GAAkB,IAAI,CAAC;QAC9B,gBAAW,GAAQ,IAAI,CAAC;QACxB,YAAO,GAAkB,IAAI,CAAC;QAC9B,UAAK,GAAW,EAAE,CAAC;QAGjB,IAAI,CAAC,IAAI,GAAG,IAAI,CAAC;QACjB,IAAI,CAAC,EAAE,GAAG,MAAM,EAAE,CAAC;QACnB,IAAI,CAAC,MAAM,GAAG,MAAM,CAAC;QACrB,IAAI,CAAC,MAAM,GAAG,MAAM,CAAC;QACrB,IAAI,CAAC,QAAQ,GAAG,QAAQ,CAAC;QAEzB,IAAI,CAAC,SAAS,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;IAC9B,CAAC;IAED,aAAa,CAAC,UAAgB;QAC5B,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,UAAU,CAAC,CAAC;IAC9B,CAAC;IAED,GAAG,CAAC,IAAsC;QACxC,MAAM,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC,OAAO,CAAC,CAAC,GAAG,EAAE,EAAE;YAChC,IAAI,GAAG,IAAI,IAAI,EAAE;gBACf,aAAa;gBACb,IAAI,CAAC,GAAG,CAAC,GAAG,IAAI,CAAC,GAAG,CAAC,CAAC;aACvB;QACH,CAAC,CAAC,CAAC;IACL,CAAC;IAED,MAAM;QACJ,OAAO;YACL,IAAI,EAAE,IAAI,CAAC,IAAI;YACf,EAAE,EAAE,IAAI,CAAC,EAAE;YACX,IAAI,EAAE,IAAI,CAAC,QAAS;YACpB,MAAM,EAAE,IAAI,CAAC,MAAM;YACnB,MAAM,EAAE,IAAI,CAAC,MAAM;YACnB,WAAW,EAAE,IAAI,CAAC,WAAW;YAC7B,QAAQ,EAAE,IAAI,CAAC,QAAQ;YACvB,KAAK,EAAE,IAAI,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC,UAAU,EAAE,EAAE,CAAC,UAAU,CAAC,MAAM,EAAE,CAAC;YAC1D,OAAO,EAAE,IAAI,CAAC,OAAO;YACrB,SAAS,EAAE,IAAI,CAAC,SAAS;YACzB,OAAO,EAAE,IAAI,CAAC,OAAO;SACtB,CAAC;IACJ,CAAC;CACF;AAED,MAAM,OAAO,YAAa,SAAQ,IAAI;IACpC,YAAY,IAAY,EAAE,SAAc,IAAI,EAAE,SAAc,IAAI,EAAE,WAAgC,EAAE;QAClG,KAAK,CAAC,IAAI,EAAE,MAAM,EAAE,MAAM,EAAE,QAAQ,CAAC,CAAC;QACtC,IAAI,CAAC,QAAQ,GAAG,QAAQ,CAAC,SAAS,CAAC;IACrC,CAAC;CACF;AAED,MAAM,OAAO,kBAAmB,SAAQ,IAAI;IAU1C,YAAY,IAAY,EAAE,SAAc,IAAI,EAAE,SAAc,IAAI,EAAE,WAAgC,EAAE;QAClG,KAAK,CAAC,IAAI,EAAE,MAAM,EAAE,MAAM,EAAE,QAAQ,CAAC,CAAC;QAVxC,aAAQ,GAAkB,IAAI,CAAC;QAC/B,iBAAY,GAAkB,IAAI,CAAC;QACnC,qBAAgB,GAAkB,IAAI,CAAC;QACvC,WAAM,GAAkB,IAAI,CAAC;QAC7B,SAAI,GAAkB,IAAI,CAAC;QAC3B,UAAK,GAAkB,IAAI,CAAC;QAC5B,oBAAe,GAA+B,IAAI,CAAC;QACnD,cAAS,GAAkB,IAAI,CAAC;QAI9B,IAAI,CAAC,QAAQ,GAAG,QAAQ,CAAC,eAAe,CAAC;IAC3C,CAAC;IAEQ,MAAM;QACb,MAAM,QAAQ,GAAG,KAAK,CAAC,MAAM,EAAE,CAAC;QAChC,OAAO;YACL,GAAG,QAAQ;YACX,QAAQ,EAAE,IAAI,CAAC,QAAQ;YACvB,YAAY,EAAE,IAAI,CAAC,YAAY;YAC/B,gBAAgB,EAAE,IAAI,CAAC,gBAAgB;YACvC,MAAM,EAAE,IAAI,CAAC,MAAM;YACnB,IAAI,EAAE,IAAI,CAAC,IAAI;YACf,KAAK,EAAE,IAAI,CAAC,KAAK;YACjB,eAAe,EAAE,IAAI,CAAC,eAAe;YACrC,SAAS,EAAE,IAAI,CAAC,SAAS;SAC1B,CAAC;IACJ,CAAC;CACF"}

package/lib/tracing/tracer.d.ts

@@ -0,0 +1,18 @@
+import { StepType } from "./enums.js";
+declare function trace(fn: Function, stepType?: StepType, stepName?: string): Function;
+export declare function addChatCompletionStepToTrace({ name, inputs, output, latency, tokens, promptTokens, completionTokens, model, modelParameters, rawOutput, metadata, provider, }: {
+    name: string;
+    inputs: any;
+    output: any;
+    latency: number;
+    tokens?: number | null;
+    promptTokens?: number | null;
+    completionTokens?: number | null;
+    model?: string | null;
+    modelParameters?: Record<string, any> | null;
+    rawOutput?: string | null;
+    metadata?: Record<string, any>;
+    provider?: string;
+}): void;
+export default trace;
+//# sourceMappingURL=tracer.d.ts.map

package/lib/tracing/tracer.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"tracer.d.ts","sourceRoot":"","sources":["../../src/lib/tracing/tracer.ts"],"names":[],"mappings":"AAIA,OAAO,EAAE,QAAQ,EAAE,MAAM,SAAS,CAAC;AA6GnC,iBAAS,KAAK,CAAC,EAAE,EAAE,QAAQ,EAAE,QAAQ,GAAE,QAA6B,EAAE,QAAQ,CAAC,EAAE,MAAM,GAAG,QAAQ,CAkBjG;AAED,wBAAgB,4BAA4B,CAAC,EAC3C,IAAI,EACJ,MAAM,EACN,MAAM,EACN,OAAO,EACP,MAAa,EACb,YAAmB,EACnB,gBAAuB,EACvB,KAAY,EACZ,eAAsB,EACtB,SAAgB,EAChB,QAAa,EACb,QAAmB,GACpB,EAAE;IACD,IAAI,EAAE,MAAM,CAAC;IACb,MAAM,EAAE,GAAG,CAAC;IACZ,MAAM,EAAE,GAAG,CAAC;IACZ,OAAO,EAAE,MAAM,CAAC;IAChB,MAAM,CAAC,EAAE,MAAM,GAAG,IAAI,CAAC;IACvB,YAAY,CAAC,EAAE,MAAM,GAAG,IAAI,CAAC;IAC7B,gBAAgB,CAAC,EAAE,MAAM,GAAG,IAAI,CAAC;IACjC,KAAK,CAAC,EAAE,MAAM,GAAG,IAAI,CAAC;IACtB,eAAe,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,GAAG,CAAC,GAAG,IAAI,CAAC;IAC7C,SAAS,CAAC,EAAE,MAAM,GAAG,IAAI,CAAC;IAC1B,QAAQ,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,GAAG,CAAC,CAAC;IAC/B,QAAQ,CAAC,EAAE,MAAM,CAAC;CACnB,QAgBA;AA4BD,eAAe,KAAK,CAAC"}

package/lib/tracing/tracer.js

@@ -0,0 +1,155 @@
+"use strict";
+// tracing/tracer.ts
+var __importDefault = (this && this.__importDefault) || function (mod) {
+    return (mod && mod.__esModule) ? mod : { "default": mod };
+};
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.addChatCompletionStepToTrace = void 0;
+const traces_1 = require("./traces.js");
+const steps_1 = require("./steps.js");
+const enums_1 = require("./enums.js");
+const index_1 = __importDefault(require("../../index.js"));
+let currentTrace = null;
+const publish = process.env['OPENLAYER_DISABLE_PUBLISH'] != 'true';
+let client = null;
+if (publish) {
+    console.debug('Publishing is enabled');
+    client = new index_1.default();
+}
+function getCurrentTrace() {
+    return currentTrace;
+}
+function setCurrentTrace(trace) {
+    currentTrace = trace;
+}
+// Function to create a new step
+const stepStack = [];
+function createStep(name, stepType = enums_1.StepType.USER_CALL, inputs, output, metadata = null) {
+    metadata = metadata || {};
+    let newStep;
+    if (stepType === enums_1.StepType.CHAT_COMPLETION) {
+        newStep = new steps_1.ChatCompletionStep(name, inputs, output, metadata);
+    }
+    else {
+        newStep = new steps_1.UserCallStep(name, inputs, output, metadata);
+    }
+    newStep.startTime = performance.now();
+    const parentStep = getCurrentStep();
+    const isRootStep = parentStep === null;
+    if (isRootStep) {
+        console.debug('Starting a new trace...');
+        console.debug(`Adding step ${name} as the root step`);
+        const currentTrace = new traces_1.Trace();
+        setCurrentTrace(currentTrace);
+        currentTrace.addStep(newStep);
+    }
+    else {
+        console.debug(`Adding step ${name} as a nested step to ${parentStep.name}`);
+        currentTrace = getCurrentTrace();
+        parentStep.addNestedStep(newStep);
+    }
+    stepStack.push(newStep);
+    const endStep = () => {
+        newStep.endTime = performance.now();
+        newStep.latency = newStep.endTime - newStep.startTime;
+        stepStack.pop(); // Remove the current step from the stack
+        if (isRootStep) {
+            console.debug('Ending the trace...');
+            const traceData = getCurrentTrace();
+            // Post process trace and get the input variable names
+            const { traceData: processedTraceData, inputVariableNames } = postProcessTrace(traceData);
+            if (publish && process.env['OPENLAYER_INFERENCE_PIPELINE_ID']) {
+                client.inferencePipelines.data.stream(process.env['OPENLAYER_INFERENCE_PIPELINE_ID'], {
+                    config: {
+                        outputColumnName: 'output',
+                        inputVariableNames: inputVariableNames,
+                        groundTruthColumnName: 'groundTruth',
+                        latencyColumnName: 'latency',
+                        costColumnName: 'cost',
+                        timestampColumnName: 'inferenceTimestamp',
+                        inferenceIdColumnName: 'inferenceId',
+                        numOfTokenColumnName: 'tokens',
+                    },
+                    rows: [processedTraceData],
+                });
+            }
+            // Reset the entire trace state
+            setCurrentTrace(null);
+            stepStack.length = 0; // Clear the step stack
+        }
+        else {
+            console.debug(`Ending step ${name}`);
+        }
+    };
+    return [newStep, endStep];
+}
+function getCurrentStep() {
+    const currentStep = stepStack.length > 0 ? stepStack[stepStack.length - 1] : null;
+    return currentStep;
+}
+function getParamNames(func) {
+    const STRIP_COMMENTS = /((\/\/.*$)|(\/\*[\s\S]*?\*\/))/gm;
+    const ARGUMENT_NAMES = /([^\s,]+)/g;
+    const fnStr = func.toString().replace(STRIP_COMMENTS, '');
+    const result = fnStr.slice(fnStr.indexOf('(') + 1, fnStr.indexOf(')')).match(ARGUMENT_NAMES);
+    return result || [];
+}
+// Higher-order function to trace synchronous or asynchronous functions
+function trace(fn, stepType = enums_1.StepType.USER_CALL, stepName) {
+    return async function (...args) {
+        const name = stepName || fn.name;
+        const paramNames = getParamNames(fn);
+        const inputs = Object.fromEntries(paramNames.map((name, index) => [name, args[index]]));
+        const [step, endStep] = createStep(name, stepType, args);
+        try {
+            const result = await fn(...args);
+            step.log({ inputs, output: result });
+            return result;
+        }
+        catch (error) {
+            step.log({ inputs, metadata: { error: error.message } });
+            throw error;
+        }
+        finally {
+            endStep();
+        }
+    };
+}
+function addChatCompletionStepToTrace({ name, inputs, output, latency, tokens = null, promptTokens = null, completionTokens = null, model = null, modelParameters = null, rawOutput = null, metadata = {}, provider = 'OpenAI', }) {
+    const [step, endStep] = createStep(name, enums_1.StepType.CHAT_COMPLETION, inputs, output, metadata);
+    if (step instanceof steps_1.ChatCompletionStep) {
+        step.provider = provider;
+        step.promptTokens = promptTokens;
+        step.completionTokens = completionTokens;
+        step.tokens = tokens;
+        step.model = model;
+        step.modelParameters = modelParameters;
+        step.rawOutput = rawOutput;
+        step.latency = latency;
+    }
+    step.log({ inputs, output, metadata });
+    endStep();
+}
+exports.addChatCompletionStepToTrace = addChatCompletionStepToTrace;
+function postProcessTrace(traceObj) {
+    const rootStep = traceObj.steps[0];
+    const input_variables = rootStep.inputs;
+    const inputVariableNames = input_variables ? Object.keys(input_variables) : [];
+    const processed_steps = traceObj.toJSON();
+    const traceData = {
+        inferenceTimestamp: Date.now(),
+        inferenceId: rootStep.id.toString(),
+        output: rootStep.output,
+        groundTruth: rootStep.groundTruth,
+        latency: rootStep.latency,
+        cost: 0,
+        tokens: 0,
+        steps: processed_steps,
+    };
+    if (input_variables) {
+        Object.assign(traceData, input_variables);
+    }
+    return { traceData, inputVariableNames };
+}
+exports.default = trace;
+//# sourceMappingURL=tracer.js.map

package/lib/tracing/tracer.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"tracer.js","sourceRoot":"","sources":["../../src/lib/tracing/tracer.ts"],"names":[],"mappings":";AAAA,oBAAoB;;;;;;AAEpB,wCAAiC;AACjC,sCAAiE;AACjE,sCAAmC;AACnC,2DAAoC;AAEpC,IAAI,YAAY,GAAiB,IAAI,CAAC;AAEtC,MAAM,OAAO,GAAG,OAAO,CAAC,GAAG,CAAC,2BAA2B,CAAC,IAAI,MAAM,CAAC;AACnE,IAAI,MAAM,GAAqB,IAAI,CAAC;AACpC,IAAI,OAAO,EAAE;IACX,OAAO,CAAC,KAAK,CAAC,uBAAuB,CAAC,CAAC;IACvC,MAAM,GAAG,IAAI,eAAS,EAAE,CAAC;CAC1B;AAED,SAAS,eAAe;IACtB,OAAO,YAAY,CAAC;AACtB,CAAC;AAED,SAAS,eAAe,CAAC,KAAmB;IAC1C,YAAY,GAAG,KAAK,CAAC;AACvB,CAAC;AAED,gCAAgC;AAChC,MAAM,SAAS,GAAW,EAAE,CAAC;AAE7B,SAAS,UAAU,CACjB,IAAY,EACZ,WAAqB,gBAAQ,CAAC,SAAS,EACvC,MAAY,EACZ,MAAY,EACZ,WAAuC,IAAI;IAE3C,QAAQ,GAAG,QAAQ,IAAI,EAAE,CAAC;IAC1B,IAAI,OAAa,CAAC;IAClB,IAAI,QAAQ,KAAK,gBAAQ,CAAC,eAAe,EAAE;QACzC,OAAO,GAAG,IAAI,0BAAkB,CAAC,IAAI,EAAE,MAAM,EAAE,MAAM,EAAE,QAAQ,CAAC,CAAC;KAClE;SAAM;QACL,OAAO,GAAG,IAAI,oBAAY,CAAC,IAAI,EAAE,MAAM,EAAE,MAAM,EAAE,QAAQ,CAAC,CAAC;KAC5D;IACD,OAAO,CAAC,SAAS,GAAG,WAAW,CAAC,GAAG,EAAE,CAAC;IAEtC,MAAM,UAAU,GAAG,cAAc,EAAE,CAAC;IACpC,MAAM,UAAU,GAAG,UAAU,KAAK,IAAI,CAAC;IAEvC,IAAI,UAAU,EAAE;QACd,OAAO,CAAC,KAAK,CAAC,yBAAyB,CAAC,CAAC;QACzC,OAAO,CAAC,KAAK,CAAC,eAAe,IAAI,mBAAmB,CAAC,CAAC;QACtD,MAAM,YAAY,GAAG,IAAI,cAAK,EAAE,CAAC;QACjC,eAAe,CAAC,YAAY,CAAC,CAAC;QAC9B,YAAY,CAAC,OAAO,CAAC,OAAO,CAAC,CAAC;KAC/B;SAAM;QACL,OAAO,CAAC,KAAK,CAAC,eAAe,IAAI,wBAAwB,UAAW,CAAC,IAAI,EAAE,CAAC,CAAC;QAC7E,YAAY,GAAG,eAAe,EAAG,CAAC;QAClC,UAAW,CAAC,aAAa,CAAC,OAAO,CAAC,CAAC;KACpC;IAED,SAAS,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;IAExB,MAAM,OAAO,GAAG,GAAG,EAAE;QACnB,OAAO,CAAC,OAAO,GAAG,WAAW,CAAC,GAAG,EAAE,CAAC;QACpC,OAAO,CAAC,OAAO,GAAG,OAAO,CAAC,OAAO,GAAG,OAAO,CAAC,SAAS,CAAC;QAEtD,SAAS,CAAC,GAAG,EAAE,CAAC,CAAC,yCAAyC;QAE1D,IAAI,UAAU,EAAE;YACd,OAAO,CAAC,KAAK,CAAC,qBAAqB,CAAC,CAAC;YACrC,MAAM,SAAS,GAAG,eAAe,EAAE,CAAC;YACpC,sDAAsD;YACtD,MAAM,EAAE,SAAS,EAAE,kBAAkB,EAAE,kBAAkB,EAAE,GAAG,gBAAgB,CAAC,SAAU,CAAC,CAAC;YAE3F,IAAI,OAAO,IAAI,OAAO,CAAC,GAAG,CAAC,iCAAiC,CAAC,EAAE;gBAC7D,MAAO,CAAC,kBAAkB,CAAC,IAAI,CAAC,MAAM,CAAC,OAAO,CAAC,GAAG,CAAC,iCAAiC,CAAC,EAAE;oBACrF,MAAM,EAAE;wBACN,gBAAgB,EAAE,QAAQ;wBAC1B,kBAAkB,EAAE,kBAAkB;wBACtC,qBAAqB,EAAE,aAAa;wBACpC,iBAAiB,EAAE,SAAS;wBAC5B,cAAc,EAAE,MAAM;wBACtB,mBAAmB,EAAE,oBAAoB;wBACzC,qBAAqB,EAAE,aAAa;wBACpC,oBAAoB,EAAE,QAAQ;qBAC/B;oBACD,IAAI,EAAE,CAAC,kBAAkB,CAAC;iBAC3B,CAAC,CAAC;aACJ;YAED,+BAA+B;YAC/B,eAAe,CAAC,IAAI,CAAC,CAAC;YACtB,SAAS,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC,uBAAuB;SAC9C;aAAM;YACL,OAAO,CAAC,KAAK,CAAC,eAAe,IAAI,EAAE,CAAC,CAAC;SACtC;IACH,CAAC,CAAC;IAEF,OAAO,CAAC,OAAO,EAAE,OAAO,CAAC,CAAC;AAC5B,CAAC;AAED,SAAS,cAAc;IACrB,MAAM,WAAW,GAAG,SAAS,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC,CAAC,SAAS,CAAC,SAAS,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC;IAClF,OAAO,WAAW,CAAC;AACrB,CAAC;AAED,SAAS,aAAa,CAAC,IAAc;IACnC,MAAM,cAAc,GAAG,kCAAkC,CAAC;IAC1D,MAAM,cAAc,GAAG,YAAY,CAAC;IACpC,MAAM,KAAK,GAAG,IAAI,CAAC,QAAQ,EAAE,CAAC,OAAO,CAAC,cAAc,EAAE,EAAE,CAAC,CAAC;IAC1D,MAAM,MAAM,GAAG,KAAK,CAAC,KAAK,CAAC,KAAK,CAAC,OAAO,CAAC,GAAG,CAAC,GAAG,CAAC,EAAE,KAAK,CAAC,OAAO,CAAC,GAAG,CAAC,CAAC,CAAC,KAAK,CAAC,cAAc,CAAC,CAAC;IAC7F,OAAO,MAAM,IAAI,EAAE,CAAC;AACtB,CAAC;AAED,uEAAuE;AACvE,SAAS,KAAK,CAAC,EAAY,EAAE,WAAqB,gBAAQ,CAAC,SAAS,EAAE,QAAiB;IACrF,OAAO,KAAK,WAAW,GAAG,IAAW;QACnC,MAAM,IAAI,GAAG,QAAQ,IAAI,EAAE,CAAC,IAAI,CAAC;QACjC,MAAM,UAAU,GAAG,aAAa,CAAC,EAAE,CAAC,CAAC;QACrC,MAAM,MAAM,GAAG,MAAM,CAAC,WAAW,CAAC,UAAU,CAAC,GAAG,CAAC,CAAC,IAAI,EAAE,KAAK,EAAE,EAAE,CAAC,CAAC,IAAI,EAAE,IAAI,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,CAAC;QACxF,MAAM,CAAC,IAAI,EAAE,OAAO,CAAC,GAAG,UAAU,CAAC,IAAI,EAAE,QAAQ,EAAE,IAAI,CAAC,CAAC;QAEzD,IAAI;YACF,MAAM,MAAM,GAAG,MAAM,EAAE,CAAC,GAAG,IAAI,CAAC,CAAC;YACjC,IAAI,CAAC,GAAG,CAAC,EAAE,MAAM,EAAE,MAAM,EAAE,MAAM,EAAE,CAAC,CAAC;YACrC,OAAO,MAAM,CAAC;SACf;QAAC,OAAO,KAAU,EAAE;YACnB,IAAI,CAAC,GAAG,CAAC,EAAE,MAAM,EAAE,QAAQ,EAAE,EAAE,KAAK,EAAE,KAAK,CAAC,OAAO,EAAE,EAAE,CAAC,CAAC;YACzD,MAAM,KAAK,CAAC;SACb;gBAAS;YACR,OAAO,EAAE,CAAC;SACX;IACH,CAAC,CAAC;AACJ,CAAC;AAED,SAAgB,4BAA4B,CAAC,EAC3C,IAAI,EACJ,MAAM,EACN,MAAM,EACN,OAAO,EACP,MAAM,GAAG,IAAI,EACb,YAAY,GAAG,IAAI,EACnB,gBAAgB,GAAG,IAAI,EACvB,KAAK,GAAG,IAAI,EACZ,eAAe,GAAG,IAAI,EACtB,SAAS,GAAG,IAAI,EAChB,QAAQ,GAAG,EAAE,EACb,QAAQ,GAAG,QAAQ,GAcpB;IACC,MAAM,CAAC,IAAI,EAAE,OAAO,CAAC,GAAG,UAAU,CAAC,IAAI,EAAE,gBAAQ,CAAC,eAAe,EAAE,MAAM,EAAE,MAAM,EAAE,QAAQ,CAAC,CAAC;IAE7F,IAAI,IAAI,YAAY,0BAAkB,EAAE;QACtC,IAAI,CAAC,QAAQ,GAAG,QAAQ,CAAC;QACzB,IAAI,CAAC,YAAY,GAAG,YAAY,CAAC;QACjC,IAAI,CAAC,gBAAgB,GAAG,gBAAgB,CAAC;QACzC,IAAI,CAAC,MAAM,GAAG,MAAM,CAAC;QACrB,IAAI,CAAC,KAAK,GAAG,KAAK,CAAC;QACnB,IAAI,CAAC,eAAe,GAAG,eAAe,CAAC;QACvC,IAAI,CAAC,SAAS,GAAG,SAAS,CAAC;QAC3B,IAAI,CAAC,OAAO,GAAG,OAAO,CAAC;KACxB;IAED,IAAI,CAAC,GAAG,CAAC,EAAE,MAAM,EAAE,MAAM,EAAE,QAAQ,EAAE,CAAC,CAAC;IACvC,OAAO,EAAE,CAAC;AACZ,CAAC;AA1CD,oEA0CC;AAED,SAAS,gBAAgB,CAAC,QAAe;IACvC,MAAM,QAAQ,GAAG,QAAQ,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC;IAEnC,MAAM,eAAe,GAAG,QAAS,CAAC,MAAM,CAAC;IACzC,MAAM,kBAAkB,GAAG,eAAe,CAAC,CAAC,CAAC,MAAM,CAAC,IAAI,CAAC,eAAe,CAAC,CAAC,CAAC,CAAC,EAAE,CAAC;IAE/E,MAAM,eAAe,GAAG,QAAQ,CAAC,MAAM,EAAE,CAAC;IAE1C,MAAM,SAAS,GAAG;QAChB,kBAAkB,EAAE,IAAI,CAAC,GAAG,EAAE;QAC9B,WAAW,EAAE,QAAS,CAAC,EAAE,CAAC,QAAQ,EAAE;QACpC,MAAM,EAAE,QAAS,CAAC,MAAM;QACxB,WAAW,EAAE,QAAS,CAAC,WAAW;QAClC,OAAO,EAAE,QAAS,CAAC,OAAO;QAC1B,IAAI,EAAE,CAAC;QACP,MAAM,EAAE,CAAC;QACT,KAAK,EAAE,eAAe;KACvB,CAAC;IAEF,IAAI,eAAe,EAAE;QACnB,MAAM,CAAC,MAAM,CAAC,SAAS,EAAE,eAAe,CAAC,CAAC;KAC3C;IAED,OAAO,EAAE,SAAS,EAAE,kBAAkB,EAAE,CAAC;AAC3C,CAAC;AAED,kBAAe,KAAK,CAAC"}

package/lib/tracing/tracer.mjs

@@ -0,0 +1,148 @@
+// tracing/tracer.ts
+import { Trace } from "./traces.mjs";
+import { UserCallStep, ChatCompletionStep } from "./steps.mjs";
+import { StepType } from "./enums.mjs";
+import Openlayer from "../../index.mjs";
+let currentTrace = null;
+const publish = process.env['OPENLAYER_DISABLE_PUBLISH'] != 'true';
+let client = null;
+if (publish) {
+    console.debug('Publishing is enabled');
+    client = new Openlayer();
+}
+function getCurrentTrace() {
+    return currentTrace;
+}
+function setCurrentTrace(trace) {
+    currentTrace = trace;
+}
+// Function to create a new step
+const stepStack = [];
+function createStep(name, stepType = StepType.USER_CALL, inputs, output, metadata = null) {
+    metadata = metadata || {};
+    let newStep;
+    if (stepType === StepType.CHAT_COMPLETION) {
+        newStep = new ChatCompletionStep(name, inputs, output, metadata);
+    }
+    else {
+        newStep = new UserCallStep(name, inputs, output, metadata);
+    }
+    newStep.startTime = performance.now();
+    const parentStep = getCurrentStep();
+    const isRootStep = parentStep === null;
+    if (isRootStep) {
+        console.debug('Starting a new trace...');
+        console.debug(`Adding step ${name} as the root step`);
+        const currentTrace = new Trace();
+        setCurrentTrace(currentTrace);
+        currentTrace.addStep(newStep);
+    }
+    else {
+        console.debug(`Adding step ${name} as a nested step to ${parentStep.name}`);
+        currentTrace = getCurrentTrace();
+        parentStep.addNestedStep(newStep);
+    }
+    stepStack.push(newStep);
+    const endStep = () => {
+        newStep.endTime = performance.now();
+        newStep.latency = newStep.endTime - newStep.startTime;
+        stepStack.pop(); // Remove the current step from the stack
+        if (isRootStep) {
+            console.debug('Ending the trace...');
+            const traceData = getCurrentTrace();
+            // Post process trace and get the input variable names
+            const { traceData: processedTraceData, inputVariableNames } = postProcessTrace(traceData);
+            if (publish && process.env['OPENLAYER_INFERENCE_PIPELINE_ID']) {
+                client.inferencePipelines.data.stream(process.env['OPENLAYER_INFERENCE_PIPELINE_ID'], {
+                    config: {
+                        outputColumnName: 'output',
+                        inputVariableNames: inputVariableNames,
+                        groundTruthColumnName: 'groundTruth',
+                        latencyColumnName: 'latency',
+                        costColumnName: 'cost',
+                        timestampColumnName: 'inferenceTimestamp',
+                        inferenceIdColumnName: 'inferenceId',
+                        numOfTokenColumnName: 'tokens',
+                    },
+                    rows: [processedTraceData],
+                });
+            }
+            // Reset the entire trace state
+            setCurrentTrace(null);
+            stepStack.length = 0; // Clear the step stack
+        }
+        else {
+            console.debug(`Ending step ${name}`);
+        }
+    };
+    return [newStep, endStep];
+}
+function getCurrentStep() {
+    const currentStep = stepStack.length > 0 ? stepStack[stepStack.length - 1] : null;
+    return currentStep;
+}
+function getParamNames(func) {
+    const STRIP_COMMENTS = /((\/\/.*$)|(\/\*[\s\S]*?\*\/))/gm;
+    const ARGUMENT_NAMES = /([^\s,]+)/g;
+    const fnStr = func.toString().replace(STRIP_COMMENTS, '');
+    const result = fnStr.slice(fnStr.indexOf('(') + 1, fnStr.indexOf(')')).match(ARGUMENT_NAMES);
+    return result || [];
+}
+// Higher-order function to trace synchronous or asynchronous functions
+function trace(fn, stepType = StepType.USER_CALL, stepName) {
+    return async function (...args) {
+        const name = stepName || fn.name;
+        const paramNames = getParamNames(fn);
+        const inputs = Object.fromEntries(paramNames.map((name, index) => [name, args[index]]));
+        const [step, endStep] = createStep(name, stepType, args);
+        try {
+            const result = await fn(...args);
+            step.log({ inputs, output: result });
+            return result;
+        }
+        catch (error) {
+            step.log({ inputs, metadata: { error: error.message } });
+            throw error;
+        }
+        finally {
+            endStep();
+        }
+    };
+}
+export function addChatCompletionStepToTrace({ name, inputs, output, latency, tokens = null, promptTokens = null, completionTokens = null, model = null, modelParameters = null, rawOutput = null, metadata = {}, provider = 'OpenAI', }) {
+    const [step, endStep] = createStep(name, StepType.CHAT_COMPLETION, inputs, output, metadata);
+    if (step instanceof ChatCompletionStep) {
+        step.provider = provider;
+        step.promptTokens = promptTokens;
+        step.completionTokens = completionTokens;
+        step.tokens = tokens;
+        step.model = model;
+        step.modelParameters = modelParameters;
+        step.rawOutput = rawOutput;
+        step.latency = latency;
+    }
+    step.log({ inputs, output, metadata });
+    endStep();
+}
+function postProcessTrace(traceObj) {
+    const rootStep = traceObj.steps[0];
+    const input_variables = rootStep.inputs;
+    const inputVariableNames = input_variables ? Object.keys(input_variables) : [];
+    const processed_steps = traceObj.toJSON();
+    const traceData = {
+        inferenceTimestamp: Date.now(),
+        inferenceId: rootStep.id.toString(),
+        output: rootStep.output,
+        groundTruth: rootStep.groundTruth,
+        latency: rootStep.latency,
+        cost: 0,
+        tokens: 0,
+        steps: processed_steps,
+    };
+    if (input_variables) {
+        Object.assign(traceData, input_variables);
+    }
+    return { traceData, inputVariableNames };
+}
+export default trace;
+//# sourceMappingURL=tracer.mjs.map

package/lib/tracing/tracer.mjs.map

@@ -0,0 +1 @@
+{"version":3,"file":"tracer.mjs","sourceRoot":"","sources":["../../src/lib/tracing/tracer.ts"],"names":[],"mappings":"AAAA,oBAAoB;OAEb,EAAE,KAAK,EAAE;OACT,EAAQ,YAAY,EAAE,kBAAkB,EAAE;OAC1C,EAAE,QAAQ,EAAE;OACZ,SAAS;AAEhB,IAAI,YAAY,GAAiB,IAAI,CAAC;AAEtC,MAAM,OAAO,GAAG,OAAO,CAAC,GAAG,CAAC,2BAA2B,CAAC,IAAI,MAAM,CAAC;AACnE,IAAI,MAAM,GAAqB,IAAI,CAAC;AACpC,IAAI,OAAO,EAAE;IACX,OAAO,CAAC,KAAK,CAAC,uBAAuB,CAAC,CAAC;IACvC,MAAM,GAAG,IAAI,SAAS,EAAE,CAAC;CAC1B;AAED,SAAS,eAAe;IACtB,OAAO,YAAY,CAAC;AACtB,CAAC;AAED,SAAS,eAAe,CAAC,KAAmB;IAC1C,YAAY,GAAG,KAAK,CAAC;AACvB,CAAC;AAED,gCAAgC;AAChC,MAAM,SAAS,GAAW,EAAE,CAAC;AAE7B,SAAS,UAAU,CACjB,IAAY,EACZ,WAAqB,QAAQ,CAAC,SAAS,EACvC,MAAY,EACZ,MAAY,EACZ,WAAuC,IAAI;IAE3C,QAAQ,GAAG,QAAQ,IAAI,EAAE,CAAC;IAC1B,IAAI,OAAa,CAAC;IAClB,IAAI,QAAQ,KAAK,QAAQ,CAAC,eAAe,EAAE;QACzC,OAAO,GAAG,IAAI,kBAAkB,CAAC,IAAI,EAAE,MAAM,EAAE,MAAM,EAAE,QAAQ,CAAC,CAAC;KAClE;SAAM;QACL,OAAO,GAAG,IAAI,YAAY,CAAC,IAAI,EAAE,MAAM,EAAE,MAAM,EAAE,QAAQ,CAAC,CAAC;KAC5D;IACD,OAAO,CAAC,SAAS,GAAG,WAAW,CAAC,GAAG,EAAE,CAAC;IAEtC,MAAM,UAAU,GAAG,cAAc,EAAE,CAAC;IACpC,MAAM,UAAU,GAAG,UAAU,KAAK,IAAI,CAAC;IAEvC,IAAI,UAAU,EAAE;QACd,OAAO,CAAC,KAAK,CAAC,yBAAyB,CAAC,CAAC;QACzC,OAAO,CAAC,KAAK,CAAC,eAAe,IAAI,mBAAmB,CAAC,CAAC;QACtD,MAAM,YAAY,GAAG,IAAI,KAAK,EAAE,CAAC;QACjC,eAAe,CAAC,YAAY,CAAC,CAAC;QAC9B,YAAY,CAAC,OAAO,CAAC,OAAO,CAAC,CAAC;KAC/B;SAAM;QACL,OAAO,CAAC,KAAK,CAAC,eAAe,IAAI,wBAAwB,UAAW,CAAC,IAAI,EAAE,CAAC,CAAC;QAC7E,YAAY,GAAG,eAAe,EAAG,CAAC;QAClC,UAAW,CAAC,aAAa,CAAC,OAAO,CAAC,CAAC;KACpC;IAED,SAAS,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;IAExB,MAAM,OAAO,GAAG,GAAG,EAAE;QACnB,OAAO,CAAC,OAAO,GAAG,WAAW,CAAC,GAAG,EAAE,CAAC;QACpC,OAAO,CAAC,OAAO,GAAG,OAAO,CAAC,OAAO,GAAG,OAAO,CAAC,SAAS,CAAC;QAEtD,SAAS,CAAC,GAAG,EAAE,CAAC,CAAC,yCAAyC;QAE1D,IAAI,UAAU,EAAE;YACd,OAAO,CAAC,KAAK,CAAC,qBAAqB,CAAC,CAAC;YACrC,MAAM,SAAS,GAAG,eAAe,EAAE,CAAC;YACpC,sDAAsD;YACtD,MAAM,EAAE,SAAS,EAAE,kBAAkB,EAAE,kBAAkB,EAAE,GAAG,gBAAgB,CAAC,SAAU,CAAC,CAAC;YAE3F,IAAI,OAAO,IAAI,OAAO,CAAC,GAAG,CAAC,iCAAiC,CAAC,EAAE;gBAC7D,MAAO,CAAC,kBAAkB,CAAC,IAAI,CAAC,MAAM,CAAC,OAAO,CAAC,GAAG,CAAC,iCAAiC,CAAC,EAAE;oBACrF,MAAM,EAAE;wBACN,gBAAgB,EAAE,QAAQ;wBAC1B,kBAAkB,EAAE,kBAAkB;wBACtC,qBAAqB,EAAE,aAAa;wBACpC,iBAAiB,EAAE,SAAS;wBAC5B,cAAc,EAAE,MAAM;wBACtB,mBAAmB,EAAE,oBAAoB;wBACzC,qBAAqB,EAAE,aAAa;wBACpC,oBAAoB,EAAE,QAAQ;qBAC/B;oBACD,IAAI,EAAE,CAAC,kBAAkB,CAAC;iBAC3B,CAAC,CAAC;aACJ;YAED,+BAA+B;YAC/B,eAAe,CAAC,IAAI,CAAC,CAAC;YACtB,SAAS,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC,uBAAuB;SAC9C;aAAM;YACL,OAAO,CAAC,KAAK,CAAC,eAAe,IAAI,EAAE,CAAC,CAAC;SACtC;IACH,CAAC,CAAC;IAEF,OAAO,CAAC,OAAO,EAAE,OAAO,CAAC,CAAC;AAC5B,CAAC;AAED,SAAS,cAAc;IACrB,MAAM,WAAW,GAAG,SAAS,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC,CAAC,SAAS,CAAC,SAAS,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC;IAClF,OAAO,WAAW,CAAC;AACrB,CAAC;AAED,SAAS,aAAa,CAAC,IAAc;IACnC,MAAM,cAAc,GAAG,kCAAkC,CAAC;IAC1D,MAAM,cAAc,GAAG,YAAY,CAAC;IACpC,MAAM,KAAK,GAAG,IAAI,CAAC,QAAQ,EAAE,CAAC,OAAO,CAAC,cAAc,EAAE,EAAE,CAAC,CAAC;IAC1D,MAAM,MAAM,GAAG,KAAK,CAAC,KAAK,CAAC,KAAK,CAAC,OAAO,CAAC,GAAG,CAAC,GAAG,CAAC,EAAE,KAAK,CAAC,OAAO,CAAC,GAAG,CAAC,CAAC,CAAC,KAAK,CAAC,cAAc,CAAC,CAAC;IAC7F,OAAO,MAAM,IAAI,EAAE,CAAC;AACtB,CAAC;AAED,uEAAuE;AACvE,SAAS,KAAK,CAAC,EAAY,EAAE,WAAqB,QAAQ,CAAC,SAAS,EAAE,QAAiB;IACrF,OAAO,KAAK,WAAW,GAAG,IAAW;QACnC,MAAM,IAAI,GAAG,QAAQ,IAAI,EAAE,CAAC,IAAI,CAAC;QACjC,MAAM,UAAU,GAAG,aAAa,CAAC,EAAE,CAAC,CAAC;QACrC,MAAM,MAAM,GAAG,MAAM,CAAC,WAAW,CAAC,UAAU,CAAC,GAAG,CAAC,CAAC,IAAI,EAAE,KAAK,EAAE,EAAE,CAAC,CAAC,IAAI,EAAE,IAAI,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,CAAC;QACxF,MAAM,CAAC,IAAI,EAAE,OAAO,CAAC,GAAG,UAAU,CAAC,IAAI,EAAE,QAAQ,EAAE,IAAI,CAAC,CAAC;QAEzD,IAAI;YACF,MAAM,MAAM,GAAG,MAAM,EAAE,CAAC,GAAG,IAAI,CAAC,CAAC;YACjC,IAAI,CAAC,GAAG,CAAC,EAAE,MAAM,EAAE,MAAM,EAAE,MAAM,EAAE,CAAC,CAAC;YACrC,OAAO,MAAM,CAAC;SACf;QAAC,OAAO,KAAU,EAAE;YACnB,IAAI,CAAC,GAAG,CAAC,EAAE,MAAM,EAAE,QAAQ,EAAE,EAAE,KAAK,EAAE,KAAK,CAAC,OAAO,EAAE,EAAE,CAAC,CAAC;YACzD,MAAM,KAAK,CAAC;SACb;gBAAS;YACR,OAAO,EAAE,CAAC;SACX;IACH,CAAC,CAAC;AACJ,CAAC;AAED,MAAM,UAAU,4BAA4B,CAAC,EAC3C,IAAI,EACJ,MAAM,EACN,MAAM,EACN,OAAO,EACP,MAAM,GAAG,IAAI,EACb,YAAY,GAAG,IAAI,EACnB,gBAAgB,GAAG,IAAI,EACvB,KAAK,GAAG,IAAI,EACZ,eAAe,GAAG,IAAI,EACtB,SAAS,GAAG,IAAI,EAChB,QAAQ,GAAG,EAAE,EACb,QAAQ,GAAG,QAAQ,GAcpB;IACC,MAAM,CAAC,IAAI,EAAE,OAAO,CAAC,GAAG,UAAU,CAAC,IAAI,EAAE,QAAQ,CAAC,eAAe,EAAE,MAAM,EAAE,MAAM,EAAE,QAAQ,CAAC,CAAC;IAE7F,IAAI,IAAI,YAAY,kBAAkB,EAAE;QACtC,IAAI,CAAC,QAAQ,GAAG,QAAQ,CAAC;QACzB,IAAI,CAAC,YAAY,GAAG,YAAY,CAAC;QACjC,IAAI,CAAC,gBAAgB,GAAG,gBAAgB,CAAC;QACzC,IAAI,CAAC,MAAM,GAAG,MAAM,CAAC;QACrB,IAAI,CAAC,KAAK,GAAG,KAAK,CAAC;QACnB,IAAI,CAAC,eAAe,GAAG,eAAe,CAAC;QACvC,IAAI,CAAC,SAAS,GAAG,SAAS,CAAC;QAC3B,IAAI,CAAC,OAAO,GAAG,OAAO,CAAC;KACxB;IAED,IAAI,CAAC,GAAG,CAAC,EAAE,MAAM,EAAE,MAAM,EAAE,QAAQ,EAAE,CAAC,CAAC;IACvC,OAAO,EAAE,CAAC;AACZ,CAAC;AAED,SAAS,gBAAgB,CAAC,QAAe;IACvC,MAAM,QAAQ,GAAG,QAAQ,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC;IAEnC,MAAM,eAAe,GAAG,QAAS,CAAC,MAAM,CAAC;IACzC,MAAM,kBAAkB,GAAG,eAAe,CAAC,CAAC,CAAC,MAAM,CAAC,IAAI,CAAC,eAAe,CAAC,CAAC,CAAC,CAAC,EAAE,CAAC;IAE/E,MAAM,eAAe,GAAG,QAAQ,CAAC,MAAM,EAAE,CAAC;IAE1C,MAAM,SAAS,GAAG;QAChB,kBAAkB,EAAE,IAAI,CAAC,GAAG,EAAE;QAC9B,WAAW,EAAE,QAAS,CAAC,EAAE,CAAC,QAAQ,EAAE;QACpC,MAAM,EAAE,QAAS,CAAC,MAAM;QACxB,WAAW,EAAE,QAAS,CAAC,WAAW;QAClC,OAAO,EAAE,QAAS,CAAC,OAAO;QAC1B,IAAI,EAAE,CAAC;QACP,MAAM,EAAE,CAAC;QACT,KAAK,EAAE,eAAe;KACvB,CAAC;IAEF,IAAI,eAAe,EAAE;QACnB,MAAM,CAAC,MAAM,CAAC,SAAS,EAAE,eAAe,CAAC,CAAC;KAC3C;IAED,OAAO,EAAE,SAAS,EAAE,kBAAkB,EAAE,CAAC;AAC3C,CAAC;AAED,eAAe,KAAK,CAAC"}

package/lib/tracing/traces.d.ts

@@ -0,0 +1,9 @@
+import { Step } from "./steps.js";
+export declare class Trace {
+    steps: Step[];
+    private currentStep;
+    constructor();
+    addStep(step: Step): void;
+    toJSON(): Array<Record<string, any>>;
+}
+//# sourceMappingURL=traces.d.ts.map

package/lib/tracing/traces.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"traces.d.ts","sourceRoot":"","sources":["../../src/lib/tracing/traces.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,IAAI,EAAE,MAAM,SAAS,CAAC;AAE/B,qBAAa,KAAK;IACT,KAAK,EAAE,IAAI,EAAE,CAAC;IACrB,OAAO,CAAC,WAAW,CAAc;;IAO1B,OAAO,CAAC,IAAI,EAAE,IAAI,GAAG,IAAI;IAIzB,MAAM,IAAI,KAAK,CAAC,MAAM,CAAC,MAAM,EAAE,GAAG,CAAC,CAAC;CAG5C"}

package/lib/tracing/traces.js

@@ -0,0 +1,17 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.Trace = void 0;
+class Trace {
+    constructor() {
+        this.steps = [];
+        this.currentStep = null;
+    }
+    addStep(step) {
+        this.steps.push(step);
+    }
+    toJSON() {
+        return this.steps.map((step) => step.toJSON());
+    }
+}
+exports.Trace = Trace;
+//# sourceMappingURL=traces.js.map

package/lib/tracing/traces.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"traces.js","sourceRoot":"","sources":["../../src/lib/tracing/traces.ts"],"names":[],"mappings":";;;AAEA,MAAa,KAAK;IAIhB;QACE,IAAI,CAAC,KAAK,GAAG,EAAE,CAAC;QAChB,IAAI,CAAC,WAAW,GAAG,IAAI,CAAC;IAC1B,CAAC;IAEM,OAAO,CAAC,IAAU;QACvB,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;IACxB,CAAC;IAEM,MAAM;QACX,OAAO,IAAI,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,CAAC,MAAM,EAAE,CAAC,CAAC;IACjD,CAAC;CACF;AAhBD,sBAgBC"}

package/lib/tracing/traces.mjs

@@ -0,0 +1,13 @@
+export class Trace {
+    constructor() {
+        this.steps = [];
+        this.currentStep = null;
+    }
+    addStep(step) {
+        this.steps.push(step);
+    }
+    toJSON() {
+        return this.steps.map((step) => step.toJSON());
+    }
+}
+//# sourceMappingURL=traces.mjs.map

package/lib/tracing/traces.mjs.map

@@ -0,0 +1 @@
+{"version":3,"file":"traces.mjs","sourceRoot":"","sources":["../../src/lib/tracing/traces.ts"],"names":[],"mappings":"AAEA,MAAM,OAAO,KAAK;IAIhB;QACE,IAAI,CAAC,KAAK,GAAG,EAAE,CAAC;QAChB,IAAI,CAAC,WAAW,GAAG,IAAI,CAAC;IAC1B,CAAC;IAEM,OAAO,CAAC,IAAU;QACvB,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;IACxB,CAAC;IAEM,MAAM;QACX,OAAO,IAAI,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,CAAC,MAAM,EAAE,CAAC,CAAC;IACjD,CAAC;CACF"}

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "openlayer",
-  "version": "0.4.0",
+  "version": "0.7.0",
   "description": "The official TypeScript library for the Openlayer API",
   "author": "Openlayer <support@openlayer.com>",
   "types": "./index.d.ts",
@@ -22,6 +22,7 @@
     "fix": "eslint --fix --ext ts,js ."
   },
   "dependencies": {
+    "@langchain/core": "^0.3.3",
     "@types/node": "^18.11.18",
     "@types/node-fetch": "^2.6.4",
     "abort-controller": "^3.0.0",

package/resources/index.d.ts

@@ -1,4 +1,5 @@
 export { Commits } from "./commits/commits.js";
-export { InferencePipelines } from "./inference-pipelines/inference-pipelines.js";
+export { InferencePipelineRetrieveResponse, InferencePipelineUpdateResponse, InferencePipelineUpdateParams, InferencePipelines, } from "./inference-pipelines/inference-pipelines.js";
 export { ProjectCreateResponse, ProjectListResponse, ProjectCreateParams, ProjectListParams, Projects, } from "./projects/projects.js";
+export { Storage } from "./storage/storage.js";
 //# sourceMappingURL=index.d.ts.map

package/resources/index.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../src/resources/index.ts"],"names":[],"mappings":"AAEA,OAAO,EAAE,OAAO,EAAE,MAAM,mBAAmB,CAAC;AAC5C,OAAO,EAAE,kBAAkB,EAAE,MAAM,2CAA2C,CAAC;AAC/E,OAAO,EACL,qBAAqB,EACrB,mBAAmB,EACnB,mBAAmB,EACnB,iBAAiB,EACjB,QAAQ,GACT,MAAM,qBAAqB,CAAC"}
+{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../src/resources/index.ts"],"names":[],"mappings":"AAEA,OAAO,EAAE,OAAO,EAAE,MAAM,mBAAmB,CAAC;AAC5C,OAAO,EACL,iCAAiC,EACjC,+BAA+B,EAC/B,6BAA6B,EAC7B,kBAAkB,GACnB,MAAM,2CAA2C,CAAC;AACnD,OAAO,EACL,qBAAqB,EACrB,mBAAmB,EACnB,mBAAmB,EACnB,iBAAiB,EACjB,QAAQ,GACT,MAAM,qBAAqB,CAAC;AAC7B,OAAO,EAAE,OAAO,EAAE,MAAM,mBAAmB,CAAC"}

package/resources/index.js

@@ -1,11 +1,13 @@
 "use strict";
 // File generated from our OpenAPI spec by Stainless. See CONTRIBUTING.md for details.
 Object.defineProperty(exports, "__esModule", { value: true });
-exports.Projects = exports.InferencePipelines = exports.Commits = void 0;
+exports.Storage = exports.Projects = exports.InferencePipelines = exports.Commits = void 0;
 var commits_1 = require("./commits/commits.js");
 Object.defineProperty(exports, "Commits", { enumerable: true, get: function () { return commits_1.Commits; } });
 var inference_pipelines_1 = require("./inference-pipelines/inference-pipelines.js");
 Object.defineProperty(exports, "InferencePipelines", { enumerable: true, get: function () { return inference_pipelines_1.InferencePipelines; } });
 var projects_1 = require("./projects/projects.js");
 Object.defineProperty(exports, "Projects", { enumerable: true, get: function () { return projects_1.Projects; } });
+var storage_1 = require("./storage/storage.js");
+Object.defineProperty(exports, "Storage", { enumerable: true, get: function () { return storage_1.Storage; } });
 //# sourceMappingURL=index.js.map

package/resources/index.js.map

@@ -1 +1 @@
-{"version":3,"file":"index.js","sourceRoot":"","sources":["../src/resources/index.ts"],"names":[],"mappings":";AAAA,sFAAsF;;;AAEtF,gDAA4C;AAAnC,kGAAA,OAAO,OAAA;AAChB,oFAA+E;AAAtE,yHAAA,kBAAkB,OAAA;AAC3B,mDAM6B;AAD3B,oGAAA,QAAQ,OAAA"}
+{"version":3,"file":"index.js","sourceRoot":"","sources":["../src/resources/index.ts"],"names":[],"mappings":";AAAA,sFAAsF;;;AAEtF,gDAA4C;AAAnC,kGAAA,OAAO,OAAA;AAChB,oFAKmD;AADjD,yHAAA,kBAAkB,OAAA;AAEpB,mDAM6B;AAD3B,oGAAA,QAAQ,OAAA;AAEV,gDAA4C;AAAnC,kGAAA,OAAO,OAAA"}

package/resources/index.mjs

@@ -1,5 +1,6 @@
 // File generated from our OpenAPI spec by Stainless. See CONTRIBUTING.md for details.
 export { Commits } from "./commits/commits.mjs";
-export { InferencePipelines } from "./inference-pipelines/inference-pipelines.mjs";
+export { InferencePipelines, } from "./inference-pipelines/inference-pipelines.mjs";
 export { Projects, } from "./projects/projects.mjs";
+export { Storage } from "./storage/storage.mjs";
 //# sourceMappingURL=index.mjs.map

package/resources/index.mjs.map

@@ -1 +1 @@
-{"version":3,"file":"index.mjs","sourceRoot":"","sources":["../src/resources/index.ts"],"names":[],"mappings":"AAAA,sFAAsF;OAE/E,EAAE,OAAO,EAAE;OACX,EAAE,kBAAkB,EAAE;OACtB,EAKL,QAAQ,GACT"}
+{"version":3,"file":"index.mjs","sourceRoot":"","sources":["../src/resources/index.ts"],"names":[],"mappings":"AAAA,sFAAsF;OAE/E,EAAE,OAAO,EAAE;OACX,EAIL,kBAAkB,GACnB;OACM,EAKL,QAAQ,GACT;OACM,EAAE,OAAO,EAAE"}

package/resources/inference-pipelines/data.d.ts

@@ -3,7 +3,7 @@ import * as Core from "../../core.js";
 import * as DataAPI from "./data.js";
 export declare class Data extends APIResource {
     /**
-     * Create an inference data point in an inference pipeline.
+     * Publish an inference data point to an inference pipeline.
      */
     stream(inferencePipelineId: string, body: DataStreamParams, options?: Core.RequestOptions): Core.APIPromise<DataStreamResponse>;
 }

package/resources/inference-pipelines/data.js

@@ -5,7 +5,7 @@ exports.Data = void 0;
 const resource_1 = require("../../resource.js");
 class Data extends resource_1.APIResource {
     /**
-     * Create an inference data point in an inference pipeline.
+     * Publish an inference data point to an inference pipeline.
      */
     stream(inferencePipelineId, body, options) {
         return this._client.post(`/inference-pipelines/${inferencePipelineId}/data-stream`, { body, ...options });

package/resources/inference-pipelines/data.mjs

@@ -2,7 +2,7 @@
 import { APIResource } from "../../resource.mjs";
 export class Data extends APIResource {
     /**
-     * Create an inference data point in an inference pipeline.
+     * Publish an inference data point to an inference pipeline.
      */
     stream(inferencePipelineId, body, options) {
         return this._client.post(`/inference-pipelines/${inferencePipelineId}/data-stream`, { body, ...options });

package/resources/inference-pipelines/index.d.ts

@@ -1,5 +1,5 @@
 export { DataStreamResponse, DataStreamParams, Data } from "./data.js";
-export { InferencePipelines } from "./inference-pipelines.js";
+export { InferencePipelineRetrieveResponse, InferencePipelineUpdateResponse, InferencePipelineUpdateParams, InferencePipelines, } from "./inference-pipelines.js";
 export { RowUpdateResponse, RowUpdateParams, Rows } from "./rows.js";
 export { TestResultListResponse, TestResultListParams, TestResults } from "./test-results.js";
 //# sourceMappingURL=index.d.ts.map

package/resources/inference-pipelines/index.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../../src/resources/inference-pipelines/index.ts"],"names":[],"mappings":"AAEA,OAAO,EAAE,kBAAkB,EAAE,gBAAgB,EAAE,IAAI,EAAE,MAAM,QAAQ,CAAC;AACpE,OAAO,EAAE,kBAAkB,EAAE,MAAM,uBAAuB,CAAC;AAC3D,OAAO,EAAE,iBAAiB,EAAE,eAAe,EAAE,IAAI,EAAE,MAAM,QAAQ,CAAC;AAClE,OAAO,EAAE,sBAAsB,EAAE,oBAAoB,EAAE,WAAW,EAAE,MAAM,gBAAgB,CAAC"}
+{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../../src/resources/inference-pipelines/index.ts"],"names":[],"mappings":"AAEA,OAAO,EAAE,kBAAkB,EAAE,gBAAgB,EAAE,IAAI,EAAE,MAAM,QAAQ,CAAC;AACpE,OAAO,EACL,iCAAiC,EACjC,+BAA+B,EAC/B,6BAA6B,EAC7B,kBAAkB,GACnB,MAAM,uBAAuB,CAAC;AAC/B,OAAO,EAAE,iBAAiB,EAAE,eAAe,EAAE,IAAI,EAAE,MAAM,QAAQ,CAAC;AAClE,OAAO,EAAE,sBAAsB,EAAE,oBAAoB,EAAE,WAAW,EAAE,MAAM,gBAAgB,CAAC"}

package/resources/inference-pipelines/index.js.map

@@ -1 +1 @@
-{"version":3,"file":"index.js","sourceRoot":"","sources":["../../src/resources/inference-pipelines/index.ts"],"names":[],"mappings":";AAAA,sFAAsF;;;AAEtF,kCAAoE;AAArB,4FAAA,IAAI,OAAA;AACnD,gEAA2D;AAAlD,yHAAA,kBAAkB,OAAA;AAC3B,kCAAkE;AAArB,4FAAA,IAAI,OAAA;AACjD,kDAA2F;AAApC,2GAAA,WAAW,OAAA"}
+{"version":3,"file":"index.js","sourceRoot":"","sources":["../../src/resources/inference-pipelines/index.ts"],"names":[],"mappings":";AAAA,sFAAsF;;;AAEtF,kCAAoE;AAArB,4FAAA,IAAI,OAAA;AACnD,gEAK+B;AAD7B,yHAAA,kBAAkB,OAAA;AAEpB,kCAAkE;AAArB,4FAAA,IAAI,OAAA;AACjD,kDAA2F;AAApC,2GAAA,WAAW,OAAA"}

package/resources/inference-pipelines/index.mjs

@@ -1,6 +1,6 @@
 // File generated from our OpenAPI spec by Stainless. See CONTRIBUTING.md for details.
 export { Data } from "./data.mjs";
-export { InferencePipelines } from "./inference-pipelines.mjs";
+export { InferencePipelines, } from "./inference-pipelines.mjs";
 export { Rows } from "./rows.mjs";
 export { TestResults } from "./test-results.mjs";
 //# sourceMappingURL=index.mjs.map

package/resources/inference-pipelines/index.mjs.map

@@ -1 +1 @@
-{"version":3,"file":"index.mjs","sourceRoot":"","sources":["../../src/resources/inference-pipelines/index.ts"],"names":[],"mappings":"AAAA,sFAAsF;OAE/E,EAAwC,IAAI,EAAE;OAC9C,EAAE,kBAAkB,EAAE;OACtB,EAAsC,IAAI,EAAE;OAC5C,EAAgD,WAAW,EAAE"}
+{"version":3,"file":"index.mjs","sourceRoot":"","sources":["../../src/resources/inference-pipelines/index.ts"],"names":[],"mappings":"AAAA,sFAAsF;OAE/E,EAAwC,IAAI,EAAE;OAC9C,EAIL,kBAAkB,GACnB;OACM,EAAsC,IAAI,EAAE;OAC5C,EAAgD,WAAW,EAAE"}