trodo-node 1.1.0 → 2.1.0

package/dist/esm/otel/autoInstrument.js

@@ -0,0 +1,248 @@
+import { getActiveContext } from './context.js';
+function hrToIso(hr) {
+    if (!hr)
+        return undefined;
+    const ms = hr[0] * 1000 + Math.floor(hr[1] / 1e6);
+    return new Date(ms).toISOString();
+}
+function inferKind(attrs) {
+    if (!attrs)
+        return 'generic';
+    if (attrs['gen_ai.tool.name'])
+        return 'tool';
+    if (attrs['gen_ai.operation.name'] || attrs['gen_ai.request.model'])
+        return 'llm';
+    if (attrs['db.system'] || attrs['retrieval.query'])
+        return 'retrieval';
+    return 'generic';
+}
+function pickNum(attrs, keys) {
+    for (const k of keys) {
+        const v = attrs[k];
+        if (typeof v === 'number')
+            return v;
+        if (typeof v === 'string' && v !== '' && !isNaN(Number(v)))
+            return Number(v);
+    }
+    return undefined;
+}
+function pickStr(attrs, keys) {
+    for (const k of keys) {
+        const v = attrs[k];
+        if (typeof v === 'string' && v.length)
+            return v;
+    }
+    return undefined;
+}
+/**
+ * Translate an OTel span to our TrodoSpan shape using GenAI semantic
+ * conventions (gen_ai.*). Accepts both stable and legacy/experimental keys.
+ */
+export function otelSpanToTrodoSpan(otel) {
+    const ctx = getActiveContext();
+    if (!ctx)
+        return null; // outside of wrapAgent — drop
+    const sc = otel.spanContext?.();
+    if (!sc)
+        return null;
+    const attrs = otel.attributes || {};
+    const kind = inferKind(attrs);
+    const startedAt = hrToIso(otel.startTime) || new Date().toISOString();
+    const endedAt = hrToIso(otel.endTime);
+    const durationMs = otel.startTime && otel.endTime
+        ? Math.max(0, (otel.endTime[0] - otel.startTime[0]) * 1000 +
+            (otel.endTime[1] - otel.startTime[1]) / 1e6)
+        : undefined;
+    const model = pickStr(attrs, [
+        'gen_ai.request.model',
+        'gen_ai.response.model',
+        'llm.request.model',
+    ]);
+    const provider = pickStr(attrs, ['gen_ai.system', 'llm.vendor']);
+    const inputTokens = pickNum(attrs, [
+        'gen_ai.usage.input_tokens',
+        'gen_ai.usage.prompt_tokens',
+        'llm.usage.prompt_tokens',
+    ]);
+    const outputTokens = pickNum(attrs, [
+        'gen_ai.usage.output_tokens',
+        'gen_ai.usage.completion_tokens',
+        'llm.usage.completion_tokens',
+    ]);
+    const temperature = pickNum(attrs, ['gen_ai.request.temperature']);
+    const toolName = pickStr(attrs, ['gen_ai.tool.name']);
+    const input = pickStr(attrs, ['gen_ai.prompt', 'llm.prompts']);
+    const output = pickStr(attrs, ['gen_ai.completion', 'llm.completion']);
+    const span = {
+        span_id: sc.spanId,
+        run_id: ctx.runId,
+        parent_span_id: otel.parentSpanId || ctx.spanId,
+        kind,
+        name: otel.name,
+        status: otel.status?.code === 2 ? 'error' : 'ok',
+        started_at: startedAt,
+        ended_at: endedAt,
+        duration_ms: durationMs,
+        model,
+        provider,
+        input_tokens: inputTokens,
+        output_tokens: outputTokens,
+        temperature,
+        tool_name: toolName,
+        input,
+        output,
+        attributes: attrs,
+    };
+    return span;
+}
+/**
+ * Adapter that lets the OTel SDK push spans into our processor.
+ * Register with `sdk.addSpanProcessor(otelAdapter(processor))` in your app
+ * bootstrap.
+ */
+export function otelAdapter(processor) {
+    return {
+        onStart() {
+            /* noop */
+        },
+        onEnd(otel) {
+            const span = otelSpanToTrodoSpan(otel);
+            if (span)
+                processor.enqueueSpan(span);
+        },
+        async forceFlush() {
+            await processor.forceFlush();
+        },
+        async shutdown() {
+            await processor.shutdown();
+        },
+    };
+}
+function buildInstrumentors() {
+    return [
+        {
+            id: 'anthropic',
+            factory: () => {
+                // eslint-disable-next-line @typescript-eslint/no-var-requires
+                const m = require('@opentelemetry/instrumentation-anthropic');
+                return new m.AnthropicInstrumentation();
+            },
+        },
+        {
+            id: 'openai',
+            factory: () => {
+                // eslint-disable-next-line @typescript-eslint/no-var-requires
+                const m = require('@opentelemetry/instrumentation-openai');
+                return new m.OpenAIInstrumentation();
+            },
+        },
+        {
+            id: 'langchain',
+            factory: () => {
+                // eslint-disable-next-line @typescript-eslint/no-var-requires
+                const m = require('@opentelemetry/instrumentation-langchain');
+                return new m.LangChainInstrumentation();
+            },
+        },
+        {
+            id: 'bedrock',
+            factory: () => {
+                // eslint-disable-next-line @typescript-eslint/no-var-requires
+                const m = require('@opentelemetry/instrumentation-bedrock');
+                return new m.BedrockInstrumentation();
+            },
+        },
+        {
+            id: 'cohere',
+            factory: () => {
+                // eslint-disable-next-line @typescript-eslint/no-var-requires
+                const m = require('@opentelemetry/instrumentation-cohere');
+                return new m.CohereInstrumentation();
+            },
+        },
+        {
+            id: 'google-generativeai',
+            factory: () => {
+                // eslint-disable-next-line @typescript-eslint/no-var-requires
+                const m = require('@opentelemetry/instrumentation-google-generativeai');
+                return new m.GoogleGenerativeAIInstrumentation();
+            },
+        },
+        {
+            id: 'vertexai',
+            factory: () => {
+                // eslint-disable-next-line @typescript-eslint/no-var-requires
+                const m = require('@opentelemetry/instrumentation-vertexai');
+                return new m.VertexAIInstrumentation();
+            },
+        },
+        {
+            id: 'llama-index',
+            factory: () => {
+                // eslint-disable-next-line @typescript-eslint/no-var-requires
+                const m = require('@opentelemetry/instrumentation-llamaindex');
+                return new m.LlamaIndexInstrumentation();
+            },
+        },
+        {
+            id: 'http',
+            factory: () => {
+                // eslint-disable-next-line @typescript-eslint/no-var-requires
+                const m = require('@opentelemetry/instrumentation-http');
+                return new m.HttpInstrumentation();
+            },
+        },
+        {
+            id: 'fetch',
+            factory: () => {
+                // eslint-disable-next-line @typescript-eslint/no-var-requires
+                const m = require('@opentelemetry/instrumentation-fetch');
+                return new m.FetchInstrumentation();
+            },
+        },
+    ];
+}
+/**
+ * Best-effort: register OTel auto-instrumentations if the user has them
+ * installed. This is optional — users can also configure OTel themselves and
+ * pass `otelAdapter(processor)` as a span processor.
+ *
+ * Returns the list of instrumentation ids that were successfully registered.
+ *
+ * Vercel AI SDK note: if you set `experimental_telemetry: { isEnabled: true,
+ * tracer: trace.getTracer('ai') }` on generateText/streamText calls, those
+ * spans flow through the SDK-registered OTel SpanProcessor automatically.
+ */
+export function enableAutoInstrument(opts) {
+    const disabled = new Set(opts.disable || []);
+    const active = [];
+    try {
+        // eslint-disable-next-line @typescript-eslint/no-var-requires
+        const { NodeSDK } = require('@opentelemetry/sdk-node');
+        // eslint-disable-next-line @typescript-eslint/no-var-requires
+        const { Resource } = require('@opentelemetry/resources');
+        const instrumentations = [];
+        for (const spec of buildInstrumentors()) {
+            if (disabled.has(spec.id))
+                continue;
+            try {
+                instrumentations.push(spec.factory());
+                active.push(spec.id);
+            }
+            catch {
+                /* package not installed — skip */
+            }
+        }
+        const sdk = new NodeSDK({
+            resource: new Resource({ 'service.name': 'trodo-agent' }),
+            spanProcessor: otelAdapter(opts.processor),
+            instrumentations,
+        });
+        sdk.start();
+    }
+    catch {
+        // @opentelemetry/sdk-node not installed; wrapAgent + withSpan still work.
+    }
+    return active;
+}
+//# sourceMappingURL=autoInstrument.js.map

package/dist/esm/otel/autoInstrument.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"autoInstrument.js","sourceRoot":"","sources":["../../../src/otel/autoInstrument.ts"],"names":[],"mappings":"AAWA,OAAO,EAAE,gBAAgB,EAAE,MAAM,cAAc,CAAC;AA+BhD,SAAS,OAAO,CAAC,EAAgC;IAC/C,IAAI,CAAC,EAAE;QAAE,OAAO,SAAS,CAAC;IAC1B,MAAM,EAAE,GAAG,EAAE,CAAC,CAAC,CAAC,GAAG,IAAI,GAAG,IAAI,CAAC,KAAK,CAAC,EAAE,CAAC,CAAC,CAAC,GAAG,GAAG,CAAC,CAAC;IAClD,OAAO,IAAI,IAAI,CAAC,EAAE,CAAC,CAAC,WAAW,EAAE,CAAC;AACpC,CAAC;AAED,SAAS,SAAS,CAAC,KAA0C;IAC3D,IAAI,CAAC,KAAK;QAAE,OAAO,SAAS,CAAC;IAC7B,IAAI,KAAK,CAAC,kBAAkB,CAAC;QAAE,OAAO,MAAM,CAAC;IAC7C,IAAI,KAAK,CAAC,uBAAuB,CAAC,IAAI,KAAK,CAAC,sBAAsB,CAAC;QAAE,OAAO,KAAK,CAAC;IAClF,IAAI,KAAK,CAAC,WAAW,CAAC,IAAI,KAAK,CAAC,iBAAiB,CAAC;QAAE,OAAO,WAAW,CAAC;IACvE,OAAO,SAAS,CAAC;AACnB,CAAC;AAED,SAAS,OAAO,CACd,KAA8B,EAC9B,IAAc;IAEd,KAAK,MAAM,CAAC,IAAI,IAAI,EAAE,CAAC;QACrB,MAAM,CAAC,GAAG,KAAK,CAAC,CAAC,CAAC,CAAC;QACnB,IAAI,OAAO,CAAC,KAAK,QAAQ;YAAE,OAAO,CAAC,CAAC;QACpC,IAAI,OAAO,CAAC,KAAK,QAAQ,IAAI,CAAC,KAAK,EAAE,IAAI,CAAC,KAAK,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC;YAAE,OAAO,MAAM,CAAC,CAAC,CAAC,CAAC;IAC/E,CAAC;IACD,OAAO,SAAS,CAAC;AACnB,CAAC;AAED,SAAS,OAAO,CACd,KAA8B,EAC9B,IAAc;IAEd,KAAK,MAAM,CAAC,IAAI,IAAI,EAAE,CAAC;QACrB,MAAM,CAAC,GAAG,KAAK,CAAC,CAAC,CAAC,CAAC;QACnB,IAAI,OAAO,CAAC,KAAK,QAAQ,IAAI,CAAC,CAAC,MAAM;YAAE,OAAO,CAAC,CAAC;IAClD,CAAC;IACD,OAAO,SAAS,CAAC;AACnB,CAAC;AAED;;;GAGG;AACH,MAAM,UAAU,mBAAmB,CAAC,IAAkB;IACpD,MAAM,GAAG,GAAG,gBAAgB,EAAE,CAAC;IAC/B,IAAI,CAAC,GAAG;QAAE,OAAO,IAAI,CAAC,CAAC,8BAA8B;IACrD,MAAM,EAAE,GAAG,IAAI,CAAC,WAAW,EAAE,EAAE,CAAC;IAChC,IAAI,CAAC,EAAE;QAAE,OAAO,IAAI,CAAC;IAErB,MAAM,KAAK,GAAG,IAAI,CAAC,UAAU,IAAI,EAAE,CAAC;IACpC,MAAM,IAAI,GAAG,SAAS,CAAC,KAAK,CAAC,CAAC;IAC9B,MAAM,SAAS,GAAG,OAAO,CAAC,IAAI,CAAC,SAAS,CAAC,IAAI,IAAI,IAAI,EAAE,CAAC,WAAW,EAAE,CAAC;IACtE,MAAM,OAAO,GAAG,OAAO,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;IACtC,MAAM,UAAU,GACd,IAAI,CAAC,SAAS,IAAI,IAAI,CAAC,OAAO;QAC5B,CAAC,CAAC,IAAI,CAAC,GAAG,CACN,CAAC,EACD,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC,CAAC,GAAG,IAAI,CAAC,SAAS,CAAC,CAAC,CAAC,CAAC,GAAG,IAAI;YAC1C,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC,CAAC,GAAG,IAAI,CAAC,SAAS,CAAC,CAAC,CAAC,CAAC,GAAG,GAAG,CAC9C;QACH,CAAC,CAAC,SAAS,CAAC;IAEhB,MAAM,KAAK,GAAG,OAAO,CAAC,KAAK,EAAE;QAC3B,sBAAsB;QACtB,uBAAuB;QACvB,mBAAmB;KACpB,CAAC,CAAC;IACH,MAAM,QAAQ,GAAG,OAAO,CAAC,KAAK,EAAE,CAAC,eAAe,EAAE,YAAY,CAAC,CAAC,CAAC;IACjE,MAAM,WAAW,GAAG,OAAO,CAAC,KAAK,EAAE;QACjC,2BAA2B;QAC3B,4BAA4B;QAC5B,yBAAyB;KAC1B,CAAC,CAAC;IACH,MAAM,YAAY,GAAG,OAAO,CAAC,KAAK,EAAE;QAClC,4BAA4B;QAC5B,gCAAgC;QAChC,6BAA6B;KAC9B,CAAC,CAAC;IACH,MAAM,WAAW,GAAG,OAAO,CAAC,KAAK,EAAE,CAAC,4BAA4B,CAAC,CAAC,CAAC;IACnE,MAAM,QAAQ,GAAG,OAAO,CAAC,KAAK,EAAE,CAAC,kBAAkB,CAAC,CAAC,CAAC;IACtD,MAAM,KAAK,GAAG,OAAO,CAAC,KAAK,EAAE,CAAC,eAAe,EAAE,aAAa,CAAC,CAAC,CAAC;IAC/D,MAAM,MAAM,GAAG,OAAO,CAAC,KAAK,EAAE,CAAC,mBAAmB,EAAE,gBAAgB,CAAC,CAAC,CAAC;IAEvE,MAAM,IAAI,GAAc;QACtB,OAAO,EAAE,EAAE,CAAC,MAAM;QAClB,MAAM,EAAE,GAAG,CAAC,KAAK;QACjB,cAAc,EAAE,IAAI,CAAC,YAAY,IAAI,GAAG,CAAC,MAAM;QAC/C,IAAI;QACJ,IAAI,EAAE,IAAI,CAAC,IAAI;QACf,MAAM,EAAE,IAAI,CAAC,MAAM,EAAE,IAAI,KAAK,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC,IAAI;QAChD,UAAU,EAAE,SAAS;QACrB,QAAQ,EAAE,OAAO;QACjB,WAAW,EAAE,UAAU;QACvB,KAAK;QACL,QAAQ;QACR,YAAY,EAAE,WAAW;QACzB,aAAa,EAAE,YAAY;QAC3B,WAAW;QACX,SAAS,EAAE,QAAQ;QACnB,KAAK;QACL,MAAM;QACN,UAAU,EAAE,KAAK;KAClB,CAAC;IACF,OAAO,IAAI,CAAC;AACd,CAAC;AAED;;;;GAIG;AACH,MAAM,UAAU,WAAW,CAAC,SAA6B;IAMvD,OAAO;QACL,OAAO;YACL,UAAU;QACZ,CAAC;QACD,KAAK,CAAC,IAAkB;YACtB,MAAM,IAAI,GAAG,mBAAmB,CAAC,IAAI,CAAC,CAAC;YACvC,IAAI,IAAI;gBAAE,SAAS,CAAC,WAAW,CAAC,IAAI,CAAC,CAAC;QACxC,CAAC;QACD,KAAK,CAAC,UAAU;YACd,MAAM,SAAS,CAAC,UAAU,EAAE,CAAC;QAC/B,CAAC;QACD,KAAK,CAAC,QAAQ;YACZ,MAAM,SAAS,CAAC,QAAQ,EAAE,CAAC;QAC7B,CAAC;KACF,CAAC;AACJ,CAAC;AAOD,SAAS,kBAAkB;IACzB,OAAO;QACL;YACE,EAAE,EAAE,WAAW;YACf,OAAO,EAAE,GAAG,EAAE;gBACZ,8DAA8D;gBAC9D,MAAM,CAAC,GAAG,OAAO,CAAC,0CAA0C,CAAC,CAAC;gBAC9D,OAAO,IAAI,CAAC,CAAC,wBAAwB,EAAE,CAAC;YAC1C,CAAC;SACF;QACD;YACE,EAAE,EAAE,QAAQ;YACZ,OAAO,EAAE,GAAG,EAAE;gBACZ,8DAA8D;gBAC9D,MAAM,CAAC,GAAG,OAAO,CAAC,uCAAuC,CAAC,CAAC;gBAC3D,OAAO,IAAI,CAAC,CAAC,qBAAqB,EAAE,CAAC;YACvC,CAAC;SACF;QACD;YACE,EAAE,EAAE,WAAW;YACf,OAAO,EAAE,GAAG,EAAE;gBACZ,8DAA8D;gBAC9D,MAAM,CAAC,GAAG,OAAO,CAAC,0CAA0C,CAAC,CAAC;gBAC9D,OAAO,IAAI,CAAC,CAAC,wBAAwB,EAAE,CAAC;YAC1C,CAAC;SACF;QACD;YACE,EAAE,EAAE,SAAS;YACb,OAAO,EAAE,GAAG,EAAE;gBACZ,8DAA8D;gBAC9D,MAAM,CAAC,GAAG,OAAO,CAAC,wCAAwC,CAAC,CAAC;gBAC5D,OAAO,IAAI,CAAC,CAAC,sBAAsB,EAAE,CAAC;YACxC,CAAC;SACF;QACD;YACE,EAAE,EAAE,QAAQ;YACZ,OAAO,EAAE,GAAG,EAAE;gBACZ,8DAA8D;gBAC9D,MAAM,CAAC,GAAG,OAAO,CAAC,uCAAuC,CAAC,CAAC;gBAC3D,OAAO,IAAI,CAAC,CAAC,qBAAqB,EAAE,CAAC;YACvC,CAAC;SACF;QACD;YACE,EAAE,EAAE,qBAAqB;YACzB,OAAO,EAAE,GAAG,EAAE;gBACZ,8DAA8D;gBAC9D,MAAM,CAAC,GAAG,OAAO,CAAC,oDAAoD,CAAC,CAAC;gBACxE,OAAO,IAAI,CAAC,CAAC,iCAAiC,EAAE,CAAC;YACnD,CAAC;SACF;QACD;YACE,EAAE,EAAE,UAAU;YACd,OAAO,EAAE,GAAG,EAAE;gBACZ,8DAA8D;gBAC9D,MAAM,CAAC,GAAG,OAAO,CAAC,yCAAyC,CAAC,CAAC;gBAC7D,OAAO,IAAI,CAAC,CAAC,uBAAuB,EAAE,CAAC;YACzC,CAAC;SACF;QACD;YACE,EAAE,EAAE,aAAa;YACjB,OAAO,EAAE,GAAG,EAAE;gBACZ,8DAA8D;gBAC9D,MAAM,CAAC,GAAG,OAAO,CAAC,2CAA2C,CAAC,CAAC;gBAC/D,OAAO,IAAI,CAAC,CAAC,yBAAyB,EAAE,CAAC;YAC3C,CAAC;SACF;QACD;YACE,EAAE,EAAE,MAAM;YACV,OAAO,EAAE,GAAG,EAAE;gBACZ,8DAA8D;gBAC9D,MAAM,CAAC,GAAG,OAAO,CAAC,qCAAqC,CAAC,CAAC;gBACzD,OAAO,IAAI,CAAC,CAAC,mBAAmB,EAAE,CAAC;YACrC,CAAC;SACF;QACD;YACE,EAAE,EAAE,OAAO;YACX,OAAO,EAAE,GAAG,EAAE;gBACZ,8DAA8D;gBAC9D,MAAM,CAAC,GAAG,OAAO,CAAC,sCAAsC,CAAC,CAAC;gBAC1D,OAAO,IAAI,CAAC,CAAC,oBAAoB,EAAE,CAAC;YACtC,CAAC;SACF;KACF,CAAC;AACJ,CAAC;AAED;;;;;;;;;;GAUG;AACH,MAAM,UAAU,oBAAoB,CAAC,IAA2B;IAC9D,MAAM,QAAQ,GAAG,IAAI,GAAG,CAAC,IAAI,CAAC,OAAO,IAAI,EAAE,CAAC,CAAC;IAC7C,MAAM,MAAM,GAAwB,EAAE,CAAC;IACvC,IAAI,CAAC;QACH,8DAA8D;QAC9D,MAAM,EAAE,OAAO,EAAE,GAAG,OAAO,CAAC,yBAAyB,CAAC,CAAC;QACvD,8DAA8D;QAC9D,MAAM,EAAE,QAAQ,EAAE,GAAG,OAAO,CAAC,0BAA0B,CAAC,CAAC;QACzD,MAAM,gBAAgB,GAAc,EAAE,CAAC;QACvC,KAAK,MAAM,IAAI,IAAI,kBAAkB,EAAE,EAAE,CAAC;YACxC,IAAI,QAAQ,CAAC,GAAG,CAAC,IAAI,CAAC,EAAE,CAAC;gBAAE,SAAS;YACpC,IAAI,CAAC;gBACH,gBAAgB,CAAC,IAAI,CAAC,IAAI,CAAC,OAAO,EAAE,CAAC,CAAC;gBACtC,MAAM,CAAC,IAAI,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;YACvB,CAAC;YAAC,MAAM,CAAC;gBACP,kCAAkC;YACpC,CAAC;QACH,CAAC;QAED,MAAM,GAAG,GAAG,IAAI,OAAO,CAAC;YACtB,QAAQ,EAAE,IAAI,QAAQ,CAAC,EAAE,cAAc,EAAE,aAAa,EAAE,CAAC;YACzD,aAAa,EAAE,WAAW,CAAC,IAAI,CAAC,SAAS,CAAC;YAC1C,gBAAgB;SACjB,CAAC,CAAC;QACH,GAAG,CAAC,KAAK,EAAE,CAAC;IACd,CAAC;IAAC,MAAM,CAAC;QACP,0EAA0E;IAC5E,CAAC;IACD,OAAO,MAAM,CAAC;AAChB,CAAC"}

package/dist/esm/otel/context.d.ts

@@ -0,0 +1,26 @@
+/**
+ * Run/span active-context — used by wrapAgent, joinRun, and withSpan to
+ * thread parent_span_id and run_id through nested function calls without
+ * the user touching IDs.
+ *
+ * Backed by AsyncLocalStorage in Node (Node >= 14). Falls back to a global
+ * store in other runtimes (best-effort — only the top of each call chain
+ * is preserved).
+ *
+ * When a user registers a real OpenTelemetry SDK, their OTel ContextManager
+ * supersedes this for cross-package propagation. This lightweight store is
+ * the baseline so wrapAgent works without any external OTel dependency.
+ */
+import type { TrodoSpanProcessor } from './processor.js';
+export interface ActiveSpanContext {
+    runId: string;
+    spanId: string;
+    parentSpanId: string | null;
+    teamSiteId: string;
+    processor: TrodoSpanProcessor;
+}
+export declare function getActiveContext(): ActiveSpanContext | null;
+export declare function runWithContext<R>(ctx: ActiveSpanContext, fn: () => R): R;
+/** Async variant — context is preserved across awaits via AsyncLocalStorage. */
+export declare function runWithContextAsync<R>(ctx: ActiveSpanContext, fn: () => Promise<R>): Promise<R>;
+//# sourceMappingURL=context.d.ts.map

package/dist/esm/otel/context.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"context.d.ts","sourceRoot":"","sources":["../../../src/otel/context.ts"],"names":[],"mappings":"AAAA;;;;;;;;;;;;GAYG;AACH,OAAO,KAAK,EAAE,kBAAkB,EAAE,MAAM,gBAAgB,CAAC;AAEzD,MAAM,WAAW,iBAAiB;IAChC,KAAK,EAAE,MAAM,CAAC;IACd,MAAM,EAAE,MAAM,CAAC;IACf,YAAY,EAAE,MAAM,GAAG,IAAI,CAAC;IAC5B,UAAU,EAAE,MAAM,CAAC;IACnB,SAAS,EAAE,kBAAkB,CAAC;CAC/B;AAsBD,wBAAgB,gBAAgB,IAAI,iBAAiB,GAAG,IAAI,CAG3D;AAED,wBAAgB,cAAc,CAAC,CAAC,EAAE,GAAG,EAAE,iBAAiB,EAAE,EAAE,EAAE,MAAM,CAAC,GAAG,CAAC,CASxE;AAED,gFAAgF;AAChF,wBAAgB,mBAAmB,CAAC,CAAC,EACnC,GAAG,EAAE,iBAAiB,EACtB,EAAE,EAAE,MAAM,OAAO,CAAC,CAAC,CAAC,GACnB,OAAO,CAAC,CAAC,CAAC,CAOZ"}

package/dist/esm/otel/context.js

@@ -0,0 +1,44 @@
+let AsyncLocalStorageCtor = null;
+try {
+    // eslint-disable-next-line @typescript-eslint/no-var-requires
+    AsyncLocalStorageCtor = require('node:async_hooks').AsyncLocalStorage;
+}
+catch {
+    try {
+        // eslint-disable-next-line @typescript-eslint/no-var-requires
+        AsyncLocalStorageCtor = require('async_hooks').AsyncLocalStorage;
+    }
+    catch {
+        AsyncLocalStorageCtor = null;
+    }
+}
+const als = AsyncLocalStorageCtor ? new AsyncLocalStorageCtor() : null;
+let fallbackStore = null;
+export function getActiveContext() {
+    if (als)
+        return als.getStore() ?? null;
+    return fallbackStore;
+}
+export function runWithContext(ctx, fn) {
+    if (als)
+        return als.run(ctx, fn);
+    const prev = fallbackStore;
+    fallbackStore = ctx;
+    try {
+        return fn();
+    }
+    finally {
+        fallbackStore = prev;
+    }
+}
+/** Async variant — context is preserved across awaits via AsyncLocalStorage. */
+export function runWithContextAsync(ctx, fn) {
+    if (als)
+        return als.run(ctx, fn);
+    const prev = fallbackStore;
+    fallbackStore = ctx;
+    return fn().finally(() => {
+        fallbackStore = prev;
+    });
+}
+//# sourceMappingURL=context.js.map

package/dist/esm/otel/context.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"context.js","sourceRoot":"","sources":["../../../src/otel/context.ts"],"names":[],"mappings":"AAuBA,IAAI,qBAAqB,GAGb,IAAI,CAAC;AAEjB,IAAI,CAAC;IACH,8DAA8D;IAC9D,qBAAqB,GAAG,OAAO,CAAC,kBAAkB,CAAC,CAAC,iBAAiB,CAAC;AACxE,CAAC;AAAC,MAAM,CAAC;IACP,IAAI,CAAC;QACH,8DAA8D;QAC9D,qBAAqB,GAAG,OAAO,CAAC,aAAa,CAAC,CAAC,iBAAiB,CAAC;IACnE,CAAC;IAAC,MAAM,CAAC;QACP,qBAAqB,GAAG,IAAI,CAAC;IAC/B,CAAC;AACH,CAAC;AAED,MAAM,GAAG,GAAG,qBAAqB,CAAC,CAAC,CAAC,IAAI,qBAAqB,EAAqB,CAAC,CAAC,CAAC,IAAI,CAAC;AAC1F,IAAI,aAAa,GAA6B,IAAI,CAAC;AAEnD,MAAM,UAAU,gBAAgB;IAC9B,IAAI,GAAG;QAAE,OAAO,GAAG,CAAC,QAAQ,EAAE,IAAI,IAAI,CAAC;IACvC,OAAO,aAAa,CAAC;AACvB,CAAC;AAED,MAAM,UAAU,cAAc,CAAI,GAAsB,EAAE,EAAW;IACnE,IAAI,GAAG;QAAE,OAAO,GAAG,CAAC,GAAG,CAAC,GAAG,EAAE,EAAE,CAAC,CAAC;IACjC,MAAM,IAAI,GAAG,aAAa,CAAC;IAC3B,aAAa,GAAG,GAAG,CAAC;IACpB,IAAI,CAAC;QACH,OAAO,EAAE,EAAE,CAAC;IACd,CAAC;YAAS,CAAC;QACT,aAAa,GAAG,IAAI,CAAC;IACvB,CAAC;AACH,CAAC;AAED,gFAAgF;AAChF,MAAM,UAAU,mBAAmB,CACjC,GAAsB,EACtB,EAAoB;IAEpB,IAAI,GAAG;QAAE,OAAO,GAAG,CAAC,GAAG,CAAC,GAAG,EAAE,EAAE,CAAC,CAAC;IACjC,MAAM,IAAI,GAAG,aAAa,CAAC;IAC3B,aAAa,GAAG,GAAG,CAAC;IACpB,OAAO,EAAE,EAAE,CAAC,OAAO,CAAC,GAAG,EAAE;QACvB,aAAa,GAAG,IAAI,CAAC;IACvB,CAAC,CAAC,CAAC;AACL,CAAC"}

package/dist/esm/otel/helpers.d.ts

@@ -0,0 +1,119 @@
+import { type WithSpanOptions } from './wrapAgent.js';
+import type { TrodoSpanProcessor } from './processor.js';
+export interface ToolOptions {
+    name?: string;
+    kind?: 'tool' | 'retrieval' | 'generic';
+}
+type AnyFn<Args extends unknown[], R> = (...args: Args) => Promise<R> | R;
+/**
+ * Wrap a function as a tool span — supports two forms:
+ *
+ *   // name-first (uselemma-compatible):
+ *   const fetchUser = trodo.tool('fetch_user', async (uid) => { ... });
+ *
+ *   // function-first (backward-compatible):
+ *   const fetchUser = trodo.tool(async (uid) => { ... }, { name: 'fetch_user' });
+ */
+export declare function tool<Args extends unknown[], R>(name: string, fn: AnyFn<Args, R>): (...args: Args) => Promise<R>;
+export declare function tool<Args extends unknown[], R>(fn: AnyFn<Args, R>, options?: ToolOptions): (...args: Args) => Promise<R>;
+/**
+ * Wrap a function as a generic span. Mirrors the uselemma `trace()` helper.
+ *
+ *   const prepare = trodo.trace('prepare', async (input) => { ... });
+ */
+export declare function trace<Args extends unknown[], R>(name: string, fn: AnyFn<Args, R>, options?: WithSpanOptions): (...args: Args) => Promise<R>;
+/** Wrap a retriever / vector search as a kind='retrieval' span. */
+export declare function retrieval<Args extends unknown[], R>(name: string, fn: AnyFn<Args, R>, options?: WithSpanOptions): (...args: Args) => Promise<R>;
+export interface LlmOptions extends WithSpanOptions {
+    model?: string;
+    provider?: string;
+    temperature?: number;
+    extractUsage?: (result: unknown) => {
+        inputTokens?: number;
+        outputTokens?: number;
+    };
+}
+/**
+ * Wrap an LLM call — auto-extracts tokens from OpenAI/Anthropic/Gemini
+ * response shapes.
+ *
+ *   const answer = trodo.llm('answer', callOpenAI, {
+ *     model: 'gpt-4o-mini', provider: 'openai',
+ *   });
+ */
+export declare function llm<Args extends unknown[], R>(name: string, fn: AnyFn<Args, R>, options?: LlmOptions): (...args: Args) => Promise<R>;
+/**
+ * Return a raw OTel tracer — the Trodo processor is already subscribed.
+ * Use for advanced cases where you want to emit spans with the raw OTel
+ * API; they flow through to Trodo like any other span.
+ */
+export declare function getTracer(name?: string): unknown;
+export interface TrackLlmCallParams {
+    model?: string;
+    provider?: string;
+    inputTokens?: number;
+    outputTokens?: number;
+    prompt?: unknown;
+    completion?: unknown;
+    temperature?: number;
+    cost?: number;
+    name?: string;
+    metadata?: Record<string, unknown>;
+}
+/**
+ * Record a one-shot LLM span for a raw-HTTP caller. Opens and immediately
+ * closes a span(kind='llm') populated with the model + token counts +
+ * prompt/completion. No-op outside an active run context.
+ *
+ * Usage:
+ *   const resp = await fetch(url, { body: JSON.stringify(body) }).then(r => r.json());
+ *   await trodo.trackLlmCall({
+ *     model: 'gemini-2.5-flash',
+ *     provider: 'google',
+ *     inputTokens: resp.usageMetadata.promptTokenCount,
+ *     outputTokens: resp.usageMetadata.candidatesTokenCount,
+ *     prompt: body,
+ *     completion: resp,
+ *   });
+ */
+export declare function trackLlmCall(params: TrackLlmCallParams): Promise<void>;
+/** Minimal subset of Express Request/Response/NextFn we actually touch. */
+interface ExpressReq {
+    headers: Record<string, string | string[] | undefined>;
+    method?: string;
+    url?: string;
+    originalUrl?: string;
+    path?: string;
+}
+interface ExpressRes {
+    statusCode?: number;
+    on?: (event: string, cb: () => void) => void;
+}
+type NextFn = (err?: unknown) => void;
+export interface ExpressMiddlewareDeps {
+    processor: TrodoSpanProcessor;
+    teamSiteId: string;
+}
+/**
+ * Return an Express/Connect middleware that auto-joins inbound runs.
+ *
+ * If the inbound request has X-Trodo-Run-Id, every span created while
+ * handling the request nests under the caller's run. Otherwise the
+ * handler runs with no active context (no-op tracking).
+ *
+ * Usage:
+ *   import express from 'express';
+ *   import trodo from 'trodo-node';
+ *   trodo.init({ siteId: '...' });
+ *   const app = express();
+ *   app.use(trodo.expressMiddleware());
+ */
+export declare function expressMiddleware(deps: ExpressMiddlewareDeps): (req: ExpressReq, res: ExpressRes, next: NextFn) => void;
+/**
+ * Return HTTP headers carrying the current run/span context. Use when
+ * making outbound HTTP calls to downstream services so they can joinRun
+ * instead of creating their own runs.
+ */
+export declare function propagationHeaders(): Record<string, string>;
+export {};
+//# sourceMappingURL=helpers.d.ts.map

package/dist/esm/otel/helpers.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"helpers.d.ts","sourceRoot":"","sources":["../../../src/otel/helpers.ts"],"names":[],"mappings":"AASA,OAAO,EAIL,KAAK,eAAe,EACrB,MAAM,gBAAgB,CAAC;AACxB,OAAO,KAAK,EAAE,kBAAkB,EAAE,MAAM,gBAAgB,CAAC;AAEzD,MAAM,WAAW,WAAW;IAC1B,IAAI,CAAC,EAAE,MAAM,CAAC;IACd,IAAI,CAAC,EAAE,MAAM,GAAG,WAAW,GAAG,SAAS,CAAC;CACzC;AAED,KAAK,KAAK,CAAC,IAAI,SAAS,OAAO,EAAE,EAAE,CAAC,IAAI,CAAC,GAAG,IAAI,EAAE,IAAI,KAAK,OAAO,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC;AAuC1E;;;;;;;;GAQG;AACH,wBAAgB,IAAI,CAAC,IAAI,SAAS,OAAO,EAAE,EAAE,CAAC,EAC5C,IAAI,EAAE,MAAM,EACZ,EAAE,EAAE,KAAK,CAAC,IAAI,EAAE,CAAC,CAAC,GACjB,CAAC,GAAG,IAAI,EAAE,IAAI,KAAK,OAAO,CAAC,CAAC,CAAC,CAAC;AACjC,wBAAgB,IAAI,CAAC,IAAI,SAAS,OAAO,EAAE,EAAE,CAAC,EAC5C,EAAE,EAAE,KAAK,CAAC,IAAI,EAAE,CAAC,CAAC,EAClB,OAAO,CAAC,EAAE,WAAW,GACpB,CAAC,GAAG,IAAI,EAAE,IAAI,KAAK,OAAO,CAAC,CAAC,CAAC,CAAC;AAiBjC;;;;GAIG;AACH,wBAAgB,KAAK,CAAC,IAAI,SAAS,OAAO,EAAE,EAAE,CAAC,EAC7C,IAAI,EAAE,MAAM,EACZ,EAAE,EAAE,KAAK,CAAC,IAAI,EAAE,CAAC,CAAC,EAClB,OAAO,GAAE,eAAoB,GAC5B,CAAC,GAAG,IAAI,EAAE,IAAI,KAAK,OAAO,CAAC,CAAC,CAAC,CAE/B;AAED,mEAAmE;AACnE,wBAAgB,SAAS,CAAC,IAAI,SAAS,OAAO,EAAE,EAAE,CAAC,EACjD,IAAI,EAAE,MAAM,EACZ,EAAE,EAAE,KAAK,CAAC,IAAI,EAAE,CAAC,CAAC,EAClB,OAAO,GAAE,eAAoB,GAC5B,CAAC,GAAG,IAAI,EAAE,IAAI,KAAK,OAAO,CAAC,CAAC,CAAC,CAE/B;AAED,MAAM,WAAW,UAAW,SAAQ,eAAe;IACjD,KAAK,CAAC,EAAE,MAAM,CAAC;IACf,QAAQ,CAAC,EAAE,MAAM,CAAC;IAClB,WAAW,CAAC,EAAE,MAAM,CAAC;IACrB,YAAY,CAAC,EAAE,CAAC,MAAM,EAAE,OAAO,KAAK;QAAE,WAAW,CAAC,EAAE,MAAM,CAAC;QAAC,YAAY,CAAC,EAAE,MAAM,CAAA;KAAE,CAAC;CACrF;AAmCD;;;;;;;GAOG;AACH,wBAAgB,GAAG,CAAC,IAAI,SAAS,OAAO,EAAE,EAAE,CAAC,EAC3C,IAAI,EAAE,MAAM,EACZ,EAAE,EAAE,KAAK,CAAC,IAAI,EAAE,CAAC,CAAC,EAClB,OAAO,GAAE,UAAe,GACvB,CAAC,GAAG,IAAI,EAAE,IAAI,KAAK,OAAO,CAAC,CAAC,CAAC,CAsB/B;AAED;;;;GAIG;AACH,wBAAgB,SAAS,CAAC,IAAI,SAAU,GAAG,OAAO,CAKjD;AAED,MAAM,WAAW,kBAAkB;IACjC,KAAK,CAAC,EAAE,MAAM,CAAC;IACf,QAAQ,CAAC,EAAE,MAAM,CAAC;IAClB,WAAW,CAAC,EAAE,MAAM,CAAC;IACrB,YAAY,CAAC,EAAE,MAAM,CAAC;IACtB,MAAM,CAAC,EAAE,OAAO,CAAC;IACjB,UAAU,CAAC,EAAE,OAAO,CAAC;IACrB,WAAW,CAAC,EAAE,MAAM,CAAC;IACrB,IAAI,CAAC,EAAE,MAAM,CAAC;IACd,IAAI,CAAC,EAAE,MAAM,CAAC;IACd,QAAQ,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,CAAC;CACpC;AAED;;;;;;;;;;;;;;;GAeG;AACH,wBAAsB,YAAY,CAAC,MAAM,EAAE,kBAAkB,GAAG,OAAO,CAAC,IAAI,CAAC,CA0B5E;AAUD,2EAA2E;AAC3E,UAAU,UAAU;IAClB,OAAO,EAAE,MAAM,CAAC,MAAM,EAAE,MAAM,GAAG,MAAM,EAAE,GAAG,SAAS,CAAC,CAAC;IACvD,MAAM,CAAC,EAAE,MAAM,CAAC;IAChB,GAAG,CAAC,EAAE,MAAM,CAAC;IACb,WAAW,CAAC,EAAE,MAAM,CAAC;IACrB,IAAI,CAAC,EAAE,MAAM,CAAC;CACf;AACD,UAAU,UAAU;IAClB,UAAU,CAAC,EAAE,MAAM,CAAC;IACpB,EAAE,CAAC,EAAE,CAAC,KAAK,EAAE,MAAM,EAAE,EAAE,EAAE,MAAM,IAAI,KAAK,IAAI,CAAC;CAC9C;AACD,KAAK,MAAM,GAAG,CAAC,GAAG,CAAC,EAAE,OAAO,KAAK,IAAI,CAAC;AAEtC,MAAM,WAAW,qBAAqB;IACpC,SAAS,EAAE,kBAAkB,CAAC;IAC9B,UAAU,EAAE,MAAM,CAAC;CACpB;AAED;;;;;;;;;;;;;GAaG;AACH,wBAAgB,iBAAiB,CAC/B,IAAI,EAAE,qBAAqB,GAC1B,CAAC,GAAG,EAAE,UAAU,EAAE,GAAG,EAAE,UAAU,EAAE,IAAI,EAAE,MAAM,KAAK,IAAI,CA+C1D;AAED;;;;GAIG;AACH,wBAAgB,kBAAkB,IAAI,MAAM,CAAC,MAAM,EAAE,MAAM,CAAC,CAM3D"}