@coalex-ai/sdk 0.4.1 → 0.5.1

package/dist/auto-instrument.d.ts

@@ -0,0 +1,14 @@
+import type { TracerProvider } from "@opentelemetry/api";
+export type InstrumentationStatus = "success" | "not_installed" | "already_instrumented" | "error" | "excluded";
+export interface AutoInstrumentOptions {
+    tracerProvider?: TracerProvider;
+    suppressErrors?: boolean;
+    includeLibraries?: string[];
+    excludeLibraries?: string[];
+}
+export declare function autoInstrument(options?: AutoInstrumentOptions): Record<string, InstrumentationStatus>;
+export declare function autoInstrumentAsync(options?: AutoInstrumentOptions): Promise<Record<string, InstrumentationStatus>>;
+export declare function instrumentOpenAI(tracerProvider?: TracerProvider): boolean;
+export declare function instrumentLangChain(tracerProvider?: TracerProvider): boolean;
+export declare function instrumentAnthropic(tracerProvider?: TracerProvider): boolean;
+//# sourceMappingURL=auto-instrument.d.ts.map

package/dist/auto-instrument.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"auto-instrument.d.ts","sourceRoot":"","sources":["../src/auto-instrument.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,cAAc,EAAE,MAAM,oBAAoB,CAAC;AAEzD,MAAM,MAAM,qBAAqB,GAC7B,SAAS,GACT,eAAe,GACf,sBAAsB,GACtB,OAAO,GACP,UAAU,CAAC;AAEf,MAAM,WAAW,qBAAqB;IACpC,cAAc,CAAC,EAAE,cAAc,CAAC;IAChC,cAAc,CAAC,EAAE,OAAO,CAAC;IACzB,gBAAgB,CAAC,EAAE,MAAM,EAAE,CAAC;IAC5B,gBAAgB,CAAC,EAAE,MAAM,EAAE,CAAC;CAC7B;AAwED,wBAAgB,cAAc,CAC5B,OAAO,CAAC,EAAE,qBAAqB,GAC9B,MAAM,CAAC,MAAM,EAAE,qBAAqB,CAAC,CAuCvC;AAED,wBAAsB,mBAAmB,CACvC,OAAO,CAAC,EAAE,qBAAqB,GAC9B,OAAO,CAAC,MAAM,CAAC,MAAM,EAAE,qBAAqB,CAAC,CAAC,CAoBhD;AAED,wBAAgB,gBAAgB,CAAC,cAAc,CAAC,EAAE,cAAc,GAAG,OAAO,CAMzE;AAED,wBAAgB,mBAAmB,CAAC,cAAc,CAAC,EAAE,cAAc,GAAG,OAAO,CAM5E;AAED,wBAAgB,mBAAmB,CAAC,cAAc,CAAC,EAAE,cAAc,GAAG,OAAO,CAM5E"}

package/dist/auto-instrument.js

@@ -0,0 +1,129 @@
+const INSTRUMENTORS = [
+    {
+        name: "openai",
+        package: "@arizeai/openinference-instrumentation-openai",
+        class: "OpenAIInstrumentation",
+        description: "OpenAI SDK",
+    },
+    {
+        name: "langchain",
+        package: "@arizeai/openinference-instrumentation-langchain",
+        class: "LangChainInstrumentation",
+        description: "LangChain",
+    },
+    {
+        name: "anthropic",
+        package: "@arizeai/openinference-instrumentation-anthropic",
+        class: "AnthropicInstrumentation",
+        description: "Anthropic Claude",
+    },
+];
+async function instrumentLibrary(config, tracerProvider, suppressErrors = true) {
+    try {
+        let mod;
+        try {
+            mod = await import(config.package);
+        }
+        catch {
+            return "not_installed";
+        }
+        const InstrumentorClass = mod[config.class];
+        if (!InstrumentorClass) {
+            console.warn(`[coalex] Found ${config.package} but missing ${config.class} export`);
+            return "error";
+        }
+        const instrumentor = new InstrumentorClass();
+        if (instrumentor._isInstrumentedByOpenTelemetry || instrumentor.isEnabled?.()) {
+            return "already_instrumented";
+        }
+        if (tracerProvider) {
+            instrumentor.setTracerProvider(tracerProvider);
+        }
+        instrumentor.enable();
+        console.info(`[coalex] Successfully instrumented ${config.description} (${config.name})`);
+        return "success";
+    }
+    catch (e) {
+        const msg = e instanceof Error ? e.message : String(e);
+        if (suppressErrors) {
+            console.warn(`[coalex] Failed to instrument ${config.name}: ${msg}`);
+            return "error";
+        }
+        throw e;
+    }
+}
+export function autoInstrument(options) {
+    const results = {};
+    for (const config of INSTRUMENTORS) {
+        if (options?.includeLibraries && !options.includeLibraries.includes(config.name)) {
+            continue;
+        }
+        if (options?.excludeLibraries?.includes(config.name)) {
+            results[config.name] = "excluded";
+            continue;
+        }
+        // Synchronous attempt — dynamic import is async but instrumentors
+        // typically need to be set up before any LLM calls, so we try sync first.
+        // For true async support, users should await autoInstrumentAsync().
+        try {
+            // eslint-disable-next-line @typescript-eslint/no-require-imports
+            const mod = require(config.package);
+            const InstrumentorClass = mod[config.class];
+            if (!InstrumentorClass) {
+                results[config.name] = "error";
+                continue;
+            }
+            const instrumentor = new InstrumentorClass();
+            if (instrumentor._isInstrumentedByOpenTelemetry || instrumentor.isEnabled?.()) {
+                results[config.name] = "already_instrumented";
+                continue;
+            }
+            if (options?.tracerProvider) {
+                instrumentor.setTracerProvider(options.tracerProvider);
+            }
+            instrumentor.enable();
+            results[config.name] = "success";
+        }
+        catch {
+            results[config.name] = "not_installed";
+        }
+    }
+    return results;
+}
+export async function autoInstrumentAsync(options) {
+    const suppressErrors = options?.suppressErrors ?? true;
+    const results = {};
+    for (const config of INSTRUMENTORS) {
+        if (options?.includeLibraries && !options.includeLibraries.includes(config.name)) {
+            continue;
+        }
+        if (options?.excludeLibraries?.includes(config.name)) {
+            results[config.name] = "excluded";
+            continue;
+        }
+        results[config.name] = await instrumentLibrary(config, options?.tracerProvider, suppressErrors);
+    }
+    return results;
+}
+export function instrumentOpenAI(tracerProvider) {
+    const result = autoInstrument({
+        tracerProvider,
+        includeLibraries: ["openai"],
+    });
+    return result.openai === "success";
+}
+export function instrumentLangChain(tracerProvider) {
+    const result = autoInstrument({
+        tracerProvider,
+        includeLibraries: ["langchain"],
+    });
+    return result.langchain === "success";
+}
+export function instrumentAnthropic(tracerProvider) {
+    const result = autoInstrument({
+        tracerProvider,
+        includeLibraries: ["anthropic"],
+    });
+    return result.anthropic === "success";
+}
+//# sourceMappingURL=auto-instrument.js.map

package/dist/auto-instrument.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"auto-instrument.js","sourceRoot":"","sources":["../src/auto-instrument.ts"],"names":[],"mappings":"AAuBA,MAAM,aAAa,GAAyB;IAC1C;QACE,IAAI,EAAE,QAAQ;QACd,OAAO,EAAE,+CAA+C;QACxD,KAAK,EAAE,uBAAuB;QAC9B,WAAW,EAAE,YAAY;KAC1B;IACD;QACE,IAAI,EAAE,WAAW;QACjB,OAAO,EAAE,kDAAkD;QAC3D,KAAK,EAAE,0BAA0B;QACjC,WAAW,EAAE,WAAW;KACzB;IACD;QACE,IAAI,EAAE,WAAW;QACjB,OAAO,EAAE,kDAAkD;QAC3D,KAAK,EAAE,0BAA0B;QACjC,WAAW,EAAE,kBAAkB;KAChC;CACF,CAAC;AAEF,KAAK,UAAU,iBAAiB,CAC9B,MAA0B,EAC1B,cAA+B,EAC/B,cAAc,GAAG,IAAI;IAErB,IAAI,CAAC;QACH,IAAI,GAAQ,CAAC;QACb,IAAI,CAAC;YACH,GAAG,GAAG,MAAM,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC;QACrC,CAAC;QAAC,MAAM,CAAC;YACP,OAAO,eAAe,CAAC;QACzB,CAAC;QAED,MAAM,iBAAiB,GAAG,GAAG,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC;QAC5C,IAAI,CAAC,iBAAiB,EAAE,CAAC;YACvB,OAAO,CAAC,IAAI,CAAC,kBAAkB,MAAM,CAAC,OAAO,gBAAgB,MAAM,CAAC,KAAK,SAAS,CAAC,CAAC;YACpF,OAAO,OAAO,CAAC;QACjB,CAAC;QAED,MAAM,YAAY,GAAG,IAAI,iBAAiB,EAAE,CAAC;QAE7C,IAAI,YAAY,CAAC,8BAA8B,IAAI,YAAY,CAAC,SAAS,EAAE,EAAE,EAAE,CAAC;YAC9E,OAAO,sBAAsB,CAAC;QAChC,CAAC;QAED,IAAI,cAAc,EAAE,CAAC;YACnB,YAAY,CAAC,iBAAiB,CAAC,cAAc,CAAC,CAAC;QACjD,CAAC;QACD,YAAY,CAAC,MAAM,EAAE,CAAC;QAEtB,OAAO,CAAC,IAAI,CAAC,sCAAsC,MAAM,CAAC,WAAW,KAAK,MAAM,CAAC,IAAI,GAAG,CAAC,CAAC;QAC1F,OAAO,SAAS,CAAC;IACnB,CAAC;IAAC,OAAO,CAAU,EAAE,CAAC;QACpB,MAAM,GAAG,GAAG,CAAC,YAAY,KAAK,CAAC,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC;QACvD,IAAI,cAAc,EAAE,CAAC;YACnB,OAAO,CAAC,IAAI,CAAC,iCAAiC,MAAM,CAAC,IAAI,KAAK,GAAG,EAAE,CAAC,CAAC;YACrE,OAAO,OAAO,CAAC;QACjB,CAAC;QACD,MAAM,CAAC,CAAC;IACV,CAAC;AACH,CAAC;AAED,MAAM,UAAU,cAAc,CAC5B,OAA+B;IAE/B,MAAM,OAAO,GAA0C,EAAE,CAAC;IAE1D,KAAK,MAAM,MAAM,IAAI,aAAa,EAAE,CAAC;QACnC,IAAI,OAAO,EAAE,gBAAgB,IAAI,CAAC,OAAO,CAAC,gBAAgB,CAAC,QAAQ,CAAC,MAAM,CAAC,IAAI,CAAC,EAAE,CAAC;YACjF,SAAS;QACX,CAAC;QACD,IAAI,OAAO,EAAE,gBAAgB,EAAE,QAAQ,CAAC,MAAM,CAAC,IAAI,CAAC,EAAE,CAAC;YACrD,OAAO,CAAC,MAAM,CAAC,IAAI,CAAC,GAAG,UAAU,CAAC;YAClC,SAAS;QACX,CAAC;QAED,kEAAkE;QAClE,0EAA0E;QAC1E,oEAAoE;QACpE,IAAI,CAAC;YACH,iEAAiE;YACjE,MAAM,GAAG,GAAG,OAAO,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC;YACpC,MAAM,iBAAiB,GAAG,GAAG,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC;YAC5C,IAAI,CAAC,iBAAiB,EAAE,CAAC;gBACvB,OAAO,CAAC,MAAM,CAAC,IAAI,CAAC,GAAG,OAAO,CAAC;gBAC/B,SAAS;YACX,CAAC;YACD,MAAM,YAAY,GAAG,IAAI,iBAAiB,EAAE,CAAC;YAC7C,IAAI,YAAY,CAAC,8BAA8B,IAAI,YAAY,CAAC,SAAS,EAAE,EAAE,EAAE,CAAC;gBAC9E,OAAO,CAAC,MAAM,CAAC,IAAI,CAAC,GAAG,sBAAsB,CAAC;gBAC9C,SAAS;YACX,CAAC;YACD,IAAI,OAAO,EAAE,cAAc,EAAE,CAAC;gBAC5B,YAAY,CAAC,iBAAiB,CAAC,OAAO,CAAC,cAAc,CAAC,CAAC;YACzD,CAAC;YACD,YAAY,CAAC,MAAM,EAAE,CAAC;YACtB,OAAO,CAAC,MAAM,CAAC,IAAI,CAAC,GAAG,SAAS,CAAC;QACnC,CAAC;QAAC,MAAM,CAAC;YACP,OAAO,CAAC,MAAM,CAAC,IAAI,CAAC,GAAG,eAAe,CAAC;QACzC,CAAC;IACH,CAAC;IAED,OAAO,OAAO,CAAC;AACjB,CAAC;AAED,MAAM,CAAC,KAAK,UAAU,mBAAmB,CACvC,OAA+B;IAE/B,MAAM,cAAc,GAAG,OAAO,EAAE,cAAc,IAAI,IAAI,CAAC;IACvD,MAAM,OAAO,GAA0C,EAAE,CAAC;IAE1D,KAAK,MAAM,MAAM,IAAI,aAAa,EAAE,CAAC;QACnC,IAAI,OAAO,EAAE,gBAAgB,IAAI,CAAC,OAAO,CAAC,gBAAgB,CAAC,QAAQ,CAAC,MAAM,CAAC,IAAI,CAAC,EAAE,CAAC;YACjF,SAAS;QACX,CAAC;QACD,IAAI,OAAO,EAAE,gBAAgB,EAAE,QAAQ,CAAC,MAAM,CAAC,IAAI,CAAC,EAAE,CAAC;YACrD,OAAO,CAAC,MAAM,CAAC,IAAI,CAAC,GAAG,UAAU,CAAC;YAClC,SAAS;QACX,CAAC;QACD,OAAO,CAAC,MAAM,CAAC,IAAI,CAAC,GAAG,MAAM,iBAAiB,CAC5C,MAAM,EACN,OAAO,EAAE,cAAc,EACvB,cAAc,CACf,CAAC;IACJ,CAAC;IAED,OAAO,OAAO,CAAC;AACjB,CAAC;AAED,MAAM,UAAU,gBAAgB,CAAC,cAA+B;IAC9D,MAAM,MAAM,GAAG,cAAc,CAAC;QAC5B,cAAc;QACd,gBAAgB,EAAE,CAAC,QAAQ,CAAC;KAC7B,CAAC,CAAC;IACH,OAAO,MAAM,CAAC,MAAM,KAAK,SAAS,CAAC;AACrC,CAAC;AAED,MAAM,UAAU,mBAAmB,CAAC,cAA+B;IACjE,MAAM,MAAM,GAAG,cAAc,CAAC;QAC5B,cAAc;QACd,gBAAgB,EAAE,CAAC,WAAW,CAAC;KAChC,CAAC,CAAC;IACH,OAAO,MAAM,CAAC,SAAS,KAAK,SAAS,CAAC;AACxC,CAAC;AAED,MAAM,UAAU,mBAAmB,CAAC,cAA+B;IACjE,MAAM,MAAM,GAAG,cAAc,CAAC;QAC5B,cAAc;QACd,gBAAgB,EAAE,CAAC,WAAW,CAAC;KAChC,CAAC,CAAC;IACH,OAAO,MAAM,CAAC,SAAS,KAAK,SAAS,CAAC;AACxC,CAAC"}

package/dist/config.d.ts

@@ -0,0 +1,5 @@
+export interface CoalexConfig {
+    readonly endpoint: string;
+    readonly serviceName: string;
+}
+//# sourceMappingURL=config.d.ts.map

package/dist/config.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"config.d.ts","sourceRoot":"","sources":["../src/config.ts"],"names":[],"mappings":"AAAA,MAAM,WAAW,YAAY;IAC3B,QAAQ,CAAC,QAAQ,EAAE,MAAM,CAAC;IAC1B,QAAQ,CAAC,WAAW,EAAE,MAAM,CAAC;CAC9B"}

package/dist/config.js

@@ -0,0 +1,2 @@
+export {};
+//# sourceMappingURL=config.js.map

package/dist/config.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"config.js","sourceRoot":"","sources":["../src/config.ts"],"names":[],"mappings":""}

package/dist/context.d.ts

@@ -0,0 +1,8 @@
+export interface CoalexContextOptions {
+    agentId: string;
+    requestId?: string;
+    version?: string;
+}
+export declare function coalexContext<T>(options: CoalexContextOptions, fn: () => Promise<T> | T): Promise<T>;
+export declare function coalexContextSync<T>(options: CoalexContextOptions, fn: () => T): T;
+//# sourceMappingURL=context.d.ts.map

package/dist/context.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"context.d.ts","sourceRoot":"","sources":["../src/context.ts"],"names":[],"mappings":"AAEA,MAAM,WAAW,oBAAoB;IACnC,OAAO,EAAE,MAAM,CAAC;IAChB,SAAS,CAAC,EAAE,MAAM,CAAC;IACnB,OAAO,CAAC,EAAE,MAAM,CAAC;CAClB;AAED,wBAAsB,aAAa,CAAC,CAAC,EACnC,OAAO,EAAE,oBAAoB,EAC7B,EAAE,EAAE,MAAM,OAAO,CAAC,CAAC,CAAC,GAAG,CAAC,GACvB,OAAO,CAAC,CAAC,CAAC,CAwBZ;AAED,wBAAgB,iBAAiB,CAAC,CAAC,EACjC,OAAO,EAAE,oBAAoB,EAC7B,EAAE,EAAE,MAAM,CAAC,GACV,CAAC,CAwBH"}

package/dist/context.js

@@ -0,0 +1,52 @@
+import { trace, SpanStatusCode } from "@opentelemetry/api";
+export async function coalexContext(options, fn) {
+    const tracer = trace.getTracer("coalex");
+    const attrs = { "coalex.agent_id": options.agentId };
+    if (options.requestId !== undefined) {
+        attrs["coalex.request_id"] = options.requestId;
+    }
+    if (options.version !== undefined) {
+        attrs["coalex.agent_version"] = options.version;
+    }
+    return tracer.startActiveSpan("coalex.invocation", { attributes: attrs }, async (span) => {
+        try {
+            const result = await fn();
+            span.setStatus({ code: SpanStatusCode.OK });
+            return result;
+        }
+        catch (e) {
+            span.setStatus({ code: SpanStatusCode.ERROR, message: String(e) });
+            span.recordException(e instanceof Error ? e : new Error(String(e)));
+            throw e;
+        }
+        finally {
+            span.end();
+        }
+    });
+}
+export function coalexContextSync(options, fn) {
+    const tracer = trace.getTracer("coalex");
+    const attrs = { "coalex.agent_id": options.agentId };
+    if (options.requestId !== undefined) {
+        attrs["coalex.request_id"] = options.requestId;
+    }
+    if (options.version !== undefined) {
+        attrs["coalex.agent_version"] = options.version;
+    }
+    return tracer.startActiveSpan("coalex.invocation", { attributes: attrs }, (span) => {
+        try {
+            const result = fn();
+            span.setStatus({ code: SpanStatusCode.OK });
+            return result;
+        }
+        catch (e) {
+            span.setStatus({ code: SpanStatusCode.ERROR, message: String(e) });
+            span.recordException(e instanceof Error ? e : new Error(String(e)));
+            throw e;
+        }
+        finally {
+            span.end();
+        }
+    });
+}
+//# sourceMappingURL=context.js.map

package/dist/context.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"context.js","sourceRoot":"","sources":["../src/context.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,KAAK,EAAE,cAAc,EAAE,MAAM,oBAAoB,CAAC;AAQ3D,MAAM,CAAC,KAAK,UAAU,aAAa,CACjC,OAA6B,EAC7B,EAAwB;IAExB,MAAM,MAAM,GAAG,KAAK,CAAC,SAAS,CAAC,QAAQ,CAAC,CAAC;IAEzC,MAAM,KAAK,GAA2B,EAAE,iBAAiB,EAAE,OAAO,CAAC,OAAO,EAAE,CAAC;IAC7E,IAAI,OAAO,CAAC,SAAS,KAAK,SAAS,EAAE,CAAC;QACpC,KAAK,CAAC,mBAAmB,CAAC,GAAG,OAAO,CAAC,SAAS,CAAC;IACjD,CAAC;IACD,IAAI,OAAO,CAAC,OAAO,KAAK,SAAS,EAAE,CAAC;QAClC,KAAK,CAAC,sBAAsB,CAAC,GAAG,OAAO,CAAC,OAAO,CAAC;IAClD,CAAC;IAED,OAAO,MAAM,CAAC,eAAe,CAAC,mBAAmB,EAAE,EAAE,UAAU,EAAE,KAAK,EAAE,EAAE,KAAK,EAAE,IAAI,EAAE,EAAE;QACvF,IAAI,CAAC;YACH,MAAM,MAAM,GAAG,MAAM,EAAE,EAAE,CAAC;YAC1B,IAAI,CAAC,SAAS,CAAC,EAAE,IAAI,EAAE,cAAc,CAAC,EAAE,EAAE,CAAC,CAAC;YAC5C,OAAO,MAAM,CAAC;QAChB,CAAC;QAAC,OAAO,CAAC,EAAE,CAAC;YACX,IAAI,CAAC,SAAS,CAAC,EAAE,IAAI,EAAE,cAAc,CAAC,KAAK,EAAE,OAAO,EAAE,MAAM,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC;YACnE,IAAI,CAAC,eAAe,CAAC,CAAC,YAAY,KAAK,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,IAAI,KAAK,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC;YACpE,MAAM,CAAC,CAAC;QACV,CAAC;gBAAS,CAAC;YACT,IAAI,CAAC,GAAG,EAAE,CAAC;QACb,CAAC;IACH,CAAC,CAAC,CAAC;AACL,CAAC;AAED,MAAM,UAAU,iBAAiB,CAC/B,OAA6B,EAC7B,EAAW;IAEX,MAAM,MAAM,GAAG,KAAK,CAAC,SAAS,CAAC,QAAQ,CAAC,CAAC;IAEzC,MAAM,KAAK,GAA2B,EAAE,iBAAiB,EAAE,OAAO,CAAC,OAAO,EAAE,CAAC;IAC7E,IAAI,OAAO,CAAC,SAAS,KAAK,SAAS,EAAE,CAAC;QACpC,KAAK,CAAC,mBAAmB,CAAC,GAAG,OAAO,CAAC,SAAS,CAAC;IACjD,CAAC;IACD,IAAI,OAAO,CAAC,OAAO,KAAK,SAAS,EAAE,CAAC;QAClC,KAAK,CAAC,sBAAsB,CAAC,GAAG,OAAO,CAAC,OAAO,CAAC;IAClD,CAAC;IAED,OAAO,MAAM,CAAC,eAAe,CAAC,mBAAmB,EAAE,EAAE,UAAU,EAAE,KAAK,EAAE,EAAE,CAAC,IAAI,EAAE,EAAE;QACjF,IAAI,CAAC;YACH,MAAM,MAAM,GAAG,EAAE,EAAE,CAAC;YACpB,IAAI,CAAC,SAAS,CAAC,EAAE,IAAI,EAAE,cAAc,CAAC,EAAE,EAAE,CAAC,CAAC;YAC5C,OAAO,MAAM,CAAC;QAChB,CAAC;QAAC,OAAO,CAAC,EAAE,CAAC;YACX,IAAI,CAAC,SAAS,CAAC,EAAE,IAAI,EAAE,cAAc,CAAC,KAAK,EAAE,OAAO,EAAE,MAAM,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC;YACnE,IAAI,CAAC,eAAe,CAAC,CAAC,YAAY,KAAK,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,IAAI,KAAK,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC;YACpE,MAAM,CAAC,CAAC;QACV,CAAC;gBAAS,CAAC;YACT,IAAI,CAAC,GAAG,EAAE,CAAC;QACb,CAAC;IACH,CAAC,CAAC,CAAC;AACL,CAAC"}

package/dist/evaluate.d.ts

@@ -0,0 +1,14 @@
+export declare const KNOWN_METRICS: Set<string>;
+export interface EvaluationDecision {
+    readonly status: "auto_approved" | "escalated" | "rejected";
+    readonly riskScore: number;
+    readonly escalationId?: string;
+}
+export interface EvaluateOptions {
+    traceId: string;
+    input: Record<string, unknown>;
+    output: Record<string, unknown>;
+    metrics: Record<string, string[]>;
+}
+export declare function evaluate(options: EvaluateOptions): Promise<EvaluationDecision>;
+//# sourceMappingURL=evaluate.d.ts.map

package/dist/evaluate.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"evaluate.d.ts","sourceRoot":"","sources":["../src/evaluate.ts"],"names":[],"mappings":"AAEA,eAAO,MAAM,aAAa,aASxB,CAAC;AAEH,MAAM,WAAW,kBAAkB;IACjC,QAAQ,CAAC,MAAM,EAAE,eAAe,GAAG,WAAW,GAAG,UAAU,CAAC;IAC5D,QAAQ,CAAC,SAAS,EAAE,MAAM,CAAC;IAC3B,QAAQ,CAAC,YAAY,CAAC,EAAE,MAAM,CAAC;CAChC;AAED,MAAM,WAAW,eAAe;IAC9B,OAAO,EAAE,MAAM,CAAC;IAChB,KAAK,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,CAAC;IAC/B,MAAM,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,CAAC;IAChC,OAAO,EAAE,MAAM,CAAC,MAAM,EAAE,MAAM,EAAE,CAAC,CAAC;CACnC;AAED,wBAAsB,QAAQ,CAAC,OAAO,EAAE,eAAe,GAAG,OAAO,CAAC,kBAAkB,CAAC,CAqCpF"}

package/dist/evaluate.js

@@ -0,0 +1,45 @@
+import { getConfig, getApiKey } from "./index.js";
+export const KNOWN_METRICS = new Set([
+    "f1",
+    "word_overlap",
+    "bleu",
+    "rouge_l",
+    "semantic_similarity",
+    "exact_match",
+    "levenshtein",
+    "contains",
+]);
+export async function evaluate(options) {
+    const config = getConfig();
+    const apiKey = getApiKey();
+    for (const [field, metricList] of Object.entries(options.metrics)) {
+        const unknown = metricList.filter((m) => !KNOWN_METRICS.has(m));
+        if (unknown.length > 0) {
+            throw new Error(`Unknown metrics for field '${field}': ${unknown.join(", ")}`);
+        }
+    }
+    const payload = {
+        trace_id: options.traceId,
+        input: options.input,
+        output: options.output,
+        metrics: options.metrics,
+    };
+    const resp = await fetch(`${config.endpoint}/api/v1/evaluate`, {
+        method: "POST",
+        headers: {
+            "Content-Type": "application/json",
+            Authorization: `Bearer ${apiKey}`,
+        },
+        body: JSON.stringify(payload),
+    });
+    if (!resp.ok) {
+        throw new Error(`evaluate() failed: ${resp.status} ${resp.statusText}`);
+    }
+    const data = await resp.json();
+    return {
+        status: data.status,
+        riskScore: data.risk_score,
+        escalationId: data.escalation_id ?? undefined,
+    };
+}
+//# sourceMappingURL=evaluate.js.map

package/dist/evaluate.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"evaluate.js","sourceRoot":"","sources":["../src/evaluate.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,SAAS,EAAE,SAAS,EAAE,MAAM,YAAY,CAAC;AAElD,MAAM,CAAC,MAAM,aAAa,GAAG,IAAI,GAAG,CAAC;IACnC,IAAI;IACJ,cAAc;IACd,MAAM;IACN,SAAS;IACT,qBAAqB;IACrB,aAAa;IACb,aAAa;IACb,UAAU;CACX,CAAC,CAAC;AAeH,MAAM,CAAC,KAAK,UAAU,QAAQ,CAAC,OAAwB;IACrD,MAAM,MAAM,GAAG,SAAS,EAAE,CAAC;IAC3B,MAAM,MAAM,GAAG,SAAS,EAAE,CAAC;IAE3B,KAAK,MAAM,CAAC,KAAK,EAAE,UAAU,CAAC,IAAI,MAAM,CAAC,OAAO,CAAC,OAAO,CAAC,OAAO,CAAC,EAAE,CAAC;QAClE,MAAM,OAAO,GAAG,UAAU,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,aAAa,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,CAAC;QAChE,IAAI,OAAO,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;YACvB,MAAM,IAAI,KAAK,CAAC,8BAA8B,KAAK,MAAM,OAAO,CAAC,IAAI,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;QACjF,CAAC;IACH,CAAC;IAED,MAAM,OAAO,GAAG;QACd,QAAQ,EAAE,OAAO,CAAC,OAAO;QACzB,KAAK,EAAE,OAAO,CAAC,KAAK;QACpB,MAAM,EAAE,OAAO,CAAC,MAAM;QACtB,OAAO,EAAE,OAAO,CAAC,OAAO;KACzB,CAAC;IAEF,MAAM,IAAI,GAAG,MAAM,KAAK,CAAC,GAAG,MAAM,CAAC,QAAQ,kBAAkB,EAAE;QAC7D,MAAM,EAAE,MAAM;QACd,OAAO,EAAE;YACP,cAAc,EAAE,kBAAkB;YAClC,aAAa,EAAE,UAAU,MAAM,EAAE;SAClC;QACD,IAAI,EAAE,IAAI,CAAC,SAAS,CAAC,OAAO,CAAC;KAC9B,CAAC,CAAC;IAEH,IAAI,CAAC,IAAI,CAAC,EAAE,EAAE,CAAC;QACb,MAAM,IAAI,KAAK,CAAC,sBAAsB,IAAI,CAAC,MAAM,IAAI,IAAI,CAAC,UAAU,EAAE,CAAC,CAAC;IAC1E,CAAC;IAED,MAAM,IAAI,GAAG,MAAM,IAAI,CAAC,IAAI,EAAE,CAAC;IAC/B,OAAO;QACL,MAAM,EAAE,IAAI,CAAC,MAAM;QACnB,SAAS,EAAE,IAAI,CAAC,UAAU;QAC1B,YAAY,EAAE,IAAI,CAAC,aAAa,IAAI,SAAS;KAC9C,CAAC;AACJ,CAAC"}

package/dist/ext/base.d.ts

@@ -0,0 +1,3 @@
+import { type Tracer } from "@opentelemetry/api";
+export declare function getTracer(): Tracer;
+//# sourceMappingURL=base.d.ts.map

package/dist/ext/base.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"base.d.ts","sourceRoot":"","sources":["../../src/ext/base.ts"],"names":[],"mappings":"AAAA,OAAO,EAAS,KAAK,MAAM,EAAE,MAAM,oBAAoB,CAAC;AAExD,wBAAgB,SAAS,IAAI,MAAM,CAElC"}

package/dist/ext/base.js

@@ -0,0 +1,5 @@
+import { trace } from "@opentelemetry/api";
+export function getTracer() {
+    return trace.getTracer("coalex.ext");
+}
+//# sourceMappingURL=base.js.map

package/dist/ext/base.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"base.js","sourceRoot":"","sources":["../../src/ext/base.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,KAAK,EAAe,MAAM,oBAAoB,CAAC;AAExD,MAAM,UAAU,SAAS;IACvB,OAAO,KAAK,CAAC,SAAS,CAAC,YAAY,CAAC,CAAC;AACvC,CAAC"}

package/dist/ext/embedding.d.ts

@@ -0,0 +1,16 @@
+export interface EmbeddingSpanOptions {
+    name?: string;
+    modelName?: string;
+}
+export declare function embeddingSpan<A extends unknown[], R>(options: EmbeddingSpanOptions, fn: (...args: A) => R | Promise<R>): (...args: A) => Promise<R>;
+export declare class EmbeddingSpan {
+    private readonly spanName;
+    private readonly modelName?;
+    private span;
+    constructor(name?: string, modelName?: string);
+    start(): void;
+    setText(text: string): void;
+    setVector(vector: number[], text?: string): void;
+    end(): void;
+}
+//# sourceMappingURL=embedding.d.ts.map

package/dist/ext/embedding.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"embedding.d.ts","sourceRoot":"","sources":["../../src/ext/embedding.ts"],"names":[],"mappings":"AAGA,MAAM,WAAW,oBAAoB;IACnC,IAAI,CAAC,EAAE,MAAM,CAAC;IACd,SAAS,CAAC,EAAE,MAAM,CAAC;CACpB;AAED,wBAAgB,aAAa,CAAC,CAAC,SAAS,OAAO,EAAE,EAAE,CAAC,EAClD,OAAO,EAAE,oBAAoB,EAC7B,EAAE,EAAE,CAAC,GAAG,IAAI,EAAE,CAAC,KAAK,CAAC,GAAG,OAAO,CAAC,CAAC,CAAC,GACjC,CAAC,GAAG,IAAI,EAAE,CAAC,KAAK,OAAO,CAAC,CAAC,CAAC,CAgC5B;AAED,qBAAa,aAAa;IACxB,OAAO,CAAC,QAAQ,CAAC,QAAQ,CAAS;IAClC,OAAO,CAAC,QAAQ,CAAC,SAAS,CAAC,CAAS;IACpC,OAAO,CAAC,IAAI,CAAqB;gBAErB,IAAI,SAAc,EAAE,SAAS,CAAC,EAAE,MAAM;IAKlD,KAAK,IAAI,IAAI;IASb,OAAO,CAAC,IAAI,EAAE,MAAM,GAAG,IAAI;IAI3B,SAAS,CAAC,MAAM,EAAE,MAAM,EAAE,EAAE,IAAI,SAAK,GAAG,IAAI;IAO5C,GAAG,IAAI,IAAI;CAGZ"}

package/dist/ext/embedding.js

@@ -0,0 +1,62 @@
+import { getTracer } from "./base.js";
+export function embeddingSpan(options, fn) {
+    const spanName = options.name ?? "embedding";
+    return async (...args) => {
+        const tracer = getTracer();
+        return tracer.startActiveSpan(spanName, async (span) => {
+            try {
+                span.setAttribute("openinference.span.kind", "EMBEDDING");
+                if (options.modelName) {
+                    span.setAttribute("embedding.model_name", options.modelName);
+                }
+                const text = args.find((a) => typeof a === "string");
+                if (typeof text === "string") {
+                    span.setAttribute("input.value", text);
+                }
+                const result = await fn(...args);
+                if (Array.isArray(result) &&
+                    result.length > 0 &&
+                    typeof result[0] === "number") {
+                    const encoded = JSON.stringify([
+                        { "embedding.vector": result, "embedding.text": String(text ?? "") },
+                    ]);
+                    span.setAttribute("embedding.embeddings", encoded);
+                }
+                return result;
+            }
+            finally {
+                span.end();
+            }
+        });
+    };
+}
+export class EmbeddingSpan {
+    spanName;
+    modelName;
+    span = null;
+    constructor(name = "embedding", modelName) {
+        this.spanName = name;
+        this.modelName = modelName;
+    }
+    start() {
+        const tracer = getTracer();
+        this.span = tracer.startSpan(this.spanName);
+        this.span.setAttribute("openinference.span.kind", "EMBEDDING");
+        if (this.modelName) {
+            this.span.setAttribute("embedding.model_name", this.modelName);
+        }
+    }
+    setText(text) {
+        this.span?.setAttribute("input.value", text);
+    }
+    setVector(vector, text = "") {
+        const encoded = JSON.stringify([
+            { "embedding.vector": vector, "embedding.text": text },
+        ]);
+        this.span?.setAttribute("embedding.embeddings", encoded);
+    }
+    end() {
+        this.span?.end();
+    }
+}
+//# sourceMappingURL=embedding.js.map

package/dist/ext/embedding.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"embedding.js","sourceRoot":"","sources":["../../src/ext/embedding.ts"],"names":[],"mappings":"AACA,OAAO,EAAE,SAAS,EAAE,MAAM,WAAW,CAAC;AAOtC,MAAM,UAAU,aAAa,CAC3B,OAA6B,EAC7B,EAAkC;IAElC,MAAM,QAAQ,GAAG,OAAO,CAAC,IAAI,IAAI,WAAW,CAAC;IAE7C,OAAO,KAAK,EAAE,GAAG,IAAO,EAAc,EAAE;QACtC,MAAM,MAAM,GAAG,SAAS,EAAE,CAAC;QAC3B,OAAO,MAAM,CAAC,eAAe,CAAC,QAAQ,EAAE,KAAK,EAAE,IAAI,EAAE,EAAE;YACrD,IAAI,CAAC;gBACH,IAAI,CAAC,YAAY,CAAC,yBAAyB,EAAE,WAAW,CAAC,CAAC;gBAC1D,IAAI,OAAO,CAAC,SAAS,EAAE,CAAC;oBACtB,IAAI,CAAC,YAAY,CAAC,sBAAsB,EAAE,OAAO,CAAC,SAAS,CAAC,CAAC;gBAC/D,CAAC;gBACD,MAAM,IAAI,GAAG,IAAI,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,OAAO,CAAC,KAAK,QAAQ,CAAC,CAAC;gBACrD,IAAI,OAAO,IAAI,KAAK,QAAQ,EAAE,CAAC;oBAC7B,IAAI,CAAC,YAAY,CAAC,aAAa,EAAE,IAAI,CAAC,CAAC;gBACzC,CAAC;gBACD,MAAM,MAAM,GAAG,MAAM,EAAE,CAAC,GAAG,IAAI,CAAC,CAAC;gBACjC,IACE,KAAK,CAAC,OAAO,CAAC,MAAM,CAAC;oBACrB,MAAM,CAAC,MAAM,GAAG,CAAC;oBACjB,OAAO,MAAM,CAAC,CAAC,CAAC,KAAK,QAAQ,EAC7B,CAAC;oBACD,MAAM,OAAO,GAAG,IAAI,CAAC,SAAS,CAAC;wBAC7B,EAAE,kBAAkB,EAAE,MAAM,EAAE,gBAAgB,EAAE,MAAM,CAAC,IAAI,IAAI,EAAE,CAAC,EAAE;qBACrE,CAAC,CAAC;oBACH,IAAI,CAAC,YAAY,CAAC,sBAAsB,EAAE,OAAO,CAAC,CAAC;gBACrD,CAAC;gBACD,OAAO,MAAM,CAAC;YAChB,CAAC;oBAAS,CAAC;gBACT,IAAI,CAAC,GAAG,EAAE,CAAC;YACb,CAAC;QACH,CAAC,CAAC,CAAC;IACL,CAAC,CAAC;AACJ,CAAC;AAED,MAAM,OAAO,aAAa;IACP,QAAQ,CAAS;IACjB,SAAS,CAAU;IAC5B,IAAI,GAAgB,IAAI,CAAC;IAEjC,YAAY,IAAI,GAAG,WAAW,EAAE,SAAkB;QAChD,IAAI,CAAC,QAAQ,GAAG,IAAI,CAAC;QACrB,IAAI,CAAC,SAAS,GAAG,SAAS,CAAC;IAC7B,CAAC;IAED,KAAK;QACH,MAAM,MAAM,GAAG,SAAS,EAAE,CAAC;QAC3B,IAAI,CAAC,IAAI,GAAG,MAAM,CAAC,SAAS,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;QAC5C,IAAI,CAAC,IAAI,CAAC,YAAY,CAAC,yBAAyB,EAAE,WAAW,CAAC,CAAC;QAC/D,IAAI,IAAI,CAAC,SAAS,EAAE,CAAC;YACnB,IAAI,CAAC,IAAI,CAAC,YAAY,CAAC,sBAAsB,EAAE,IAAI,CAAC,SAAS,CAAC,CAAC;QACjE,CAAC;IACH,CAAC;IAED,OAAO,CAAC,IAAY;QAClB,IAAI,CAAC,IAAI,EAAE,YAAY,CAAC,aAAa,EAAE,IAAI,CAAC,CAAC;IAC/C,CAAC;IAED,SAAS,CAAC,MAAgB,EAAE,IAAI,GAAG,EAAE;QACnC,MAAM,OAAO,GAAG,IAAI,CAAC,SAAS,CAAC;YAC7B,EAAE,kBAAkB,EAAE,MAAM,EAAE,gBAAgB,EAAE,IAAI,EAAE;SACvD,CAAC,CAAC;QACH,IAAI,CAAC,IAAI,EAAE,YAAY,CAAC,sBAAsB,EAAE,OAAO,CAAC,CAAC;IAC3D,CAAC;IAED,GAAG;QACD,IAAI,CAAC,IAAI,EAAE,GAAG,EAAE,CAAC;IACnB,CAAC;CACF"}

package/dist/ext/guardrail.d.ts

@@ -0,0 +1,6 @@
+export interface GuardrailSpanOptions {
+    name?: string;
+    guardrailName?: string;
+}
+export declare function guardrailSpan<A extends unknown[], R>(options: GuardrailSpanOptions, fn: (...args: A) => R | Promise<R>): (...args: A) => Promise<R>;
+//# sourceMappingURL=guardrail.d.ts.map

package/dist/ext/guardrail.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"guardrail.d.ts","sourceRoot":"","sources":["../../src/ext/guardrail.ts"],"names":[],"mappings":"AAGA,MAAM,WAAW,oBAAoB;IACnC,IAAI,CAAC,EAAE,MAAM,CAAC;IACd,aAAa,CAAC,EAAE,MAAM,CAAC;CACxB;AAED,wBAAgB,aAAa,CAAC,CAAC,SAAS,OAAO,EAAE,EAAE,CAAC,EAClD,OAAO,EAAE,oBAAoB,EAC7B,EAAE,EAAE,CAAC,GAAG,IAAI,EAAE,CAAC,KAAK,CAAC,GAAG,OAAO,CAAC,CAAC,CAAC,GACjC,CAAC,GAAG,IAAI,EAAE,CAAC,KAAK,OAAO,CAAC,CAAC,CAAC,CA2B5B"}

package/dist/ext/guardrail.js

@@ -0,0 +1,32 @@
+import { SpanStatusCode } from "@opentelemetry/api";
+import { getTracer } from "./base.js";
+export function guardrailSpan(options, fn) {
+    const spanName = options.name ?? "guardrail";
+    const logicalName = options.guardrailName ?? spanName;
+    return async (...args) => {
+        const tracer = getTracer();
+        return tracer.startActiveSpan(spanName, async (span) => {
+            try {
+                span.setAttribute("openinference.span.kind", "GUARDRAIL");
+                span.setAttribute("guardrail.name", logicalName);
+                const result = await fn(...args);
+                span.setAttribute("guardrail.result", "pass");
+                if (typeof result === "number") {
+                    span.setAttribute("guardrail.score", result);
+                }
+                span.setStatus({ code: SpanStatusCode.OK });
+                return result;
+            }
+            catch (e) {
+                span.setAttribute("guardrail.result", "fail");
+                span.setStatus({ code: SpanStatusCode.ERROR, message: String(e) });
+                span.recordException(e instanceof Error ? e : new Error(String(e)));
+                throw e;
+            }
+            finally {
+                span.end();
+            }
+        });
+    };
+}
+//# sourceMappingURL=guardrail.js.map

package/dist/ext/guardrail.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"guardrail.js","sourceRoot":"","sources":["../../src/ext/guardrail.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,cAAc,EAAE,MAAM,oBAAoB,CAAC;AACpD,OAAO,EAAE,SAAS,EAAE,MAAM,WAAW,CAAC;AAOtC,MAAM,UAAU,aAAa,CAC3B,OAA6B,EAC7B,EAAkC;IAElC,MAAM,QAAQ,GAAG,OAAO,CAAC,IAAI,IAAI,WAAW,CAAC;IAC7C,MAAM,WAAW,GAAG,OAAO,CAAC,aAAa,IAAI,QAAQ,CAAC;IAEtD,OAAO,KAAK,EAAE,GAAG,IAAO,EAAc,EAAE;QACtC,MAAM,MAAM,GAAG,SAAS,EAAE,CAAC;QAC3B,OAAO,MAAM,CAAC,eAAe,CAAC,QAAQ,EAAE,KAAK,EAAE,IAAI,EAAE,EAAE;YACrD,IAAI,CAAC;gBACH,IAAI,CAAC,YAAY,CAAC,yBAAyB,EAAE,WAAW,CAAC,CAAC;gBAC1D,IAAI,CAAC,YAAY,CAAC,gBAAgB,EAAE,WAAW,CAAC,CAAC;gBACjD,MAAM,MAAM,GAAG,MAAM,EAAE,CAAC,GAAG,IAAI,CAAC,CAAC;gBACjC,IAAI,CAAC,YAAY,CAAC,kBAAkB,EAAE,MAAM,CAAC,CAAC;gBAC9C,IAAI,OAAO,MAAM,KAAK,QAAQ,EAAE,CAAC;oBAC/B,IAAI,CAAC,YAAY,CAAC,iBAAiB,EAAE,MAAM,CAAC,CAAC;gBAC/C,CAAC;gBACD,IAAI,CAAC,SAAS,CAAC,EAAE,IAAI,EAAE,cAAc,CAAC,EAAE,EAAE,CAAC,CAAC;gBAC5C,OAAO,MAAM,CAAC;YAChB,CAAC;YAAC,OAAO,CAAC,EAAE,CAAC;gBACX,IAAI,CAAC,YAAY,CAAC,kBAAkB,EAAE,MAAM,CAAC,CAAC;gBAC9C,IAAI,CAAC,SAAS,CAAC,EAAE,IAAI,EAAE,cAAc,CAAC,KAAK,EAAE,OAAO,EAAE,MAAM,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC;gBACnE,IAAI,CAAC,eAAe,CAAC,CAAC,YAAY,KAAK,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,IAAI,KAAK,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC;gBACpE,MAAM,CAAC,CAAC;YACV,CAAC;oBAAS,CAAC;gBACT,IAAI,CAAC,GAAG,EAAE,CAAC;YACb,CAAC;QACH,CAAC,CAAC,CAAC;IACL,CAAC,CAAC;AACJ,CAAC"}

package/dist/ext/index.d.ts

@@ -0,0 +1,12 @@
+export { getTracer } from "./base.js";
+export { toolSpan } from "./tool.js";
+export type { ToolSpanOptions } from "./tool.js";
+export { embeddingSpan, EmbeddingSpan } from "./embedding.js";
+export type { EmbeddingSpanOptions } from "./embedding.js";
+export { retrievalSpan, RetrievalSpan, encodeDocuments } from "./retrieval.js";
+export type { Document, RetrievalSpanOptions } from "./retrieval.js";
+export { rerankerSpan, RerankerSpan } from "./reranker.js";
+export type { RerankerSpanOptions } from "./reranker.js";
+export { guardrailSpan } from "./guardrail.js";
+export type { GuardrailSpanOptions } from "./guardrail.js";
+//# sourceMappingURL=index.d.ts.map

package/dist/ext/index.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../../src/ext/index.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,SAAS,EAAE,MAAM,WAAW,CAAC;AACtC,OAAO,EAAE,QAAQ,EAAE,MAAM,WAAW,CAAC;AACrC,YAAY,EAAE,eAAe,EAAE,MAAM,WAAW,CAAC;AACjD,OAAO,EAAE,aAAa,EAAE,aAAa,EAAE,MAAM,gBAAgB,CAAC;AAC9D,YAAY,EAAE,oBAAoB,EAAE,MAAM,gBAAgB,CAAC;AAC3D,OAAO,EAAE,aAAa,EAAE,aAAa,EAAE,eAAe,EAAE,MAAM,gBAAgB,CAAC;AAC/E,YAAY,EAAE,QAAQ,EAAE,oBAAoB,EAAE,MAAM,gBAAgB,CAAC;AACrE,OAAO,EAAE,YAAY,EAAE,YAAY,EAAE,MAAM,eAAe,CAAC;AAC3D,YAAY,EAAE,mBAAmB,EAAE,MAAM,eAAe,CAAC;AACzD,OAAO,EAAE,aAAa,EAAE,MAAM,gBAAgB,CAAC;AAC/C,YAAY,EAAE,oBAAoB,EAAE,MAAM,gBAAgB,CAAC"}

package/dist/ext/index.js

@@ -0,0 +1,7 @@
+export { getTracer } from "./base.js";
+export { toolSpan } from "./tool.js";
+export { embeddingSpan, EmbeddingSpan } from "./embedding.js";
+export { retrievalSpan, RetrievalSpan, encodeDocuments } from "./retrieval.js";
+export { rerankerSpan, RerankerSpan } from "./reranker.js";
+export { guardrailSpan } from "./guardrail.js";
+//# sourceMappingURL=index.js.map

package/dist/ext/index.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"index.js","sourceRoot":"","sources":["../../src/ext/index.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,SAAS,EAAE,MAAM,WAAW,CAAC;AACtC,OAAO,EAAE,QAAQ,EAAE,MAAM,WAAW,CAAC;AAErC,OAAO,EAAE,aAAa,EAAE,aAAa,EAAE,MAAM,gBAAgB,CAAC;AAE9D,OAAO,EAAE,aAAa,EAAE,aAAa,EAAE,eAAe,EAAE,MAAM,gBAAgB,CAAC;AAE/E,OAAO,EAAE,YAAY,EAAE,YAAY,EAAE,MAAM,eAAe,CAAC;AAE3D,OAAO,EAAE,aAAa,EAAE,MAAM,gBAAgB,CAAC"}

package/dist/ext/reranker.d.ts

@@ -0,0 +1,20 @@
+import { type Document } from "./retrieval.js";
+export interface RerankerSpanOptions {
+    name?: string;
+    modelName?: string;
+    topK?: number;
+}
+export declare function rerankerSpan<A extends unknown[], R>(options: RerankerSpanOptions, fn: (...args: A) => R | Promise<R>): (...args: A) => Promise<R>;
+export declare class RerankerSpan {
+    private readonly spanName;
+    private readonly modelName?;
+    private readonly topK?;
+    private span;
+    constructor(name?: string, modelName?: string, topK?: number);
+    start(): void;
+    setQuery(query: string): void;
+    setInputDocuments(docs: Document[]): void;
+    setOutputDocuments(docs: Document[]): void;
+    end(): void;
+}
+//# sourceMappingURL=reranker.d.ts.map

package/dist/ext/reranker.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"reranker.d.ts","sourceRoot":"","sources":["../../src/ext/reranker.ts"],"names":[],"mappings":"AAEA,OAAO,EAAE,KAAK,QAAQ,EAAmB,MAAM,gBAAgB,CAAC;AAEhE,MAAM,WAAW,mBAAmB;IAClC,IAAI,CAAC,EAAE,MAAM,CAAC;IACd,SAAS,CAAC,EAAE,MAAM,CAAC;IACnB,IAAI,CAAC,EAAE,MAAM,CAAC;CACf;AAED,wBAAgB,YAAY,CAAC,CAAC,SAAS,OAAO,EAAE,EAAE,CAAC,EACjD,OAAO,EAAE,mBAAmB,EAC5B,EAAE,EAAE,CAAC,GAAG,IAAI,EAAE,CAAC,KAAK,CAAC,GAAG,OAAO,CAAC,CAAC,CAAC,GACjC,CAAC,GAAG,IAAI,EAAE,CAAC,KAAK,OAAO,CAAC,CAAC,CAAC,CA4C5B;AAED,qBAAa,YAAY;IACvB,OAAO,CAAC,QAAQ,CAAC,QAAQ,CAAS;IAClC,OAAO,CAAC,QAAQ,CAAC,SAAS,CAAC,CAAS;IACpC,OAAO,CAAC,QAAQ,CAAC,IAAI,CAAC,CAAS;IAC/B,OAAO,CAAC,IAAI,CAAqB;gBAErB,IAAI,SAAa,EAAE,SAAS,CAAC,EAAE,MAAM,EAAE,IAAI,CAAC,EAAE,MAAM;IAMhE,KAAK,IAAI,IAAI;IAYb,QAAQ,CAAC,KAAK,EAAE,MAAM,GAAG,IAAI;IAI7B,iBAAiB,CAAC,IAAI,EAAE,QAAQ,EAAE,GAAG,IAAI;IAIzC,kBAAkB,CAAC,IAAI,EAAE,QAAQ,EAAE,GAAG,IAAI;IAI1C,GAAG,IAAI,IAAI;CAGZ"}

package/dist/ext/reranker.js

@@ -0,0 +1,72 @@
+import { getTracer } from "./base.js";
+import { encodeDocuments } from "./retrieval.js";
+export function rerankerSpan(options, fn) {
+    const spanName = options.name ?? "reranker";
+    return async (...args) => {
+        const tracer = getTracer();
+        return tracer.startActiveSpan(spanName, async (span) => {
+            try {
+                span.setAttribute("openinference.span.kind", "RERANKER");
+                if (options.modelName) {
+                    span.setAttribute("reranker.model_name", options.modelName);
+                }
+                if (options.topK !== undefined) {
+                    span.setAttribute("reranker.top_k", options.topK);
+                }
+                const query = args.find((a) => typeof a === "string");
+                if (typeof query === "string") {
+                    span.setAttribute("reranker.query", query);
+                }
+                const inputDocs = args.find((a) => Array.isArray(a) && a.length > 0 && a[0]?.content !== undefined);
+                if (inputDocs) {
+                    span.setAttribute("reranker.input_documents", encodeDocuments(inputDocs));
+                }
+                const result = await fn(...args);
+                if (Array.isArray(result) &&
+                    result.length > 0 &&
+                    result[0]?.content !== undefined) {
+                    span.setAttribute("reranker.output_documents", encodeDocuments(result));
+                }
+                return result;
+            }
+            finally {
+                span.end();
+            }
+        });
+    };
+}
+export class RerankerSpan {
+    spanName;
+    modelName;
+    topK;
+    span = null;
+    constructor(name = "reranker", modelName, topK) {
+        this.spanName = name;
+        this.modelName = modelName;
+        this.topK = topK;
+    }
+    start() {
+        const tracer = getTracer();
+        this.span = tracer.startSpan(this.spanName);
+        this.span.setAttribute("openinference.span.kind", "RERANKER");
+        if (this.modelName) {
+            this.span.setAttribute("reranker.model_name", this.modelName);
+        }
+        if (this.topK !== undefined) {
+            this.span.setAttribute("reranker.top_k", this.topK);
+        }
+    }
+    setQuery(query) {
+        this.span?.setAttribute("reranker.query", query);
+    }
+    setInputDocuments(docs) {
+        this.span?.setAttribute("reranker.input_documents", encodeDocuments(docs));
+    }
+    setOutputDocuments(docs) {
+        this.span?.setAttribute("reranker.output_documents", encodeDocuments(docs));
+    }
+    end() {
+        this.span?.end();
+    }
+}
+//# sourceMappingURL=reranker.js.map

package/dist/ext/reranker.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"reranker.js","sourceRoot":"","sources":["../../src/ext/reranker.ts"],"names":[],"mappings":"AACA,OAAO,EAAE,SAAS,EAAE,MAAM,WAAW,CAAC;AACtC,OAAO,EAAiB,eAAe,EAAE,MAAM,gBAAgB,CAAC;AAQhE,MAAM,UAAU,YAAY,CAC1B,OAA4B,EAC5B,EAAkC;IAElC,MAAM,QAAQ,GAAG,OAAO,CAAC,IAAI,IAAI,UAAU,CAAC;IAE5C,OAAO,KAAK,EAAE,GAAG,IAAO,EAAc,EAAE;QACtC,MAAM,MAAM,GAAG,SAAS,EAAE,CAAC;QAC3B,OAAO,MAAM,CAAC,eAAe,CAAC,QAAQ,EAAE,KAAK,EAAE,IAAI,EAAE,EAAE;YACrD,IAAI,CAAC;gBACH,IAAI,CAAC,YAAY,CAAC,yBAAyB,EAAE,UAAU,CAAC,CAAC;gBACzD,IAAI,OAAO,CAAC,SAAS,EAAE,CAAC;oBACtB,IAAI,CAAC,YAAY,CAAC,qBAAqB,EAAE,OAAO,CAAC,SAAS,CAAC,CAAC;gBAC9D,CAAC;gBACD,IAAI,OAAO,CAAC,IAAI,KAAK,SAAS,EAAE,CAAC;oBAC/B,IAAI,CAAC,YAAY,CAAC,gBAAgB,EAAE,OAAO,CAAC,IAAI,CAAC,CAAC;gBACpD,CAAC;gBACD,MAAM,KAAK,GAAG,IAAI,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,OAAO,CAAC,KAAK,QAAQ,CAAC,CAAC;gBACtD,IAAI,OAAO,KAAK,KAAK,QAAQ,EAAE,CAAC;oBAC9B,IAAI,CAAC,YAAY,CAAC,gBAAgB,EAAE,KAAK,CAAC,CAAC;gBAC7C,CAAC;gBACD,MAAM,SAAS,GAAG,IAAI,CAAC,IAAI,CACzB,CAAC,CAAC,EAAE,EAAE,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,MAAM,GAAG,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,EAAE,OAAO,KAAK,SAAS,CACvE,CAAC;gBACF,IAAI,SAAS,EAAE,CAAC;oBACd,IAAI,CAAC,YAAY,CACf,0BAA0B,EAC1B,eAAe,CAAC,SAAuB,CAAC,CACzC,CAAC;gBACJ,CAAC;gBACD,MAAM,MAAM,GAAG,MAAM,EAAE,CAAC,GAAG,IAAI,CAAC,CAAC;gBACjC,IACE,KAAK,CAAC,OAAO,CAAC,MAAM,CAAC;oBACrB,MAAM,CAAC,MAAM,GAAG,CAAC;oBACjB,MAAM,CAAC,CAAC,CAAC,EAAE,OAAO,KAAK,SAAS,EAChC,CAAC;oBACD,IAAI,CAAC,YAAY,CACf,2BAA2B,EAC3B,eAAe,CAAC,MAAoB,CAAC,CACtC,CAAC;gBACJ,CAAC;gBACD,OAAO,MAAM,CAAC;YAChB,CAAC;oBAAS,CAAC;gBACT,IAAI,CAAC,GAAG,EAAE,CAAC;YACb,CAAC;QACH,CAAC,CAAC,CAAC;IACL,CAAC,CAAC;AACJ,CAAC;AAED,MAAM,OAAO,YAAY;IACN,QAAQ,CAAS;IACjB,SAAS,CAAU;IACnB,IAAI,CAAU;IACvB,IAAI,GAAgB,IAAI,CAAC;IAEjC,YAAY,IAAI,GAAG,UAAU,EAAE,SAAkB,EAAE,IAAa;QAC9D,IAAI,CAAC,QAAQ,GAAG,IAAI,CAAC;QACrB,IAAI,CAAC,SAAS,GAAG,SAAS,CAAC;QAC3B,IAAI,CAAC,IAAI,GAAG,IAAI,CAAC;IACnB,CAAC;IAED,KAAK;QACH,MAAM,MAAM,GAAG,SAAS,EAAE,CAAC;QAC3B,IAAI,CAAC,IAAI,GAAG,MAAM,CAAC,SAAS,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;QAC5C,IAAI,CAAC,IAAI,CAAC,YAAY,CAAC,yBAAyB,EAAE,UAAU,CAAC,CAAC;QAC9D,IAAI,IAAI,CAAC,SAAS,EAAE,CAAC;YACnB,IAAI,CAAC,IAAI,CAAC,YAAY,CAAC,qBAAqB,EAAE,IAAI,CAAC,SAAS,CAAC,CAAC;QAChE,CAAC;QACD,IAAI,IAAI,CAAC,IAAI,KAAK,SAAS,EAAE,CAAC;YAC5B,IAAI,CAAC,IAAI,CAAC,YAAY,CAAC,gBAAgB,EAAE,IAAI,CAAC,IAAI,CAAC,CAAC;QACtD,CAAC;IACH,CAAC;IAED,QAAQ,CAAC,KAAa;QACpB,IAAI,CAAC,IAAI,EAAE,YAAY,CAAC,gBAAgB,EAAE,KAAK,CAAC,CAAC;IACnD,CAAC;IAED,iBAAiB,CAAC,IAAgB;QAChC,IAAI,CAAC,IAAI,EAAE,YAAY,CAAC,0BAA0B,EAAE,eAAe,CAAC,IAAI,CAAC,CAAC,CAAC;IAC7E,CAAC;IAED,kBAAkB,CAAC,IAAgB;QACjC,IAAI,CAAC,IAAI,EAAE,YAAY,CAAC,2BAA2B,EAAE,eAAe,CAAC,IAAI,CAAC,CAAC,CAAC;IAC9E,CAAC;IAED,GAAG;QACD,IAAI,CAAC,IAAI,EAAE,GAAG,EAAE,CAAC;IACnB,CAAC;CACF"}