shield360 1.0.0

package/dist/instrumentation/letta/index.d.ts

@@ -0,0 +1,10 @@
+import { InstrumentationBase, InstrumentationConfig, InstrumentationModuleDefinition } from '@opentelemetry/instrumentation';
+export interface LettaInstrumentationConfig extends InstrumentationConfig {
+}
+export default class Shield360LettaInstrumentation extends InstrumentationBase {
+    constructor(config?: LettaInstrumentationConfig);
+    protected init(): void | InstrumentationModuleDefinition | InstrumentationModuleDefinition[];
+    manualPatch(letta: any): void;
+    private _patchClass;
+    private _unpatchClass;
+}

package/dist/instrumentation/letta/index.js

@@ -0,0 +1,122 @@
+"use strict";
+var __importDefault = (this && this.__importDefault) || function (mod) {
+    return (mod && mod.__esModule) ? mod : { "default": mod };
+};
+Object.defineProperty(exports, "__esModule", { value: true });
+const instrumentation_1 = require("@opentelemetry/instrumentation");
+const api_1 = require("@opentelemetry/api");
+const constant_1 = require("../../constant");
+const wrapper_1 = __importDefault(require("./wrapper"));
+const AGENT_METHODS = [
+    ['create', 'letta.create'],
+    ['retrieve', 'letta.retrieve'],
+    ['update', 'letta.modify'],
+    ['delete', 'letta.delete'],
+    ['list', 'letta.list'],
+];
+const MESSAGE_METHODS = [
+    ['create', 'letta.create_message'],
+    ['stream', 'letta.create_stream'],
+    ['createAsync', 'letta.create_async'],
+    ['list', 'letta.list'],
+    ['cancel', 'letta.cancel'],
+    ['reset', 'letta.reset'],
+];
+const MEMORY_METHODS = [
+    ['create', 'letta.create'],
+    ['retrieve', 'letta.retrieve'],
+    ['update', 'letta.modify'],
+    ['delete', 'letta.delete'],
+    ['list', 'letta.list'],
+];
+const AGENT_BLOCK_METHODS = [
+    ['retrieve', 'letta.retrieve'],
+    ['update', 'letta.modify'],
+    ['list', 'letta.list'],
+    ['attach', 'letta.attach'],
+    ['detach', 'letta.detach'],
+];
+const AGENT_ATTACH_METHODS = [
+    ['list', 'letta.list'],
+    ['attach', 'letta.attach'],
+    ['detach', 'letta.detach'],
+];
+const PASSAGE_METHODS = [
+    ['create', 'letta.create'],
+    ['list', 'letta.list'],
+    ['delete', 'letta.delete'],
+];
+const LETTA_MODULES = [
+    { file: 'resources/agents/agents.js', className: 'Agents', methods: AGENT_METHODS, agentScoped: true },
+    { file: 'resources/agents/messages.js', className: 'Messages', methods: MESSAGE_METHODS, agentScoped: true },
+    { file: 'resources/blocks/blocks.js', className: 'Blocks', methods: MEMORY_METHODS, agentScoped: false },
+    { file: 'resources/agents/blocks.js', className: 'Blocks', methods: AGENT_BLOCK_METHODS, agentScoped: true },
+    { file: 'resources/tools.js', className: 'Tools', methods: MEMORY_METHODS, agentScoped: false },
+    { file: 'resources/agents/tools.js', className: 'Tools', methods: AGENT_ATTACH_METHODS, agentScoped: true },
+    { file: 'resources/folders/folders.js', className: 'Folders', methods: MEMORY_METHODS, agentScoped: false },
+    { file: 'resources/agents/folders.js', className: 'Folders', methods: AGENT_ATTACH_METHODS, agentScoped: true },
+    { file: 'resources/agents/passages.js', className: 'Passages', methods: PASSAGE_METHODS, agentScoped: true },
+];
+const PACKAGE = '@letta-ai/letta-client';
+const VERSIONS = ['>=0.1.0'];
+class Shield360LettaInstrumentation extends instrumentation_1.InstrumentationBase {
+    constructor(config = {}) {
+        super(`${constant_1.INSTRUMENTATION_PREFIX}/instrumentation-letta`, '1.0.0', config);
+    }
+    init() {
+        // The Letta client exposes resource classes as nested objects (client.agents,
+        // client.agents.messages, ...), each defined in its own internal module file.
+        // require-in-the-middle reaches those files via the `files` array, so every
+        // resource class is registered as an InstrumentationNodeModuleFile.
+        const files = LETTA_MODULES.map((mod) => new instrumentation_1.InstrumentationNodeModuleFile(`${PACKAGE}/${mod.file}`, VERSIONS, (moduleExports, moduleVersion) => {
+            this._patchClass(moduleExports?.[mod.className], mod.methods, mod.agentScoped, moduleVersion);
+            return moduleExports;
+        }, (moduleExports) => {
+            if (moduleExports !== undefined)
+                this._unpatchClass(moduleExports?.[mod.className], mod.methods);
+        }));
+        const module = new instrumentation_1.InstrumentationNodeModuleDefinition(PACKAGE, VERSIONS, (moduleExports) => moduleExports, () => { }, files);
+        return [module];
+    }
+    manualPatch(letta) {
+        // Fallback entry point. Best-effort: look each resource class up by name on the
+        // object passed in; the isWrapped guard in _patchClass keeps repeat calls safe.
+        for (const mod of LETTA_MODULES) {
+            this._patchClass(letta?.[mod.className], mod.methods, mod.agentScoped);
+        }
+    }
+    _patchClass(target, methods, agentScoped, moduleVersion) {
+        try {
+            const proto = target?.prototype;
+            if (!proto)
+                return;
+            for (const [method, endpoint] of methods) {
+                if (typeof proto[method] !== 'function')
+                    continue;
+                if ((0, instrumentation_1.isWrapped)(proto[method]))
+                    continue;
+                this._wrap(proto, method, wrapper_1.default._patchOperation(this.tracer, endpoint, agentScoped, moduleVersion));
+            }
+        }
+        catch (e) {
+            api_1.diag.error('letta instrumentation: error in _patch method', e);
+        }
+    }
+    _unpatchClass(target, methods) {
+        try {
+            const proto = target?.prototype;
+            if (!proto)
+                return;
+            for (const [method] of methods) {
+                if (typeof proto[method] === 'function' && (0, instrumentation_1.isWrapped)(proto[method])) {
+                    this._unwrap(proto, method);
+                }
+            }
+        }
+        catch {
+            /* ignore */
+        }
+    }
+}
+exports.default = Shield360LettaInstrumentation;
+//# sourceMappingURL=index.js.map

package/dist/instrumentation/letta/index.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"index.js","sourceRoot":"","sources":["../../../src/instrumentation/letta/index.ts"],"names":[],"mappings":";;;;;AAAA,oEAOwC;AACxC,4CAA0C;AAC1C,6CAAwD;AACxD,wDAAqC;AAqBrC,MAAM,aAAa,GAA4B;IAC7C,CAAC,QAAQ,EAAE,cAAc,CAAC;IAC1B,CAAC,UAAU,EAAE,gBAAgB,CAAC;IAC9B,CAAC,QAAQ,EAAE,cAAc,CAAC;IAC1B,CAAC,QAAQ,EAAE,cAAc,CAAC;IAC1B,CAAC,MAAM,EAAE,YAAY,CAAC;CACvB,CAAC;AAEF,MAAM,eAAe,GAA4B;IAC/C,CAAC,QAAQ,EAAE,sBAAsB,CAAC;IAClC,CAAC,QAAQ,EAAE,qBAAqB,CAAC;IACjC,CAAC,aAAa,EAAE,oBAAoB,CAAC;IACrC,CAAC,MAAM,EAAE,YAAY,CAAC;IACtB,CAAC,QAAQ,EAAE,cAAc,CAAC;IAC1B,CAAC,OAAO,EAAE,aAAa,CAAC;CACzB,CAAC;AAEF,MAAM,cAAc,GAA4B;IAC9C,CAAC,QAAQ,EAAE,cAAc,CAAC;IAC1B,CAAC,UAAU,EAAE,gBAAgB,CAAC;IAC9B,CAAC,QAAQ,EAAE,cAAc,CAAC;IAC1B,CAAC,QAAQ,EAAE,cAAc,CAAC;IAC1B,CAAC,MAAM,EAAE,YAAY,CAAC;CACvB,CAAC;AAEF,MAAM,mBAAmB,GAA4B;IACnD,CAAC,UAAU,EAAE,gBAAgB,CAAC;IAC9B,CAAC,QAAQ,EAAE,cAAc,CAAC;IAC1B,CAAC,MAAM,EAAE,YAAY,CAAC;IACtB,CAAC,QAAQ,EAAE,cAAc,CAAC;IAC1B,CAAC,QAAQ,EAAE,cAAc,CAAC;CAC3B,CAAC;AAEF,MAAM,oBAAoB,GAA4B;IACpD,CAAC,MAAM,EAAE,YAAY,CAAC;IACtB,CAAC,QAAQ,EAAE,cAAc,CAAC;IAC1B,CAAC,QAAQ,EAAE,cAAc,CAAC;CAC3B,CAAC;AAEF,MAAM,eAAe,GAA4B;IAC/C,CAAC,QAAQ,EAAE,cAAc,CAAC;IAC1B,CAAC,MAAM,EAAE,YAAY,CAAC;IACtB,CAAC,QAAQ,EAAE,cAAc,CAAC;CAC3B,CAAC;AAEF,MAAM,aAAa,GAAkB;IACnC,EAAE,IAAI,EAAE,4BAA4B,EAAE,SAAS,EAAE,QAAQ,EAAE,OAAO,EAAE,aAAa,EAAE,WAAW,EAAE,IAAI,EAAE;IACtG,EAAE,IAAI,EAAE,8BAA8B,EAAE,SAAS,EAAE,UAAU,EAAE,OAAO,EAAE,eAAe,EAAE,WAAW,EAAE,IAAI,EAAE;IAC5G,EAAE,IAAI,EAAE,4BAA4B,EAAE,SAAS,EAAE,QAAQ,EAAE,OAAO,EAAE,cAAc,EAAE,WAAW,EAAE,KAAK,EAAE;IACxG,EAAE,IAAI,EAAE,4BAA4B,EAAE,SAAS,EAAE,QAAQ,EAAE,OAAO,EAAE,mBAAmB,EAAE,WAAW,EAAE,IAAI,EAAE;IAC5G,EAAE,IAAI,EAAE,oBAAoB,EAAE,SAAS,EAAE,OAAO,EAAE,OAAO,EAAE,cAAc,EAAE,WAAW,EAAE,KAAK,EAAE;IAC/F,EAAE,IAAI,EAAE,2BAA2B,EAAE,SAAS,EAAE,OAAO,EAAE,OAAO,EAAE,oBAAoB,EAAE,WAAW,EAAE,IAAI,EAAE;IAC3G,EAAE,IAAI,EAAE,8BAA8B,EAAE,SAAS,EAAE,SAAS,EAAE,OAAO,EAAE,cAAc,EAAE,WAAW,EAAE,KAAK,EAAE;IAC3G,EAAE,IAAI,EAAE,6BAA6B,EAAE,SAAS,EAAE,SAAS,EAAE,OAAO,EAAE,oBAAoB,EAAE,WAAW,EAAE,IAAI,EAAE;IAC/G,EAAE,IAAI,EAAE,8BAA8B,EAAE,SAAS,EAAE,UAAU,EAAE,OAAO,EAAE,eAAe,EAAE,WAAW,EAAE,IAAI,EAAE;CAC7G,CAAC;AAEF,MAAM,OAAO,GAAG,wBAAwB,CAAC;AACzC,MAAM,QAAQ,GAAG,CAAC,SAAS,CAAC,CAAC;AAE7B,MAAqB,6BAA8B,SAAQ,qCAAmB;IAC5E,YAAY,SAAqC,EAAE;QACjD,KAAK,CAAC,GAAG,iCAAsB,wBAAwB,EAAE,OAAO,EAAE,MAAM,CAAC,CAAC;IAC5E,CAAC;IAES,IAAI;QACZ,8EAA8E;QAC9E,8EAA8E;QAC9E,4EAA4E;QAC5E,oEAAoE;QACpE,MAAM,KAAK,GAAG,aAAa,CAAC,GAAG,CAC7B,CAAC,GAAG,EAAE,EAAE,CACN,IAAI,+CAA6B,CAC/B,GAAG,OAAO,IAAI,GAAG,CAAC,IAAI,EAAE,EACxB,QAAQ,EACR,CAAC,aAAkB,EAAE,aAAsB,EAAE,EAAE;YAC7C,IAAI,CAAC,WAAW,CAAC,aAAa,EAAE,CAAC,GAAG,CAAC,SAAS,CAAC,EAAE,GAAG,CAAC,OAAO,EAAE,GAAG,CAAC,WAAW,EAAE,aAAa,CAAC,CAAC;YAC9F,OAAO,aAAa,CAAC;QACvB,CAAC,EACD,CAAC,aAAkB,EAAE,EAAE;YACrB,IAAI,aAAa,KAAK,SAAS;gBAAE,IAAI,CAAC,aAAa,CAAC,aAAa,EAAE,CAAC,GAAG,CAAC,SAAS,CAAC,EAAE,GAAG,CAAC,OAAO,CAAC,CAAC;QACnG,CAAC,CACF,CACJ,CAAC;QAEF,MAAM,MAAM,GAAG,IAAI,qDAAmC,CACpD,OAAO,EACP,QAAQ,EACR,CAAC,aAAkB,EAAE,EAAE,CAAC,aAAa,EACrC,GAAG,EAAE,GAAE,CAAC,EACR,KAAK,CACN,CAAC;QAEF,OAAO,CAAC,MAAM,CAAC,CAAC;IAClB,CAAC;IAEM,WAAW,CAAC,KAAU;QAC3B,gFAAgF;QAChF,gFAAgF;QAChF,KAAK,MAAM,GAAG,IAAI,aAAa,EAAE,CAAC;YAChC,IAAI,CAAC,WAAW,CAAC,KAAK,EAAE,CAAC,GAAG,CAAC,SAAS,CAAC,EAAE,GAAG,CAAC,OAAO,EAAE,GAAG,CAAC,WAAW,CAAC,CAAC;QACzE,CAAC;IACH,CAAC;IAEO,WAAW,CACjB,MAAW,EACX,OAAgC,EAChC,WAAoB,EACpB,aAAsB;QAEtB,IAAI,CAAC;YACH,MAAM,KAAK,GAAG,MAAM,EAAE,SAAS,CAAC;YAChC,IAAI,CAAC,KAAK;gBAAE,OAAO;YACnB,KAAK,MAAM,CAAC,MAAM,EAAE,QAAQ,CAAC,IAAI,OAAO,EAAE,CAAC;gBACzC,IAAI,OAAO,KAAK,CAAC,MAAM,CAAC,KAAK,UAAU;oBAAE,SAAS;gBAClD,IAAI,IAAA,2BAAS,EAAC,KAAK,CAAC,MAAM,CAAC,CAAC;oBAAE,SAAS;gBACvC,IAAI,CAAC,KAAK,CAAC,KAAK,EAAE,MAAM,EAAE,iBAAY,CAAC,eAAe,CAAC,IAAI,CAAC,MAAM,EAAE,QAAQ,EAAE,WAAW,EAAE,aAAa,CAAC,CAAC,CAAC;YAC7G,CAAC;QACH,CAAC;QAAC,OAAO,CAAC,EAAE,CAAC;YACX,UAAI,CAAC,KAAK,CAAC,+CAA+C,EAAE,CAAC,CAAC,CAAC;QACjE,CAAC;IACH,CAAC;IAEO,aAAa,CAAC,MAAW,EAAE,OAAgC;QACjE,IAAI,CAAC;YACH,MAAM,KAAK,GAAG,MAAM,EAAE,SAAS,CAAC;YAChC,IAAI,CAAC,KAAK;gBAAE,OAAO;YACnB,KAAK,MAAM,CAAC,MAAM,CAAC,IAAI,OAAO,EAAE,CAAC;gBAC/B,IAAI,OAAO,KAAK,CAAC,MAAM,CAAC,KAAK,UAAU,IAAI,IAAA,2BAAS,EAAC,KAAK,CAAC,MAAM,CAAC,CAAC,EAAE,CAAC;oBACpE,IAAI,CAAC,OAAO,CAAC,KAAK,EAAE,MAAM,CAAC,CAAC;gBAC9B,CAAC;YACH,CAAC;QACH,CAAC;QAAC,MAAM,CAAC;YACP,YAAY;QACd,CAAC;IACH,CAAC;CACF;AA5ED,gDA4EC"}

package/dist/instrumentation/letta/utils.d.ts

@@ -0,0 +1,24 @@
+import { Span } from '@opentelemetry/api';
+export declare const DEFAULT_MODEL = "gpt-4o";
+export declare const DEFAULT_SERVER_ADDRESS = "api.letta.com";
+export declare const DEFAULT_SERVER_PORT = 443;
+/**
+ * Operation type per Letta method name. Mirrors OPERATION_TYPE_MAP in the Python
+ * reference (sdk/python/src/shield360/instrumentation/letta/utils.py). The type is
+ * keyed on the method suffix so identical calls emit identical telemetry in both
+ * SDKs. Anything unmapped falls back to `workflow`.
+ */
+export declare const OPERATION_TYPE_MAP: Record<string, string>;
+export declare function getOperationType(endpoint: string): string;
+/**
+ * Span name generation ported from Python's get_span_name. The span name is built
+ * before the call runs, so it uses only the request: chat spans follow the
+ * provider pattern `chat {model}`; agent spans use `{operation} {agent_name}`;
+ * workflow/tool spans use the method suffix.
+ */
+export declare function getSpanName(operationType: string, endpoint: string, instance?: any, body?: any): string;
+/**
+ * Sets the common framework + Letta-specific attributes on a span. Mirrors
+ * process_letta_response + common_framework_span_attributes in the Python reference.
+ */
+export declare function processLettaResponse(span: Span, response: any, body: any, operationType: string, instance: any, durationSeconds: number, sdkVersion: string, agentId?: string): void;

package/dist/instrumentation/letta/utils.js

@@ -0,0 +1,343 @@
+"use strict";
+var __createBinding = (this && this.__createBinding) || (Object.create ? (function(o, m, k, k2) {
+    if (k2 === undefined) k2 = k;
+    var desc = Object.getOwnPropertyDescriptor(m, k);
+    if (!desc || ("get" in desc ? !m.__esModule : desc.writable || desc.configurable)) {
+      desc = { enumerable: true, get: function() { return m[k]; } };
+    }
+    Object.defineProperty(o, k2, desc);
+}) : (function(o, m, k, k2) {
+    if (k2 === undefined) k2 = k;
+    o[k2] = m[k];
+}));
+var __setModuleDefault = (this && this.__setModuleDefault) || (Object.create ? (function(o, v) {
+    Object.defineProperty(o, "default", { enumerable: true, value: v });
+}) : function(o, v) {
+    o["default"] = v;
+});
+var __importStar = (this && this.__importStar) || (function () {
+    var ownKeys = function(o) {
+        ownKeys = Object.getOwnPropertyNames || function (o) {
+            var ar = [];
+            for (var k in o) if (Object.prototype.hasOwnProperty.call(o, k)) ar[ar.length] = k;
+            return ar;
+        };
+        return ownKeys(o);
+    };
+    return function (mod) {
+        if (mod && mod.__esModule) return mod;
+        var result = {};
+        if (mod != null) for (var k = ownKeys(mod), i = 0; i < k.length; i++) if (k[i] !== "default") __createBinding(result, mod, k[i]);
+        __setModuleDefault(result, mod);
+        return result;
+    };
+})();
+var __importDefault = (this && this.__importDefault) || function (mod) {
+    return (mod && mod.__esModule) ? mod : { "default": mod };
+};
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.OPERATION_TYPE_MAP = exports.DEFAULT_SERVER_PORT = exports.DEFAULT_SERVER_ADDRESS = exports.DEFAULT_MODEL = void 0;
+exports.getOperationType = getOperationType;
+exports.getSpanName = getSpanName;
+exports.processLettaResponse = processLettaResponse;
+const semantic_conventions_1 = require("@opentelemetry/semantic-conventions");
+const config_1 = __importDefault(require("../../config"));
+const constant_1 = require("../../constant");
+const helpers_1 = __importStar(require("../../helpers"));
+const semantic_convention_1 = __importDefault(require("../../semantic-convention"));
+exports.DEFAULT_MODEL = 'gpt-4o';
+exports.DEFAULT_SERVER_ADDRESS = 'api.letta.com';
+exports.DEFAULT_SERVER_PORT = 443;
+/**
+ * Operation type per Letta method name. Mirrors OPERATION_TYPE_MAP in the Python
+ * reference (sdk/python/src/shield360/instrumentation/letta/utils.py). The type is
+ * keyed on the method suffix so identical calls emit identical telemetry in both
+ * SDKs. Anything unmapped falls back to `workflow`.
+ */
+exports.OPERATION_TYPE_MAP = {
+    create: semantic_convention_1.default.GEN_AI_OPERATION_TYPE_CREATE_AGENT,
+    retrieve: semantic_convention_1.default.GEN_AI_OPERATION_TYPE_AGENT,
+    modify: semantic_convention_1.default.GEN_AI_OPERATION_TYPE_AGENT,
+    delete: semantic_convention_1.default.GEN_AI_OPERATION_TYPE_AGENT,
+    list: semantic_convention_1.default.GEN_AI_OPERATION_TYPE_WORKFLOW,
+    create_stream: semantic_convention_1.default.GEN_AI_OPERATION_TYPE_CHAT,
+    create_message: semantic_convention_1.default.GEN_AI_OPERATION_TYPE_CHAT,
+    create_async: semantic_convention_1.default.GEN_AI_OPERATION_TYPE_CHAT,
+    cancel: semantic_convention_1.default.GEN_AI_OPERATION_TYPE_CHAT,
+    reset: semantic_convention_1.default.GEN_AI_OPERATION_TYPE_CHAT,
+    attach: semantic_convention_1.default.GEN_AI_OPERATION_TYPE_TOOLS,
+    detach: semantic_convention_1.default.GEN_AI_OPERATION_TYPE_TOOLS,
+};
+function getOperationType(endpoint) {
+    const method = endpoint.split('.').pop() || 'unknown';
+    return exports.OPERATION_TYPE_MAP[method] ?? semantic_convention_1.default.GEN_AI_OPERATION_TYPE_WORKFLOW;
+}
+/** Resolve the model name from response / request in the same priority order as Python. */
+function resolveModel(response, body) {
+    if (response?.llm_config?.model)
+        return response.llm_config.model;
+    if (body?.model)
+        return body.model;
+    return exports.DEFAULT_MODEL;
+}
+/**
+ * Span name generation ported from Python's get_span_name. The span name is built
+ * before the call runs, so it uses only the request: chat spans follow the
+ * provider pattern `chat {model}`; agent spans use `{operation} {agent_name}`;
+ * workflow/tool spans use the method suffix.
+ */
+function getSpanName(operationType, endpoint, instance, body) {
+    const method = endpoint.split('.').pop() || 'unknown';
+    if (operationType === semantic_convention_1.default.GEN_AI_OPERATION_TYPE_CHAT) {
+        return `chat ${body?.model || exports.DEFAULT_MODEL}`;
+    }
+    if (operationType === semantic_convention_1.default.GEN_AI_OPERATION_TYPE_CREATE_AGENT ||
+        operationType === semantic_convention_1.default.GEN_AI_OPERATION_TYPE_AGENT) {
+        let agentName = 'agent';
+        if (body?.name)
+            agentName = body.name;
+        else if (body?.slug)
+            agentName = body.slug;
+        else if (instance?.name)
+            agentName = instance.name;
+        else if (instance?.slug)
+            agentName = instance.slug;
+        else if (body?.agent_id)
+            agentName = `agent-${String(body.agent_id).slice(0, 8)}`;
+        if (agentName && agentName !== 'agent') {
+            agentName = agentName.replace(/ /g, '_').replace(/-/g, '_').toLowerCase();
+            if (agentName.length > 20)
+                agentName = agentName.slice(0, 20);
+        }
+        return `${operationType} ${agentName}`;
+    }
+    if (operationType === semantic_convention_1.default.GEN_AI_OPERATION_TYPE_WORKFLOW) {
+        return `workflow ${method}`;
+    }
+    if (operationType === semantic_convention_1.default.GEN_AI_OPERATION_TYPE_TOOLS) {
+        return `tool ${method}`;
+    }
+    return `${operationType} ${method}`;
+}
+function truncateContent(content) {
+    const limit = config_1.default.maxContentLength;
+    if (typeof limit === 'number' && limit > 0 && content.length > limit) {
+        return content.slice(0, limit);
+    }
+    return content;
+}
+function resolveServerInfo(instance) {
+    let baseUrl;
+    if (instance) {
+        baseUrl =
+            instance._client?.baseURL ??
+                instance._client_wrapper?._base_url ??
+                instance._client?.base_url ??
+                instance.baseURL ??
+                instance.base_url ??
+                instance._base_url;
+    }
+    if (baseUrl) {
+        try {
+            const parsed = new URL(String(baseUrl));
+            if (parsed.hostname) {
+                const port = parsed.port
+                    ? Number(parsed.port)
+                    : parsed.protocol === 'https:'
+                        ? 443
+                        : 80;
+                return [parsed.hostname, port];
+            }
+        }
+        catch {
+            /* ignore malformed base url */
+        }
+    }
+    return [exports.DEFAULT_SERVER_ADDRESS, exports.DEFAULT_SERVER_PORT];
+}
+function setLlmConfigAttributes(span, llmConfig) {
+    if (!llmConfig)
+        return;
+    if (llmConfig.model)
+        span.setAttribute(semantic_convention_1.default.GEN_AI_REQUEST_MODEL, llmConfig.model);
+    if (llmConfig.provider_name) {
+        span.setAttribute(semantic_convention_1.default.GEN_AI_REQUEST_PROVIDER, llmConfig.provider_name);
+    }
+    if (llmConfig.model_endpoint) {
+        span.setAttribute(semantic_convention_1.default.GEN_AI_ENDPOINT, llmConfig.model_endpoint);
+    }
+    if (llmConfig.temperature != null) {
+        span.setAttribute(semantic_convention_1.default.GEN_AI_REQUEST_TEMPERATURE, llmConfig.temperature);
+    }
+    if (llmConfig.max_tokens != null) {
+        span.setAttribute(semantic_convention_1.default.GEN_AI_REQUEST_MAX_TOKENS, llmConfig.max_tokens);
+    }
+    if (llmConfig.frequency_penalty != null) {
+        span.setAttribute(semantic_convention_1.default.GEN_AI_REQUEST_FREQUENCY_PENALTY, llmConfig.frequency_penalty);
+    }
+    if (llmConfig.context_window != null) {
+        span.setAttribute(semantic_convention_1.default.GEN_AI_REQUEST_CONTEXT_WINDOW, llmConfig.context_window);
+    }
+    if (llmConfig.enable_reasoner != null) {
+        span.setAttribute(semantic_convention_1.default.GEN_AI_REQUEST_ENABLE_REASONER, llmConfig.enable_reasoner);
+    }
+    if (llmConfig.reasoning_effort != null) {
+        span.setAttribute(semantic_convention_1.default.GEN_AI_REQUEST_REASONING_EFFORT, llmConfig.reasoning_effort);
+    }
+    if (llmConfig.max_reasoning_tokens != null) {
+        span.setAttribute(semantic_convention_1.default.GEN_AI_USAGE_REASONING_TOKENS, llmConfig.max_reasoning_tokens);
+    }
+    if (llmConfig.handle != null) {
+        span.setAttribute(semantic_convention_1.default.GEN_AI_MODEL_HANDLE, llmConfig.handle);
+    }
+}
+function setUsageAttributes(span, usage) {
+    if (!usage)
+        return;
+    if (usage.prompt_tokens != null) {
+        span.setAttribute(semantic_convention_1.default.GEN_AI_USAGE_INPUT_TOKENS, usage.prompt_tokens);
+    }
+    if (usage.completion_tokens != null) {
+        span.setAttribute(semantic_convention_1.default.GEN_AI_USAGE_OUTPUT_TOKENS, usage.completion_tokens);
+    }
+    if (usage.total_tokens != null) {
+        span.setAttribute(semantic_convention_1.default.GEN_AI_USAGE_TOTAL_TOKENS, usage.total_tokens);
+    }
+    if (usage.step_count != null) {
+        span.setAttribute(semantic_convention_1.default.GEN_AI_AGENT_STEP_COUNT, usage.step_count);
+    }
+    if (usage.run_ids != null) {
+        span.setAttribute(semantic_convention_1.default.GEN_AI_RUN_ID, String(usage.run_ids));
+    }
+    if (usage.steps_messages != null) {
+        span.setAttribute(semantic_convention_1.default.GEN_AI_AGENT_STEP_MESSAGES, JSON.stringify(usage.steps_messages));
+    }
+}
+function setRequestAttributes(span, body, operationType) {
+    if (!body || typeof body !== 'object')
+        return;
+    if (body.model != null)
+        span.setAttribute(semantic_convention_1.default.GEN_AI_REQUEST_MODEL, String(body.model));
+    if (body.temperature != null) {
+        span.setAttribute(semantic_convention_1.default.GEN_AI_REQUEST_TEMPERATURE, body.temperature);
+    }
+    if (body.max_tokens != null) {
+        span.setAttribute(semantic_convention_1.default.GEN_AI_REQUEST_MAX_TOKENS, body.max_tokens);
+    }
+    if (body.frequency_penalty != null) {
+        span.setAttribute(semantic_convention_1.default.GEN_AI_REQUEST_FREQUENCY_PENALTY, body.frequency_penalty);
+    }
+    if (body.stream != null)
+        span.setAttribute(semantic_convention_1.default.GEN_AI_REQUEST_IS_STREAM, body.stream);
+    else if (body.streaming != null) {
+        span.setAttribute(semantic_convention_1.default.GEN_AI_REQUEST_IS_STREAM, body.streaming);
+    }
+    if (body.run_async != null)
+        span.setAttribute(semantic_convention_1.default.GEN_AI_REQUEST_ASYNC, body.run_async);
+    if (body.return_message_sequence_no != null) {
+        span.setAttribute(semantic_convention_1.default.GEN_AI_REQUEST_RETURN_SEQUENCE_NO, body.return_message_sequence_no);
+    }
+    if (body.include_final_message != null) {
+        span.setAttribute(semantic_convention_1.default.GEN_AI_REQUEST_INCLUDE_FINAL_MESSAGE, body.include_final_message);
+    }
+    if (operationType === semantic_convention_1.default.GEN_AI_OPERATION_TYPE_CHAT && Array.isArray(body.messages)) {
+        span.setAttribute(semantic_convention_1.default.GEN_AI_REQUEST_MESSAGE_COUNT, body.messages.length);
+    }
+}
+function setLettaSpecificAttributes(span, body, response, operationType, agentId) {
+    let resolvedAgentId = agentId;
+    if (!resolvedAgentId && body?.agent_id)
+        resolvedAgentId = String(body.agent_id);
+    else if (!resolvedAgentId && response?.id)
+        resolvedAgentId = String(response.id);
+    if (resolvedAgentId)
+        span.setAttribute(semantic_convention_1.default.GEN_AI_AGENT_ID, resolvedAgentId);
+    if (response) {
+        if (response.name != null)
+            span.setAttribute(semantic_convention_1.default.GEN_AI_AGENT_NAME, String(response.name));
+        if (response.slug != null)
+            span.setAttribute(semantic_convention_1.default.GEN_AI_AGENT_SLUG, String(response.slug));
+        if (response.description != null) {
+            span.setAttribute(semantic_convention_1.default.GEN_AI_AGENT_DESCRIPTION, String(response.description));
+        }
+        if (response.agent_type != null) {
+            span.setAttribute(semantic_convention_1.default.GEN_AI_AGENT_TYPE, response.agent_type);
+        }
+        if (response.system != null) {
+            span.setAttribute(semantic_convention_1.default.GEN_AI_AGENT_INSTRUCTIONS, truncateContent(String(response.system)));
+        }
+        if (response.llm_config)
+            setLlmConfigAttributes(span, response.llm_config);
+        if (operationType === semantic_convention_1.default.GEN_AI_OPERATION_TYPE_CHAT && response.usage) {
+            setUsageAttributes(span, response.usage);
+        }
+    }
+    setRequestAttributes(span, body, operationType);
+    if (body?.name != null)
+        span.setAttribute(semantic_convention_1.default.GEN_AI_AGENT_NAME, String(body.name));
+    if (body?.slug != null)
+        span.setAttribute(semantic_convention_1.default.GEN_AI_AGENT_SLUG, String(body.slug));
+}
+function serializeContent(value) {
+    try {
+        return JSON.stringify(value);
+    }
+    catch {
+        return String(value);
+    }
+}
+function setContentAttributes(span, body, response) {
+    if (body?.messages != null) {
+        span.setAttribute(semantic_convention_1.default.GEN_AI_INPUT_MESSAGES, serializeContent(body.messages));
+    }
+    if (response?.messages != null) {
+        span.setAttribute(semantic_convention_1.default.GEN_AI_OUTPUT_MESSAGES, serializeContent(response.messages));
+    }
+}
+function calculateCost(span, response, model) {
+    const pricingInfo = config_1.default.pricingInfo;
+    if (!pricingInfo || !response)
+        return;
+    let usage = response.usage;
+    if (!usage && Array.isArray(response)) {
+        for (let i = response.length - 1; i >= 0; i -= 1) {
+            if (response[i]?.message_type === 'usage_statistics') {
+                usage = response[i];
+                break;
+            }
+        }
+    }
+    if (usage && usage.prompt_tokens != null && usage.completion_tokens != null) {
+        const actualModel = response.llm_config?.model || model;
+        const cost = helpers_1.default.getChatModelCost(actualModel, pricingInfo, usage.prompt_tokens, usage.completion_tokens);
+        span.setAttribute(semantic_convention_1.default.GEN_AI_USAGE_COST, cost);
+    }
+}
+/**
+ * Sets the common framework + Letta-specific attributes on a span. Mirrors
+ * process_letta_response + common_framework_span_attributes in the Python reference.
+ */
+function processLettaResponse(span, response, body, operationType, instance, durationSeconds, sdkVersion, agentId) {
+    const model = resolveModel(response, body);
+    const [serverAddress, serverPort] = resolveServerInfo(instance);
+    span.setAttribute(semantic_conventions_1.ATTR_TELEMETRY_SDK_NAME, constant_1.SDK_NAME);
+    span.setAttribute(semantic_convention_1.default.GEN_AI_SDK_VERSION, sdkVersion);
+    span.setAttribute(semantic_convention_1.default.GEN_AI_PROVIDER_NAME, semantic_convention_1.default.GEN_AI_SYSTEM_LETTA);
+    span.setAttribute(semantic_convention_1.default.GEN_AI_PROVIDER_NAME_OTEL, semantic_convention_1.default.GEN_AI_SYSTEM_LETTA);
+    span.setAttribute(semantic_convention_1.default.GEN_AI_OPERATION, operationType);
+    span.setAttribute(semantic_convention_1.default.GEN_AI_REQUEST_MODEL, model);
+    span.setAttribute(semantic_convention_1.default.SERVER_ADDRESS, serverAddress);
+    span.setAttribute(semantic_convention_1.default.SERVER_PORT, serverPort);
+    span.setAttribute(semantic_convention_1.default.ATTR_DEPLOYMENT_ENVIRONMENT, config_1.default.environment || '');
+    span.setAttribute(semantic_conventions_1.ATTR_SERVICE_NAME, config_1.default.applicationName || '');
+    span.setAttribute(semantic_convention_1.default.GEN_AI_CLIENT_OPERATION_DURATION, durationSeconds);
+    setLettaSpecificAttributes(span, body, response, operationType, agentId);
+    if (operationType === semantic_convention_1.default.GEN_AI_OPERATION_TYPE_CHAT && config_1.default.captureMessageContent) {
+        setContentAttributes(span, body, response);
+    }
+    if (operationType === semantic_convention_1.default.GEN_AI_OPERATION_TYPE_CHAT) {
+        calculateCost(span, response, model);
+    }
+    (0, helpers_1.applyCustomSpanAttributes)(span);
+}
+//# sourceMappingURL=utils.js.map

package/dist/instrumentation/letta/utils.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"utils.js","sourceRoot":"","sources":["../../../src/instrumentation/letta/utils.ts"],"names":[],"mappings":";;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;AAgCA,4CAGC;AAeD,kCAkCC;AAoND,oDAoCC;AA3UD,8EAAiG;AACjG,0DAA2C;AAC3C,6CAA0C;AAC1C,yDAA2E;AAC3E,oFAA2D;AAE9C,QAAA,aAAa,GAAG,QAAQ,CAAC;AACzB,QAAA,sBAAsB,GAAG,eAAe,CAAC;AACzC,QAAA,mBAAmB,GAAG,GAAG,CAAC;AAEvC;;;;;GAKG;AACU,QAAA,kBAAkB,GAA2B;IACxD,MAAM,EAAE,6BAAkB,CAAC,kCAAkC;IAC7D,QAAQ,EAAE,6BAAkB,CAAC,2BAA2B;IACxD,MAAM,EAAE,6BAAkB,CAAC,2BAA2B;IACtD,MAAM,EAAE,6BAAkB,CAAC,2BAA2B;IACtD,IAAI,EAAE,6BAAkB,CAAC,8BAA8B;IACvD,aAAa,EAAE,6BAAkB,CAAC,0BAA0B;IAC5D,cAAc,EAAE,6BAAkB,CAAC,0BAA0B;IAC7D,YAAY,EAAE,6BAAkB,CAAC,0BAA0B;IAC3D,MAAM,EAAE,6BAAkB,CAAC,0BAA0B;IACrD,KAAK,EAAE,6BAAkB,CAAC,0BAA0B;IACpD,MAAM,EAAE,6BAAkB,CAAC,2BAA2B;IACtD,MAAM,EAAE,6BAAkB,CAAC,2BAA2B;CACvD,CAAC;AAEF,SAAgB,gBAAgB,CAAC,QAAgB;IAC/C,MAAM,MAAM,GAAG,QAAQ,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC,GAAG,EAAE,IAAI,SAAS,CAAC;IACtD,OAAO,0BAAkB,CAAC,MAAM,CAAC,IAAI,6BAAkB,CAAC,8BAA8B,CAAC;AACzF,CAAC;AAED,2FAA2F;AAC3F,SAAS,YAAY,CAAC,QAAa,EAAE,IAAS;IAC5C,IAAI,QAAQ,EAAE,UAAU,EAAE,KAAK;QAAE,OAAO,QAAQ,CAAC,UAAU,CAAC,KAAK,CAAC;IAClE,IAAI,IAAI,EAAE,KAAK;QAAE,OAAO,IAAI,CAAC,KAAK,CAAC;IACnC,OAAO,qBAAa,CAAC;AACvB,CAAC;AAED;;;;;GAKG;AACH,SAAgB,WAAW,CAAC,aAAqB,EAAE,QAAgB,EAAE,QAAc,EAAE,IAAU;IAC7F,MAAM,MAAM,GAAG,QAAQ,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC,GAAG,EAAE,IAAI,SAAS,CAAC;IAEtD,IAAI,aAAa,KAAK,6BAAkB,CAAC,0BAA0B,EAAE,CAAC;QACpE,OAAO,QAAQ,IAAI,EAAE,KAAK,IAAI,qBAAa,EAAE,CAAC;IAChD,CAAC;IAED,IACE,aAAa,KAAK,6BAAkB,CAAC,kCAAkC;QACvE,aAAa,KAAK,6BAAkB,CAAC,2BAA2B,EAChE,CAAC;QACD,IAAI,SAAS,GAAG,OAAO,CAAC;QACxB,IAAI,IAAI,EAAE,IAAI;YAAE,SAAS,GAAG,IAAI,CAAC,IAAI,CAAC;aACjC,IAAI,IAAI,EAAE,IAAI;YAAE,SAAS,GAAG,IAAI,CAAC,IAAI,CAAC;aACtC,IAAI,QAAQ,EAAE,IAAI;YAAE,SAAS,GAAG,QAAQ,CAAC,IAAI,CAAC;aAC9C,IAAI,QAAQ,EAAE,IAAI;YAAE,SAAS,GAAG,QAAQ,CAAC,IAAI,CAAC;aAC9C,IAAI,IAAI,EAAE,QAAQ;YAAE,SAAS,GAAG,SAAS,MAAM,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC,KAAK,CAAC,CAAC,EAAE,CAAC,CAAC,EAAE,CAAC;QAElF,IAAI,SAAS,IAAI,SAAS,KAAK,OAAO,EAAE,CAAC;YACvC,SAAS,GAAG,SAAS,CAAC,OAAO,CAAC,IAAI,EAAE,GAAG,CAAC,CAAC,OAAO,CAAC,IAAI,EAAE,GAAG,CAAC,CAAC,WAAW,EAAE,CAAC;YAC1E,IAAI,SAAS,CAAC,MAAM,GAAG,EAAE;gBAAE,SAAS,GAAG,SAAS,CAAC,KAAK,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC;QAChE,CAAC;QACD,OAAO,GAAG,aAAa,IAAI,SAAS,EAAE,CAAC;IACzC,CAAC;IAED,IAAI,aAAa,KAAK,6BAAkB,CAAC,8BAA8B,EAAE,CAAC;QACxE,OAAO,YAAY,MAAM,EAAE,CAAC;IAC9B,CAAC;IAED,IAAI,aAAa,KAAK,6BAAkB,CAAC,2BAA2B,EAAE,CAAC;QACrE,OAAO,QAAQ,MAAM,EAAE,CAAC;IAC1B,CAAC;IAED,OAAO,GAAG,aAAa,IAAI,MAAM,EAAE,CAAC;AACtC,CAAC;AAED,SAAS,eAAe,CAAC,OAAe;IACtC,MAAM,KAAK,GAAG,gBAAe,CAAC,gBAAgB,CAAC;IAC/C,IAAI,OAAO,KAAK,KAAK,QAAQ,IAAI,KAAK,GAAG,CAAC,IAAI,OAAO,CAAC,MAAM,GAAG,KAAK,EAAE,CAAC;QACrE,OAAO,OAAO,CAAC,KAAK,CAAC,CAAC,EAAE,KAAK,CAAC,CAAC;IACjC,CAAC;IACD,OAAO,OAAO,CAAC;AACjB,CAAC;AAED,SAAS,iBAAiB,CAAC,QAAa;IACtC,IAAI,OAA2B,CAAC;IAChC,IAAI,QAAQ,EAAE,CAAC;QACb,OAAO;YACL,QAAQ,CAAC,OAAO,EAAE,OAAO;gBACzB,QAAQ,CAAC,eAAe,EAAE,SAAS;gBACnC,QAAQ,CAAC,OAAO,EAAE,QAAQ;gBAC1B,QAAQ,CAAC,OAAO;gBAChB,QAAQ,CAAC,QAAQ;gBACjB,QAAQ,CAAC,SAAS,CAAC;IACvB,CAAC;IACD,IAAI,OAAO,EAAE,CAAC;QACZ,IAAI,CAAC;YACH,MAAM,MAAM,GAAG,IAAI,GAAG,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC,CAAC;YACxC,IAAI,MAAM,CAAC,QAAQ,EAAE,CAAC;gBACpB,MAAM,IAAI,GAAG,MAAM,CAAC,IAAI;oBACtB,CAAC,CAAC,MAAM,CAAC,MAAM,CAAC,IAAI,CAAC;oBACrB,CAAC,CAAC,MAAM,CAAC,QAAQ,KAAK,QAAQ;wBAC5B,CAAC,CAAC,GAAG;wBACL,CAAC,CAAC,EAAE,CAAC;gBACT,OAAO,CAAC,MAAM,CAAC,QAAQ,EAAE,IAAI,CAAC,CAAC;YACjC,CAAC;QACH,CAAC;QAAC,MAAM,CAAC;YACP,+BAA+B;QACjC,CAAC;IACH,CAAC;IACD,OAAO,CAAC,8BAAsB,EAAE,2BAAmB,CAAC,CAAC;AACvD,CAAC;AAED,SAAS,sBAAsB,CAAC,IAAU,EAAE,SAAc;IACxD,IAAI,CAAC,SAAS;QAAE,OAAO;IACvB,IAAI,SAAS,CAAC,KAAK;QAAE,IAAI,CAAC,YAAY,CAAC,6BAAkB,CAAC,oBAAoB,EAAE,SAAS,CAAC,KAAK,CAAC,CAAC;IACjG,IAAI,SAAS,CAAC,aAAa,EAAE,CAAC;QAC5B,IAAI,CAAC,YAAY,CAAC,6BAAkB,CAAC,uBAAuB,EAAE,SAAS,CAAC,aAAa,CAAC,CAAC;IACzF,CAAC;IACD,IAAI,SAAS,CAAC,cAAc,EAAE,CAAC;QAC7B,IAAI,CAAC,YAAY,CAAC,6BAAkB,CAAC,eAAe,EAAE,SAAS,CAAC,cAAc,CAAC,CAAC;IAClF,CAAC;IACD,IAAI,SAAS,CAAC,WAAW,IAAI,IAAI,EAAE,CAAC;QAClC,IAAI,CAAC,YAAY,CAAC,6BAAkB,CAAC,0BAA0B,EAAE,SAAS,CAAC,WAAW,CAAC,CAAC;IAC1F,CAAC;IACD,IAAI,SAAS,CAAC,UAAU,IAAI,IAAI,EAAE,CAAC;QACjC,IAAI,CAAC,YAAY,CAAC,6BAAkB,CAAC,yBAAyB,EAAE,SAAS,CAAC,UAAU,CAAC,CAAC;IACxF,CAAC;IACD,IAAI,SAAS,CAAC,iBAAiB,IAAI,IAAI,EAAE,CAAC;QACxC,IAAI,CAAC,YAAY,CAAC,6BAAkB,CAAC,gCAAgC,EAAE,SAAS,CAAC,iBAAiB,CAAC,CAAC;IACtG,CAAC;IACD,IAAI,SAAS,CAAC,cAAc,IAAI,IAAI,EAAE,CAAC;QACrC,IAAI,CAAC,YAAY,CAAC,6BAAkB,CAAC,6BAA6B,EAAE,SAAS,CAAC,cAAc,CAAC,CAAC;IAChG,CAAC;IACD,IAAI,SAAS,CAAC,eAAe,IAAI,IAAI,EAAE,CAAC;QACtC,IAAI,CAAC,YAAY,CAAC,6BAAkB,CAAC,8BAA8B,EAAE,SAAS,CAAC,eAAe,CAAC,CAAC;IAClG,CAAC;IACD,IAAI,SAAS,CAAC,gBAAgB,IAAI,IAAI,EAAE,CAAC;QACvC,IAAI,CAAC,YAAY,CAAC,6BAAkB,CAAC,+BAA+B,EAAE,SAAS,CAAC,gBAAgB,CAAC,CAAC;IACpG,CAAC;IACD,IAAI,SAAS,CAAC,oBAAoB,IAAI,IAAI,EAAE,CAAC;QAC3C,IAAI,CAAC,YAAY,CAAC,6BAAkB,CAAC,6BAA6B,EAAE,SAAS,CAAC,oBAAoB,CAAC,CAAC;IACtG,CAAC;IACD,IAAI,SAAS,CAAC,MAAM,IAAI,IAAI,EAAE,CAAC;QAC7B,IAAI,CAAC,YAAY,CAAC,6BAAkB,CAAC,mBAAmB,EAAE,SAAS,CAAC,MAAM,CAAC,CAAC;IAC9E,CAAC;AACH,CAAC;AAED,SAAS,kBAAkB,CAAC,IAAU,EAAE,KAAU;IAChD,IAAI,CAAC,KAAK;QAAE,OAAO;IACnB,IAAI,KAAK,CAAC,aAAa,IAAI,IAAI,EAAE,CAAC;QAChC,IAAI,CAAC,YAAY,CAAC,6BAAkB,CAAC,yBAAyB,EAAE,KAAK,CAAC,aAAa,CAAC,CAAC;IACvF,CAAC;IACD,IAAI,KAAK,CAAC,iBAAiB,IAAI,IAAI,EAAE,CAAC;QACpC,IAAI,CAAC,YAAY,CAAC,6BAAkB,CAAC,0BAA0B,EAAE,KAAK,CAAC,iBAAiB,CAAC,CAAC;IAC5F,CAAC;IACD,IAAI,KAAK,CAAC,YAAY,IAAI,IAAI,EAAE,CAAC;QAC/B,IAAI,CAAC,YAAY,CAAC,6BAAkB,CAAC,yBAAyB,EAAE,KAAK,CAAC,YAAY,CAAC,CAAC;IACtF,CAAC;IACD,IAAI,KAAK,CAAC,UAAU,IAAI,IAAI,EAAE,CAAC;QAC7B,IAAI,CAAC,YAAY,CAAC,6BAAkB,CAAC,uBAAuB,EAAE,KAAK,CAAC,UAAU,CAAC,CAAC;IAClF,CAAC;IACD,IAAI,KAAK,CAAC,OAAO,IAAI,IAAI,EAAE,CAAC;QAC1B,IAAI,CAAC,YAAY,CAAC,6BAAkB,CAAC,aAAa,EAAE,MAAM,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC;IAC7E,CAAC;IACD,IAAI,KAAK,CAAC,cAAc,IAAI,IAAI,EAAE,CAAC;QACjC,IAAI,CAAC,YAAY,CAAC,6BAAkB,CAAC,0BAA0B,EAAE,IAAI,CAAC,SAAS,CAAC,KAAK,CAAC,cAAc,CAAC,CAAC,CAAC;IACzG,CAAC;AACH,CAAC;AAED,SAAS,oBAAoB,CAAC,IAAU,EAAE,IAAS,EAAE,aAAqB;IACxE,IAAI,CAAC,IAAI,IAAI,OAAO,IAAI,KAAK,QAAQ;QAAE,OAAO;IAC9C,IAAI,IAAI,CAAC,KAAK,IAAI,IAAI;QAAE,IAAI,CAAC,YAAY,CAAC,6BAAkB,CAAC,oBAAoB,EAAE,MAAM,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,CAAC;IACvG,IAAI,IAAI,CAAC,WAAW,IAAI,IAAI,EAAE,CAAC;QAC7B,IAAI,CAAC,YAAY,CAAC,6BAAkB,CAAC,0BAA0B,EAAE,IAAI,CAAC,WAAW,CAAC,CAAC;IACrF,CAAC;IACD,IAAI,IAAI,CAAC,UAAU,IAAI,IAAI,EAAE,CAAC;QAC5B,IAAI,CAAC,YAAY,CAAC,6BAAkB,CAAC,yBAAyB,EAAE,IAAI,CAAC,UAAU,CAAC,CAAC;IACnF,CAAC;IACD,IAAI,IAAI,CAAC,iBAAiB,IAAI,IAAI,EAAE,CAAC;QACnC,IAAI,CAAC,YAAY,CAAC,6BAAkB,CAAC,gCAAgC,EAAE,IAAI,CAAC,iBAAiB,CAAC,CAAC;IACjG,CAAC;IACD,IAAI,IAAI,CAAC,MAAM,IAAI,IAAI;QAAE,IAAI,CAAC,YAAY,CAAC,6BAAkB,CAAC,wBAAwB,EAAE,IAAI,CAAC,MAAM,CAAC,CAAC;SAChG,IAAI,IAAI,CAAC,SAAS,IAAI,IAAI,EAAE,CAAC;QAChC,IAAI,CAAC,YAAY,CAAC,6BAAkB,CAAC,wBAAwB,EAAE,IAAI,CAAC,SAAS,CAAC,CAAC;IACjF,CAAC;IACD,IAAI,IAAI,CAAC,SAAS,IAAI,IAAI;QAAE,IAAI,CAAC,YAAY,CAAC,6BAAkB,CAAC,oBAAoB,EAAE,IAAI,CAAC,SAAS,CAAC,CAAC;IACvG,IAAI,IAAI,CAAC,0BAA0B,IAAI,IAAI,EAAE,CAAC;QAC5C,IAAI,CAAC,YAAY,CACf,6BAAkB,CAAC,iCAAiC,EACpD,IAAI,CAAC,0BAA0B,CAChC,CAAC;IACJ,CAAC;IACD,IAAI,IAAI,CAAC,qBAAqB,IAAI,IAAI,EAAE,CAAC;QACvC,IAAI,CAAC,YAAY,CACf,6BAAkB,CAAC,oCAAoC,EACvD,IAAI,CAAC,qBAAqB,CAC3B,CAAC;IACJ,CAAC;IACD,IAAI,aAAa,KAAK,6BAAkB,CAAC,0BAA0B,IAAI,KAAK,CAAC,OAAO,CAAC,IAAI,CAAC,QAAQ,CAAC,EAAE,CAAC;QACpG,IAAI,CAAC,YAAY,CAAC,6BAAkB,CAAC,4BAA4B,EAAE,IAAI,CAAC,QAAQ,CAAC,MAAM,CAAC,CAAC;IAC3F,CAAC;AACH,CAAC;AAED,SAAS,0BAA0B,CACjC,IAAU,EACV,IAAS,EACT,QAAa,EACb,aAAqB,EACrB,OAAgB;IAEhB,IAAI,eAAe,GAAG,OAAO,CAAC;IAC9B,IAAI,CAAC,eAAe,IAAI,IAAI,EAAE,QAAQ;QAAE,eAAe,GAAG,MAAM,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;SAC3E,IAAI,CAAC,eAAe,IAAI,QAAQ,EAAE,EAAE;QAAE,eAAe,GAAG,MAAM,CAAC,QAAQ,CAAC,EAAE,CAAC,CAAC;IACjF,IAAI,eAAe;QAAE,IAAI,CAAC,YAAY,CAAC,6BAAkB,CAAC,eAAe,EAAE,eAAe,CAAC,CAAC;IAE5F,IAAI,QAAQ,EAAE,CAAC;QACb,IAAI,QAAQ,CAAC,IAAI,IAAI,IAAI;YAAE,IAAI,CAAC,YAAY,CAAC,6BAAkB,CAAC,iBAAiB,EAAE,MAAM,CAAC,QAAQ,CAAC,IAAI,CAAC,CAAC,CAAC;QAC1G,IAAI,QAAQ,CAAC,IAAI,IAAI,IAAI;YAAE,IAAI,CAAC,YAAY,CAAC,6BAAkB,CAAC,iBAAiB,EAAE,MAAM,CAAC,QAAQ,CAAC,IAAI,CAAC,CAAC,CAAC;QAC1G,IAAI,QAAQ,CAAC,WAAW,IAAI,IAAI,EAAE,CAAC;YACjC,IAAI,CAAC,YAAY,CAAC,6BAAkB,CAAC,wBAAwB,EAAE,MAAM,CAAC,QAAQ,CAAC,WAAW,CAAC,CAAC,CAAC;QAC/F,CAAC;QACD,IAAI,QAAQ,CAAC,UAAU,IAAI,IAAI,EAAE,CAAC;YAChC,IAAI,CAAC,YAAY,CAAC,6BAAkB,CAAC,iBAAiB,EAAE,QAAQ,CAAC,UAAU,CAAC,CAAC;QAC/E,CAAC;QACD,IAAI,QAAQ,CAAC,MAAM,IAAI,IAAI,EAAE,CAAC;YAC5B,IAAI,CAAC,YAAY,CAAC,6BAAkB,CAAC,yBAAyB,EAAE,eAAe,CAAC,MAAM,CAAC,QAAQ,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC;QAC5G,CAAC;QACD,IAAI,QAAQ,CAAC,UAAU;YAAE,sBAAsB,CAAC,IAAI,EAAE,QAAQ,CAAC,UAAU,CAAC,CAAC;QAC3E,IAAI,aAAa,KAAK,6BAAkB,CAAC,0BAA0B,IAAI,QAAQ,CAAC,KAAK,EAAE,CAAC;YACtF,kBAAkB,CAAC,IAAI,EAAE,QAAQ,CAAC,KAAK,CAAC,CAAC;QAC3C,CAAC;IACH,CAAC;IAED,oBAAoB,CAAC,IAAI,EAAE,IAAI,EAAE,aAAa,CAAC,CAAC;IAEhD,IAAI,IAAI,EAAE,IAAI,IAAI,IAAI;QAAE,IAAI,CAAC,YAAY,CAAC,6BAAkB,CAAC,iBAAiB,EAAE,MAAM,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC,CAAC;IACnG,IAAI,IAAI,EAAE,IAAI,IAAI,IAAI;QAAE,IAAI,CAAC,YAAY,CAAC,6BAAkB,CAAC,iBAAiB,EAAE,MAAM,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC,CAAC;AACrG,CAAC;AAED,SAAS,gBAAgB,CAAC,KAAU;IAClC,IAAI,CAAC;QACH,OAAO,IAAI,CAAC,SAAS,CAAC,KAAK,CAAC,CAAC;IAC/B,CAAC;IAAC,MAAM,CAAC;QACP,OAAO,MAAM,CAAC,KAAK,CAAC,CAAC;IACvB,CAAC;AACH,CAAC;AAED,SAAS,oBAAoB,CAAC,IAAU,EAAE,IAAS,EAAE,QAAa;IAChE,IAAI,IAAI,EAAE,QAAQ,IAAI,IAAI,EAAE,CAAC;QAC3B,IAAI,CAAC,YAAY,CAAC,6BAAkB,CAAC,qBAAqB,EAAE,gBAAgB,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC,CAAC;IAC/F,CAAC;IACD,IAAI,QAAQ,EAAE,QAAQ,IAAI,IAAI,EAAE,CAAC;QAC/B,IAAI,CAAC,YAAY,CAAC,6BAAkB,CAAC,sBAAsB,EAAE,gBAAgB,CAAC,QAAQ,CAAC,QAAQ,CAAC,CAAC,CAAC;IACpG,CAAC;AACH,CAAC;AAED,SAAS,aAAa,CAAC,IAAU,EAAE,QAAa,EAAE,KAAa;IAC7D,MAAM,WAAW,GAAG,gBAAe,CAAC,WAAW,CAAC;IAChD,IAAI,CAAC,WAAW,IAAI,CAAC,QAAQ;QAAE,OAAO;IAEtC,IAAI,KAAK,GAAG,QAAQ,CAAC,KAAK,CAAC;IAC3B,IAAI,CAAC,KAAK,IAAI,KAAK,CAAC,OAAO,CAAC,QAAQ,CAAC,EAAE,CAAC;QACtC,KAAK,IAAI,CAAC,GAAG,QAAQ,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC,IAAI,CAAC,EAAE,CAAC,IAAI,CAAC,EAAE,CAAC;YACjD,IAAI,QAAQ,CAAC,CAAC,CAAC,EAAE,YAAY,KAAK,kBAAkB,EAAE,CAAC;gBACrD,KAAK,GAAG,QAAQ,CAAC,CAAC,CAAC,CAAC;gBACpB,MAAM;YACR,CAAC;QACH,CAAC;IACH,CAAC;IACD,IAAI,KAAK,IAAI,KAAK,CAAC,aAAa,IAAI,IAAI,IAAI,KAAK,CAAC,iBAAiB,IAAI,IAAI,EAAE,CAAC;QAC5E,MAAM,WAAW,GAAG,QAAQ,CAAC,UAAU,EAAE,KAAK,IAAI,KAAK,CAAC;QACxD,MAAM,IAAI,GAAG,iBAAe,CAAC,gBAAgB,CAC3C,WAAW,EACX,WAAW,EACX,KAAK,CAAC,aAAa,EACnB,KAAK,CAAC,iBAAiB,CACxB,CAAC;QACF,IAAI,CAAC,YAAY,CAAC,6BAAkB,CAAC,iBAAiB,EAAE,IAAI,CAAC,CAAC;IAChE,CAAC;AACH,CAAC;AAED;;;GAGG;AACH,SAAgB,oBAAoB,CAClC,IAAU,EACV,QAAa,EACb,IAAS,EACT,aAAqB,EACrB,QAAa,EACb,eAAuB,EACvB,UAAkB,EAClB,OAAgB;IAEhB,MAAM,KAAK,GAAG,YAAY,CAAC,QAAQ,EAAE,IAAI,CAAC,CAAC;IAC3C,MAAM,CAAC,aAAa,EAAE,UAAU,CAAC,GAAG,iBAAiB,CAAC,QAAQ,CAAC,CAAC;IAEhE,IAAI,CAAC,YAAY,CAAC,8CAAuB,EAAE,mBAAQ,CAAC,CAAC;IACrD,IAAI,CAAC,YAAY,CAAC,6BAAkB,CAAC,kBAAkB,EAAE,UAAU,CAAC,CAAC;IACrE,IAAI,CAAC,YAAY,CAAC,6BAAkB,CAAC,oBAAoB,EAAE,6BAAkB,CAAC,mBAAmB,CAAC,CAAC;IACnG,IAAI,CAAC,YAAY,CAAC,6BAAkB,CAAC,yBAAyB,EAAE,6BAAkB,CAAC,mBAAmB,CAAC,CAAC;IACxG,IAAI,CAAC,YAAY,CAAC,6BAAkB,CAAC,gBAAgB,EAAE,aAAa,CAAC,CAAC;IACtE,IAAI,CAAC,YAAY,CAAC,6BAAkB,CAAC,oBAAoB,EAAE,KAAK,CAAC,CAAC;IAClE,IAAI,CAAC,YAAY,CAAC,6BAAkB,CAAC,cAAc,EAAE,aAAa,CAAC,CAAC;IACpE,IAAI,CAAC,YAAY,CAAC,6BAAkB,CAAC,WAAW,EAAE,UAAU,CAAC,CAAC;IAC9D,IAAI,CAAC,YAAY,CAAC,6BAAkB,CAAC,2BAA2B,EAAE,gBAAe,CAAC,WAAW,IAAI,EAAE,CAAC,CAAC;IACrG,IAAI,CAAC,YAAY,CAAC,wCAAiB,EAAE,gBAAe,CAAC,eAAe,IAAI,EAAE,CAAC,CAAC;IAC5E,IAAI,CAAC,YAAY,CAAC,6BAAkB,CAAC,gCAAgC,EAAE,eAAe,CAAC,CAAC;IAExF,0BAA0B,CAAC,IAAI,EAAE,IAAI,EAAE,QAAQ,EAAE,aAAa,EAAE,OAAO,CAAC,CAAC;IAEzE,IAAI,aAAa,KAAK,6BAAkB,CAAC,0BAA0B,IAAI,gBAAe,CAAC,qBAAqB,EAAE,CAAC;QAC7G,oBAAoB,CAAC,IAAI,EAAE,IAAI,EAAE,QAAQ,CAAC,CAAC;IAC7C,CAAC;IAED,IAAI,aAAa,KAAK,6BAAkB,CAAC,0BAA0B,EAAE,CAAC;QACpE,aAAa,CAAC,IAAI,EAAE,QAAQ,EAAE,KAAK,CAAC,CAAC;IACvC,CAAC;IAED,IAAA,mCAAyB,EAAC,IAAI,CAAC,CAAC;AAClC,CAAC"}

package/dist/instrumentation/letta/wrapper.d.ts

@@ -0,0 +1,37 @@
+import { Tracer } from '@opentelemetry/api';
+/**
+ * Instrumentation for the Letta agent platform (`@letta-ai/letta-client`). Each
+ * patched resource method becomes one CLIENT span whose name and operation type
+ * mirror the Python reference (sdk/python/src/shield360/instrumentation/letta): the
+ * method suffix drives the operation type via OPERATION_TYPE_MAP, and chat
+ * (message) operations stream their response through a traced async iterator.
+ */
+declare class LettaWrapper {
+    static aiSystem: string;
+    /**
+     * Wraps a Letta resource method. `endpoint` is the Python-style endpoint id
+     * (e.g. `letta.create`, `letta.create_message`) that drives the operation type
+     * and span name.
+     */
+    static _patchOperation(tracer: Tracer, endpoint: string, agentScoped: boolean, version?: string): any;
+    /**
+     * Non-chat operations return the SDK's own promise untouched. `list` yields a
+     * PagePromise whose async iterator drives auto-pagination, so flattening it here
+     * would break paging; the span is finalized from a completion callback instead.
+     */
+    private static _handleOperation;
+    /**
+     * Chat (message) operations may resolve to a streaming iterator. The resolved
+     * value is awaited so a stream can be wrapped in a traced async iterator; a
+     * non-streaming response is processed inline.
+     */
+    private static _handleChat;
+    /**
+     * Wraps a Letta message stream so chunks flow through untouched while the span
+     * accumulates chunk count, TTFT, and the collected messages. Mirrors
+     * TracedLettaStream in the Python reference.
+     */
+    private static _traceStream;
+    private static _finalizeError;
+}
+export default LettaWrapper;