@agimon-ai/po-log 0.0.0-alpha.113 → 0.0.0-alpha.115

package/dist/extensions/server.cjs

@@ -24,7 +24,8 @@ var __toESM = (mod, isNodeMode, target) => (target = mod != null ? __create(__ge
 	enumerable: true
 }) : target, mod));
 //#endregion
-const require_metricsSource = require("../metricsSource-BbcKbQJq.cjs");
+const require_metricsSource = require("../metricsSource-CF3cp-sf.cjs");
+const require_metricsStore = require("../metricsStore-N2UlsnN4.cjs");
 let _agimon_ai_po_logsink = require("@agimon-ai/po-logsink");
 let _agimon_ai_po_core_serverPlugin = require("@agimon-ai/po-core/serverPlugin");
 let _agimon_ai_po_core_extensionFile = require("@agimon-ai/po-core/extensionFile");
@@ -327,6 +328,59 @@ function createLogsinkNamespace(operations) {
 	};
 }
 //#endregion
+//#region src/services/metricsForwarder/index.ts
+const RECORD_FAILED = "Metrics could not be recorded.";
+const METRICS_ENTRY_ATTRIBUTES = new Set(Object.values(require_metricsStore.METRICS_ATTRIBUTE));
+/** Only the fixed, scalar attributes a metrics entry may carry leave the session. */
+function fixedAttributes(attributes) {
+	const kept = {};
+	for (const [key, value] of Object.entries(attributes)) {
+		if (!METRICS_ENTRY_ATTRIBUTES.has(key)) continue;
+		if (typeof value === "string" || typeof value === "number" || typeof value === "boolean") kept[key] = value;
+	}
+	return kept;
+}
+/**
+* Hands a session's sanitized telemetry records to the metrics service a host bound, in batches.
+*
+* A host that bound none, such as one that keeps the log sink, queues nothing. Records made in the same tick
+* travel together, so a session pays one call for a turn rather than one per record.
+*/
+function createMetricsForwarder(options) {
+	const now = options.now ?? Date.now;
+	let waiting = [];
+	let scheduled = false;
+	const flush = async () => {
+		scheduled = false;
+		const batch = waiting;
+		waiting = [];
+		if (batch.length === 0) return;
+		const service = options.resolve();
+		if (service === void 0) return;
+		for (let start = 0; start < batch.length; start += 500) try {
+			await service.record(batch.slice(start, start + 500));
+		} catch {
+			options.onDiagnostic?.(RECORD_FAILED);
+		}
+	};
+	return {
+		record(name, attributes) {
+			if (options.resolve() === void 0) return;
+			waiting.push({
+				at: now(),
+				event: name,
+				attributes: fixedAttributes(attributes)
+			});
+			if (scheduled) return;
+			scheduled = true;
+			queueMicrotask(() => {
+				flush().catch(() => options.onDiagnostic?.(RECORD_FAILED));
+			});
+		},
+		flush
+	};
+}
+//#endregion
 //#region src/services/telemetryEnabled/index.ts
 function isEnabled(value) {
 	return value === "1" || value?.toLowerCase() === "true" || value?.toLowerCase() === "yes";
@@ -784,9 +838,14 @@ function createServerTelemetry(options = {}) {
 //#endregion
 //#region src/extensions/workspaces/sessions/(backend)/_lib/serverRoot.ts
 const createLogServerRoot = ({ context, agent, host, signal: ownerSignal }) => {
+	const metrics = createMetricsForwarder({
+		resolve: () => (0, _agimon_ai_po_core_serverPlugin.readPoServerScope)(context)?.resolve(require_metricsStore.PO_LOG_METRICS_CONTRACT),
+		onDiagnostic: (message) => host.context.onNotice(message)
+	});
 	const runtime = createServerTelemetry({
 		workflowContext: (0, _agimon_ai_po_core_workflow.readPoWorkflowContextService)(context),
-		childSessionHookRegistry: (0, _agimon_ai_po_core_childSession.readPoChildSessionHookRegistry)(context)
+		childSessionHookRegistry: (0, _agimon_ai_po_core_childSession.readPoChildSessionHookRegistry)(context),
+		metrics
 	});
 	const sessionId = agent?.context.sessionId;
 	const logQueries = agent && sessionId ? createLogsinkOperations({
@@ -1135,6 +1194,25 @@ function toReport(report, source, dimension) {
 function isEmpty(report) {
 	return report.totals.totalTokens === 0 && report.groups.length === 0 && report.tools.length === 0;
 }
+/**
+* A bound store answers before any sink is looked for. It claims no transport: it is neither the sink's HTTP
+* API nor its worker. The store is looked up on every query, because a host binds it after this API starts.
+*/
+function preferBound(bound, sink) {
+	let answeredByStore = false;
+	return {
+		async query(params) {
+			const store = bound();
+			answeredByStore = store !== void 0;
+			return store === void 0 ? sink.query(params) : store.query(params);
+		},
+		lastTransport: () => answeredByStore ? void 0 : sink.lastTransport(),
+		instance: () => sink.instance?.(),
+		close: async () => {
+			await sink.close?.();
+		}
+	};
+}
 function createLogHubApi(options = {}) {
 	const app = new hono.Hono();
 	const identity = {
@@ -1142,7 +1220,8 @@ function createLogHubApi(options = {}) {
 		serviceName: "pi",
 		...options.env === void 0 ? {} : { env: options.env }
 	};
-	const source = options.source ?? require_metricsSource.createMetricsSource(identity);
+	const sink = options.source ?? require_metricsSource.createMetricsSource(identity);
+	const source = options.metrics === void 0 ? sink : preferBound(options.metrics, sink);
 	const issues = options.issues ?? createIssuesSource(identity);
 	const fetchImpl = options.fetchImpl ?? globalThis.fetch;
 	const sinkEndpoint = options.sinkEndpoint ?? (() => require_metricsSource.resolveMetricsEndpoint(identity));
@@ -1233,27 +1312,36 @@ function createLogHubApi(options = {}) {
 	});
 	return app;
 }
+/**
+* The package's API for a host. `metrics` resolves the store the host bound, if any, so a host that keeps its
+* own metrics answers from them and every other host keeps answering from the log sink.
+*/
+function createLogApi(metrics) {
+	return {
+		basePath: "log",
+		start(context) {
+			const env = { ...context.environment };
+			const source = require_metricsSource.createMetricsSource({
+				env,
+				packageName: PACKAGE_NAME$1,
+				serviceName: "pi"
+			});
+			const app = createLogHubApi({
+				source,
+				env,
+				...metrics === void 0 ? {} : { metrics }
+			});
+			return {
+				fetch: (request) => app.fetch(request),
+				close: () => source.close?.()
+			};
+		}
+	};
+}
 //#endregion
 //#region src/extensions/(backend)/api/log/route.server.ts
-var route_server_default = (0, _agimon_ai_po_core_extensionFile.defineRoute)({
-	basePath: "log",
-	start(context) {
-		const env = { ...context.environment };
-		const source = require_metricsSource.createMetricsSource({
-			env,
-			packageName: PACKAGE_NAME$1,
-			serviceName: "pi"
-		});
-		const app = createLogHubApi({
-			source,
-			env
-		});
-		return {
-			fetch: (request) => app.fetch(request),
-			close: () => source.close?.()
-		};
-	}
-});
+/** The store is resolved per request: a host that keeps its own metrics binds it after this API mounts. */
+var route_server_default = (0, _agimon_ai_po_core_extensionFile.defineRoute)(({ context }) => createLogApi(() => (0, _agimon_ai_po_core_serverPlugin.readPoServerScope)(context)?.resolve(require_metricsStore.PO_LOG_METRICS_CONTRACT)));
 //#endregion
 //#region src/extensions/workspaces/sessions/(backend)/command/log-metrics.server.ts
 var log_metrics_server_default = (0, _agimon_ai_po_core_extensionFile.defineServerCommand)((context) => context.root.commands[0]);