@agentuity/server 0.0.4 → 0.0.6

package/src/otel/logger.ts

@@ -1,238 +0,0 @@
-import { format } from 'node:util';
-import { safeStringify } from '@agentuity/core';
-import * as LogsAPI from '@opentelemetry/api-logs';
-import type { Logger } from '../logger';
-import ConsoleLogger from '../logger/console';
-import { getAgentContext } from '../_context';
-
-/**
- * Reference to the original console object before patching
- */
-export const __originalConsole = Object.create(console); // save the original console before we patch it
-
-export class OtelLogger implements Logger {
-	private readonly delegates: LogsAPI.Logger[];
-	private readonly context: Record<string, unknown> | undefined;
-	private readonly logger: ConsoleLogger | undefined;
-
-	constructor(
-		useConsole: boolean,
-		delegates: LogsAPI.Logger | LogsAPI.Logger[],
-		context?: Record<string, unknown> | undefined
-	) {
-		this.delegates = Array.isArray(delegates) ? delegates : [delegates];
-		this.context = context;
-		this.logger = useConsole ? new ConsoleLogger(context, false) : undefined;
-	}
-
-	addDelegate(delegate: LogsAPI.Logger) {
-		this.delegates.push(delegate);
-	}
-
-	private formatMessage(message: unknown) {
-		if (typeof message === 'string') {
-			return message;
-		}
-		try {
-			return safeStringify(message);
-		} catch {
-			// Handle circular references or other unknown stringification errors
-			return String(message);
-		}
-	}
-
-	private getAttributes(): Record<string, unknown> | undefined {
-		try {
-			const actx = getAgentContext();
-			const current = actx.current?.metadata;
-			if (current) {
-				const result: Record<string, unknown> = {
-					...(this.context ?? {}),
-				};
-				for (const [key, value] of Object.entries(current)) {
-					if (value !== null && value !== undefined) {
-						result[`@agentuity/${key}`] = value as unknown;
-					}
-				}
-				return result;
-			}
-		} catch {
-			/* fall through */
-		}
-		return this.context;
-	}
-
-	private emitToAll(severityNumber: LogsAPI.SeverityNumber, severityText: string, body: string) {
-		const attributes = this.getAttributes();
-
-		this.delegates.forEach((delegate) => {
-			try {
-				delegate.emit({
-					severityNumber,
-					severityText,
-					body,
-					attributes: attributes as LogsAPI.LogRecord['attributes'],
-				});
-			} catch (error) {
-				// Log error to console if available, but don't fail the entire operation
-				this.logger?.error('Failed to emit log to OTLP instance:', error);
-			}
-		});
-	}
-
-	debug(message: string, ...args: unknown[]) {
-		this.logger?.debug(message, ...args);
-		let body: string;
-		try {
-			body = format(this.formatMessage(message), ...args);
-		} catch {
-			// Fallback if format causes recursion
-			body = `${this.formatMessage(message)} ${args.map((arg) => String(arg)).join(' ')}`;
-		}
-		this.emitToAll(LogsAPI.SeverityNumber.DEBUG, 'DEBUG', body);
-	}
-	info(message: string, ...args: unknown[]) {
-		this.logger?.info(message, ...args);
-		let body: string;
-		try {
-			body = format(this.formatMessage(message), ...args);
-		} catch {
-			// Fallback if format causes recursion
-			body = `${this.formatMessage(message)} ${args.map((arg) => String(arg)).join(' ')}`;
-		}
-		this.emitToAll(LogsAPI.SeverityNumber.INFO, 'INFO', body);
-	}
-	warn(message: string, ...args: unknown[]) {
-		this.logger?.warn(message, ...args);
-		let body: string;
-		try {
-			body = format(this.formatMessage(message), ...args);
-		} catch {
-			// Fallback if format causes recursion
-			body = `${this.formatMessage(message)} ${args.map((arg) => String(arg)).join(' ')}`;
-		}
-		this.emitToAll(LogsAPI.SeverityNumber.WARN, 'WARN', body);
-	}
-	error(message: string, ...args: unknown[]) {
-		this.logger?.error(message, ...args);
-		let body: string;
-		try {
-			body = format(this.formatMessage(message), ...args);
-		} catch {
-			// Fallback if format causes recursion
-			body = `${this.formatMessage(message)} ${args.map((arg) => String(arg)).join(' ')}`;
-		}
-		this.emitToAll(LogsAPI.SeverityNumber.ERROR, 'ERROR', body);
-	}
-	child(opts: Record<string, unknown>) {
-		return new OtelLogger(!!this.logger, this.delegates, {
-			...(this.context ?? {}),
-			...opts,
-		});
-	}
-}
-
-/**
- * Creates a logger that integrates with OpenTelemetry
- *
- * @param useConsole - Whether to also log to the console
- * @param context - Additional context to include with log records
- * @returns A logger instance
- */
-export function createLogger(useConsole: boolean, context?: Record<string, unknown>): Logger {
-	const delegate = LogsAPI.logs.getLogger('default', undefined, {
-		scopeAttributes: context as LogsAPI.LoggerOptions['scopeAttributes'],
-	});
-	return new OtelLogger(useConsole, [delegate], context);
-}
-
-/**
- * Patches the global console object to integrate with OpenTelemetry logging
- *
- * @param attributes - Attributes to include with all console log records
- */
-export function patchConsole(enabled: boolean, attributes: Record<string, unknown>) {
-	if (!enabled) {
-		return;
-	}
-	const _patch = { ...__originalConsole };
-	const delegate = createLogger(true, attributes);
-
-	// Patch individual console methods instead of reassigning the whole object
-	_patch.log = (...args: unknown[]) => {
-		delegate.info(args[0] as string, ...args.slice(1));
-	};
-	_patch.error = (...args: unknown[]) => {
-		delegate.error(args[0] as string, ...args.slice(1));
-	};
-	_patch.warn = (...args: unknown[]) => {
-		delegate.warn(args[0] as string, ...args.slice(1));
-	};
-	_patch.debug = (...args: unknown[]) => {
-		delegate.debug(args[0] as string, ...args.slice(1));
-	};
-	_patch.info = (...args: unknown[]) => {
-		delegate.info(args[0] as string, ...args.slice(1));
-	};
-	_patch.dir = (...args: unknown[]) => {
-		let msg = '';
-		if (args.length === 1) {
-			msg = format(args[0]);
-		} else if (args.length >= 2) {
-			msg = format(args[0], args[1]);
-		} else {
-			msg = safeStringify(args);
-		}
-		delegate.debug(msg);
-	};
-	_patch.dirxml = (...args: unknown[]) => {
-		delegate.debug('dirxml:', ...args);
-	};
-	_patch.table = (...args: unknown[]) => {
-		delegate.debug('table:', ...args);
-	};
-	_patch.trace = (...args: unknown[]) => {
-		delegate.debug(args[0] as string, ...args.slice(1));
-	};
-	_patch.group = (...args: unknown[]) => {
-		delegate.debug('group:', ...args);
-	};
-	_patch.groupCollapsed = (...args: unknown[]) => {
-		delegate.debug('groupCollapsed:', ...args);
-	};
-	_patch.groupEnd = () => {
-		delegate.debug('groupEnd');
-	};
-	_patch.clear = () => {
-		/* no-op */
-	};
-	_patch.count = (...args: unknown[]) => {
-		delegate.debug('count:', ...args);
-	};
-	_patch.countReset = (...args: unknown[]) => {
-		delegate.debug('countReset:', ...args);
-	};
-	_patch.assert = (condition?: boolean, ...args: unknown[]) => {
-		if (!condition) {
-			delegate.error('assertion failed:', ...args);
-		}
-	};
-	_patch.time = (...args: unknown[]) => {
-		delegate.debug('time:', ...args);
-	};
-	_patch.timeLog = (...args: unknown[]) => {
-		delegate.debug('timeLog:', ...args);
-	};
-	_patch.timeEnd = (...args: unknown[]) => {
-		delegate.debug('timeEnd:', ...args);
-	};
-	_patch.profile = (...args: unknown[]) => {
-		delegate.debug('profile:', ...args);
-	};
-	_patch.profileEnd = (...args: unknown[]) => {
-		delegate.debug('profileEnd:', ...args);
-	};
-
-	// eslint-disable-next-line no-global-assign
-	console = globalThis.console = _patch;
-}

package/src/otel/otel.ts

@@ -1,317 +0,0 @@
-import type { SpanProcessor } from '@opentelemetry/sdk-trace-base';
-import opentelemetry, { type Meter, metrics, propagation, type Tracer } 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 type { Resource } from '@opentelemetry/resources';
-import {
-	BatchLogRecordProcessor,
-	LoggerProvider,
-	type LogRecordProcessor,
-	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 type { Logger } from '../logger';
-import { ConsoleLogRecordExporter } from './console';
-import { instrumentFetch } from './fetch';
-import { createLogger, patchConsole } from './logger';
-import { getSDKVersion } from '../_config';
-
-/**
- * Configuration for OpenTelemetry initialization
- */
-export interface OtelConfig {
-	url?: string;
-	name: string;
-	version: string;
-	bearerToken?: string;
-	orgId?: string;
-	projectId?: string;
-	deploymentId?: string;
-	environment?: string;
-	cliVersion?: string;
-	devmode?: boolean;
-	spanProcessors?: Array<SpanProcessor>;
-}
-
-/**
- * Response from OpenTelemetry initialization
- */
-export interface OtelResponse {
-	tracer: Tracer;
-	meter: Meter;
-	logger: Logger;
-	shutdown: () => Promise<void>;
-}
-
-const devmodeExportInterval = 1_000; // 1 second
-const productionExportInterval = 10_000; // 10 seconds
-
-export const createResource = (config: OtelConfig): Resource => {
-	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,
-}: {
-	url?: string;
-	headers?: Record<string, string>;
-	resource: Resource;
-}) => {
-	let processor: LogRecordProcessor;
-	let exporter: OTLPLogExporter | undefined;
-
-	if (url) {
-		exporter = new OTLPLogExporter({
-			url: `${url}/v1/logs`,
-			headers,
-			compression: CompressionAlgorithm.GZIP,
-			timeoutMillis: 10_000,
-		});
-		processor = new BatchLogRecordProcessor(exporter);
-	} else {
-		processor = new SimpleLogRecordProcessor(new ConsoleLogRecordExporter());
-	}
-	const provider = new LoggerProvider({
-		resource,
-		processors: [processor],
-	});
-	LogsAPI.logs.setGlobalLoggerProvider(provider);
-
-	return {
-		processor,
-		provider,
-		exporter,
-	};
-};
-
-export const createUserLoggerProvider = ({
-	url,
-	headers,
-	resource,
-}: {
-	url: string;
-	headers?: Record<string, string>;
-	resource: 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: OtelConfig): OtelResponse {
-	const {
-		url,
-		name,
-		version,
-		bearerToken,
-		environment = 'development',
-		orgId,
-		projectId,
-		deploymentId,
-		devmode = false,
-	} = config;
-
-	let headers: Record<string, string> | undefined;
-
-	if (bearerToken) {
-		headers = {};
-		headers.Authorization = `Bearer ${bearerToken}`;
-	}
-
-	const resource = createResource(config);
-	const loggerProvider = createAgentuityLoggerProvider({
-		url,
-		headers,
-		resource,
-	});
-	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);
-
-	// must do this after we have created the logger
-	patchConsole(!!url, attrs);
-
-	const traceExporter = url
-		? new OTLPTraceExporter({
-				url: `${url}/v1/traces`,
-				headers,
-				keepAlive: true,
-			})
-		: undefined;
-
-	const metricExporter = url
-		? 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: NodeSDK | undefined;
-
-	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 orgName = config.orgId || '';
-			const appName = `${orgName}:${projectName}`;
-
-			const traceloopHeaders: Record<string, string> = {};
-			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.info('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) {
-		logger.debug('connected to Agentuity Agent Cloud');
-	}
-
-	return { tracer, meter, logger, shutdown };
-}