@alexkroman1/aai 0.10.2 → 0.10.4

package/dist/middleware-core.d.ts

@@ -1,47 +0,0 @@
-/**
- * Pure middleware runner functions — zero runtime dependencies.
- *
- * This module is intentionally dependency-free so it can be bundled into
- * the secure-exec isolate harness (which has no access to node_modules).
- * Only `import type` statements are allowed here.
- */
-import type { HookContext, Middleware, MiddlewareBlockResult } from "./types.ts";
-/** Result from a middleware tool call interceptor. */
-export type ToolInterceptResult = {
-    type: "block";
-    reason: string;
-} | {
-    type: "result";
-    result: string;
-} | {
-    type: "args";
-    args: Record<string, unknown>;
-} | undefined;
-/**
- * Run all `beforeInput` middleware in order, piping the text through each.
- * Symmetric to {@link runOutputFilters} but for user input.
- */
-export declare function runInputFilters(middleware: readonly Middleware[], text: string, ctx: HookContext): Promise<string>;
-/**
- * Run all `beforeTurn` middleware in order. Returns a block result if any
- * middleware blocks the turn, or `undefined` to proceed.
- */
-export declare function runBeforeTurnMiddleware(middleware: readonly Middleware[], text: string, ctx: HookContext): Promise<MiddlewareBlockResult | undefined>;
-/**
- * Run all `afterTurn` middleware in reverse order.
- */
-export declare function runAfterTurnMiddleware(middleware: readonly Middleware[], text: string, ctx: HookContext): Promise<void>;
-/**
- * Run all `beforeToolCall` middleware in order. Returns a result that
- * may block execution, provide a cached result, or transform args.
- * Returns `undefined` to proceed with normal execution.
- */
-export declare function runToolCallInterceptors(middleware: readonly Middleware[], toolName: string, args: Readonly<Record<string, unknown>>, ctx: HookContext): Promise<ToolInterceptResult>;
-/**
- * Run all `afterToolCall` middleware in reverse order.
- */
-export declare function runAfterToolCallMiddleware(middleware: readonly Middleware[], toolName: string, args: Readonly<Record<string, unknown>>, result: string, ctx: HookContext): Promise<void>;
-/**
- * Run all `beforeOutput` middleware in order, piping the text through each.
- */
-export declare function runOutputFilters(middleware: readonly Middleware[], text: string, ctx: HookContext): Promise<string>;

package/dist/middleware-core.js

@@ -1,107 +0,0 @@
-//#region middleware-core.ts
-/**
-* Run all `beforeInput` middleware in order, piping the text through each.
-* Symmetric to {@link runOutputFilters} but for user input.
-*/
-async function runInputFilters(middleware, text, ctx) {
-	let filtered = text;
-	for (const mw of middleware) {
-		if (!mw.beforeInput) continue;
-		try {
-			filtered = await mw.beforeInput(filtered, ctx);
-		} catch (err) {
-			console.warn("Middleware beforeInput failed:", err);
-		}
-	}
-	return filtered;
-}
-/**
-* Run all `beforeTurn` middleware in order. Returns a block result if any
-* middleware blocks the turn, or `undefined` to proceed.
-*/
-async function runBeforeTurnMiddleware(middleware, text, ctx) {
-	for (const mw of middleware) {
-		if (!mw.beforeTurn) continue;
-		try {
-			const result = await mw.beforeTurn(text, ctx);
-			if (result && "block" in result && result.block) return result;
-		} catch (err) {
-			console.warn("Middleware beforeTurn failed:", err);
-		}
-	}
-}
-/**
-* Run all `afterTurn` middleware in reverse order.
-*/
-async function runAfterTurnMiddleware(middleware, text, ctx) {
-	for (let i = middleware.length - 1; i >= 0; i--) {
-		const mw = middleware[i];
-		if (!mw?.afterTurn) continue;
-		try {
-			await mw.afterTurn(text, ctx);
-		} catch (err) {
-			console.warn("Middleware afterTurn failed:", err);
-		}
-	}
-}
-/**
-* Run all `beforeToolCall` middleware in order. Returns a result that
-* may block execution, provide a cached result, or transform args.
-* Returns `undefined` to proceed with normal execution.
-*/
-async function runToolCallInterceptors(middleware, toolName, args, ctx) {
-	let currentArgs = args;
-	for (const mw of middleware) {
-		if (!mw.beforeToolCall) continue;
-		try {
-			const result = await mw.beforeToolCall(toolName, currentArgs, ctx);
-			if (!result) continue;
-			if ("block" in result && result.block) return {
-				type: "block",
-				reason: result.reason
-			};
-			if ("result" in result) return {
-				type: "result",
-				result: result.result
-			};
-			if ("args" in result) currentArgs = result.args;
-		} catch (err) {
-			console.warn("Middleware beforeToolCall failed:", err);
-		}
-	}
-	if (currentArgs !== args) return {
-		type: "args",
-		args: currentArgs
-	};
-}
-/**
-* Run all `afterToolCall` middleware in reverse order.
-*/
-async function runAfterToolCallMiddleware(middleware, toolName, args, result, ctx) {
-	for (let i = middleware.length - 1; i >= 0; i--) {
-		const mw = middleware[i];
-		if (!mw?.afterToolCall) continue;
-		try {
-			await mw.afterToolCall(toolName, args, result, ctx);
-		} catch (err) {
-			console.warn("Middleware afterToolCall failed:", err);
-		}
-	}
-}
-/**
-* Run all `beforeOutput` middleware in order, piping the text through each.
-*/
-async function runOutputFilters(middleware, text, ctx) {
-	let filtered = text;
-	for (const mw of middleware) {
-		if (!mw.beforeOutput) continue;
-		try {
-			filtered = await mw.beforeOutput(filtered, ctx);
-		} catch (err) {
-			console.warn("Middleware beforeOutput failed:", err);
-		}
-	}
-	return filtered;
-}
-//#endregion
-export { runAfterToolCallMiddleware, runAfterTurnMiddleware, runBeforeTurnMiddleware, runInputFilters, runOutputFilters, runToolCallInterceptors };

package/dist/middleware.d.ts

@@ -1,37 +0,0 @@
-/**
- * Middleware runner — executes middleware chains for turns, tool calls,
- * and output filtering.
- *
- * Pure runner logic lives in middleware-core.ts (isolate-safe, zero deps).
- * This module re-exports it and adds the HookInvoker interface.
- */
-import type { StepInfo } from "./types.ts";
-export { runAfterToolCallMiddleware, runAfterTurnMiddleware, runBeforeTurnMiddleware, runInputFilters, runOutputFilters, runToolCallInterceptors, type ToolInterceptResult, } from "./middleware-core.ts";
-/** Generic interface for invoking agent lifecycle hooks, including middleware. */
-export type HookInvoker = {
-    onConnect(sessionId: string, timeoutMs?: number): Promise<void>;
-    onDisconnect(sessionId: string, timeoutMs?: number): Promise<void>;
-    onTurn(sessionId: string, text: string, timeoutMs?: number): Promise<void>;
-    onError(sessionId: string, error: {
-        message: string;
-    }, timeoutMs?: number): Promise<void>;
-    onStep(sessionId: string, step: StepInfo, timeoutMs?: number): Promise<void>;
-    resolveTurnConfig(sid: string, ms?: number): Promise<{
-        maxSteps?: number;
-    } | null>;
-    filterInput?(sid: string, text: string, ms?: number): Promise<string>;
-    beforeTurn?(sid: string, text: string, ms?: number): Promise<string | undefined>;
-    afterTurn?(sid: string, text: string, ms?: number): Promise<void>;
-    interceptToolCall?(sid: string, tool: string, args: Readonly<Record<string, unknown>>, ms?: number): Promise<{
-        type: "block";
-        reason: string;
-    } | {
-        type: "result";
-        result: string;
-    } | {
-        type: "args";
-        args: Record<string, unknown>;
-    } | undefined>;
-    afterToolCall?(sid: string, tool: string, args: Readonly<Record<string, unknown>>, result: string, ms?: number): Promise<void>;
-    filterOutput?(sid: string, text: string, ms?: number): Promise<string>;
-};

package/dist/runtime.js

@@ -1,53 +0,0 @@
-import { DEFAULT_STT_SAMPLE_RATE, DEFAULT_TTS_SAMPLE_RATE } from "./protocol.js";
-import { context, trace } from "@opentelemetry/api";
-//#region runtime.ts
-/**
-* Runtime dependencies injected into the session pipeline.
-*
-* Defines the {@link Logger} interface, a default {@link consoleLogger},
-* and the {@link S2SConfig} for Speech-to-Speech endpoint configuration.
-*/
-/** Default console-backed logger. */
-const consoleLogger = {
-	info: (msg, ctx) => ctx ? console.log(msg, ctx) : console.log(msg),
-	warn: (msg, ctx) => ctx ? console.warn(msg, ctx) : console.warn(msg),
-	error: (msg, ctx) => ctx ? console.error(msg, ctx) : console.error(msg),
-	debug: (msg, ctx) => ctx ? console.debug(msg, ctx) : console.debug(msg)
-};
-/**
-* Structured JSON logger for production diagnostics. Each log entry is a
-* single-line JSON object with `timestamp`, `level`, `msg`, and any
-* caller-provided context fields. When an active OpenTelemetry span exists,
-* `trace_id` and `span_id` are included automatically.
-*/
-function jsonLog(level) {
-	return (msg, ctx) => {
-		const entry = {
-			timestamp: (/* @__PURE__ */ new Date()).toISOString(),
-			level,
-			msg
-		};
-		const span = trace.getSpan(context.active());
-		if (span) {
-			const sc = span.spanContext();
-			entry.trace_id = sc.traceId;
-			entry.span_id = sc.spanId;
-		}
-		if (ctx) Object.assign(entry, ctx);
-		(level === "error" || level === "warn" ? process.stderr : process.stdout).write(`${JSON.stringify(entry)}\n`);
-	};
-}
-const jsonLogger = {
-	info: jsonLog("info"),
-	warn: jsonLog("warn"),
-	error: jsonLog("error"),
-	debug: jsonLog("debug")
-};
-/** Default S2S endpoint configuration. */
-const DEFAULT_S2S_CONFIG = {
-	wssUrl: "wss://speech-to-speech.us.assemblyai.com/v1/realtime",
-	inputSampleRate: DEFAULT_STT_SAMPLE_RATE,
-	outputSampleRate: DEFAULT_TTS_SAMPLE_RATE
-};
-//#endregion
-export { DEFAULT_S2S_CONFIG, consoleLogger, jsonLogger };

package/dist/s2s.js

@@ -1,272 +0,0 @@
-import { consoleLogger } from "./runtime.js";
-import { s2sConnectionDuration, s2sErrorCounter, tracer } from "./telemetry.js";
-import { createNanoEvents } from "nanoevents";
-import { WebSocket } from "ws";
-//#region s2s.ts
-const uint8ToBase64 = (bytes) => Buffer.from(bytes).toString("base64");
-const base64ToUint8 = (base64) => new Uint8Array(Buffer.from(base64, "base64"));
-const WS_OPEN = 1;
-const defaultCreateS2sWebSocket = (url, opts) => new WebSocket(url, { headers: opts.headers });
-function hasStringFields(obj, ...keys) {
-	for (const k of keys) if (typeof obj[k] !== "string") return false;
-	return true;
-}
-function parseAgentTranscript(obj) {
-	if (typeof obj.text !== "string") return;
-	return {
-		type: "transcript.agent",
-		text: obj.text,
-		reply_id: typeof obj.reply_id === "string" ? obj.reply_id : "",
-		item_id: typeof obj.item_id === "string" ? obj.item_id : "",
-		interrupted: obj.interrupted === true
-	};
-}
-function parseToolCall(obj) {
-	if (typeof obj.call_id !== "string" || typeof obj.name !== "string") return;
-	const args = obj.args != null && typeof obj.args === "object" && !Array.isArray(obj.args) ? obj.args : {};
-	return {
-		type: "tool.call",
-		call_id: obj.call_id,
-		name: obj.name,
-		args
-	};
-}
-function passthrough(obj) {
-	return obj;
-}
-function requireFields(...keys) {
-	return (obj) => hasStringFields(obj, ...keys) ? obj : void 0;
-}
-const MESSAGE_VALIDATORS = new Map([
-	["session.ready", requireFields("session_id")],
-	["session.updated", passthrough],
-	["input.speech.started", passthrough],
-	["input.speech.stopped", passthrough],
-	["reply.content_part.started", passthrough],
-	["reply.content_part.done", passthrough],
-	["transcript.user.delta", requireFields("text")],
-	["transcript.user", requireFields("item_id", "text")],
-	["reply.started", requireFields("reply_id")],
-	["transcript.agent.delta", requireFields("delta")],
-	["transcript.agent", parseAgentTranscript],
-	["tool.call", parseToolCall],
-	["reply.done", (obj) => ({
-		type: "reply.done",
-		...typeof obj.status === "string" ? { status: obj.status } : {}
-	})],
-	["session.error", requireFields("code", "message")],
-	["error", requireFields("message")]
-]);
-function parseS2sMessage(obj) {
-	const type = obj.type;
-	if (typeof type !== "string") return;
-	return MESSAGE_VALIDATORS.get(type)?.(obj);
-}
-function dispatchS2sMessage(emitter, msg) {
-	switch (msg.type) {
-		case "session.ready":
-			emitter.emit("ready", { sessionId: msg.session_id });
-			break;
-		case "session.updated":
-			emitter.emit("sessionUpdated", msg);
-			break;
-		case "input.speech.started":
-			emitter.emit("speechStarted");
-			break;
-		case "input.speech.stopped":
-			emitter.emit("speechStopped");
-			break;
-		case "transcript.user.delta":
-			emitter.emit("userTranscriptDelta", { text: msg.text });
-			break;
-		case "transcript.user":
-			emitter.emit("userTranscript", {
-				itemId: msg.item_id,
-				text: msg.text
-			});
-			break;
-		case "reply.started":
-			emitter.emit("replyStarted", { replyId: msg.reply_id });
-			break;
-		case "transcript.agent.delta":
-			emitter.emit("agentTranscriptDelta", { text: msg.delta });
-			break;
-		case "transcript.agent":
-			emitter.emit("agentTranscript", {
-				text: msg.text,
-				replyId: msg.reply_id,
-				itemId: msg.item_id,
-				interrupted: msg.interrupted
-			});
-			break;
-		case "tool.call":
-			emitter.emit("toolCall", {
-				callId: msg.call_id,
-				name: msg.name,
-				args: msg.args
-			});
-			break;
-		case "reply.done":
-			emitter.emit("replyDone", msg.status ? { status: msg.status } : {});
-			break;
-		case "session.error":
-			if (msg.code === "session_not_found" || msg.code === "session_forbidden") emitter.emit("sessionExpired", {
-				code: msg.code,
-				message: msg.message
-			});
-			else emitter.emit("error", {
-				code: msg.code,
-				message: msg.message
-			});
-			break;
-		case "error":
-			emitter.emit("error", {
-				code: "connection",
-				message: msg.message
-			});
-			break;
-		case "reply.content_part.started":
-		case "reply.content_part.done": break;
-		default: break;
-	}
-}
-function connectS2s(opts) {
-	const { apiKey, config, createWebSocket, logger: log = consoleLogger } = opts;
-	return new Promise((resolve, reject) => {
-		log.info("S2S connecting", { url: config.wssUrl });
-		const connectionSpan = tracer.startSpan("s2s.connection", { attributes: { "aai.s2s.url": config.wssUrl } });
-		const connectStart = performance.now();
-		const ws = createWebSocket(config.wssUrl, { headers: { Authorization: `Bearer ${apiKey}` } });
-		const emitter = createNanoEvents();
-		let opened = false;
-		function send(msg) {
-			if (ws.readyState !== WS_OPEN) {
-				log.debug("S2S send dropped: socket not open", { type: msg.type });
-				return;
-			}
-			const json = JSON.stringify(msg);
-			if (msg.type !== "input.audio") log.info(`S2S >> ${msg.type}`, msg.type === "session.update" ? { payload: json } : void 0);
-			ws.send(json);
-		}
-		const handle = {
-			on: emitter.on.bind(emitter),
-			sendAudio(audio) {
-				if (ws.readyState !== WS_OPEN) {
-					log.debug("S2S sendAudio dropped: socket not open");
-					return;
-				}
-				ws.send(`{"type":"input.audio","audio":"${uint8ToBase64(audio)}"}`);
-			},
-			sendToolResult(callId, result) {
-				const msg = {
-					type: "tool.result",
-					call_id: callId,
-					result
-				};
-				log.info("S2S >> tool.result", {
-					call_id: callId,
-					resultLength: result.length
-				});
-				send(msg);
-			},
-			updateSession(sessionConfig) {
-				const { systemPrompt, ...rest } = sessionConfig;
-				send({
-					type: "session.update",
-					session: {
-						system_prompt: systemPrompt,
-						...rest
-					}
-				});
-			},
-			resumeSession(sessionId) {
-				send({
-					type: "session.resume",
-					session_id: sessionId
-				});
-			},
-			close() {
-				log.info("S2S closing");
-				ws.close();
-			}
-		};
-		ws.addEventListener("open", () => {
-			opened = true;
-			log.info("S2S WebSocket open");
-			connectionSpan.addEvent("ws.open");
-			resolve(handle);
-		});
-		function tryParseJson(data) {
-			try {
-				return JSON.parse(String(data));
-			} catch {
-				log.warn("S2S << invalid JSON", { data: String(data).slice(0, 200) });
-			}
-		}
-		function handleAudioFastPath(obj) {
-			if (obj.type === "reply.audio" && typeof obj.data === "string") {
-				const audioBytes = base64ToUint8(obj.data);
-				emitter.emit("audio", { audio: audioBytes });
-				return true;
-			}
-			return false;
-		}
-		function logIncoming(obj) {
-			if (obj.type === "reply.audio" || obj.type === "input.audio") return;
-			log.info(`S2S << ${obj.type}`, obj.type === "transcript.agent.delta" ? { delta: obj.delta } : void 0);
-		}
-		function handleS2sMessage(ev) {
-			const raw = tryParseJson(ev.data);
-			if (raw === void 0) return;
-			if (typeof raw !== "object" || raw === null || Array.isArray(raw)) {
-				log.warn("S2S << non-object JSON message", { type: typeof raw });
-				return;
-			}
-			const obj = raw;
-			logIncoming(obj);
-			if (handleAudioFastPath(obj)) return;
-			const parsed = parseS2sMessage(obj);
-			if (!parsed) {
-				log.warn(`S2S << unrecognised message type: ${obj.type ?? JSON.stringify(raw).slice(0, 200)}`);
-				return;
-			}
-			dispatchS2sMessage(emitter, parsed);
-		}
-		ws.addEventListener("message", handleS2sMessage);
-		ws.addEventListener("close", (ev) => {
-			log.info("S2S WebSocket closed", {
-				code: ev.code ?? 0,
-				reason: ev.reason ?? ""
-			});
-			const elapsed = (performance.now() - connectStart) / 1e3;
-			s2sConnectionDuration.record(elapsed);
-			connectionSpan.addEvent("ws.closed", {
-				"ws.close.code": ev.code ?? 0,
-				"ws.close.reason": ev.reason ?? ""
-			});
-			connectionSpan.end();
-			if (!opened) reject(/* @__PURE__ */ new Error(`WebSocket closed before open (code: ${ev.code ?? 0})`));
-			emitter.emit("close");
-		});
-		ws.addEventListener("error", (ev) => {
-			const message = typeof ev.message === "string" ? ev.message : "WebSocket error";
-			const errObj = new Error(message);
-			log.error("S2S WebSocket error", { error: errObj.message });
-			s2sErrorCounter.add(1);
-			connectionSpan.setStatus({
-				code: 2,
-				message: errObj.message
-			});
-			connectionSpan.recordException(errObj);
-			if (!opened) {
-				connectionSpan.end();
-				reject(errObj);
-			} else emitter.emit("error", {
-				code: "ws_error",
-				message: errObj.message
-			});
-		});
-	});
-}
-//#endregion
-export { connectS2s, defaultCreateS2sWebSocket };