@agentuity/runtime 0.0.51 → 0.0.52

package/dist/_services.js

@@ -0,0 +1,258 @@
+import { context, SpanKind, SpanStatusCode, trace } from '@opentelemetry/api';
+import { KeyValueStorageService, ObjectStorageService, StreamStorageService, VectorStorageService, } from '@agentuity/core';
+import { APIClient, createServerFetchAdapter, getServiceUrls } from '@agentuity/server';
+import { CompositeSessionEventProvider, LocalSessionEventProvider, JSONSessionEventProvider, HTTPSessionEventProvider, } from './services/session';
+import { CompositeEvalRunEventProvider, LocalEvalRunEventProvider, JSONEvalRunEventProvider, HTTPEvalRunEventProvider, } from './services/evalrun';
+import { injectTraceContextToHeaders } from './otel/http';
+import { getTracer } from './_server';
+import { getSDKVersion, isAuthenticated, isProduction } from './_config';
+import { DefaultSessionProvider, DefaultThreadProvider, } from './session';
+import { LocalKeyValueStorage, LocalObjectStorage, LocalStreamStorage, LocalVectorStorage, getLocalDB, normalizeProjectPath, createLocalStorageRouter, } from './services/local';
+const userAgent = `Agentuity SDK/${getSDKVersion()}`;
+const bearerKey = `Bearer ${process.env.AGENTUITY_SDK_KEY}`;
+const serviceUrls = getServiceUrls();
+const kvBaseUrl = serviceUrls.keyvalue;
+const streamBaseUrl = serviceUrls.stream;
+const vectorBaseUrl = serviceUrls.vector;
+const objectBaseUrl = serviceUrls.objectstore;
+const catalystBaseUrl = serviceUrls.catalyst;
+let adapter;
+const createFetchAdapter = (logger) => createServerFetchAdapter({
+    headers: {
+        Authorization: bearerKey,
+        'User-Agent': userAgent,
+    },
+    onBefore: async (url, options, callback) => {
+        logger.debug('before request: %s with options: %s', url, options);
+        if (!options.telemetry) {
+            return callback();
+        }
+        options.headers = { ...options.headers, ...injectTraceContextToHeaders() };
+        const tracer = getTracer() ?? trace.getTracer('agentuity');
+        const currentContext = context.active();
+        const span = tracer.startSpan(options.telemetry.name, { attributes: options.telemetry.attributes, kind: SpanKind.CLIENT }, currentContext);
+        const spanContext = trace.setSpan(currentContext, span);
+        try {
+            await context.with(spanContext, callback);
+            span.setStatus({ code: SpanStatusCode.OK });
+        }
+        catch (err) {
+            const e = err;
+            span.recordException(e);
+            span.setStatus({ code: SpanStatusCode.ERROR, message: e?.message ?? String(err) });
+            throw err;
+        }
+        finally {
+            span.end();
+        }
+    },
+    onAfter: async (url, options, result, err) => {
+        logger.debug('after request: %s (%d) => %s', url, result.response.status, err);
+        if (err) {
+            return;
+        }
+        const span = trace.getSpan(context.active());
+        switch (options.telemetry?.name) {
+            case 'agentuity.keyvalue.get': {
+                if (result.response.status === 404) {
+                    span?.addEvent('miss');
+                }
+                else if (result.response.ok) {
+                    span?.addEvent('hit');
+                }
+                break;
+            }
+            case 'agentuity.stream.create': {
+                if (result.response.ok) {
+                    const res = result.data;
+                    span?.setAttributes({
+                        'stream.id': res.id,
+                        'stream.url': `${streamBaseUrl}/${res.id}`,
+                    });
+                }
+                break;
+            }
+            case 'agentuity.stream.list': {
+                const response = result.data;
+                if (response && span) {
+                    span.setAttributes({
+                        'stream.count': response.streams.length,
+                        'stream.total': response.total,
+                    });
+                }
+                break;
+            }
+            case 'agentuity.vector.upsert': {
+                if (result.response.ok) {
+                    const data = result.data;
+                    span?.setAttributes({
+                        'vector.count': data.length,
+                    });
+                }
+                break;
+            }
+            case 'agentuity.vector.search': {
+                if (result.response.ok) {
+                    const data = result.data;
+                    span?.setAttributes({
+                        'vector.results': data.length,
+                    });
+                }
+                break;
+            }
+            case 'agentuity.vector.get': {
+                if (result.response.status === 404) {
+                    span?.addEvent('miss');
+                }
+                else if (result.response.ok) {
+                    span?.addEvent('hit');
+                }
+                break;
+            }
+            case 'agentuity.objectstore.get': {
+                if (result.response.status === 404) {
+                    span?.addEvent('miss');
+                }
+                else if (result.response.ok) {
+                    span?.addEvent('hit');
+                }
+                break;
+            }
+            case 'agentuity.objectstore.delete': {
+                if (result.response.status === 404) {
+                    span?.addEvent('not_found', { deleted: false });
+                }
+                else if (result.response.ok) {
+                    span?.addEvent('deleted', { deleted: true });
+                }
+                break;
+            }
+        }
+    },
+}, logger);
+let kv;
+let objectStore;
+let stream;
+let vector;
+let session;
+let thread;
+let sessionEvent;
+let evalRunEvent;
+// eslint-disable-next-line @typescript-eslint/no-explicit-any
+let localRouter = null;
+export function createServices(logger, config, serverUrl) {
+    const authenticated = isAuthenticated();
+    const useLocal = config?.services?.useLocal ?? false;
+    adapter = createFetchAdapter(logger);
+    // Use local services if explicitly requested OR if not authenticated
+    const shouldUseLocal = useLocal || !authenticated || process.env.AGENTUITY_FORCE_LOCAL_SERVICES === 'true';
+    if (shouldUseLocal) {
+        const db = getLocalDB();
+        const projectPath = normalizeProjectPath();
+        if (!serverUrl) {
+            throw new Error('serverUrl is required when using local services');
+        }
+        logger.info('Using local services (development only)');
+        kv = config?.services?.keyvalue || new LocalKeyValueStorage(db, projectPath);
+        objectStore = config?.services?.object || new LocalObjectStorage(db, projectPath, serverUrl);
+        stream = config?.services?.stream || new LocalStreamStorage(db, projectPath, serverUrl);
+        vector = config?.services?.vector || new LocalVectorStorage(db, projectPath);
+        session = config?.services?.session || new DefaultSessionProvider();
+        thread = config?.services?.thread || new DefaultThreadProvider();
+        sessionEvent = config?.services?.sessionEvent
+            ? new CompositeSessionEventProvider(new LocalSessionEventProvider(), config.services.sessionEvent)
+            : new LocalSessionEventProvider();
+        evalRunEvent = config?.services?.evalRunEvent
+            ? new CompositeEvalRunEventProvider(new LocalEvalRunEventProvider(), config.services.evalRunEvent)
+            : new LocalEvalRunEventProvider();
+        localRouter = createLocalStorageRouter(db, projectPath);
+        return { localRouter };
+    }
+    // Reset local router if not using local services
+    localRouter = null;
+    // At this point we must be authenticated (since !authenticated would trigger local services above)
+    kv = config?.services?.keyvalue || new KeyValueStorageService(kvBaseUrl, adapter);
+    objectStore = config?.services?.object || new ObjectStorageService(objectBaseUrl, adapter);
+    stream = config?.services?.stream || new StreamStorageService(streamBaseUrl, adapter);
+    vector = config?.services?.vector || new VectorStorageService(vectorBaseUrl, adapter);
+    session = config?.services?.session || new DefaultSessionProvider();
+    thread = config?.services?.thread || new DefaultThreadProvider();
+    // FIXME: this is turned off for now for production until we have the new changes deployed
+    sessionEvent =
+        isProduction() && process.env.AGENTUITY_CLOUD_EXPORT_DIR
+            ? new JSONSessionEventProvider(process.env.AGENTUITY_CLOUD_EXPORT_DIR)
+            : new HTTPSessionEventProvider(new APIClient(catalystBaseUrl, logger), logger);
+    new LocalSessionEventProvider();
+    if (config?.services?.sessionEvent) {
+        sessionEvent = new CompositeSessionEventProvider(sessionEvent, config.services.sessionEvent);
+    }
+    // FIXME: this is turned off for now for production until we have the new changes deployed
+    evalRunEvent =
+        isProduction() && process.env.AGENTUITY_CLOUD_EXPORT_DIR
+            ? new JSONEvalRunEventProvider(process.env.AGENTUITY_CLOUD_EXPORT_DIR)
+            : new HTTPEvalRunEventProvider(new APIClient(catalystBaseUrl, logger), logger, catalystBaseUrl);
+    new LocalEvalRunEventProvider();
+    if (config?.services?.evalRunEvent) {
+        evalRunEvent = new CompositeEvalRunEventProvider(evalRunEvent, config.services.evalRunEvent);
+    }
+    return {};
+}
+export function getThreadProvider() {
+    return thread;
+}
+export function getSessionProvider() {
+    return session;
+}
+// eslint-disable-next-line @typescript-eslint/no-explicit-any
+export function getLocalRouter() {
+    return localRouter;
+}
+export function getSessionEventProvider() {
+    return sessionEvent;
+}
+export function getEvalRunEventProvider() {
+    return evalRunEvent;
+}
+// eslint-disable-next-line @typescript-eslint/no-explicit-any
+export function registerServices(o, includeAgents = false) {
+    Object.defineProperty(o, 'kv', {
+        get: () => kv,
+        enumerable: false,
+        configurable: false,
+    });
+    Object.defineProperty(o, 'objectstore', {
+        get: () => objectStore,
+        enumerable: false,
+        configurable: false,
+    });
+    Object.defineProperty(o, 'stream', {
+        get: () => stream,
+        enumerable: false,
+        configurable: false,
+    });
+    Object.defineProperty(o, 'vector', {
+        get: () => vector,
+        enumerable: false,
+        configurable: false,
+    });
+    // Also register agent registry if requested
+    if (includeAgents) {
+        // Cache the populated registry to avoid re-creating on every access
+        // eslint-disable-next-line @typescript-eslint/no-explicit-any
+        let cachedRegistry;
+        Object.defineProperty(o, 'agent', {
+            get: () => {
+                if (!cachedRegistry) {
+                    // Lazy-load to avoid circular dependency
+                    // eslint-disable-next-line @typescript-eslint/no-require-imports
+                    const { populateAgentsRegistry } = require('./agent');
+                    cachedRegistry = populateAgentsRegistry(o);
+                }
+                return cachedRegistry;
+            },
+            enumerable: false,
+            configurable: false,
+        });
+    }
+}
+//# sourceMappingURL=_services.js.map

package/dist/_services.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"_services.js","sourceRoot":"","sources":["../src/_services.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,OAAO,EAAE,QAAQ,EAAE,cAAc,EAAE,KAAK,EAAE,MAAM,oBAAoB,CAAC;AAC9E,OAAO,EACN,sBAAsB,EACtB,oBAAoB,EACpB,oBAAoB,EACpB,oBAAoB,GAYpB,MAAM,iBAAiB,CAAC;AACzB,OAAO,EAAE,SAAS,EAAE,wBAAwB,EAAE,cAAc,EAAE,MAAM,mBAAmB,CAAC;AACxF,OAAO,EACN,6BAA6B,EAC7B,yBAAyB,EACzB,wBAAwB,EACxB,wBAAwB,GACxB,MAAM,oBAAoB,CAAC;AAC5B,OAAO,EACN,6BAA6B,EAC7B,yBAAyB,EACzB,wBAAwB,EACxB,wBAAwB,GACxB,MAAM,oBAAoB,CAAC;AAC5B,OAAO,EAAE,2BAA2B,EAAE,MAAM,aAAa,CAAC;AAC1D,OAAO,EAAE,SAAS,EAAE,MAAM,WAAW,CAAC;AACtC,OAAO,EAAE,aAAa,EAAE,eAAe,EAAE,YAAY,EAAE,MAAM,WAAW,CAAC;AAEzE,OAAO,EACN,sBAAsB,EACtB,qBAAqB,GAGrB,MAAM,WAAW,CAAC;AACnB,OAAO,EACN,oBAAoB,EACpB,kBAAkB,EAClB,kBAAkB,EAClB,kBAAkB,EAClB,UAAU,EACV,oBAAoB,EACpB,wBAAwB,GACxB,MAAM,kBAAkB,CAAC;AAE1B,MAAM,SAAS,GAAG,iBAAiB,aAAa,EAAE,EAAE,CAAC;AACrD,MAAM,SAAS,GAAG,UAAU,OAAO,CAAC,GAAG,CAAC,iBAAiB,EAAE,CAAC;AAE5D,MAAM,WAAW,GAAG,cAAc,EAAE,CAAC;AACrC,MAAM,SAAS,GAAG,WAAW,CAAC,QAAQ,CAAC;AACvC,MAAM,aAAa,GAAG,WAAW,CAAC,MAAM,CAAC;AACzC,MAAM,aAAa,GAAG,WAAW,CAAC,MAAM,CAAC;AACzC,MAAM,aAAa,GAAG,WAAW,CAAC,WAAW,CAAC;AAC9C,MAAM,eAAe,GAAG,WAAW,CAAC,QAAQ,CAAC;AAE7C,IAAI,OAAqB,CAAC;AAE1B,MAAM,kBAAkB,GAAG,CAAC,MAAc,EAAE,EAAE,CAC7C,wBAAwB,CACvB;IACC,OAAO,EAAE;QACR,aAAa,EAAE,SAAS;QACxB,YAAY,EAAE,SAAS;KACvB;IACD,QAAQ,EAAE,KAAK,EAAE,GAAG,EAAE,OAAO,EAAE,QAAQ,EAAE,EAAE;QAC1C,MAAM,CAAC,KAAK,CAAC,qCAAqC,EAAE,GAAG,EAAE,OAAO,CAAC,CAAC;QAClE,IAAI,CAAC,OAAO,CAAC,SAAS,EAAE,CAAC;YACxB,OAAO,QAAQ,EAAE,CAAC;QACnB,CAAC;QACD,OAAO,CAAC,OAAO,GAAG,EAAE,GAAG,OAAO,CAAC,OAAO,EAAE,GAAG,2BAA2B,EAAE,EAAE,CAAC;QAC3E,MAAM,MAAM,GAAG,SAAS,EAAE,IAAI,KAAK,CAAC,SAAS,CAAC,WAAW,CAAC,CAAC;QAC3D,MAAM,cAAc,GAAG,OAAO,CAAC,MAAM,EAAE,CAAC;QACxC,MAAM,IAAI,GAAG,MAAM,CAAC,SAAS,CAC5B,OAAO,CAAC,SAAS,CAAC,IAAI,EACtB,EAAE,UAAU,EAAE,OAAO,CAAC,SAAS,CAAC,UAAU,EAAE,IAAI,EAAE,QAAQ,CAAC,MAAM,EAAE,EACnE,cAAc,CACd,CAAC;QACF,MAAM,WAAW,GAAG,KAAK,CAAC,OAAO,CAAC,cAAc,EAAE,IAAI,CAAC,CAAC;QACxD,IAAI,CAAC;YACJ,MAAM,OAAO,CAAC,IAAI,CAAC,WAAW,EAAE,QAAQ,CAAC,CAAC;YAC1C,IAAI,CAAC,SAAS,CAAC,EAAE,IAAI,EAAE,cAAc,CAAC,EAAE,EAAE,CAAC,CAAC;QAC7C,CAAC;QAAC,OAAO,GAAG,EAAE,CAAC;YACd,MAAM,CAAC,GAAG,GAAY,CAAC;YACvB,IAAI,CAAC,eAAe,CAAC,CAAC,CAAC,CAAC;YACxB,IAAI,CAAC,SAAS,CAAC,EAAE,IAAI,EAAE,cAAc,CAAC,KAAK,EAAE,OAAO,EAAE,CAAC,EAAE,OAAO,IAAI,MAAM,CAAC,GAAG,CAAC,EAAE,CAAC,CAAC;YACnF,MAAM,GAAG,CAAC;QACX,CAAC;gBAAS,CAAC;YACV,IAAI,CAAC,GAAG,EAAE,CAAC;QACZ,CAAC;IACF,CAAC;IACD,OAAO,EAAE,KAAK,EAAE,GAAG,EAAE,OAAO,EAAE,MAAM,EAAE,GAAG,EAAE,EAAE;QAC5C,MAAM,CAAC,KAAK,CAAC,8BAA8B,EAAE,GAAG,EAAE,MAAM,CAAC,QAAQ,CAAC,MAAM,EAAE,GAAG,CAAC,CAAC;QAC/E,IAAI,GAAG,EAAE,CAAC;YACT,OAAO;QACR,CAAC;QACD,MAAM,IAAI,GAAG,KAAK,CAAC,OAAO,CAAC,OAAO,CAAC,MAAM,EAAE,CAAC,CAAC;QAC7C,QAAQ,OAAO,CAAC,SAAS,EAAE,IAAI,EAAE,CAAC;YACjC,KAAK,wBAAwB,CAAC,CAAC,CAAC;gBAC/B,IAAI,MAAM,CAAC,QAAQ,CAAC,MAAM,KAAK,GAAG,EAAE,CAAC;oBACpC,IAAI,EAAE,QAAQ,CAAC,MAAM,CAAC,CAAC;gBACxB,CAAC;qBAAM,IAAI,MAAM,CAAC,QAAQ,CAAC,EAAE,EAAE,CAAC;oBAC/B,IAAI,EAAE,QAAQ,CAAC,KAAK,CAAC,CAAC;gBACvB,CAAC;gBACD,MAAM;YACP,CAAC;YACD,KAAK,yBAAyB,CAAC,CAAC,CAAC;gBAChC,IAAI,MAAM,CAAC,QAAQ,CAAC,EAAE,EAAE,CAAC;oBACxB,MAAM,GAAG,GAAG,MAAM,CAAC,IAAsB,CAAC;oBAC1C,IAAI,EAAE,aAAa,CAAC;wBACnB,WAAW,EAAE,GAAG,CAAC,EAAE;wBACnB,YAAY,EAAE,GAAG,aAAa,IAAI,GAAG,CAAC,EAAE,EAAE;qBAC1C,CAAC,CAAC;gBACJ,CAAC;gBACD,MAAM;YACP,CAAC;YACD,KAAK,uBAAuB,CAAC,CAAC,CAAC;gBAC9B,MAAM,QAAQ,GAAG,MAAM,CAAC,IAA2B,CAAC;gBACpD,IAAI,QAAQ,IAAI,IAAI,EAAE,CAAC;oBACtB,IAAI,CAAC,aAAa,CAAC;wBAClB,cAAc,EAAE,QAAQ,CAAC,OAAO,CAAC,MAAM;wBACvC,cAAc,EAAE,QAAQ,CAAC,KAAK;qBAC9B,CAAC,CAAC;gBACJ,CAAC;gBACD,MAAM;YACP,CAAC;YACD,KAAK,yBAAyB,CAAC,CAAC,CAAC;gBAChC,IAAI,MAAM,CAAC,QAAQ,CAAC,EAAE,EAAE,CAAC;oBACxB,MAAM,IAAI,GAAG,MAAM,CAAC,IAA4B,CAAC;oBACjD,IAAI,EAAE,aAAa,CAAC;wBACnB,cAAc,EAAE,IAAI,CAAC,MAAM;qBAC3B,CAAC,CAAC;gBACJ,CAAC;gBACD,MAAM;YACP,CAAC;YACD,KAAK,yBAAyB,CAAC,CAAC,CAAC;gBAChC,IAAI,MAAM,CAAC,QAAQ,CAAC,EAAE,EAAE,CAAC;oBACxB,MAAM,IAAI,GAAG,MAAM,CAAC,IAA4B,CAAC;oBACjD,IAAI,EAAE,aAAa,CAAC;wBACnB,gBAAgB,EAAE,IAAI,CAAC,MAAM;qBAC7B,CAAC,CAAC;gBACJ,CAAC;gBACD,MAAM;YACP,CAAC;YACD,KAAK,sBAAsB,CAAC,CAAC,CAAC;gBAC7B,IAAI,MAAM,CAAC,QAAQ,CAAC,MAAM,KAAK,GAAG,EAAE,CAAC;oBACpC,IAAI,EAAE,QAAQ,CAAC,MAAM,CAAC,CAAC;gBACxB,CAAC;qBAAM,IAAI,MAAM,CAAC,QAAQ,CAAC,EAAE,EAAE,CAAC;oBAC/B,IAAI,EAAE,QAAQ,CAAC,KAAK,CAAC,CAAC;gBACvB,CAAC;gBACD,MAAM;YACP,CAAC;YACD,KAAK,2BAA2B,CAAC,CAAC,CAAC;gBAClC,IAAI,MAAM,CAAC,QAAQ,CAAC,MAAM,KAAK,GAAG,EAAE,CAAC;oBACpC,IAAI,EAAE,QAAQ,CAAC,MAAM,CAAC,CAAC;gBACxB,CAAC;qBAAM,IAAI,MAAM,CAAC,QAAQ,CAAC,EAAE,EAAE,CAAC;oBAC/B,IAAI,EAAE,QAAQ,CAAC,KAAK,CAAC,CAAC;gBACvB,CAAC;gBACD,MAAM;YACP,CAAC;YACD,KAAK,8BAA8B,CAAC,CAAC,CAAC;gBACrC,IAAI,MAAM,CAAC,QAAQ,CAAC,MAAM,KAAK,GAAG,EAAE,CAAC;oBACpC,IAAI,EAAE,QAAQ,CAAC,WAAW,EAAE,EAAE,OAAO,EAAE,KAAK,EAAE,CAAC,CAAC;gBACjD,CAAC;qBAAM,IAAI,MAAM,CAAC,QAAQ,CAAC,EAAE,EAAE,CAAC;oBAC/B,IAAI,EAAE,QAAQ,CAAC,SAAS,EAAE,EAAE,OAAO,EAAE,IAAI,EAAE,CAAC,CAAC;gBAC9C,CAAC;gBACD,MAAM;YACP,CAAC;QACF,CAAC;IACF,CAAC;CACD,EACD,MAAM,CACN,CAAC;AAEH,IAAI,EAAmB,CAAC;AACxB,IAAI,WAA0B,CAAC;AAC/B,IAAI,MAAqB,CAAC;AAC1B,IAAI,MAAqB,CAAC;AAC1B,IAAI,OAAwB,CAAC;AAC7B,IAAI,MAAsB,CAAC;AAC3B,IAAI,YAAkC,CAAC;AACvC,IAAI,YAAkC,CAAC;AAEvC,8DAA8D;AAC9D,IAAI,WAAW,GAAe,IAAI,CAAC;AAEnC,MAAM,UAAU,cAAc,CAAC,MAAc,EAAE,MAAkB,EAAE,SAAkB;IACpF,MAAM,aAAa,GAAG,eAAe,EAAE,CAAC;IACxC,MAAM,QAAQ,GAAG,MAAM,EAAE,QAAQ,EAAE,QAAQ,IAAI,KAAK,CAAC;IACrD,OAAO,GAAG,kBAAkB,CAAC,MAAM,CAAC,CAAC;IAErC,qEAAqE;IACrE,MAAM,cAAc,GACnB,QAAQ,IAAI,CAAC,aAAa,IAAI,OAAO,CAAC,GAAG,CAAC,8BAA8B,KAAK,MAAM,CAAC;IAErF,IAAI,cAAc,EAAE,CAAC;QACpB,MAAM,EAAE,GAAG,UAAU,EAAE,CAAC;QACxB,MAAM,WAAW,GAAG,oBAAoB,EAAE,CAAC;QAE3C,IAAI,CAAC,SAAS,EAAE,CAAC;YAChB,MAAM,IAAI,KAAK,CAAC,iDAAiD,CAAC,CAAC;QACpE,CAAC;QAED,MAAM,CAAC,IAAI,CAAC,yCAAyC,CAAC,CAAC;QAEvD,EAAE,GAAG,MAAM,EAAE,QAAQ,EAAE,QAAQ,IAAI,IAAI,oBAAoB,CAAC,EAAE,EAAE,WAAW,CAAC,CAAC;QAC7E,WAAW,GAAG,MAAM,EAAE,QAAQ,EAAE,MAAM,IAAI,IAAI,kBAAkB,CAAC,EAAE,EAAE,WAAW,EAAE,SAAS,CAAC,CAAC;QAC7F,MAAM,GAAG,MAAM,EAAE,QAAQ,EAAE,MAAM,IAAI,IAAI,kBAAkB,CAAC,EAAE,EAAE,WAAW,EAAE,SAAS,CAAC,CAAC;QACxF,MAAM,GAAG,MAAM,EAAE,QAAQ,EAAE,MAAM,IAAI,IAAI,kBAAkB,CAAC,EAAE,EAAE,WAAW,CAAC,CAAC;QAC7E,OAAO,GAAG,MAAM,EAAE,QAAQ,EAAE,OAAO,IAAI,IAAI,sBAAsB,EAAE,CAAC;QACpE,MAAM,GAAG,MAAM,EAAE,QAAQ,EAAE,MAAM,IAAI,IAAI,qBAAqB,EAAE,CAAC;QACjE,YAAY,GAAG,MAAM,EAAE,QAAQ,EAAE,YAAY;YAC5C,CAAC,CAAC,IAAI,6BAA6B,CACjC,IAAI,yBAAyB,EAAE,EAC/B,MAAM,CAAC,QAAQ,CAAC,YAAY,CAC5B;YACF,CAAC,CAAC,IAAI,yBAAyB,EAAE,CAAC;QACnC,YAAY,GAAG,MAAM,EAAE,QAAQ,EAAE,YAAY;YAC5C,CAAC,CAAC,IAAI,6BAA6B,CACjC,IAAI,yBAAyB,EAAE,EAC/B,MAAM,CAAC,QAAQ,CAAC,YAAY,CAC5B;YACF,CAAC,CAAC,IAAI,yBAAyB,EAAE,CAAC;QAEnC,WAAW,GAAG,wBAAwB,CAAC,EAAE,EAAE,WAAW,CAAC,CAAC;QAExD,OAAO,EAAE,WAAW,EAAE,CAAC;IACxB,CAAC;IAED,iDAAiD;IACjD,WAAW,GAAG,IAAI,CAAC;IAEnB,mGAAmG;IACnG,EAAE,GAAG,MAAM,EAAE,QAAQ,EAAE,QAAQ,IAAI,IAAI,sBAAsB,CAAC,SAAS,EAAE,OAAO,CAAC,CAAC;IAClF,WAAW,GAAG,MAAM,EAAE,QAAQ,EAAE,MAAM,IAAI,IAAI,oBAAoB,CAAC,aAAa,EAAE,OAAO,CAAC,CAAC;IAC3F,MAAM,GAAG,MAAM,EAAE,QAAQ,EAAE,MAAM,IAAI,IAAI,oBAAoB,CAAC,aAAa,EAAE,OAAO,CAAC,CAAC;IACtF,MAAM,GAAG,MAAM,EAAE,QAAQ,EAAE,MAAM,IAAI,IAAI,oBAAoB,CAAC,aAAa,EAAE,OAAO,CAAC,CAAC;IACtF,OAAO,GAAG,MAAM,EAAE,QAAQ,EAAE,OAAO,IAAI,IAAI,sBAAsB,EAAE,CAAC;IACpE,MAAM,GAAG,MAAM,EAAE,QAAQ,EAAE,MAAM,IAAI,IAAI,qBAAqB,EAAE,CAAC;IACjE,0FAA0F;IAC1F,YAAY;QACX,YAAY,EAAE,IAAI,OAAO,CAAC,GAAG,CAAC,0BAA0B;YACvD,CAAC,CAAC,IAAI,wBAAwB,CAAC,OAAO,CAAC,GAAG,CAAC,0BAA0B,CAAC;YACtE,CAAC,CAAC,IAAI,wBAAwB,CAAC,IAAI,SAAS,CAAC,eAAe,EAAE,MAAM,CAAC,EAAE,MAAM,CAAC,CAAC;IACjF,IAAI,yBAAyB,EAAE,CAAC;IAChC,IAAI,MAAM,EAAE,QAAQ,EAAE,YAAY,EAAE,CAAC;QACpC,YAAY,GAAG,IAAI,6BAA6B,CAAC,YAAY,EAAE,MAAM,CAAC,QAAQ,CAAC,YAAY,CAAC,CAAC;IAC9F,CAAC;IACD,0FAA0F;IAC1F,YAAY;QACX,YAAY,EAAE,IAAI,OAAO,CAAC,GAAG,CAAC,0BAA0B;YACvD,CAAC,CAAC,IAAI,wBAAwB,CAAC,OAAO,CAAC,GAAG,CAAC,0BAA0B,CAAC;YACtE,CAAC,CAAC,IAAI,wBAAwB,CAC5B,IAAI,SAAS,CAAC,eAAe,EAAE,MAAM,CAAC,EACtC,MAAM,EACN,eAAe,CACf,CAAC;IACL,IAAI,yBAAyB,EAAE,CAAC;IAChC,IAAI,MAAM,EAAE,QAAQ,EAAE,YAAY,EAAE,CAAC;QACpC,YAAY,GAAG,IAAI,6BAA6B,CAAC,YAAY,EAAE,MAAM,CAAC,QAAQ,CAAC,YAAY,CAAC,CAAC;IAC9F,CAAC;IAED,OAAO,EAAE,CAAC;AACX,CAAC;AAED,MAAM,UAAU,iBAAiB;IAChC,OAAO,MAAM,CAAC;AACf,CAAC;AAED,MAAM,UAAU,kBAAkB;IACjC,OAAO,OAAO,CAAC;AAChB,CAAC;AAED,8DAA8D;AAC9D,MAAM,UAAU,cAAc;IAC7B,OAAO,WAAW,CAAC;AACpB,CAAC;AAED,MAAM,UAAU,uBAAuB;IACtC,OAAO,YAAY,CAAC;AACrB,CAAC;AAED,MAAM,UAAU,uBAAuB;IACtC,OAAO,YAAY,CAAC;AACrB,CAAC;AAED,8DAA8D;AAC9D,MAAM,UAAU,gBAAgB,CAAC,CAAM,EAAE,aAAa,GAAG,KAAK;IAC7D,MAAM,CAAC,cAAc,CAAC,CAAC,EAAE,IAAI,EAAE;QAC9B,GAAG,EAAE,GAAG,EAAE,CAAC,EAAE;QACb,UAAU,EAAE,KAAK;QACjB,YAAY,EAAE,KAAK;KACnB,CAAC,CAAC;IACH,MAAM,CAAC,cAAc,CAAC,CAAC,EAAE,aAAa,EAAE;QACvC,GAAG,EAAE,GAAG,EAAE,CAAC,WAAW;QACtB,UAAU,EAAE,KAAK;QACjB,YAAY,EAAE,KAAK;KACnB,CAAC,CAAC;IACH,MAAM,CAAC,cAAc,CAAC,CAAC,EAAE,QAAQ,EAAE;QAClC,GAAG,EAAE,GAAG,EAAE,CAAC,MAAM;QACjB,UAAU,EAAE,KAAK;QACjB,YAAY,EAAE,KAAK;KACnB,CAAC,CAAC;IACH,MAAM,CAAC,cAAc,CAAC,CAAC,EAAE,QAAQ,EAAE;QAClC,GAAG,EAAE,GAAG,EAAE,CAAC,MAAM;QACjB,UAAU,EAAE,KAAK;QACjB,YAAY,EAAE,KAAK;KACnB,CAAC,CAAC;IAEH,4CAA4C;IAC5C,IAAI,aAAa,EAAE,CAAC;QACnB,oEAAoE;QACpE,8DAA8D;QAC9D,IAAI,cAAmB,CAAC;QACxB,MAAM,CAAC,cAAc,CAAC,CAAC,EAAE,OAAO,EAAE;YACjC,GAAG,EAAE,GAAG,EAAE;gBACT,IAAI,CAAC,cAAc,EAAE,CAAC;oBACrB,yCAAyC;oBACzC,iEAAiE;oBACjE,MAAM,EAAE,sBAAsB,EAAE,GAAG,OAAO,CAAC,SAAS,CAAC,CAAC;oBACtD,cAAc,GAAG,sBAAsB,CAAC,CAAC,CAAC,CAAC;gBAC5C,CAAC;gBACD,OAAO,cAAc,CAAC;YACvB,CAAC;YACD,UAAU,EAAE,KAAK;YACjB,YAAY,EAAE,KAAK;SACnB,CAAC,CAAC;IACJ,CAAC;AACF,CAAC"}

package/dist/_tokens.js

@@ -0,0 +1,98 @@
+import { inAgentContext, inHTTPContext, getHTTPContext } from './_context';
+import { SpanAttributes } from '@traceloop/ai-semantic-conventions';
+import { addSpanProcessor } from './_server';
+export const TOKENS_HEADER = 'x-agentuity-tokens';
+const AI_SDK_SPAN_NAME = 'ai.generateText';
+const AI_SDK_MODEL_NAME = 'ai.model.id';
+const AI_SDK_USAGE_PROMPT_TOKENS = 'ai.usage.promptTokens';
+const AI_SDK_USAGE_COMPLETION_TOKENS = 'ai.usage.completionTokens';
+const parseTokenHeader = (val) => {
+    const kv = new Map();
+    if (val) {
+        // format is: [model]:[count] [model:count]
+        const tok = val.split(' ');
+        for (const entry of tok) {
+            const [name, count] = entry.split(':');
+            if (name) {
+                kv.set(name, parseInt(count ?? '0') ?? 0);
+            }
+        }
+    }
+    return kv;
+};
+const serializeTokenHeader = (kv) => {
+    const lines = [];
+    for (const [k, v] of kv) {
+        lines.push(`${k}:${v}`);
+    }
+    return lines.join(' ');
+};
+const getTokenValue = (val) => {
+    if (val) {
+        const v = val.valueOf();
+        switch (typeof v) {
+            case 'number':
+                return v;
+            case 'string':
+                return parseInt(v, 10);
+            default:
+        }
+    }
+    return 0;
+};
+class TokenSpanProcessor {
+    onStart(_span, _context) {
+        return;
+    }
+    onEnd(span) {
+        if (inAgentContext() && inHTTPContext()) {
+            const ctx = getHTTPContext();
+            const tokenLine = ctx.res.headers.get(TOKENS_HEADER) ?? undefined;
+            const tokens = parseTokenHeader(tokenLine);
+            let mutated = false;
+            // this is AI SDK which doesn't seem to use the semantic attribute names
+            if (span.name === AI_SDK_SPAN_NAME && AI_SDK_MODEL_NAME in span.attributes) {
+                const model = span.attributes[AI_SDK_MODEL_NAME].toString();
+                let totalTokens = tokens.get(model) ?? 0;
+                if (AI_SDK_USAGE_PROMPT_TOKENS in span.attributes) {
+                    totalTokens += getTokenValue(span.attributes[AI_SDK_USAGE_PROMPT_TOKENS]);
+                }
+                if (AI_SDK_USAGE_COMPLETION_TOKENS in span.attributes) {
+                    totalTokens += getTokenValue(span.attributes[AI_SDK_USAGE_COMPLETION_TOKENS]);
+                }
+                if (totalTokens > 0) {
+                    tokens.set(model, totalTokens);
+                    mutated = true;
+                }
+            }
+            else if (SpanAttributes.LLM_SYSTEM in span.attributes &&
+                SpanAttributes.LLM_RESPONSE_MODEL in span.attributes) {
+                const model = span.attributes[SpanAttributes.LLM_RESPONSE_MODEL].toString();
+                let totalTokens = tokens.get(model) ?? 0;
+                if (SpanAttributes.LLM_USAGE_PROMPT_TOKENS in span.attributes) {
+                    totalTokens += getTokenValue(span.attributes[SpanAttributes.LLM_USAGE_PROMPT_TOKENS]);
+                }
+                if (SpanAttributes.LLM_USAGE_COMPLETION_TOKENS in span.attributes) {
+                    totalTokens += getTokenValue(span.attributes[SpanAttributes.LLM_USAGE_COMPLETION_TOKENS]);
+                }
+                if (totalTokens > 0) {
+                    tokens.set(model, totalTokens);
+                    mutated = true;
+                }
+            }
+            if (mutated) {
+                ctx.header(TOKENS_HEADER, serializeTokenHeader(tokens));
+            }
+        }
+    }
+    forceFlush() {
+        return Promise.resolve();
+    }
+    shutdown() {
+        return Promise.resolve();
+    }
+}
+export default function register() {
+    addSpanProcessor(new TokenSpanProcessor());
+}
+//# sourceMappingURL=_tokens.js.map

package/dist/_tokens.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"_tokens.js","sourceRoot":"","sources":["../src/_tokens.ts"],"names":[],"mappings":"AAGA,OAAO,EAAE,cAAc,EAAE,aAAa,EAAE,cAAc,EAAE,MAAM,YAAY,CAAC;AAC3E,OAAO,EAAE,cAAc,EAAE,MAAM,oCAAoC,CAAC;AACpE,OAAO,EAAE,gBAAgB,EAAE,MAAM,WAAW,CAAC;AAE7C,MAAM,CAAC,MAAM,aAAa,GAAG,oBAAoB,CAAC;AAElD,MAAM,gBAAgB,GAAG,iBAAiB,CAAC;AAC3C,MAAM,iBAAiB,GAAG,aAAa,CAAC;AACxC,MAAM,0BAA0B,GAAG,uBAAuB,CAAC;AAC3D,MAAM,8BAA8B,GAAG,2BAA2B,CAAC;AAEnE,MAAM,gBAAgB,GAAG,CAAC,GAAuB,EAAuB,EAAE;IACzE,MAAM,EAAE,GAAG,IAAI,GAAG,EAAkB,CAAC;IACrC,IAAI,GAAG,EAAE,CAAC;QACT,2CAA2C;QAC3C,MAAM,GAAG,GAAG,GAAG,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC;QAC3B,KAAK,MAAM,KAAK,IAAI,GAAG,EAAE,CAAC;YACzB,MAAM,CAAC,IAAI,EAAE,KAAK,CAAC,GAAG,KAAK,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC;YACvC,IAAI,IAAI,EAAE,CAAC;gBACV,EAAE,CAAC,GAAG,CAAC,IAAI,EAAE,QAAQ,CAAC,KAAK,IAAI,GAAG,CAAC,IAAI,CAAC,CAAC,CAAC;YAC3C,CAAC;QACF,CAAC;IACF,CAAC;IACD,OAAO,EAAE,CAAC;AACX,CAAC,CAAC;AAEF,MAAM,oBAAoB,GAAG,CAAC,EAAuB,EAAU,EAAE;IAChE,MAAM,KAAK,GAAa,EAAE,CAAC;IAC3B,KAAK,MAAM,CAAC,CAAC,EAAE,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC;QACzB,KAAK,CAAC,IAAI,CAAC,GAAG,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;IACzB,CAAC;IACD,OAAO,KAAK,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;AACxB,CAAC,CAAC;AAEF,MAAM,aAAa,GAAG,CAAC,GAA+B,EAAU,EAAE;IACjE,IAAI,GAAG,EAAE,CAAC;QACT,MAAM,CAAC,GAAG,GAAG,CAAC,OAAO,EAAE,CAAC;QACxB,QAAQ,OAAO,CAAC,EAAE,CAAC;YAClB,KAAK,QAAQ;gBACZ,OAAO,CAAC,CAAC;YACV,KAAK,QAAQ;gBACZ,OAAO,QAAQ,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC;YACxB,QAAQ;QACT,CAAC;IACF,CAAC;IACD,OAAO,CAAC,CAAC;AACV,CAAC,CAAC;AAEF,MAAM,kBAAkB;IACvB,OAAO,CAAC,KAAW,EAAE,QAAiB;QACrC,OAAO;IACR,CAAC;IAED,KAAK,CAAC,IAAU;QACf,IAAI,cAAc,EAAE,IAAI,aAAa,EAAE,EAAE,CAAC;YACzC,MAAM,GAAG,GAAG,cAAc,EAAE,CAAC;YAC7B,MAAM,SAAS,GAAG,GAAG,CAAC,GAAG,CAAC,OAAO,CAAC,GAAG,CAAC,aAAa,CAAC,IAAI,SAAS,CAAC;YAClE,MAAM,MAAM,GAAG,gBAAgB,CAAC,SAAS,CAAC,CAAC;YAC3C,IAAI,OAAO,GAAG,KAAK,CAAC;YACpB,wEAAwE;YACxE,IAAI,IAAI,CAAC,IAAI,KAAK,gBAAgB,IAAI,iBAAiB,IAAI,IAAI,CAAC,UAAU,EAAE,CAAC;gBAC5E,MAAM,KAAK,GAAG,IAAI,CAAC,UAAU,CAAC,iBAAiB,CAAE,CAAC,QAAQ,EAAE,CAAC;gBAC7D,IAAI,WAAW,GAAG,MAAM,CAAC,GAAG,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC;gBACzC,IAAI,0BAA0B,IAAI,IAAI,CAAC,UAAU,EAAE,CAAC;oBACnD,WAAW,IAAI,aAAa,CAAC,IAAI,CAAC,UAAU,CAAC,0BAA0B,CAAC,CAAC,CAAC;gBAC3E,CAAC;gBACD,IAAI,8BAA8B,IAAI,IAAI,CAAC,UAAU,EAAE,CAAC;oBACvD,WAAW,IAAI,aAAa,CAAC,IAAI,CAAC,UAAU,CAAC,8BAA8B,CAAC,CAAC,CAAC;gBAC/E,CAAC;gBACD,IAAI,WAAW,GAAG,CAAC,EAAE,CAAC;oBACrB,MAAM,CAAC,GAAG,CAAC,KAAK,EAAE,WAAW,CAAC,CAAC;oBAC/B,OAAO,GAAG,IAAI,CAAC;gBAChB,CAAC;YACF,CAAC;iBAAM,IACN,cAAc,CAAC,UAAU,IAAI,IAAI,CAAC,UAAU;gBAC5C,cAAc,CAAC,kBAAkB,IAAI,IAAI,CAAC,UAAU,EACnD,CAAC;gBACF,MAAM,KAAK,GAAG,IAAI,CAAC,UAAU,CAAC,cAAc,CAAC,kBAAkB,CAAE,CAAC,QAAQ,EAAE,CAAC;gBAC7E,IAAI,WAAW,GAAG,MAAM,CAAC,GAAG,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC;gBACzC,IAAI,cAAc,CAAC,uBAAuB,IAAI,IAAI,CAAC,UAAU,EAAE,CAAC;oBAC/D,WAAW,IAAI,aAAa,CAC3B,IAAI,CAAC,UAAU,CAAC,cAAc,CAAC,uBAAuB,CAAC,CACvD,CAAC;gBACH,CAAC;gBACD,IAAI,cAAc,CAAC,2BAA2B,IAAI,IAAI,CAAC,UAAU,EAAE,CAAC;oBACnE,WAAW,IAAI,aAAa,CAC3B,IAAI,CAAC,UAAU,CAAC,cAAc,CAAC,2BAA2B,CAAC,CAC3D,CAAC;gBACH,CAAC;gBACD,IAAI,WAAW,GAAG,CAAC,EAAE,CAAC;oBACrB,MAAM,CAAC,GAAG,CAAC,KAAK,EAAE,WAAW,CAAC,CAAC;oBAC/B,OAAO,GAAG,IAAI,CAAC;gBAChB,CAAC;YACF,CAAC;YACD,IAAI,OAAO,EAAE,CAAC;gBACb,GAAG,CAAC,MAAM,CAAC,aAAa,EAAE,oBAAoB,CAAC,MAAM,CAAC,CAAC,CAAC;YACzD,CAAC;QACF,CAAC;IACF,CAAC;IAED,UAAU;QACT,OAAO,OAAO,CAAC,OAAO,EAAE,CAAC;IAC1B,CAAC;IAED,QAAQ;QACP,OAAO,OAAO,CAAC,OAAO,EAAE,CAAC;IAC1B,CAAC;CACD;AAED,MAAM,CAAC,OAAO,UAAU,QAAQ;IAC/B,gBAAgB,CAAC,IAAI,kBAAkB,EAAE,CAAC,CAAC;AAC5C,CAAC"}

package/dist/_util.js

@@ -0,0 +1,54 @@
+export function returnResponse(ctx, result) {
+    if (result instanceof ReadableStream)
+        return ctx.body(result);
+    if (result instanceof Response)
+        return result;
+    if (typeof result === 'string')
+        return ctx.text(result);
+    if (typeof result === 'number' || typeof result === 'boolean')
+        return ctx.text(String(result));
+    return ctx.json(result);
+}
+/**
+ * SHA256 hash of the given values
+ *
+ * @param val one or more strings to hash
+ * @returns hash string in hex format
+ */
+export function hash(...val) {
+    const hasher = new Bun.CryptoHasher('sha256');
+    val.map((val) => hasher.update(val));
+    return hasher.digest().toHex();
+}
+/**
+ * Safely stringify an object to JSON, handling circular references
+ * @param obj - The object to stringify
+ * @returns JSON string representation
+ */
+export function safeStringify(obj) {
+    const stack = [];
+    function replacer(_key, value) {
+        if (typeof value === 'bigint') {
+            return value.toString();
+        }
+        if (typeof value === 'object' && value !== null) {
+            // Check if this object is already in our ancestor chain
+            if (stack.includes(value)) {
+                return '[Circular]';
+            }
+            // Add to stack before processing
+            stack.push(value);
+            // Process the object
+            const result = Array.isArray(value) ? [] : {};
+            for (const [k, v] of Object.entries(value)) {
+                result[k] = replacer(k, v);
+            }
+            // Remove from stack after processing
+            stack.pop();
+            return result;
+        }
+        return value;
+    }
+    return JSON.stringify(replacer('', obj));
+}
+//# sourceMappingURL=_util.js.map

package/dist/_util.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"_util.js","sourceRoot":"","sources":["../src/_util.ts"],"names":[],"mappings":"AAEA,MAAM,UAAU,cAAc,CAAC,GAAY,EAAE,MAAe;IAC3D,IAAI,MAAM,YAAY,cAAc;QAAE,OAAO,GAAG,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;IAC9D,IAAI,MAAM,YAAY,QAAQ;QAAE,OAAO,MAAM,CAAC;IAC9C,IAAI,OAAO,MAAM,KAAK,QAAQ;QAAE,OAAO,GAAG,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;IACxD,IAAI,OAAO,MAAM,KAAK,QAAQ,IAAI,OAAO,MAAM,KAAK,SAAS;QAAE,OAAO,GAAG,CAAC,IAAI,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC,CAAC;IAC/F,OAAO,GAAG,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;AACzB,CAAC;AAED;;;;;GAKG;AACH,MAAM,UAAU,IAAI,CAAC,GAAG,GAAa;IACpC,MAAM,MAAM,GAAG,IAAI,GAAG,CAAC,YAAY,CAAC,QAAQ,CAAC,CAAC;IAC9C,GAAG,CAAC,GAAG,CAAC,CAAC,GAAG,EAAE,EAAE,CAAC,MAAM,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC,CAAC;IACrC,OAAO,MAAM,CAAC,MAAM,EAAE,CAAC,KAAK,EAAE,CAAC;AAChC,CAAC;AAED;;;;GAIG;AACH,MAAM,UAAU,aAAa,CAAC,GAAY;IACzC,MAAM,KAAK,GAAc,EAAE,CAAC;IAE5B,SAAS,QAAQ,CAAC,IAAY,EAAE,KAAc;QAC7C,IAAI,OAAO,KAAK,KAAK,QAAQ,EAAE,CAAC;YAC/B,OAAO,KAAK,CAAC,QAAQ,EAAE,CAAC;QACzB,CAAC;QAED,IAAI,OAAO,KAAK,KAAK,QAAQ,IAAI,KAAK,KAAK,IAAI,EAAE,CAAC;YACjD,wDAAwD;YACxD,IAAI,KAAK,CAAC,QAAQ,CAAC,KAAK,CAAC,EAAE,CAAC;gBAC3B,OAAO,YAAY,CAAC;YACrB,CAAC;YAED,iCAAiC;YACjC,KAAK,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;YAElB,qBAAqB;YACrB,MAAM,MAAM,GAAG,KAAK,CAAC,OAAO,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;YAE9C,KAAK,MAAM,CAAC,CAAC,EAAE,CAAC,CAAC,IAAI,MAAM,CAAC,OAAO,CAAC,KAAK,CAAC,EAAE,CAAC;gBAC3C,MAAkC,CAAC,CAAC,CAAC,GAAG,QAAQ,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC;YACzD,CAAC;YAED,qCAAqC;YACrC,KAAK,CAAC,GAAG,EAAE,CAAC;YAEZ,OAAO,MAAM,CAAC;QACf,CAAC;QAED,OAAO,KAAK,CAAC;IACd,CAAC;IAED,OAAO,IAAI,CAAC,SAAS,CAAC,QAAQ,CAAC,EAAE,EAAE,GAAG,CAAC,CAAC,CAAC;AAC1C,CAAC"}

package/dist/_waituntil.js

@@ -0,0 +1,87 @@
+import { context, SpanStatusCode, trace } from '@opentelemetry/api';
+import { internal } from './logger/internal';
+let running = 0;
+/**
+ * returns true if wait until is pending
+ * @returns boolean
+ */
+export function hasWaitUntilPending() {
+    internal.debug('hasWaitUntilPending called: %d', running);
+    return running > 0;
+}
+export default class WaitUntilHandler {
+    promises;
+    tracer;
+    started;
+    hasCalledWaitUntilAll = false;
+    constructor(tracer) {
+        this.tracer = tracer;
+        this.promises = [];
+    }
+    waitUntil(promise) {
+        if (this.hasCalledWaitUntilAll) {
+            throw new Error('Cannot call waitUntil after waitUntilAll has been called');
+        }
+        running++;
+        internal.debug('wait until called, running: %d', running);
+        const currentContext = context.active();
+        // Start execution immediately, don't defer it
+        const executingPromise = (async () => {
+            if (this.started === undefined) {
+                this.started = Date.now(); /// this first execution marks the start time
+            }
+            const span = this.tracer.startSpan('waitUntil', {}, currentContext);
+            const spanContext = trace.setSpan(currentContext, span);
+            try {
+                internal.debug('starting waituntil');
+                await context.with(spanContext, async () => {
+                    const resolvedPromise = typeof promise === 'function' ? promise() : promise;
+                    return await Promise.resolve(resolvedPromise);
+                });
+                internal.debug('completed waituntil');
+                span.setStatus({ code: SpanStatusCode.OK });
+            }
+            catch (ex) {
+                span.recordException(ex);
+                span.setStatus({ code: SpanStatusCode.ERROR });
+                throw ex;
+            }
+            finally {
+                span.end();
+            }
+            // NOTE: we only decrement when the promise is removed from the array in waitUntilAll
+        })();
+        // Store the executing promise for cleanup tracking
+        this.promises.push(executingPromise);
+    }
+    hasPending() {
+        return this.promises.length > 0;
+    }
+    async waitUntilAll(logger, sessionId) {
+        internal.debug(`🔍 waitUntilAll() called for session ${sessionId} (count: %d)`, running);
+        if (this.hasCalledWaitUntilAll) {
+            throw new Error('waitUntilAll can only be called once per instance');
+        }
+        this.hasCalledWaitUntilAll = true;
+        if (this.promises.length === 0) {
+            internal.debug('No promises to wait for, executing evals directly');
+            // await this.executeEvalsForSession(logger, sessionId);
+            return;
+        }
+        internal.debug(`⏳ Waiting for ${this.promises.length} promises to complete...`);
+        try {
+            // Promises are already executing, just wait for them to complete
+            await Promise.all(this.promises);
+            const duration = Date.now() - this.started;
+            internal.debug('✅ All promises completed, marking session completed (duration %dms)', duration);
+        }
+        catch (ex) {
+            logger.error('error sending session completed', ex);
+        }
+        finally {
+            running -= this.promises.length;
+            this.promises.length = 0;
+        }
+    }
+}
+//# sourceMappingURL=_waituntil.js.map

package/dist/_waituntil.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"_waituntil.js","sourceRoot":"","sources":["../src/_waituntil.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,OAAO,EAAE,cAAc,EAAe,KAAK,EAAE,MAAM,oBAAoB,CAAC;AAEjF,OAAO,EAAE,QAAQ,EAAE,MAAM,mBAAmB,CAAC;AAE7C,IAAI,OAAO,GAAG,CAAC,CAAC;AAEhB;;;GAGG;AACH,MAAM,UAAU,mBAAmB;IAClC,QAAQ,CAAC,KAAK,CAAC,gCAAgC,EAAE,OAAO,CAAC,CAAC;IAC1D,OAAO,OAAO,GAAG,CAAC,CAAC;AACpB,CAAC;AAED,MAAM,CAAC,OAAO,OAAO,gBAAgB;IAC5B,QAAQ,CAAkB;IAC1B,MAAM,CAAS;IACf,OAAO,CAAqB;IAC5B,qBAAqB,GAAG,KAAK,CAAC;IAEtC,YAAmB,MAAc;QAChC,IAAI,CAAC,MAAM,GAAG,MAAM,CAAC;QACrB,IAAI,CAAC,QAAQ,GAAG,EAAE,CAAC;IACpB,CAAC;IAEM,SAAS,CAAC,OAAqD;QACrE,IAAI,IAAI,CAAC,qBAAqB,EAAE,CAAC;YAChC,MAAM,IAAI,KAAK,CAAC,0DAA0D,CAAC,CAAC;QAC7E,CAAC;QACD,OAAO,EAAE,CAAC;QACV,QAAQ,CAAC,KAAK,CAAC,gCAAgC,EAAE,OAAO,CAAC,CAAC;QAC1D,MAAM,cAAc,GAAG,OAAO,CAAC,MAAM,EAAE,CAAC;QAExC,8CAA8C;QAC9C,MAAM,gBAAgB,GAAG,CAAC,KAAK,IAAI,EAAE;YACpC,IAAI,IAAI,CAAC,OAAO,KAAK,SAAS,EAAE,CAAC;gBAChC,IAAI,CAAC,OAAO,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC,CAAC,6CAA6C;YACzE,CAAC;YACD,MAAM,IAAI,GAAG,IAAI,CAAC,MAAM,CAAC,SAAS,CAAC,WAAW,EAAE,EAAE,EAAE,cAAc,CAAC,CAAC;YACpE,MAAM,WAAW,GAAG,KAAK,CAAC,OAAO,CAAC,cAAc,EAAE,IAAI,CAAC,CAAC;YACxD,IAAI,CAAC;gBACJ,QAAQ,CAAC,KAAK,CAAC,oBAAoB,CAAC,CAAC;gBACrC,MAAM,OAAO,CAAC,IAAI,CAAC,WAAW,EAAE,KAAK,IAAI,EAAE;oBAC1C,MAAM,eAAe,GAAG,OAAO,OAAO,KAAK,UAAU,CAAC,CAAC,CAAC,OAAO,EAAE,CAAC,CAAC,CAAC,OAAO,CAAC;oBAC5E,OAAO,MAAM,OAAO,CAAC,OAAO,CAAC,eAAe,CAAC,CAAC;gBAC/C,CAAC,CAAC,CAAC;gBACH,QAAQ,CAAC,KAAK,CAAC,qBAAqB,CAAC,CAAC;gBACtC,IAAI,CAAC,SAAS,CAAC,EAAE,IAAI,EAAE,cAAc,CAAC,EAAE,EAAE,CAAC,CAAC;YAC7C,CAAC;YAAC,OAAO,EAAW,EAAE,CAAC;gBACtB,IAAI,CAAC,eAAe,CAAC,EAAW,CAAC,CAAC;gBAClC,IAAI,CAAC,SAAS,CAAC,EAAE,IAAI,EAAE,cAAc,CAAC,KAAK,EAAE,CAAC,CAAC;gBAC/C,MAAM,EAAE,CAAC;YACV,CAAC;oBAAS,CAAC;gBACV,IAAI,CAAC,GAAG,EAAE,CAAC;YACZ,CAAC;YACD,qFAAqF;QACtF,CAAC,CAAC,EAAE,CAAC;QAEL,mDAAmD;QACnD,IAAI,CAAC,QAAQ,CAAC,IAAI,CAAC,gBAAgB,CAAC,CAAC;IACtC,CAAC;IAEM,UAAU;QAChB,OAAO,IAAI,CAAC,QAAQ,CAAC,MAAM,GAAG,CAAC,CAAC;IACjC,CAAC;IAEM,KAAK,CAAC,YAAY,CAAC,MAAc,EAAE,SAAiB;QAC1D,QAAQ,CAAC,KAAK,CAAC,wCAAwC,SAAS,cAAc,EAAE,OAAO,CAAC,CAAC;QAEzF,IAAI,IAAI,CAAC,qBAAqB,EAAE,CAAC;YAChC,MAAM,IAAI,KAAK,CAAC,mDAAmD,CAAC,CAAC;QACtE,CAAC;QACD,IAAI,CAAC,qBAAqB,GAAG,IAAI,CAAC;QAElC,IAAI,IAAI,CAAC,QAAQ,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;YAChC,QAAQ,CAAC,KAAK,CAAC,mDAAmD,CAAC,CAAC;YACpE,wDAAwD;YACxD,OAAO;QACR,CAAC;QAED,QAAQ,CAAC,KAAK,CAAC,iBAAiB,IAAI,CAAC,QAAQ,CAAC,MAAM,0BAA0B,CAAC,CAAC;QAChF,IAAI,CAAC;YACJ,iEAAiE;YACjE,MAAM,OAAO,CAAC,GAAG,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;YACjC,MAAM,QAAQ,GAAG,IAAI,CAAC,GAAG,EAAE,GAAI,IAAI,CAAC,OAAkB,CAAC;YACvD,QAAQ,CAAC,KAAK,CACb,qEAAqE,EACrE,QAAQ,CACR,CAAC;QACH,CAAC;QAAC,OAAO,EAAE,EAAE,CAAC;YACb,MAAM,CAAC,KAAK,CAAC,iCAAiC,EAAE,EAAE,CAAC,CAAC;QACrD,CAAC;gBAAS,CAAC;YACV,OAAO,IAAI,IAAI,CAAC,QAAQ,CAAC,MAAM,CAAC;YAChC,IAAI,CAAC,QAAQ,CAAC,MAAM,GAAG,CAAC,CAAC;QAC1B,CAAC;IACF,CAAC;CACD"}