@alexkroman1/aai 0.10.2 → 0.10.4

package/dist/{testing-MRl3SXsI.js → testing-BonJtfHJ.js}

@@ -1,9 +1,9 @@
-import { t as createDirectExecutor } from "./direct-executor-Ca0wt5H0.js";
-import { createSqliteKv } from "./sqlite-kv.js";
-import { createSqliteVectorStore, createTestEmbedFn } from "./sqlite-vector.js";
-import { mkdtempSync } from "node:fs";
-import { tmpdir } from "node:os";
-import { join } from "node:path";
+import "./types.js";
+import { i as createUnstorageKv, t as createRuntime } from "./direct-executor-BfHrDdPL.js";
+import { resolve } from "node:path";
+import { createStorage } from "unstorage";
+import "nanoevents";
+import { vi } from "vitest";
 //#region _mock-ws.ts
 /**
 * A mock WebSocket implementation for testing.
@@ -157,6 +157,14 @@ function installMockWebSocket() {
 	};
 }
 //#endregion
+//#region _test-utils.ts
+/** Yield to the microtask queue so pending promises settle. */
+function flush() {
+	return new Promise((r) => queueMicrotask(r));
+}
+vi.fn(), vi.fn(), vi.fn(), vi.fn();
+resolve(import.meta.dirname, "__fixtures__");
+//#endregion
 //#region testing.ts
 /**
 * Testing utilities for AAI agents.
@@ -317,7 +325,6 @@ var TestHarness = class {
 	/** @internal */
 	_sessionId;
 	_messages = [];
-	_onStepCalls = [];
 	_onTurnCalls = [];
 	_connected = false;
 	/** @internal */
@@ -329,10 +336,6 @@ var TestHarness = class {
 	get messages() {
 		return this._messages;
 	}
-	/** All `onStep` hook invocations recorded so far. */
-	get steps() {
-		return this._onStepCalls;
-	}
 	/** All `onTurn` hook invocations (the text argument) recorded so far. */
 	get turns() {
 		return this._onTurnCalls;
@@ -345,7 +348,7 @@ var TestHarness = class {
 	async connect() {
 		if (this._connected) return;
 		this._connected = true;
-		await this._executor.hookInvoker.onConnect(this._sessionId);
+		await this._executor.hooks.callHook("connect", this._sessionId);
 	}
 	/**
 	* Fire the `onDisconnect` lifecycle hook and clean up session state.
@@ -353,12 +356,12 @@ var TestHarness = class {
 	async disconnect() {
 		if (!this._connected) return;
 		this._connected = false;
-		await this._executor.hookInvoker.onDisconnect(this._sessionId);
+		await this._executor.hooks.callHook("disconnect", this._sessionId);
 	}
 	/**
 	* Execute a single tool by name with the given arguments.
 	*
-	* The tool runs with full agent context (env, state, kv, vector, messages).
+	* The tool runs with full agent context (env, state, kv, messages).
 	* The call is **not** recorded in conversation history — use {@link turn}
 	* for that.
 	*
@@ -384,7 +387,7 @@ var TestHarness = class {
 	* 1. Fires `onConnect` if this is the first turn
 	* 2. Adds the user message to conversation history
 	* 3. Fires the `onTurn` hook
-	* 4. Executes each tool call in order, firing `onStep` for each
+	* 4. Executes each tool call in order
 	* 5. Returns a {@link TurnResult} with assertion helpers
 	*
 	* @param text - The user's spoken/typed input.
@@ -407,10 +410,9 @@ var TestHarness = class {
 			content: text
 		});
 		this._onTurnCalls.push(text);
-		await this._executor.hookInvoker.onTurn(this._sessionId, text);
+		await this._executor.hooks.callHook("turn", this._sessionId, text);
 		const recorded = [];
-		for (let i = 0; i < toolCalls.length; i++) {
-			const tc = toolCalls[i];
+		for (const tc of toolCalls) {
 			const result = await this._executor.executeTool(tc.tool, tc.args, this._sessionId, this._messages);
 			recorded.push({
 				toolName: tc.tool,
@@ -421,16 +423,6 @@ var TestHarness = class {
 				role: "tool",
 				content: result
 			});
-			const step = {
-				stepNumber: i + 1,
-				toolCalls: [{
-					toolName: tc.tool,
-					args: tc.args
-				}],
-				text: ""
-			};
-			this._onStepCalls.push(step);
-			await this._executor.hookInvoker.onStep(this._sessionId, step);
 		}
 		return new TurnResult(text, recorded);
 	}
@@ -459,25 +451,14 @@ var TestHarness = class {
 	/**
 	* Reset conversation state: clears messages, step/turn history.
 	*
-	* Does **not** reset KV or vector store — create a new harness for that.
+	* Does **not** reset KV store — create a new harness for that.
 	*/
 	reset() {
 		this._messages = [];
-		this._onStepCalls = [];
 		this._onTurnCalls = [];
 	}
 };
 /**
-* Create a SQLite-vec backed vector store with deterministic test embeddings.
-* Uses a temp directory that is unique per call for test isolation.
-*/
-function createTestVectorStore() {
-	return createSqliteVectorStore({
-		path: join(mkdtempSync(join(tmpdir(), "aai-test-vec-")), "vectors.db"),
-		embedFn: createTestEmbedFn()
-	});
-}
-/**
 * Create a test harness for unit-testing an agent.
 *
 * The harness wraps the agent's tool definitions and lifecycle hooks,
@@ -485,7 +466,7 @@ function createTestVectorStore() {
 * conversations — all without audio, network, or an LLM.
 *
 * @param agent - The agent definition returned by `defineAgent()`.
-* @param options - Optional environment, KV, and vector store overrides.
+* @param options - Optional environment and KV store overrides.
 * @returns A {@link TestHarness} instance.
 *
 * @example
@@ -507,13 +488,12 @@ function createTestVectorStore() {
 * @public
 */
 function createTestHarness(agent, options = {}) {
-	const { env = {}, kv = createSqliteKv({ path: ":memory:" }), vector = createTestVectorStore() } = options;
-	return new TestHarness(createDirectExecutor({
+	const { env = {}, kv = createUnstorageKv({ storage: createStorage() }) } = options;
+	return new TestHarness(createRuntime({
 		agent,
 		env,
-		kv,
-		vector
+		kv
 	}), `test-${Date.now()}`);
 }
 //#endregion
-export { installMockWebSocket as a, MockWebSocket as i, TurnResult as n, createTestHarness as r, TestHarness as t };
+export { MockWebSocket as a, flush as i, TurnResult as n, installMockWebSocket as o, createTestHarness as r, TestHarness as t };

package/dist/testing.d.ts

@@ -35,11 +35,11 @@
  *
  * @packageDocumentation
  */
-import { type DirectExecutor } from "./direct-executor.ts";
+import { type Runtime } from "./direct-executor.ts";
 import type { Kv } from "./kv.ts";
-import type { AgentDef, Message, StepInfo } from "./types.ts";
-import type { VectorStore } from "./vector.ts";
+import type { AgentDef, Message } from "./types.ts";
 export { installMockWebSocket, MockWebSocket } from "./_mock-ws.ts";
+export { flush } from "./_test-utils.ts";
 /**
  * A single tool call recorded during a turn.
  *
@@ -137,8 +137,6 @@ export type TestHarnessOptions = {
     env?: Record<string, string>;
     /** KV store instance. Defaults to an in-memory SQLite store. */
     kv?: Kv;
-    /** Vector store instance. Defaults to an in-memory SQLite store. */
-    vector?: VectorStore;
 };
 /**
  * A tool call to execute during a simulated turn.
@@ -179,19 +177,16 @@ export type TurnToolCall = {
  */
 export declare class TestHarness {
     /** @internal */
-    readonly _executor: DirectExecutor;
+    readonly _executor: Runtime;
     /** @internal */
     readonly _sessionId: string;
     private _messages;
-    private _onStepCalls;
     private _onTurnCalls;
     private _connected;
     /** @internal */
-    constructor(executor: DirectExecutor, sessionId: string);
+    constructor(executor: Runtime, sessionId: string);
     /** Conversation messages accumulated across turns. */
     get messages(): readonly Message[];
-    /** All `onStep` hook invocations recorded so far. */
-    get steps(): readonly StepInfo[];
     /** All `onTurn` hook invocations (the text argument) recorded so far. */
     get turns(): readonly string[];
     /**
@@ -207,7 +202,7 @@ export declare class TestHarness {
     /**
      * Execute a single tool by name with the given arguments.
      *
-     * The tool runs with full agent context (env, state, kv, vector, messages).
+     * The tool runs with full agent context (env, state, kv, messages).
      * The call is **not** recorded in conversation history — use {@link turn}
      * for that.
      *
@@ -231,7 +226,7 @@ export declare class TestHarness {
      * 1. Fires `onConnect` if this is the first turn
      * 2. Adds the user message to conversation history
      * 3. Fires the `onTurn` hook
-     * 4. Executes each tool call in order, firing `onStep` for each
+     * 4. Executes each tool call in order
      * 5. Returns a {@link TurnResult} with assertion helpers
      *
      * @param text - The user's spoken/typed input.
@@ -263,7 +258,7 @@ export declare class TestHarness {
     /**
      * Reset conversation state: clears messages, step/turn history.
      *
-     * Does **not** reset KV or vector store — create a new harness for that.
+     * Does **not** reset KV store — create a new harness for that.
      */
     reset(): void;
 }
@@ -275,7 +270,7 @@ export declare class TestHarness {
  * conversations — all without audio, network, or an LLM.
  *
  * @param agent - The agent definition returned by `defineAgent()`.
- * @param options - Optional environment, KV, and vector store overrides.
+ * @param options - Optional environment and KV store overrides.
  * @returns A {@link TestHarness} instance.
  *
  * @example

package/dist/testing.js

@@ -1,2 +1,2 @@
-import { a as installMockWebSocket, i as MockWebSocket, n as TurnResult, r as createTestHarness, t as TestHarness } from "./testing-MRl3SXsI.js";
-export { MockWebSocket, TestHarness, TurnResult, createTestHarness, installMockWebSocket };
+import { a as MockWebSocket, i as flush, n as TurnResult, o as installMockWebSocket, r as createTestHarness, t as TestHarness } from "./testing-BonJtfHJ.js";
+export { MockWebSocket, TestHarness, TurnResult, createTestHarness, flush, installMockWebSocket };

package/dist/types.d.ts

@@ -3,154 +3,6 @@
  */
 import { z } from "zod";
 import type { Kv } from "./kv.ts";
-import type { VectorStore } from "./vector.ts";
-/**
- * Result returned by a `beforeTurn` middleware to block a turn.
- * @public
- */
-export type MiddlewareBlockResult = {
-    block: true;
-    reason: string;
-};
-/**
- * Result returned by a `beforeToolCall` middleware hook to short-circuit tool execution.
- *
- * Return `{ result: string }` to skip execution and use a cached/synthetic result.
- * Return `{ block: true; reason: string }` to deny the tool call.
- * Return `{ args: Record<string, unknown> }` to transform the arguments.
- * Return `undefined` to proceed normally.
- *
- * @public
- */
-export type ToolCallInterceptResult = {
-    result: string;
-} | {
-    block: true;
-    reason: string;
-} | {
-    args: Record<string, unknown>;
-} | undefined;
-/**
- * Composable middleware for the agent lifecycle.
- *
- * Middleware can intercept turns, tool calls, and output at well-defined
- * points. Multiple middleware compose in array order: the first middleware
- * in the array runs first for "before" hooks and last for "after" hooks.
- *
- * ## Middleware Composition Order
- *
- * **"Before" hooks** run in array order (first middleware → last):
- * `beforeInput` → `beforeTurn` → `beforeToolCall` → `beforeOutput`
- *
- * **"After" hooks** run in reverse array order (last middleware → first):
- * `afterToolCall` → `afterTurn`
- *
- * ### Execution flow for a user turn
- *
- * ```
- * User text arrives
- *   │
- *   ▼
- * beforeInput  — transform/redact input text (piped through each middleware)
- *   │
- *   ▼
- * beforeTurn   — block or allow the turn (short-circuits on first block)
- *   │
- *   ▼  (if not blocked)
- * LLM reasoning + tool calls
- *   │  ┌─ beforeToolCall (per call, short-circuits on block/result)
- *   │  └─ afterToolCall  (per call, reverse order)
- *   │
- *   ▼
- * beforeOutput — transform/redact LLM output text (piped through each middleware)
- *   │
- *   ▼
- * afterTurn    — cleanup/logging (reverse order)
- * ```
- *
- * ### Short-circuit behavior
- *
- * - **`beforeTurn`**: If *any* middleware returns `{ block: true, reason }`,
- *   the turn is blocked and subsequent `beforeTurn` hooks do **not** run.
- *   When a turn is blocked, `afterTurn` hooks do **not** fire (the turn
- *   never started).
- * - **`beforeToolCall`**: If *any* middleware blocks or returns a cached
- *   result, subsequent `beforeToolCall` hooks do **not** run. Arg transforms
- *   accumulate across middleware.
- * - **`beforeInput`** / **`beforeOutput`**: These *always* run all middleware
- *   in sequence (no short-circuit). Each receives the output of the previous.
- *
- * ### Ordering example
- *
- * ```ts
- * middleware: [auditTrail, hipaaGuardrails]
- * ```
- * - `beforeInput`: auditTrail runs first, then hipaaGuardrails
- * - `beforeTurn`: auditTrail runs first; if it blocks, hipaaGuardrails is skipped
- * - `afterTurn`: hipaaGuardrails runs first (reverse), then auditTrail
- *
- * If you want the guardrail to run *before* the audit trail logs, place it
- * first in the array: `[hipaaGuardrails, auditTrail]`.
- *
- * @typeParam S - The shape of per-session state. Defaults to `any` so that
- *   reusable, state-agnostic middleware (e.g. loggers, rate-limiters) can be
- *   authored without threading a state generic. Use an explicit generic
- *   (`Middleware<MyState>`) when the middleware reads or writes session state.
- *
- * @public
- */
-export type Middleware<S = any> = {
-    /** Human-readable name for logging and debugging. */
-    name: string;
-    /**
-     * Filters user input text before it reaches the LLM. Return the
-     * (possibly modified) text. Runs on every user transcript, before
-     * `beforeTurn`.
-     *
-     * Use this to redact PII (SSNs, emails, patient IDs) from speech
-     * transcripts so the LLM never sees sensitive data.
-     *
-     * Middleware is piped in array order: the output of one filter becomes
-     * the input of the next.
-     *
-     * @example
-     * ```ts
-     * beforeInput: (text) =>
-     *   text.replace(/\b\d{3}[-.]?\d{2}[-.]?\d{4}\b/g, "[SSN REDACTED]")
-     * ```
-     */
-    beforeInput?: (text: string, ctx: HookContext<S>) => string | Promise<string>;
-    /**
-     * Runs before each user turn. Can block the turn by returning
-     * `{ block: true, reason: "..." }`. Return `undefined` to proceed.
-     *
-     * Receives the text *after* `beforeInput` filtering has been applied.
-     *
-     * When a turn is blocked, subsequent `beforeTurn` middleware is skipped,
-     * and `afterTurn` hooks do **not** fire.
-     */
-    beforeTurn?: (text: string, ctx: HookContext<S>) => MiddlewareBlockResult | void | undefined | Promise<MiddlewareBlockResult | void | undefined>;
-    /**
-     * Runs after each user turn completes (after all steps finish).
-     * Runs in reverse array order.
-     */
-    afterTurn?: (text: string, ctx: HookContext<S>) => void | Promise<void>;
-    /**
-     * Runs before each tool call. Can approve, deny, transform args, or
-     * return a cached result.
-     */
-    beforeToolCall?: (toolName: string, args: Readonly<Record<string, unknown>>, ctx: HookContext<S>) => ToolCallInterceptResult | undefined | Promise<ToolCallInterceptResult | undefined>;
-    /**
-     * Runs after each tool call completes. Useful for caching results,
-     * logging, or analytics. Runs in reverse array order.
-     */
-    afterToolCall?: (toolName: string, args: Readonly<Record<string, unknown>>, result: string, ctx: HookContext<S>) => void | Promise<void>;
-    /**
-     * Filters agent text output before it is sent to TTS. Return the
-     * (possibly modified) text. Runs on every agent transcript chunk.
-     */
-    beforeOutput?: (text: string, ctx: HookContext<S>) => string | Promise<string>;
-};
 /**
  * Identifier for a built-in server-side tool.
  *
@@ -161,12 +13,11 @@ export type Middleware<S = any> = {
  * - `"visit_webpage"` — Fetch a URL and return its content as clean text.
  * - `"fetch_json"` — Call a REST API endpoint and return the JSON response.
  * - `"run_code"` — Execute JavaScript in a sandbox for calculations and data processing.
- * - `"vector_search"` — Search the agent's RAG knowledge base for relevant documents.
  * - `"memory"` — Persistent KV memory: save_memory, recall_memory, list_memories, forget_memory.
  *
  * @public
  */
-export type BuiltinTool = "web_search" | "visit_webpage" | "fetch_json" | "run_code" | "vector_search" | "memory";
+export type BuiltinTool = "web_search" | "visit_webpage" | "fetch_json" | "run_code" | "memory";
 /**
  * How the LLM should select tools during a turn.
  *
@@ -228,23 +79,8 @@ export type ToolContext<S = Record<string, unknown>> = {
     state: S;
     /** Key-value store scoped to this agent deployment. */
     kv: Kv;
-    /** Vector store scoped to this agent deployment. */
-    vector: VectorStore;
     /** Read-only snapshot of conversation messages so far. */
     messages: readonly Message[];
-    /**
-     * Push an intermediate update to the client UI before the tool finishes.
-     *
-     * Use this to send progressive data so the UI can render partial results
-     * immediately (e.g. a loading card, preview, or streaming data) instead
-     * of waiting for the full tool result.
-     *
-     * The data is serialized to JSON and delivered as a `tool_call_update`
-     * event on the client. Use `useToolCallUpdate` in the UI to consume it.
-     *
-     * No-op in sandbox (platform) mode.
-     */
-    sendUpdate(data: unknown): void;
     /**
      * SSRF-safe fetch function.
      *
@@ -258,17 +94,29 @@ export type ToolContext<S = Record<string, unknown>> = {
     sessionId: string;
 };
 /**
- * Context passed to lifecycle hooks (`onConnect`, `onTurn`, etc.).
- *
- * Same as {@link ToolContext} but without `messages`, since hooks
- * run outside the tool execution flow.
+ * Context passed to lifecycle hooks.
  *
  * @typeParam S - The shape of per-session state created by the agent's
  *   `state` factory. Defaults to `Record<string, unknown>`.
  *
  * @public
  */
-export type HookContext<S = Record<string, unknown>> = Omit<ToolContext<S>, "messages" | "sendUpdate">;
+export type HookContext<S = Record<string, unknown>> = {
+    /** Environment variables declared in the agent config. */
+    env: Readonly<Record<string, string>>;
+    /** Mutable per-session state created by the agent's `state` factory. */
+    state: S;
+    /** Key-value store scoped to this agent deployment. */
+    kv: Kv;
+    /**
+     * SSRF-safe fetch function.
+     * In self-hosted mode this calls the network directly (with SSRF protection).
+     * In platform mode this is proxied through the sidecar.
+     */
+    fetch: typeof globalThis.fetch;
+    /** Unique identifier for the current session. */
+    sessionId: string;
+};
 /**
  * Definition of a custom tool that the agent can invoke.
  *
@@ -344,17 +192,17 @@ export { defineTool as tool };
  * Create a typed `defineTool` helper with the session state type baked in.
  *
  * When tools need access to typed session state, you'd normally have to write
- * verbose generics on every `defineTool` call. `createToolFactory` eliminates
+ * verbose generics on every `defineTool` call. `defineToolFactory` eliminates
  * that boilerplate by returning a `defineTool` variant that already knows `S`.
  *
  * @example
  * ```ts
- * import { createToolFactory, defineAgent } from "aai";
+ * import { defineToolFactory, defineAgent } from "aai";
  * import { z } from "zod";
  *
  * interface PortfolioState { holdings: Map<string, number> }
  *
- * const tool = createToolFactory<PortfolioState>();
+ * const tool = defineToolFactory<PortfolioState>();
  *
  * export default defineAgent<PortfolioState>({
  *   name: "portfolio",
@@ -374,7 +222,7 @@ export { defineTool as tool };
  *
  * @public
  */
-export declare function createToolFactory<S = Record<string, unknown>>(): <P extends z.ZodObject<z.ZodRawShape>>(def: ToolDef<P, S>) => ToolDef<P, S>;
+export declare function defineToolFactory<S = Record<string, unknown>>(): <P extends z.ZodObject<z.ZodRawShape>>(def: ToolDef<P, S>) => ToolDef<P, S>;
 /**
  * A mapping of tool names to their result types.
  *
@@ -413,25 +261,6 @@ export declare function createToolFactory<S = Record<string, unknown>>(): <P ext
  * @public
  */
 export type ToolResultMap<T extends Record<string, unknown> = Record<string, unknown>> = T;
-/**
- * Information about a completed agentic step, passed to the `onStep` hook.
- *
- * Each turn may consist of multiple steps (up to `maxSteps`). A step
- * represents one LLM invocation that may include tool calls and text output.
- *
- * @public
- */
-export type StepInfo = {
-    /** 1-based step index within the current turn. */
-    stepNumber: number;
-    /** Tool calls made during this step. */
-    toolCalls: readonly {
-        toolName: string;
-        args: Readonly<Record<string, unknown>>;
-    }[];
-    /** LLM text output for this step. */
-    text: string;
-};
 /**
  * Options passed to {@link defineAgent} to configure an agent.
  *
@@ -485,21 +314,6 @@ export type AgentOptions<S = Record<string, unknown>> = {
     tools?: Readonly<Record<string, ToolDef<z.ZodObject<z.ZodRawShape>, NoInfer<S>>>>;
     /** Factory that creates fresh per-session state. Called once per connection. */
     state?: () => S;
-    /**
-     * Enable automatic session state persistence across reconnects.
-     *
-     * When enabled, session state, conversation history, and the S2S session ID
-     * are saved to KV on disconnect and restored when the client reconnects with
-     * `?sessionId=<old-session-id>` in the WebSocket URL.
-     *
-     * - `true` — enable with default TTL (1 hour)
-     * - `{ ttl: number }` — enable with custom TTL in milliseconds
-     *
-     * Requires the agent's `state` return value to be JSON-serializable.
-     */
-    persistence?: boolean | {
-        ttl?: number;
-    };
     /** Called when a new session connects. */
     onConnect?: (ctx: HookContext<S>) => void | Promise<void>;
     /** Called when a session disconnects. */
@@ -508,15 +322,6 @@ export type AgentOptions<S = Record<string, unknown>> = {
     onError?: (error: Error, ctx?: HookContext<S>) => void;
     /** Called after a complete turn (all steps finished). */
     onTurn?: (text: string, ctx: HookContext<S>) => void | Promise<void>;
-    /** Called after each agentic step completes. */
-    onStep?: (step: StepInfo, ctx: HookContext<S>) => void | Promise<void>;
-    /**
-     * Composable middleware that intercepts turns, tool calls, and output.
-     *
-     * Middleware runs in array order for "before" hooks (first to last)
-     * and reverse order for "after" hooks (last to first).
-     */
-    middleware?: readonly Middleware<S>[];
     /**
      * Close the S2S connection after this many milliseconds of inactivity.
      * Inactivity means no audio, transcripts, or tool calls in either direction.
@@ -543,8 +348,8 @@ export declare const DEFAULT_GREETING: string;
  *
  * Core fields (`name`, `instructions`, `greeting`, `maxSteps`, `tools`)
  * are resolved to their final values with defaults applied. Optional
- * behavioral fields (hooks, middleware, `sttPrompt`, etc.) remain
- * optional — `undefined` means "not configured."
+ * behavioral fields (hooks, `sttPrompt`, etc.) remain optional —
+ * `undefined` means "not configured."
  *
  * @public
  */
@@ -558,16 +363,10 @@ export type AgentDef<S = Record<string, unknown>> = {
     builtinTools?: readonly BuiltinTool[];
     tools: Readonly<Record<string, ToolDef<z.ZodObject<z.ZodRawShape>, S>>>;
     state?: () => S;
-    /** Resolved persistence config, or `undefined` if disabled. */
-    persistence?: {
-        ttl: number;
-    };
     onConnect?: (ctx: HookContext<S>) => void | Promise<void>;
     onDisconnect?: (ctx: HookContext<S>) => void | Promise<void>;
     onError?: (error: Error, ctx?: HookContext<S>) => void;
     onTurn?: (text: string, ctx: HookContext<S>) => void | Promise<void>;
-    onStep?: (step: StepInfo, ctx: HookContext<S>) => void | Promise<void>;
-    middleware?: readonly Middleware<S>[];
     idleTimeoutMs?: number;
 };
 /** @internal Zod schema for {@link BuiltinTool}. Exported for reuse in internal schemas. */
@@ -576,7 +375,6 @@ export declare const BuiltinToolSchema: z.ZodEnum<{
     visit_webpage: "visit_webpage";
     fetch_json: "fetch_json";
     run_code: "run_code";
-    vector_search: "vector_search";
     memory: "memory";
 }>;
 /** @internal Zod schema for {@link ToolChoice}. Exported for reuse in internal schemas. */

package/dist/types.js

@@ -38,17 +38,17 @@ function defineTool(def) {
 * Create a typed `defineTool` helper with the session state type baked in.
 *
 * When tools need access to typed session state, you'd normally have to write
-* verbose generics on every `defineTool` call. `createToolFactory` eliminates
+* verbose generics on every `defineTool` call. `defineToolFactory` eliminates
 * that boilerplate by returning a `defineTool` variant that already knows `S`.
 *
 * @example
 * ```ts
-* import { createToolFactory, defineAgent } from "aai";
+* import { defineToolFactory, defineAgent } from "aai";
 * import { z } from "zod";
 *
 * interface PortfolioState { holdings: Map<string, number> }
 *
-* const tool = createToolFactory<PortfolioState>();
+* const tool = defineToolFactory<PortfolioState>();
 *
 * export default defineAgent<PortfolioState>({
 *   name: "portfolio",
@@ -68,7 +68,7 @@ function defineTool(def) {
 *
 * @public
 */
-function createToolFactory() {
+function defineToolFactory() {
 	return (def) => def;
 }
 /**
@@ -100,7 +100,6 @@ const BuiltinToolSchema = z.enum([
 	"visit_webpage",
 	"fetch_json",
 	"run_code",
-	"vector_search",
 	"memory"
 ]);
 /** @internal Zod schema for {@link ToolChoice}. Exported for reuse in internal schemas. */
@@ -117,8 +116,6 @@ const ToolDefSchema = z.object({
 	parameters: z.custom((val) => val === void 0 || val instanceof z.ZodType, "Expected a Zod schema").optional(),
 	execute: z.function()
 });
-/** Default TTL for persisted session data: 1 hour. */
-const DEFAULT_PERSIST_TTL = 36e5;
 const AgentOptionsSchema = z.object({
 	name: z.string().min(1, "Agent name must be non-empty"),
 	instructions: z.string().optional(),
@@ -129,21 +126,10 @@ const AgentOptionsSchema = z.object({
 	builtinTools: z.array(BuiltinToolSchema).optional(),
 	tools: z.record(z.string(), ToolDefSchema).optional(),
 	state: z.function().optional(),
-	persistence: z.union([z.literal(true), z.object({ ttl: z.number().int().positive().optional() })]).optional(),
 	onConnect: z.function().optional(),
 	onDisconnect: z.function().optional(),
 	onError: z.function().optional(),
 	onTurn: z.function().optional(),
-	onStep: z.function().optional(),
-	middleware: z.array(z.object({
-		name: z.string().min(1, "Middleware name must be non-empty"),
-		beforeInput: z.function().optional(),
-		beforeTurn: z.function().optional(),
-		afterTurn: z.function().optional(),
-		beforeToolCall: z.function().optional(),
-		afterToolCall: z.function().optional(),
-		beforeOutput: z.function().optional()
-	})).optional(),
 	idleTimeoutMs: z.number().nonnegative().optional()
 });
 /**
@@ -178,15 +164,13 @@ const AgentOptionsSchema = z.object({
 */
 function defineAgent(options) {
 	AgentOptionsSchema.parse(options);
-	const persistence = options.persistence ? { ttl: typeof options.persistence === "object" ? options.persistence.ttl ?? DEFAULT_PERSIST_TTL : DEFAULT_PERSIST_TTL } : void 0;
 	return {
 		...options,
 		instructions: options.instructions ?? DEFAULT_INSTRUCTIONS,
 		greeting: options.greeting ?? "Hey there. I'm a voice assistant. What can I help you with?",
 		maxSteps: options.maxSteps ?? 5,
-		tools: options.tools ?? {},
-		persistence
+		tools: options.tools ?? {}
 	};
 }
 //#endregion
-export { BuiltinToolSchema, DEFAULT_GREETING, DEFAULT_INSTRUCTIONS, ToolChoiceSchema, createToolFactory, defineAgent, defineTool, defineTool as tool };
+export { BuiltinToolSchema, DEFAULT_GREETING, DEFAULT_INSTRUCTIONS, ToolChoiceSchema, defineAgent, defineTool, defineTool as tool, defineToolFactory };