@agentuity/runtime 0.0.51 → 0.0.53

package/dist/otel/otel.js

@@ -0,0 +1,233 @@
+import opentelemetry, { metrics, propagation } from '@opentelemetry/api';
+import * as LogsAPI from '@opentelemetry/api-logs';
+import { getNodeAutoInstrumentations } from '@opentelemetry/auto-instrumentations-node';
+import { CompositePropagator, W3CBaggagePropagator, W3CTraceContextPropagator, } from '@opentelemetry/core';
+import { OTLPLogExporter } from '@opentelemetry/exporter-logs-otlp-http';
+import { OTLPMetricExporter } from '@opentelemetry/exporter-metrics-otlp-http';
+import { OTLPTraceExporter } from '@opentelemetry/exporter-trace-otlp-http';
+import { HostMetrics } from '@opentelemetry/host-metrics';
+import { CompressionAlgorithm } from '@opentelemetry/otlp-exporter-base';
+import { resourceFromAttributes } from '@opentelemetry/resources';
+import { BatchLogRecordProcessor, LoggerProvider, SimpleLogRecordProcessor, } from '@opentelemetry/sdk-logs';
+import { MeterProvider, PeriodicExportingMetricReader } from '@opentelemetry/sdk-metrics';
+import { NodeSDK } from '@opentelemetry/sdk-node';
+import { ATTR_SERVICE_NAME, ATTR_SERVICE_VERSION } from '@opentelemetry/semantic-conventions';
+import { initialize } from '@traceloop/node-server-sdk';
+import { ConsoleLogRecordExporter } from './console';
+import { instrumentFetch } from './fetch';
+import { createLogger, patchConsole } from './logger';
+import { getSDKVersion, isAuthenticated } from '../_config';
+import { JSONLLogExporter, JSONLTraceExporter, JSONLMetricExporter } from './exporters';
+const devmodeExportInterval = 1_000; // 1 second
+const productionExportInterval = 10_000; // 10 seconds
+export const createResource = (config) => {
+    const { name, version, orgId, projectId, deploymentId, environment, devmode, cliVersion } = config;
+    const sdkVersion = getSDKVersion();
+    return resourceFromAttributes({
+        [ATTR_SERVICE_NAME]: name,
+        [ATTR_SERVICE_VERSION]: version,
+        '@agentuity/orgId': orgId ?? 'unknown',
+        '@agentuity/projectId': projectId ?? 'unknown',
+        '@agentuity/deploymentId': deploymentId ?? 'unknown',
+        '@agentuity/env': environment,
+        '@agentuity/devmode': devmode,
+        '@agentuity/sdkVersion': sdkVersion ?? 'unknown',
+        '@agentuity/cliVersion': cliVersion ?? 'unknown',
+    });
+};
+export const createAgentuityLoggerProvider = ({ url, headers, resource, logLevel, jsonlBasePath, }) => {
+    let processor;
+    let exporter;
+    if (jsonlBasePath) {
+        exporter = new JSONLLogExporter(jsonlBasePath);
+        processor = new BatchLogRecordProcessor(exporter);
+    }
+    else if (url) {
+        // Original OTLP export behavior
+        const otlpExporter = new OTLPLogExporter({
+            url: `${url}/v1/logs`,
+            headers,
+            compression: CompressionAlgorithm.GZIP,
+            timeoutMillis: 10_000,
+        });
+        exporter = otlpExporter;
+        processor = new BatchLogRecordProcessor(otlpExporter);
+    }
+    else {
+        processor = new SimpleLogRecordProcessor(new ConsoleLogRecordExporter(logLevel));
+    }
+    const provider = new LoggerProvider({
+        resource,
+        processors: [processor],
+    });
+    LogsAPI.logs.setGlobalLoggerProvider(provider);
+    return {
+        processor,
+        provider,
+        exporter,
+    };
+};
+export const createUserLoggerProvider = ({ url, headers, resource, }) => {
+    const exporter = new OTLPLogExporter({
+        url: `${url}/v1/logs`,
+        headers,
+        compression: CompressionAlgorithm.GZIP,
+        timeoutMillis: 10_000,
+    });
+    const processor = new BatchLogRecordProcessor(exporter);
+    const provider = new LoggerProvider({
+        resource,
+        processors: [processor],
+    });
+    return {
+        provider,
+        exporter,
+        processor,
+    };
+};
+/**
+ * Registers and initializes OpenTelemetry with the specified configuration
+ *
+ * @param config - The configuration for OpenTelemetry
+ * @returns An object containing the tracer, logger, and shutdown function
+ */
+export function registerOtel(config) {
+    const { url, name, version, bearerToken, environment = 'development', orgId, projectId, deploymentId, devmode = false, logLevel = 'warn', jsonlBasePath = undefined, } = config;
+    let headers;
+    if (bearerToken) {
+        headers = {};
+        headers.Authorization = `Bearer ${bearerToken}`;
+    }
+    const resource = createResource(config);
+    const loggerProvider = createAgentuityLoggerProvider({
+        url,
+        headers,
+        resource,
+        logLevel,
+        jsonlBasePath,
+    });
+    const attrs = {
+        '@agentuity/orgId': orgId ?? 'unknown',
+        '@agentuity/projectId': projectId ?? 'unknown',
+        '@agentuity/deploymentId': deploymentId ?? 'unknown',
+        '@agentuity/env': environment,
+        '@agentuity/devmode': devmode,
+        '@agentuity/language': 'javascript',
+    };
+    const logger = createLogger(!!url, attrs, logLevel);
+    // must do this after we have created the logger
+    patchConsole(!!url, attrs, logLevel);
+    const traceExporter = url
+        ? jsonlBasePath
+            ? new JSONLTraceExporter(jsonlBasePath)
+            : new OTLPTraceExporter({
+                url: `${url}/v1/traces`,
+                headers,
+                keepAlive: true,
+            })
+        : undefined;
+    const metricExporter = url
+        ? jsonlBasePath
+            ? new JSONLMetricExporter(jsonlBasePath)
+            : new OTLPMetricExporter({
+                url: `${url}/v1/metrics`,
+                headers,
+                keepAlive: true,
+            })
+        : undefined;
+    // Create a separate metric reader for the NodeSDK
+    const sdkMetricReader = url && metricExporter
+        ? new PeriodicExportingMetricReader({
+            exporter: metricExporter,
+            exportTimeoutMillis: devmode ? devmodeExportInterval : productionExportInterval,
+            exportIntervalMillis: devmode ? devmodeExportInterval : productionExportInterval,
+        })
+        : undefined;
+    // Create a separate metric reader for the MeterProvider
+    const hostMetricReader = url && metricExporter
+        ? new PeriodicExportingMetricReader({
+            exporter: metricExporter,
+            exportTimeoutMillis: devmode ? devmodeExportInterval : productionExportInterval,
+            exportIntervalMillis: devmode ? devmodeExportInterval : productionExportInterval,
+        })
+        : undefined;
+    const meterProvider = hostMetricReader
+        ? new MeterProvider({
+            resource,
+            readers: [hostMetricReader],
+        })
+        : undefined;
+    if (meterProvider) {
+        metrics.setGlobalMeterProvider(meterProvider);
+    }
+    const hostMetrics = meterProvider ? new HostMetrics({ meterProvider }) : undefined;
+    let running = false;
+    let instrumentationSDK;
+    if (url) {
+        const propagator = new CompositePropagator({
+            propagators: [new W3CTraceContextPropagator(), new W3CBaggagePropagator()],
+        });
+        propagation.setGlobalPropagator(propagator);
+        instrumentFetch();
+        instrumentationSDK = new NodeSDK({
+            logRecordProcessor: loggerProvider.processor,
+            traceExporter,
+            metricReader: sdkMetricReader,
+            instrumentations: [getNodeAutoInstrumentations()],
+            resource,
+            textMapPropagator: propagator,
+            spanProcessors: config.spanProcessors,
+        });
+        instrumentationSDK.start();
+        hostMetrics?.start();
+        try {
+            const projectName = config.projectId || '';
+            const orgId = config.orgId || '';
+            const appName = `${orgId}:${projectName}`;
+            const traceloopHeaders = {};
+            if (bearerToken) {
+                traceloopHeaders.Authorization = `Bearer ${bearerToken}`;
+            }
+            initialize({
+                appName,
+                baseUrl: url,
+                headers: traceloopHeaders,
+                disableBatch: devmode,
+                logLevel: 'error',
+                tracingEnabled: false, // Disable traceloop's own tracing (equivalent to Python's telemetryEnabled: false)
+                // Note: JavaScript SDK doesn't support resourceAttributes like Python
+            });
+            logger.debug(`Telemetry initialized with app_name: ${appName}`);
+            logger.debug('Telemetry configured successfully');
+        }
+        catch (error) {
+            logger.warn('Telemetry not available, skipping automatic instrumentation', {
+                error: error instanceof Error ? error.message : String(error),
+            });
+        }
+        running = true;
+    }
+    const tracer = opentelemetry.trace.getTracer(name, version);
+    const meter = metrics.getMeter(name, version);
+    const shutdown = async () => {
+        if (running) {
+            running = false;
+            logger.debug('shutting down OpenTelemetry');
+            await loggerProvider.provider
+                .forceFlush()
+                .catch((e) => logger.warn('error in forceFlush of otel provider. %s', e));
+            await loggerProvider.exporter
+                ?.shutdown()
+                .catch((e) => !devmode && logger.warn('error in shutdown of otel exporter. %s', e));
+            await instrumentationSDK
+                ?.shutdown()
+                .catch((e) => !devmode && logger.warn('error in shutdown of otel instrumentation. %s', e));
+            logger.debug('shut down OpenTelemetry');
+        }
+    };
+    if (url && isAuthenticated()) {
+        logger.debug('connected to Agentuity Agent Cloud');
+    }
+    return { tracer, meter, logger, shutdown };
+}
+//# sourceMappingURL=otel.js.map

package/dist/otel/otel.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"otel.js","sourceRoot":"","sources":["../../src/otel/otel.ts"],"names":[],"mappings":"AACA,OAAO,aAAa,EAAE,EAAc,OAAO,EAAE,WAAW,EAAe,MAAM,oBAAoB,CAAC;AAClG,OAAO,KAAK,OAAO,MAAM,yBAAyB,CAAC;AACnD,OAAO,EAAE,2BAA2B,EAAE,MAAM,2CAA2C,CAAC;AACxF,OAAO,EACN,mBAAmB,EACnB,oBAAoB,EACpB,yBAAyB,GACzB,MAAM,qBAAqB,CAAC;AAC7B,OAAO,EAAE,eAAe,EAAE,MAAM,wCAAwC,CAAC;AACzE,OAAO,EAAE,kBAAkB,EAAE,MAAM,2CAA2C,CAAC;AAC/E,OAAO,EAAE,iBAAiB,EAAE,MAAM,yCAAyC,CAAC;AAC5E,OAAO,EAAE,WAAW,EAAE,MAAM,6BAA6B,CAAC;AAC1D,OAAO,EAAE,oBAAoB,EAAE,MAAM,mCAAmC,CAAC;AACzE,OAAO,EAAE,sBAAsB,EAAE,MAAM,0BAA0B,CAAC;AAElE,OAAO,EACN,uBAAuB,EACvB,cAAc,EAEd,wBAAwB,GACxB,MAAM,yBAAyB,CAAC;AACjC,OAAO,EAAE,aAAa,EAAE,6BAA6B,EAAE,MAAM,4BAA4B,CAAC;AAC1F,OAAO,EAAE,OAAO,EAAE,MAAM,yBAAyB,CAAC;AAClD,OAAO,EAAE,iBAAiB,EAAE,oBAAoB,EAAE,MAAM,qCAAqC,CAAC;AAC9F,OAAO,EAAE,UAAU,EAAE,MAAM,4BAA4B,CAAC;AAExD,OAAO,EAAE,wBAAwB,EAAE,MAAM,WAAW,CAAC;AACrD,OAAO,EAAE,eAAe,EAAE,MAAM,SAAS,CAAC;AAC1C,OAAO,EAAE,YAAY,EAAE,YAAY,EAAE,MAAM,UAAU,CAAC;AACtD,OAAO,EAAE,aAAa,EAAE,eAAe,EAAE,MAAM,YAAY,CAAC;AAE5D,OAAO,EAAE,gBAAgB,EAAE,kBAAkB,EAAE,mBAAmB,EAAE,MAAM,aAAa,CAAC;AA+BxF,MAAM,qBAAqB,GAAG,KAAK,CAAC,CAAC,WAAW;AAChD,MAAM,wBAAwB,GAAG,MAAM,CAAC,CAAC,aAAa;AAEtD,MAAM,CAAC,MAAM,cAAc,GAAG,CAAC,MAAkB,EAAY,EAAE;IAC9D,MAAM,EAAE,IAAI,EAAE,OAAO,EAAE,KAAK,EAAE,SAAS,EAAE,YAAY,EAAE,WAAW,EAAE,OAAO,EAAE,UAAU,EAAE,GACxF,MAAM,CAAC;IAER,MAAM,UAAU,GAAG,aAAa,EAAE,CAAC;IAEnC,OAAO,sBAAsB,CAAC;QAC7B,CAAC,iBAAiB,CAAC,EAAE,IAAI;QACzB,CAAC,oBAAoB,CAAC,EAAE,OAAO;QAC/B,kBAAkB,EAAE,KAAK,IAAI,SAAS;QACtC,sBAAsB,EAAE,SAAS,IAAI,SAAS;QAC9C,yBAAyB,EAAE,YAAY,IAAI,SAAS;QACpD,gBAAgB,EAAE,WAAW;QAC7B,oBAAoB,EAAE,OAAO;QAC7B,uBAAuB,EAAE,UAAU,IAAI,SAAS;QAChD,uBAAuB,EAAE,UAAU,IAAI,SAAS;KAChD,CAAC,CAAC;AACJ,CAAC,CAAC;AAEF,MAAM,CAAC,MAAM,6BAA6B,GAAG,CAAC,EAC7C,GAAG,EACH,OAAO,EACP,QAAQ,EACR,QAAQ,EACR,aAAa,GAOb,EAAE,EAAE;IACJ,IAAI,SAA6B,CAAC;IAClC,IAAI,QAAwD,CAAC;IAE7D,IAAI,aAAa,EAAE,CAAC;QACnB,QAAQ,GAAG,IAAI,gBAAgB,CAAC,aAAa,CAAC,CAAC;QAC/C,SAAS,GAAG,IAAI,uBAAuB,CAAC,QAAQ,CAAC,CAAC;IACnD,CAAC;SAAM,IAAI,GAAG,EAAE,CAAC;QAChB,gCAAgC;QAChC,MAAM,YAAY,GAAG,IAAI,eAAe,CAAC;YACxC,GAAG,EAAE,GAAG,GAAG,UAAU;YACrB,OAAO;YACP,WAAW,EAAE,oBAAoB,CAAC,IAAI;YACtC,aAAa,EAAE,MAAM;SACrB,CAAC,CAAC;QACH,QAAQ,GAAG,YAAY,CAAC;QACxB,SAAS,GAAG,IAAI,uBAAuB,CAAC,YAAY,CAAC,CAAC;IACvD,CAAC;SAAM,CAAC;QACP,SAAS,GAAG,IAAI,wBAAwB,CAAC,IAAI,wBAAwB,CAAC,QAAQ,CAAC,CAAC,CAAC;IAClF,CAAC;IACD,MAAM,QAAQ,GAAG,IAAI,cAAc,CAAC;QACnC,QAAQ;QACR,UAAU,EAAE,CAAC,SAAS,CAAC;KACvB,CAAC,CAAC;IACH,OAAO,CAAC,IAAI,CAAC,uBAAuB,CAAC,QAAQ,CAAC,CAAC;IAE/C,OAAO;QACN,SAAS;QACT,QAAQ;QACR,QAAQ;KACR,CAAC;AACH,CAAC,CAAC;AAEF,MAAM,CAAC,MAAM,wBAAwB,GAAG,CAAC,EACxC,GAAG,EACH,OAAO,EACP,QAAQ,GAKR,EAAE,EAAE;IACJ,MAAM,QAAQ,GAAG,IAAI,eAAe,CAAC;QACpC,GAAG,EAAE,GAAG,GAAG,UAAU;QACrB,OAAO;QACP,WAAW,EAAE,oBAAoB,CAAC,IAAI;QACtC,aAAa,EAAE,MAAM;KACrB,CAAC,CAAC;IACH,MAAM,SAAS,GAAG,IAAI,uBAAuB,CAAC,QAAQ,CAAC,CAAC;IACxD,MAAM,QAAQ,GAAG,IAAI,cAAc,CAAC;QACnC,QAAQ;QACR,UAAU,EAAE,CAAC,SAAS,CAAC;KACvB,CAAC,CAAC;IACH,OAAO;QACN,QAAQ;QACR,QAAQ;QACR,SAAS;KACT,CAAC;AACH,CAAC,CAAC;AAEF;;;;;GAKG;AACH,MAAM,UAAU,YAAY,CAAC,MAAkB;IAC9C,MAAM,EACL,GAAG,EACH,IAAI,EACJ,OAAO,EACP,WAAW,EACX,WAAW,GAAG,aAAa,EAC3B,KAAK,EACL,SAAS,EACT,YAAY,EACZ,OAAO,GAAG,KAAK,EACf,QAAQ,GAAG,MAAM,EACjB,aAAa,GAAG,SAAS,GACzB,GAAG,MAAM,CAAC;IAEX,IAAI,OAA2C,CAAC;IAEhD,IAAI,WAAW,EAAE,CAAC;QACjB,OAAO,GAAG,EAAE,CAAC;QACb,OAAO,CAAC,aAAa,GAAG,UAAU,WAAW,EAAE,CAAC;IACjD,CAAC;IAED,MAAM,QAAQ,GAAG,cAAc,CAAC,MAAM,CAAC,CAAC;IACxC,MAAM,cAAc,GAAG,6BAA6B,CAAC;QACpD,GAAG;QACH,OAAO;QACP,QAAQ;QACR,QAAQ;QACR,aAAa;KACb,CAAC,CAAC;IACH,MAAM,KAAK,GAAG;QACb,kBAAkB,EAAE,KAAK,IAAI,SAAS;QACtC,sBAAsB,EAAE,SAAS,IAAI,SAAS;QAC9C,yBAAyB,EAAE,YAAY,IAAI,SAAS;QACpD,gBAAgB,EAAE,WAAW;QAC7B,oBAAoB,EAAE,OAAO;QAC7B,qBAAqB,EAAE,YAAY;KACnC,CAAC;IACF,MAAM,MAAM,GAAG,YAAY,CAAC,CAAC,CAAC,GAAG,EAAE,KAAK,EAAE,QAAQ,CAAC,CAAC;IAEpD,gDAAgD;IAChD,YAAY,CAAC,CAAC,CAAC,GAAG,EAAE,KAAK,EAAE,QAAQ,CAAC,CAAC;IAErC,MAAM,aAAa,GAAG,GAAG;QACxB,CAAC,CAAC,aAAa;YACd,CAAC,CAAC,IAAI,kBAAkB,CAAC,aAAa,CAAC;YACvC,CAAC,CAAC,IAAI,iBAAiB,CAAC;gBACtB,GAAG,EAAE,GAAG,GAAG,YAAY;gBACvB,OAAO;gBACP,SAAS,EAAE,IAAI;aACf,CAAC;QACJ,CAAC,CAAC,SAAS,CAAC;IAEb,MAAM,cAAc,GAAG,GAAG;QACzB,CAAC,CAAC,aAAa;YACd,CAAC,CAAC,IAAI,mBAAmB,CAAC,aAAa,CAAC;YACxC,CAAC,CAAC,IAAI,kBAAkB,CAAC;gBACvB,GAAG,EAAE,GAAG,GAAG,aAAa;gBACxB,OAAO;gBACP,SAAS,EAAE,IAAI;aACf,CAAC;QACJ,CAAC,CAAC,SAAS,CAAC;IAEb,kDAAkD;IAClD,MAAM,eAAe,GACpB,GAAG,IAAI,cAAc;QACpB,CAAC,CAAC,IAAI,6BAA6B,CAAC;YAClC,QAAQ,EAAE,cAAc;YACxB,mBAAmB,EAAE,OAAO,CAAC,CAAC,CAAC,qBAAqB,CAAC,CAAC,CAAC,wBAAwB;YAC/E,oBAAoB,EAAE,OAAO,CAAC,CAAC,CAAC,qBAAqB,CAAC,CAAC,CAAC,wBAAwB;SAChF,CAAC;QACH,CAAC,CAAC,SAAS,CAAC;IAEd,wDAAwD;IACxD,MAAM,gBAAgB,GACrB,GAAG,IAAI,cAAc;QACpB,CAAC,CAAC,IAAI,6BAA6B,CAAC;YAClC,QAAQ,EAAE,cAAc;YACxB,mBAAmB,EAAE,OAAO,CAAC,CAAC,CAAC,qBAAqB,CAAC,CAAC,CAAC,wBAAwB;YAC/E,oBAAoB,EAAE,OAAO,CAAC,CAAC,CAAC,qBAAqB,CAAC,CAAC,CAAC,wBAAwB;SAChF,CAAC;QACH,CAAC,CAAC,SAAS,CAAC;IAEd,MAAM,aAAa,GAAG,gBAAgB;QACrC,CAAC,CAAC,IAAI,aAAa,CAAC;YAClB,QAAQ;YACR,OAAO,EAAE,CAAC,gBAAgB,CAAC;SAC3B,CAAC;QACH,CAAC,CAAC,SAAS,CAAC;IAEb,IAAI,aAAa,EAAE,CAAC;QACnB,OAAO,CAAC,sBAAsB,CAAC,aAAa,CAAC,CAAC;IAC/C,CAAC;IAED,MAAM,WAAW,GAAG,aAAa,CAAC,CAAC,CAAC,IAAI,WAAW,CAAC,EAAE,aAAa,EAAE,CAAC,CAAC,CAAC,CAAC,SAAS,CAAC;IAEnF,IAAI,OAAO,GAAG,KAAK,CAAC;IACpB,IAAI,kBAAuC,CAAC;IAE5C,IAAI,GAAG,EAAE,CAAC;QACT,MAAM,UAAU,GAAG,IAAI,mBAAmB,CAAC;YAC1C,WAAW,EAAE,CAAC,IAAI,yBAAyB,EAAE,EAAE,IAAI,oBAAoB,EAAE,CAAC;SAC1E,CAAC,CAAC;QACH,WAAW,CAAC,mBAAmB,CAAC,UAAU,CAAC,CAAC;QAE5C,eAAe,EAAE,CAAC;QAClB,kBAAkB,GAAG,IAAI,OAAO,CAAC;YAChC,kBAAkB,EAAE,cAAc,CAAC,SAAS;YAC5C,aAAa;YACb,YAAY,EAAE,eAAe;YAC7B,gBAAgB,EAAE,CAAC,2BAA2B,EAAE,CAAC;YACjD,QAAQ;YACR,iBAAiB,EAAE,UAAU;YAC7B,cAAc,EAAE,MAAM,CAAC,cAAc;SACrC,CAAC,CAAC;QACH,kBAAkB,CAAC,KAAK,EAAE,CAAC;QAC3B,WAAW,EAAE,KAAK,EAAE,CAAC;QAErB,IAAI,CAAC;YACJ,MAAM,WAAW,GAAG,MAAM,CAAC,SAAS,IAAI,EAAE,CAAC;YAC3C,MAAM,KAAK,GAAG,MAAM,CAAC,KAAK,IAAI,EAAE,CAAC;YACjC,MAAM,OAAO,GAAG,GAAG,KAAK,IAAI,WAAW,EAAE,CAAC;YAE1C,MAAM,gBAAgB,GAA2B,EAAE,CAAC;YACpD,IAAI,WAAW,EAAE,CAAC;gBACjB,gBAAgB,CAAC,aAAa,GAAG,UAAU,WAAW,EAAE,CAAC;YAC1D,CAAC;YAED,UAAU,CAAC;gBACV,OAAO;gBACP,OAAO,EAAE,GAAG;gBACZ,OAAO,EAAE,gBAAgB;gBACzB,YAAY,EAAE,OAAO;gBACrB,QAAQ,EAAE,OAAO;gBACjB,cAAc,EAAE,KAAK,EAAE,mFAAmF;gBAC1G,sEAAsE;aACtE,CAAC,CAAC;YACH,MAAM,CAAC,KAAK,CAAC,wCAAwC,OAAO,EAAE,CAAC,CAAC;YAChE,MAAM,CAAC,KAAK,CAAC,mCAAmC,CAAC,CAAC;QACnD,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YAChB,MAAM,CAAC,IAAI,CAAC,6DAA6D,EAAE;gBAC1E,KAAK,EAAE,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,KAAK,CAAC;aAC7D,CAAC,CAAC;QACJ,CAAC;QACD,OAAO,GAAG,IAAI,CAAC;IAChB,CAAC;IAED,MAAM,MAAM,GAAG,aAAa,CAAC,KAAK,CAAC,SAAS,CAAC,IAAI,EAAE,OAAO,CAAC,CAAC;IAC5D,MAAM,KAAK,GAAG,OAAO,CAAC,QAAQ,CAAC,IAAI,EAAE,OAAO,CAAC,CAAC;IAE9C,MAAM,QAAQ,GAAG,KAAK,IAAI,EAAE;QAC3B,IAAI,OAAO,EAAE,CAAC;YACb,OAAO,GAAG,KAAK,CAAC;YAChB,MAAM,CAAC,KAAK,CAAC,6BAA6B,CAAC,CAAC;YAC5C,MAAM,cAAc,CAAC,QAAQ;iBAC3B,UAAU,EAAE;iBACZ,KAAK,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,MAAM,CAAC,IAAI,CAAC,0CAA0C,EAAE,CAAC,CAAC,CAAC,CAAC;YAC3E,MAAM,cAAc,CAAC,QAAQ;gBAC5B,EAAE,QAAQ,EAAE;iBACX,KAAK,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,OAAO,IAAI,MAAM,CAAC,IAAI,CAAC,wCAAwC,EAAE,CAAC,CAAC,CAAC,CAAC;YACrF,MAAM,kBAAkB;gBACvB,EAAE,QAAQ,EAAE;iBACX,KAAK,CACL,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,OAAO,IAAI,MAAM,CAAC,IAAI,CAAC,+CAA+C,EAAE,CAAC,CAAC,CAClF,CAAC;YACH,MAAM,CAAC,KAAK,CAAC,yBAAyB,CAAC,CAAC;QACzC,CAAC;IACF,CAAC,CAAC;IAEF,IAAI,GAAG,IAAI,eAAe,EAAE,EAAE,CAAC;QAC9B,MAAM,CAAC,KAAK,CAAC,oCAAoC,CAAC,CAAC;IACpD,CAAC;IAED,OAAO,EAAE,MAAM,EAAE,KAAK,EAAE,MAAM,EAAE,QAAQ,EAAE,CAAC;AAC5C,CAAC"}

package/dist/router.js

@@ -0,0 +1,350 @@
+/* eslint-disable @typescript-eslint/no-explicit-any */
+/* eslint-disable @typescript-eslint/no-empty-object-type */
+import { Hono } from 'hono';
+import { stream as honoStream, streamSSE as honoStreamSSE } from 'hono/streaming';
+import { upgradeWebSocket as honoUpgradeWebSocket } from 'hono/bun';
+import { hash, returnResponse } from './_util';
+import { getAgentAsyncLocalStorage } from './_context';
+import { parseEmail } from './io/email';
+export const createRouter = () => {
+    const router = new Hono();
+    // tslint:disable-next-line:no-any no-unused-variable
+    // biome-ignore lint:no-any
+    const _router = router;
+    for (const method of ['get', 'put', 'post', 'delete', 'options', 'patch']) {
+        const _originalInvoker = _router[method].bind(router);
+        _router[method] = (path, ...args) => {
+            if (args.length === 1) {
+                const arg = args[0];
+                // Check if it's a middleware (not our agent handler)
+                if (typeof arg === 'function' && arg.length === 2) {
+                    // Middleware only: (path, middleware)
+                    return _originalInvoker(path, arg);
+                }
+                // 2-arg: (path, handler)
+                const handler = arg;
+                const wrapper = async (c) => {
+                    let result = handler(c);
+                    if (result instanceof Promise)
+                        result = await result;
+                    // If handler returns a Response (e.g., websocket upgrade), return it unchanged
+                    if (result instanceof Response)
+                        return result;
+                    return returnResponse(c, result);
+                };
+                return _originalInvoker(path, wrapper);
+            }
+            else {
+                // 3-arg: (path, middleware, handler)
+                const middleware = args[0];
+                const handler = args[1];
+                const wrapper = async (c) => {
+                    let result = handler(c);
+                    if (result instanceof Promise)
+                        result = await result;
+                    // If handler returns a Response (e.g., websocket upgrade), return it unchanged
+                    if (result instanceof Response) {
+                        return result;
+                    }
+                    return returnResponse(c, result);
+                };
+                return _originalInvoker(path, middleware, wrapper);
+            }
+        };
+    }
+    // shim in special routes
+    _router.email = (address, ...args) => {
+        let middleware;
+        let handler;
+        if (args.length === 1) {
+            handler = args[0];
+        }
+        else {
+            middleware = args[0];
+            handler = args[1];
+        }
+        const id = hash(address);
+        const path = `/${id}`;
+        // registerEmailHandler(address)
+        const wrapper = async (c) => {
+            const contentType = (c.req.header('content-type') || '').trim().toLowerCase();
+            if (!contentType.includes('message/rfc822')) {
+                return c.text('Bad Request: Content-Type must be message/rfc822', 400);
+            }
+            const arrayBuffer = await c.req.arrayBuffer();
+            const buffer = Buffer.from(arrayBuffer);
+            const email = await parseEmail(buffer);
+            let result = handler(email, c);
+            if (result instanceof Promise)
+                result = await result;
+            if (result === undefined) {
+                return c.text('OK', 200);
+            }
+            return returnResponse(c, result);
+        };
+        if (middleware) {
+            return router.post(path, middleware, wrapper);
+        }
+        else {
+            return router.post(path, wrapper);
+        }
+    };
+    _router.sms = ({ number }, handler) => {
+        const id = hash(number);
+        const path = `/${id}`;
+        // registerSMSHandler(number)
+        const wrapper = async (c) => {
+            let result = handler(c);
+            if (result instanceof Promise)
+                result = await result;
+            return returnResponse(c, result);
+        };
+        return router.post(path, wrapper);
+    };
+    _router.cron = (schedule, handler) => {
+        const id = hash(schedule);
+        const path = `/${id}`;
+        // registerCronHandler(schedule)
+        const wrapper = async (c) => {
+            let result = handler(c);
+            if (result instanceof Promise)
+                result = await result;
+            return returnResponse(c, result);
+        };
+        return router.post(path, wrapper);
+    };
+    _router.stream = (path, ...args) => {
+        let middleware;
+        let handler;
+        if (args.length === 1) {
+            handler = args[0];
+        }
+        else {
+            middleware = args[0];
+            handler = args[1];
+        }
+        const wrapper = (c) => {
+            // Capture the AgentContext from the request
+            const asyncLocalStorage = getAgentAsyncLocalStorage();
+            const capturedContext = asyncLocalStorage.getStore();
+            return honoStream(c, async (s) => {
+                const runInContext = async () => {
+                    try {
+                        let streamResult = handler(c);
+                        if (streamResult instanceof Promise)
+                            streamResult = await streamResult;
+                        await s.pipe(streamResult);
+                    }
+                    catch (err) {
+                        c.var.logger.error('Stream error:', err);
+                        throw err;
+                    }
+                };
+                if (capturedContext) {
+                    await asyncLocalStorage.run(capturedContext, runInContext);
+                }
+                else {
+                    await runInContext();
+                }
+            });
+        };
+        if (middleware) {
+            return router.get(path, middleware, wrapper);
+        }
+        else {
+            return router.get(path, wrapper);
+        }
+    };
+    _router.websocket = (path, ...args) => {
+        let middleware;
+        let handler;
+        if (args.length === 1) {
+            handler = args[0];
+        }
+        else {
+            middleware = args[0];
+            handler = args[1];
+        }
+        const wrapper = honoUpgradeWebSocket((c) => {
+            let openHandler;
+            let messageHandler;
+            let closeHandler;
+            let initialized = false;
+            // Capture the AgentContext from the upgrade request
+            const asyncLocalStorage = getAgentAsyncLocalStorage();
+            const capturedContext = asyncLocalStorage.getStore();
+            const wsConnection = {
+                onOpen: (h) => {
+                    openHandler = h;
+                },
+                onMessage: (h) => {
+                    messageHandler = h;
+                },
+                onClose: (h) => {
+                    closeHandler = h;
+                },
+                send: (_data) => {
+                    // This will be bound to the actual ws in the handlers
+                },
+            };
+            const setupResult = handler(c);
+            const setupFn = typeof setupResult === 'function' ? setupResult : null;
+            // Call setup IMMEDIATELY during upgrade, not in onOpen
+            // This allows the user's code to register handlers before events fire
+            if (setupFn) {
+                if (capturedContext) {
+                    asyncLocalStorage.run(capturedContext, () => setupFn(wsConnection));
+                }
+                else {
+                    setupFn(wsConnection);
+                }
+                initialized = true;
+            }
+            return {
+                onOpen: async (event, ws) => {
+                    try {
+                        // Bind the real ws.send now that we have the actual websocket
+                        wsConnection.send = (data) => ws.send(data);
+                        if (openHandler) {
+                            // Run handler in captured context
+                            const handler = openHandler;
+                            if (capturedContext) {
+                                await asyncLocalStorage.run(capturedContext, () => handler(event));
+                            }
+                            else {
+                                await handler(event);
+                            }
+                        }
+                    }
+                    catch (err) {
+                        c.var.logger?.error('WebSocket onOpen error:', err);
+                        throw err;
+                    }
+                },
+                onMessage: async (event, ws) => {
+                    try {
+                        // Lazy initialization fallback (shouldn't normally happen)
+                        if (!initialized && setupFn) {
+                            wsConnection.send = (data) => ws.send(data);
+                            if (capturedContext) {
+                                await asyncLocalStorage.run(capturedContext, async () => {
+                                    const result = setupFn(wsConnection);
+                                    if (result instanceof Promise)
+                                        await result;
+                                });
+                            }
+                            else {
+                                const result = setupFn(wsConnection);
+                                if (result instanceof Promise)
+                                    await result;
+                            }
+                            initialized = true;
+                        }
+                        if (messageHandler) {
+                            // Run handler in captured context
+                            const handler = messageHandler;
+                            if (capturedContext) {
+                                await asyncLocalStorage.run(capturedContext, () => handler(event));
+                            }
+                            else {
+                                await handler(event);
+                            }
+                        }
+                    }
+                    catch (err) {
+                        c.var.logger?.error('WebSocket onMessage error:', err);
+                        throw err;
+                    }
+                },
+                onClose: async (event, _ws) => {
+                    try {
+                        if (closeHandler) {
+                            // Run handler in captured context
+                            const handler = closeHandler;
+                            if (capturedContext) {
+                                await asyncLocalStorage.run(capturedContext, () => handler(event));
+                            }
+                            else {
+                                await handler(event);
+                            }
+                        }
+                    }
+                    catch (err) {
+                        c.var.logger?.error('WebSocket onClose error:', err);
+                    }
+                },
+            };
+        });
+        // wrapper is what upgradeWebSocket(...) returned. Force arity=2 so our get shim
+        // recognizes it as middleware and does not wrap/convert undefined -> 200.
+        const wsMiddleware = (c, next) => wrapper(c, next);
+        if (middleware) {
+            // Compose into a single middleware to avoid the 3-arg route which treats the
+            // second function as a handler and wraps it.
+            const composed = async (c, next) => {
+                return middleware(c, async () => {
+                    await wsMiddleware(c, next);
+                });
+            };
+            return router.get(path, composed);
+        }
+        else {
+            return router.get(path, wsMiddleware);
+        }
+    };
+    _router.sse = (path, ...args) => {
+        let middleware;
+        let handler;
+        if (args.length === 1) {
+            handler = args[0];
+        }
+        else {
+            middleware = args[0];
+            handler = args[1];
+        }
+        const wrapper = (c) => {
+            // Capture the AgentContext from the request
+            const asyncLocalStorage = getAgentAsyncLocalStorage();
+            const capturedContext = asyncLocalStorage.getStore();
+            return honoStreamSSE(c, async (stream) => {
+                // Wrap the stream to intercept write() calls
+                const wrappedStream = {
+                    ...stream,
+                    write: async (data) => {
+                        // Convert simple write to writeSSE format
+                        if (typeof data === 'string' ||
+                            typeof data === 'number' ||
+                            typeof data === 'boolean') {
+                            return stream.writeSSE({ data: String(data) });
+                        }
+                        else if (typeof data === 'object' && data !== null) {
+                            // If it's already an SSE message object, pass it through
+                            return stream.writeSSE(data);
+                        }
+                        return stream.writeSSE({ data: String(data) });
+                    },
+                    writeSSE: stream.writeSSE.bind(stream),
+                    onAbort: stream.onAbort.bind(stream),
+                    close: stream.close?.bind(stream),
+                };
+                const runInContext = async () => {
+                    await handler(c)(wrappedStream);
+                };
+                if (capturedContext) {
+                    await asyncLocalStorage.run(capturedContext, runInContext);
+                }
+                else {
+                    await runInContext();
+                }
+            });
+        };
+        if (middleware) {
+            return router.get(path, middleware, wrapper);
+        }
+        else {
+            return router.get(path, wrapper);
+        }
+    };
+    return router;
+};
+//# sourceMappingURL=router.js.map

package/dist/router.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"router.js","sourceRoot":"","sources":["../src/router.ts"],"names":[],"mappings":"AAAA,uDAAuD;AACvD,4DAA4D;AAC5D,OAAO,EAAgB,IAAI,EAAmD,MAAM,MAAM,CAAC;AAC3F,OAAO,EAAE,MAAM,IAAI,UAAU,EAAE,SAAS,IAAI,aAAa,EAAE,MAAM,gBAAgB,CAAC;AAClF,OAAO,EAAE,gBAAgB,IAAI,oBAAoB,EAAE,MAAM,UAAU,CAAC;AACpE,OAAO,EAAE,IAAI,EAAE,cAAc,EAAE,MAAM,SAAS,CAAC;AAE/C,OAAO,EAAE,yBAAyB,EAAE,MAAM,YAAY,CAAC;AACvD,OAAO,EAAE,UAAU,EAAc,MAAM,YAAY,CAAC;AA4EpD,MAAM,CAAC,MAAM,YAAY,GAAG,GAG1B,EAAE;IACH,MAAM,MAAM,GAAG,IAAI,IAAI,EAAQ,CAAC;IAChC,qDAAqD;IACrD,2BAA2B;IAC3B,MAAM,OAAO,GAAG,MAAa,CAAC;IAE9B,KAAK,MAAM,MAAM,IAAI,CAAC,KAAK,EAAE,KAAK,EAAE,MAAM,EAAE,QAAQ,EAAE,SAAS,EAAE,OAAO,CAAC,EAAE,CAAC;QAC3E,MAAM,gBAAgB,GAAG,OAAO,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;QACtD,OAAO,CAAC,MAAM,CAAC,GAAG,CAAC,IAAY,EAAE,GAAG,IAAW,EAAE,EAAE;YAClD,IAAI,IAAI,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;gBACvB,MAAM,GAAG,GAAG,IAAI,CAAC,CAAC,CAAC,CAAC;gBACpB,qDAAqD;gBACrD,IAAI,OAAO,GAAG,KAAK,UAAU,IAAI,GAAG,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;oBACnD,sCAAsC;oBACtC,OAAO,gBAAgB,CAAC,IAAI,EAAE,GAAG,CAAC,CAAC;gBACpC,CAAC;gBACD,yBAAyB;gBACzB,MAAM,OAAO,GAAG,GAAG,CAAC;gBACpB,MAAM,OAAO,GAAG,KAAK,EAAE,CAAU,EAAqB,EAAE;oBACvD,IAAI,MAAM,GAAG,OAAO,CAAC,CAAC,CAAC,CAAC;oBACxB,IAAI,MAAM,YAAY,OAAO;wBAAE,MAAM,GAAG,MAAM,MAAM,CAAC;oBACrD,+EAA+E;oBAC/E,IAAI,MAAM,YAAY,QAAQ;wBAAE,OAAO,MAAM,CAAC;oBAC9C,OAAO,cAAc,CAAC,CAAC,EAAE,MAAM,CAAC,CAAC;gBAClC,CAAC,CAAC;gBACF,OAAO,gBAAgB,CAAC,IAAI,EAAE,OAAO,CAAC,CAAC;YACxC,CAAC;iBAAM,CAAC;gBACP,qCAAqC;gBACrC,MAAM,UAAU,GAAG,IAAI,CAAC,CAAC,CAAC,CAAC;gBAC3B,MAAM,OAAO,GAAG,IAAI,CAAC,CAAC,CAAC,CAAC;gBACxB,MAAM,OAAO,GAAG,KAAK,EAAE,CAAU,EAAqB,EAAE;oBACvD,IAAI,MAAM,GAAG,OAAO,CAAC,CAAC,CAAC,CAAC;oBACxB,IAAI,MAAM,YAAY,OAAO;wBAAE,MAAM,GAAG,MAAM,MAAM,CAAC;oBACrD,+EAA+E;oBAC/E,IAAI,MAAM,YAAY,QAAQ,EAAE,CAAC;wBAChC,OAAO,MAAM,CAAC;oBACf,CAAC;oBACD,OAAO,cAAc,CAAC,CAAC,EAAE,MAAM,CAAC,CAAC;gBAClC,CAAC,CAAC;gBACF,OAAO,gBAAgB,CAAC,IAAI,EAAE,UAAU,EAAE,OAAO,CAAC,CAAC;YACpD,CAAC;QACF,CAAC,CAAC;IACH,CAAC;IAED,yBAAyB;IACzB,OAAO,CAAC,KAAK,GAAG,CAAC,OAAe,EAAE,GAAG,IAAW,EAAE,EAAE;QACnD,IAAI,UAAyC,CAAC;QAC9C,IAAI,OAAqB,CAAC;QAE1B,IAAI,IAAI,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;YACvB,OAAO,GAAG,IAAI,CAAC,CAAC,CAAC,CAAC;QACnB,CAAC;aAAM,CAAC;YACP,UAAU,GAAG,IAAI,CAAC,CAAC,CAAC,CAAC;YACrB,OAAO,GAAG,IAAI,CAAC,CAAC,CAAC,CAAC;QACnB,CAAC;QAED,MAAM,EAAE,GAAG,IAAI,CAAC,OAAO,CAAC,CAAC;QACzB,MAAM,IAAI,GAAG,IAAI,EAAE,EAAE,CAAC;QACtB,gCAAgC;QAChC,MAAM,OAAO,GAAG,KAAK,EAAE,CAAU,EAAqB,EAAE;YACvD,MAAM,WAAW,GAAG,CAAC,CAAC,CAAC,GAAG,CAAC,MAAM,CAAC,cAAc,CAAC,IAAI,EAAE,CAAC,CAAC,IAAI,EAAE,CAAC,WAAW,EAAE,CAAC;YAC9E,IAAI,CAAC,WAAW,CAAC,QAAQ,CAAC,gBAAgB,CAAC,EAAE,CAAC;gBAC7C,OAAO,CAAC,CAAC,IAAI,CAAC,kDAAkD,EAAE,GAAG,CAAC,CAAC;YACxE,CAAC;YAED,MAAM,WAAW,GAAG,MAAM,CAAC,CAAC,GAAG,CAAC,WAAW,EAAE,CAAC;YAC9C,MAAM,MAAM,GAAG,MAAM,CAAC,IAAI,CAAC,WAAW,CAAC,CAAC;YAExC,MAAM,KAAK,GAAG,MAAM,UAAU,CAAC,MAAM,CAAC,CAAC;YAEvC,IAAI,MAAM,GAAG,OAAO,CAAC,KAAK,EAAE,CAAC,CAAC,CAAC;YAC/B,IAAI,MAAM,YAAY,OAAO;gBAAE,MAAM,GAAG,MAAM,MAAM,CAAC;YAErD,IAAI,MAAM,KAAK,SAAS,EAAE,CAAC;gBAC1B,OAAO,CAAC,CAAC,IAAI,CAAC,IAAI,EAAE,GAAG,CAAC,CAAC;YAC1B,CAAC;YACD,OAAO,cAAc,CAAC,CAAC,EAAE,MAAM,CAAC,CAAC;QAClC,CAAC,CAAC;QAEF,IAAI,UAAU,EAAE,CAAC;YAChB,OAAO,MAAM,CAAC,IAAI,CAAC,IAAI,EAAE,UAAU,EAAE,OAAO,CAAC,CAAC;QAC/C,CAAC;aAAM,CAAC;YACP,OAAO,MAAM,CAAC,IAAI,CAAC,IAAI,EAAE,OAAO,CAAC,CAAC;QACnC,CAAC;IACF,CAAC,CAAC;IAEF,OAAO,CAAC,GAAG,GAAG,CAAC,EAAE,MAAM,EAAsB,EAAE,OAAqB,EAAE,EAAE;QACvE,MAAM,EAAE,GAAG,IAAI,CAAC,MAAM,CAAC,CAAC;QACxB,MAAM,IAAI,GAAG,IAAI,EAAE,EAAE,CAAC;QACtB,6BAA6B;QAC7B,MAAM,OAAO,GAAG,KAAK,EAAE,CAAU,EAAqB,EAAE;YACvD,IAAI,MAAM,GAAG,OAAO,CAAC,CAAC,CAAC,CAAC;YACxB,IAAI,MAAM,YAAY,OAAO;gBAAE,MAAM,GAAG,MAAM,MAAM,CAAC;YACrD,OAAO,cAAc,CAAC,CAAC,EAAE,MAAM,CAAC,CAAC;QAClC,CAAC,CAAC;QACF,OAAO,MAAM,CAAC,IAAI,CAAC,IAAI,EAAE,OAAO,CAAC,CAAC;IACnC,CAAC,CAAC;IAEF,OAAO,CAAC,IAAI,GAAG,CAAC,QAAgB,EAAE,OAAqB,EAAE,EAAE;QAC1D,MAAM,EAAE,GAAG,IAAI,CAAC,QAAQ,CAAC,CAAC;QAC1B,MAAM,IAAI,GAAG,IAAI,EAAE,EAAE,CAAC;QACtB,gCAAgC;QAChC,MAAM,OAAO,GAAG,KAAK,EAAE,CAAU,EAAqB,EAAE;YACvD,IAAI,MAAM,GAAG,OAAO,CAAC,CAAC,CAAC,CAAC;YACxB,IAAI,MAAM,YAAY,OAAO;gBAAE,MAAM,GAAG,MAAM,MAAM,CAAC;YACrD,OAAO,cAAc,CAAC,CAAC,EAAE,MAAM,CAAC,CAAC;QAClC,CAAC,CAAC;QACF,OAAO,MAAM,CAAC,IAAI,CAAC,IAAI,EAAE,OAAO,CAAC,CAAC;IACnC,CAAC,CAAC;IAEF,OAAO,CAAC,MAAM,GAAG,CAAC,IAAY,EAAE,GAAG,IAAW,EAAE,EAAE;QACjD,IAAI,UAAyC,CAAC;QAC9C,IAAI,OAAsB,CAAC;QAE3B,IAAI,IAAI,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;YACvB,OAAO,GAAG,IAAI,CAAC,CAAC,CAAC,CAAC;QACnB,CAAC;aAAM,CAAC;YACP,UAAU,GAAG,IAAI,CAAC,CAAC,CAAC,CAAC;YACrB,OAAO,GAAG,IAAI,CAAC,CAAC,CAAC,CAAC;QACnB,CAAC;QAED,MAAM,OAAO,GAAG,CAAC,CAAU,EAAE,EAAE;YAC9B,4CAA4C;YAC5C,MAAM,iBAAiB,GAAG,yBAAyB,EAAE,CAAC;YACtD,MAAM,eAAe,GAAG,iBAAiB,CAAC,QAAQ,EAAE,CAAC;YAErD,OAAO,UAAU,CAAC,CAAC,EAAE,KAAK,EAAE,CAAM,EAAE,EAAE;gBACrC,MAAM,YAAY,GAAG,KAAK,IAAI,EAAE;oBAC/B,IAAI,CAAC;wBACJ,IAAI,YAAY,GAAG,OAAO,CAAC,CAAC,CAAC,CAAC;wBAC9B,IAAI,YAAY,YAAY,OAAO;4BAAE,YAAY,GAAG,MAAM,YAAY,CAAC;wBACvE,MAAM,CAAC,CAAC,IAAI,CAAC,YAAY,CAAC,CAAC;oBAC5B,CAAC;oBAAC,OAAO,GAAG,EAAE,CAAC;wBACd,CAAC,CAAC,GAAG,CAAC,MAAM,CAAC,KAAK,CAAC,eAAe,EAAE,GAAG,CAAC,CAAC;wBACzC,MAAM,GAAG,CAAC;oBACX,CAAC;gBACF,CAAC,CAAC;gBAEF,IAAI,eAAe,EAAE,CAAC;oBACrB,MAAM,iBAAiB,CAAC,GAAG,CAAC,eAAe,EAAE,YAAY,CAAC,CAAC;gBAC5D,CAAC;qBAAM,CAAC;oBACP,MAAM,YAAY,EAAE,CAAC;gBACtB,CAAC;YACF,CAAC,CAAC,CAAC;QACJ,CAAC,CAAC;QAEF,IAAI,UAAU,EAAE,CAAC;YAChB,OAAO,MAAM,CAAC,GAAG,CAAC,IAAI,EAAE,UAAU,EAAE,OAAO,CAAC,CAAC;QAC9C,CAAC;aAAM,CAAC;YACP,OAAO,MAAM,CAAC,GAAG,CAAC,IAAI,EAAE,OAAO,CAAC,CAAC;QAClC,CAAC;IACF,CAAC,CAAC;IAEF,OAAO,CAAC,SAAS,GAAG,CAAC,IAAY,EAAE,GAAG,IAAW,EAAE,EAAE;QACpD,IAAI,UAAyC,CAAC;QAC9C,IAAI,OAAyB,CAAC;QAE9B,IAAI,IAAI,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;YACvB,OAAO,GAAG,IAAI,CAAC,CAAC,CAAC,CAAC;QACnB,CAAC;aAAM,CAAC;YACP,UAAU,GAAG,IAAI,CAAC,CAAC,CAAC,CAAC;YACrB,OAAO,GAAG,IAAI,CAAC,CAAC,CAAC,CAAC;QACnB,CAAC;QAED,MAAM,OAAO,GAAG,oBAAoB,CAAC,CAAC,CAAU,EAAE,EAAE;YACnD,IAAI,WAA+D,CAAC;YACpE,IAAI,cAAkE,CAAC;YACvE,IAAI,YAAgE,CAAC;YACrE,IAAI,WAAW,GAAG,KAAK,CAAC;YAExB,oDAAoD;YACpD,MAAM,iBAAiB,GAAG,yBAAyB,EAAE,CAAC;YACtD,MAAM,eAAe,GAAG,iBAAiB,CAAC,QAAQ,EAAE,CAAC;YAErD,MAAM,YAAY,GAAwB;gBACzC,MAAM,EAAE,CAAC,CAAC,EAAE,EAAE;oBACb,WAAW,GAAG,CAAC,CAAC;gBACjB,CAAC;gBACD,SAAS,EAAE,CAAC,CAAC,EAAE,EAAE;oBAChB,cAAc,GAAG,CAAC,CAAC;gBACpB,CAAC;gBACD,OAAO,EAAE,CAAC,CAAC,EAAE,EAAE;oBACd,YAAY,GAAG,CAAC,CAAC;gBAClB,CAAC;gBACD,IAAI,EAAE,CAAC,KAAwC,EAAE,EAAE;oBAClD,sDAAsD;gBACvD,CAAC;aACD,CAAC;YAEF,MAAM,WAAW,GAAG,OAAO,CAAC,CAAC,CAAC,CAAC;YAC/B,MAAM,OAAO,GAAG,OAAO,WAAW,KAAK,UAAU,CAAC,CAAC,CAAC,WAAW,CAAC,CAAC,CAAC,IAAI,CAAC;YAEvE,uDAAuD;YACvD,sEAAsE;YACtE,IAAI,OAAO,EAAE,CAAC;gBACb,IAAI,eAAe,EAAE,CAAC;oBACrB,iBAAiB,CAAC,GAAG,CAAC,eAAe,EAAE,GAAG,EAAE,CAAC,OAAO,CAAC,YAAY,CAAC,CAAC,CAAC;gBACrE,CAAC;qBAAM,CAAC;oBACP,OAAO,CAAC,YAAY,CAAC,CAAC;gBACvB,CAAC;gBACD,WAAW,GAAG,IAAI,CAAC;YACpB,CAAC;YAED,OAAO;gBACN,MAAM,EAAE,KAAK,EAAE,KAAU,EAAE,EAAO,EAAE,EAAE;oBACrC,IAAI,CAAC;wBACJ,8DAA8D;wBAC9D,YAAY,CAAC,IAAI,GAAG,CAAC,IAAI,EAAE,EAAE,CAAC,EAAE,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;wBAE5C,IAAI,WAAW,EAAE,CAAC;4BACjB,kCAAkC;4BAClC,MAAM,OAAO,GAAG,WAAW,CAAC;4BAC5B,IAAI,eAAe,EAAE,CAAC;gCACrB,MAAM,iBAAiB,CAAC,GAAG,CAAC,eAAe,EAAE,GAAG,EAAE,CAAC,OAAO,CAAC,KAAK,CAAC,CAAC,CAAC;4BACpE,CAAC;iCAAM,CAAC;gCACP,MAAM,OAAO,CAAC,KAAK,CAAC,CAAC;4BACtB,CAAC;wBACF,CAAC;oBACF,CAAC;oBAAC,OAAO,GAAG,EAAE,CAAC;wBACd,CAAC,CAAC,GAAG,CAAC,MAAM,EAAE,KAAK,CAAC,yBAAyB,EAAE,GAAG,CAAC,CAAC;wBACpD,MAAM,GAAG,CAAC;oBACX,CAAC;gBACF,CAAC;gBACD,SAAS,EAAE,KAAK,EAAE,KAAU,EAAE,EAAO,EAAE,EAAE;oBACxC,IAAI,CAAC;wBACJ,2DAA2D;wBAC3D,IAAI,CAAC,WAAW,IAAI,OAAO,EAAE,CAAC;4BAC7B,YAAY,CAAC,IAAI,GAAG,CAAC,IAAI,EAAE,EAAE,CAAC,EAAE,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;4BAC5C,IAAI,eAAe,EAAE,CAAC;gCACrB,MAAM,iBAAiB,CAAC,GAAG,CAAC,eAAe,EAAE,KAAK,IAAI,EAAE;oCACvD,MAAM,MAAM,GAAG,OAAO,CAAC,YAAY,CAAC,CAAC;oCACrC,IAAI,MAAM,YAAY,OAAO;wCAAE,MAAM,MAAM,CAAC;gCAC7C,CAAC,CAAC,CAAC;4BACJ,CAAC;iCAAM,CAAC;gCACP,MAAM,MAAM,GAAG,OAAO,CAAC,YAAY,CAAC,CAAC;gCACrC,IAAI,MAAM,YAAY,OAAO;oCAAE,MAAM,MAAM,CAAC;4BAC7C,CAAC;4BACD,WAAW,GAAG,IAAI,CAAC;wBACpB,CAAC;wBACD,IAAI,cAAc,EAAE,CAAC;4BACpB,kCAAkC;4BAClC,MAAM,OAAO,GAAG,cAAc,CAAC;4BAC/B,IAAI,eAAe,EAAE,CAAC;gCACrB,MAAM,iBAAiB,CAAC,GAAG,CAAC,eAAe,EAAE,GAAG,EAAE,CAAC,OAAO,CAAC,KAAK,CAAC,CAAC,CAAC;4BACpE,CAAC;iCAAM,CAAC;gCACP,MAAM,OAAO,CAAC,KAAK,CAAC,CAAC;4BACtB,CAAC;wBACF,CAAC;oBACF,CAAC;oBAAC,OAAO,GAAG,EAAE,CAAC;wBACd,CAAC,CAAC,GAAG,CAAC,MAAM,EAAE,KAAK,CAAC,4BAA4B,EAAE,GAAG,CAAC,CAAC;wBACvD,MAAM,GAAG,CAAC;oBACX,CAAC;gBACF,CAAC;gBACD,OAAO,EAAE,KAAK,EAAE,KAAU,EAAE,GAAQ,EAAE,EAAE;oBACvC,IAAI,CAAC;wBACJ,IAAI,YAAY,EAAE,CAAC;4BAClB,kCAAkC;4BAClC,MAAM,OAAO,GAAG,YAAY,CAAC;4BAC7B,IAAI,eAAe,EAAE,CAAC;gCACrB,MAAM,iBAAiB,CAAC,GAAG,CAAC,eAAe,EAAE,GAAG,EAAE,CAAC,OAAO,CAAC,KAAK,CAAC,CAAC,CAAC;4BACpE,CAAC;iCAAM,CAAC;gCACP,MAAM,OAAO,CAAC,KAAK,CAAC,CAAC;4BACtB,CAAC;wBACF,CAAC;oBACF,CAAC;oBAAC,OAAO,GAAG,EAAE,CAAC;wBACd,CAAC,CAAC,GAAG,CAAC,MAAM,EAAE,KAAK,CAAC,0BAA0B,EAAE,GAAG,CAAC,CAAC;oBACtD,CAAC;gBACF,CAAC;aACD,CAAC;QACH,CAAC,CAAC,CAAC;QAEH,gFAAgF;QAChF,0EAA0E;QAC1E,MAAM,YAAY,GAAsB,CAAC,CAAC,EAAE,IAAI,EAAE,EAAE,CAClD,OAAwC,CAAC,CAAC,EAAE,IAAI,CAAC,CAAC;QAEpD,IAAI,UAAU,EAAE,CAAC;YAChB,6EAA6E;YAC7E,6CAA6C;YAC7C,MAAM,QAAQ,GAAsB,KAAK,EAAE,CAAC,EAAE,IAAI,EAAE,EAAE;gBACrD,OAAO,UAAU,CAAC,CAAC,EAAE,KAAK,IAAI,EAAE;oBAC/B,MAAM,YAAY,CAAC,CAAC,EAAE,IAAI,CAAC,CAAC;gBAC7B,CAAC,CAAC,CAAC;YACJ,CAAC,CAAC;YACF,OAAO,MAAM,CAAC,GAAG,CAAC,IAAI,EAAE,QAAQ,CAAC,CAAC;QACnC,CAAC;aAAM,CAAC;YACP,OAAO,MAAM,CAAC,GAAG,CAAC,IAAI,EAAE,YAAY,CAAC,CAAC;QACvC,CAAC;IACF,CAAC,CAAC;IAEF,OAAO,CAAC,GAAG,GAAG,CAAC,IAAY,EAAE,GAAG,IAAW,EAAE,EAAE;QAC9C,IAAI,UAAyC,CAAC;QAC9C,IAAI,OAAmB,CAAC;QAExB,IAAI,IAAI,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;YACvB,OAAO,GAAG,IAAI,CAAC,CAAC,CAAC,CAAC;QACnB,CAAC;aAAM,CAAC;YACP,UAAU,GAAG,IAAI,CAAC,CAAC,CAAC,CAAC;YACrB,OAAO,GAAG,IAAI,CAAC,CAAC,CAAC,CAAC;QACnB,CAAC;QAED,MAAM,OAAO,GAAG,CAAC,CAAU,EAAE,EAAE;YAC9B,4CAA4C;YAC5C,MAAM,iBAAiB,GAAG,yBAAyB,EAAE,CAAC;YACtD,MAAM,eAAe,GAAG,iBAAiB,CAAC,QAAQ,EAAE,CAAC;YAErD,OAAO,aAAa,CAAC,CAAC,EAAE,KAAK,EAAE,MAAW,EAAE,EAAE;gBAC7C,6CAA6C;gBAC7C,MAAM,aAAa,GAAG;oBACrB,GAAG,MAAM;oBACT,KAAK,EAAE,KAAK,EAAE,IAAS,EAAE,EAAE;wBAC1B,0CAA0C;wBAC1C,IACC,OAAO,IAAI,KAAK,QAAQ;4BACxB,OAAO,IAAI,KAAK,QAAQ;4BACxB,OAAO,IAAI,KAAK,SAAS,EACxB,CAAC;4BACF,OAAO,MAAM,CAAC,QAAQ,CAAC,EAAE,IAAI,EAAE,MAAM,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;wBAChD,CAAC;6BAAM,IAAI,OAAO,IAAI,KAAK,QAAQ,IAAI,IAAI,KAAK,IAAI,EAAE,CAAC;4BACtD,yDAAyD;4BACzD,OAAO,MAAM,CAAC,QAAQ,CAAC,IAAI,CAAC,CAAC;wBAC9B,CAAC;wBACD,OAAO,MAAM,CAAC,QAAQ,CAAC,EAAE,IAAI,EAAE,MAAM,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;oBAChD,CAAC;oBACD,QAAQ,EAAE,MAAM,CAAC,QAAQ,CAAC,IAAI,CAAC,MAAM,CAAC;oBACtC,OAAO,EAAE,MAAM,CAAC,OAAO,CAAC,IAAI,CAAC,MAAM,CAAC;oBACpC,KAAK,EAAE,MAAM,CAAC,KAAK,EAAE,IAAI,CAAC,MAAM,CAAC;iBACjC,CAAC;gBAEF,MAAM,YAAY,GAAG,KAAK,IAAI,EAAE;oBAC/B,MAAM,OAAO,CAAC,CAAC,CAAC,CAAC,aAAa,CAAC,CAAC;gBACjC,CAAC,CAAC;gBAEF,IAAI,eAAe,EAAE,CAAC;oBACrB,MAAM,iBAAiB,CAAC,GAAG,CAAC,eAAe,EAAE,YAAY,CAAC,CAAC;gBAC5D,CAAC;qBAAM,CAAC;oBACP,MAAM,YAAY,EAAE,CAAC;gBACtB,CAAC;YACF,CAAC,CAAC,CAAC;QACJ,CAAC,CAAC;QAEF,IAAI,UAAU,EAAE,CAAC;YAChB,OAAO,MAAM,CAAC,GAAG,CAAC,IAAI,EAAE,UAAU,EAAE,OAAO,CAAC,CAAC;QAC9C,CAAC;aAAM,CAAC;YACP,OAAO,MAAM,CAAC,GAAG,CAAC,IAAI,EAAE,OAAO,CAAC,CAAC;QAClC,CAAC;IACF,CAAC,CAAC;IAEF,OAAO,MAAuC,CAAC;AAChD,CAAC,CAAC"}

package/dist/services/evalrun/composite.js

@@ -0,0 +1,26 @@
+/**
+ * An implementation of the EvalRunEventProvider which wraps multiple providers
+ */
+export class CompositeEvalRunEventProvider {
+    providers;
+    constructor(...providers) {
+        this.providers = providers;
+    }
+    /**
+     * called when the eval run starts
+     *
+     * @param event EvalRunStartEvent
+     */
+    async start(event) {
+        await Promise.allSettled(this.providers.map((provider) => provider.start(event)));
+    }
+    /**
+     * called when the eval run completes
+     *
+     * @param event EvalRunCompleteEvent
+     */
+    async complete(event) {
+        await Promise.allSettled(this.providers.map((provider) => provider.complete(event)));
+    }
+}
+//# sourceMappingURL=composite.js.map

package/dist/services/evalrun/composite.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"composite.js","sourceRoot":"","sources":["../../../src/services/evalrun/composite.ts"],"names":[],"mappings":"AAMA;;GAEG;AACH,MAAM,OAAO,6BAA6B;IACjC,SAAS,CAAyB;IAE1C,YAAY,GAAG,SAAiC;QAC/C,IAAI,CAAC,SAAS,GAAG,SAAS,CAAC;IAC5B,CAAC;IAED;;;;OAIG;IACH,KAAK,CAAC,KAAK,CAAC,KAAwB;QACnC,MAAM,OAAO,CAAC,UAAU,CAAC,IAAI,CAAC,SAAS,CAAC,GAAG,CAAC,CAAC,QAAQ,EAAE,EAAE,CAAC,QAAQ,CAAC,KAAK,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC;IACnF,CAAC;IAED;;;;OAIG;IACH,KAAK,CAAC,QAAQ,CAAC,KAA2B;QACzC,MAAM,OAAO,CAAC,UAAU,CAAC,IAAI,CAAC,SAAS,CAAC,GAAG,CAAC,CAAC,QAAQ,EAAE,EAAE,CAAC,QAAQ,CAAC,QAAQ,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC;IACtF,CAAC;CACD"}