@alexkroman1/aai 0.10.1 → 0.10.3

package/dist/s2s-M7JqtgFw.js

@@ -0,0 +1,272 @@
+import { n as consoleLogger } from "./runtime-CxcwaK68.js";
+import { c as s2sConnectionDuration, h as tracer, l as s2sErrorCounter } from "./telemetry-CJlaDFNc.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 { defaultCreateS2sWebSocket as n, connectS2s as t };

package/dist/s2s.js

@@ -1,272 +1,2 @@
-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
+import { n as defaultCreateS2sWebSocket, t as connectS2s } from "./s2s-M7JqtgFw.js";
 export { connectS2s, defaultCreateS2sWebSocket };

package/dist/server.d.ts

@@ -5,10 +5,9 @@
  * Calls `createDirectExecutor` + `wireSessionSocket` directly — no
  * intermediary needed.
  */
-import type { Kv } from "./kv.ts";
+import { type Storage } from "unstorage";
 import type { Logger, S2SConfig } from "./runtime.ts";
 import type { AgentDef } from "./types.ts";
-import type { VectorStore } from "./vector.ts";
 /**
  * Configuration for a self-hosted agent server created by {@link createServer}.
  *
@@ -19,10 +18,11 @@ export type ServerOptions = {
     agent: AgentDef<any>;
     /** Environment variables. Defaults to `process.env`. */
     env?: Record<string, string>;
-    /** KV store. Defaults to SQLite-backed (`.aai/local.db`). */
-    kv?: Kv;
-    /** Vector store. Defaults to SQLite-backed (`.aai/local.db`). */
-    vector?: VectorStore;
+    /**
+     * Unstorage instance for KV and vector storage. Defaults to in-memory.
+     * Configure with an S3/R2/filesystem driver for persistence.
+     */
+    storage?: Storage;
     /** HTML to serve at `GET /`. */
     clientHtml?: string;
     /** Directory containing built client files (index.html + assets/). */