@amodalai/runtime 0.1.26 → 0.2.0

package/dist/src/tools/store-tools.js

@@ -0,0 +1,224 @@
+/**
+ * @license
+ * Copyright 2026 Amodal Labs, Inc.
+ * SPDX-License-Identifier: MIT
+ */
+/**
+ * Store tools rewritten for the new ToolRegistry.
+ *
+ * These replace the BaseDeclarativeTool subclasses in core with Zod-based
+ * ToolDefinition objects. Each function returns a ToolDefinition that can
+ * be registered on the ToolRegistry.
+ *
+ * Three tools per store:
+ * - store_{name}       — write a single document
+ * - store_{name}_batch — write multiple documents
+ * - query_store        — query/get documents from any store
+ */
+import { z } from 'zod';
+import { StoreError } from '../errors.js';
+// ---------------------------------------------------------------------------
+// Helpers
+// ---------------------------------------------------------------------------
+/**
+ * Convert a store name (kebab-case) to a tool name.
+ * Example: "active-alerts" → "store_active_alerts"
+ */
+export function storeToToolName(storeName) {
+    return `store_${storeName.replace(/-/g, '_')}`;
+}
+/**
+ * Resolve a key template (e.g., "{alert_id}") from payload values.
+ */
+function resolveKey(template, payload) {
+    return template.replace(/\{(\w+)\}/g, (_match, field) => {
+        const value = payload[field];
+        if (value === undefined || value === null) {
+            throw new StoreError(`Missing required field "${field}" for key template "${template}"`, {
+                store: '',
+                operation: 'resolve_key',
+                context: { template, field },
+            });
+        }
+        return String(value);
+    });
+}
+/**
+ * Build a Zod schema from a store's entity field definitions.
+ */
+function buildStoreZodSchema(store) {
+    const shape = {};
+    for (const [name, field] of Object.entries(store.entity.schema)) {
+        shape[name] = fieldToZod(field);
+    }
+    return z.object(shape).passthrough();
+}
+function fieldToZod(field) {
+    let schema;
+    switch (field.type) {
+        case 'string':
+            schema = z.string();
+            break;
+        case 'number': {
+            let numSchema = z.number();
+            if (field.min !== undefined)
+                numSchema = numSchema.min(field.min);
+            if (field.max !== undefined)
+                numSchema = numSchema.max(field.max);
+            schema = numSchema;
+            break;
+        }
+        case 'boolean':
+            schema = z.boolean();
+            break;
+        case 'datetime':
+            schema = z.string();
+            break;
+        case 'enum':
+            schema = field.values && field.values.length > 0
+                // eslint-disable-next-line @typescript-eslint/no-unsafe-type-assertion -- guarded by length > 0 check
+                ? z.enum(field.values)
+                : z.string();
+            break;
+        case 'array':
+            schema = z.array(field.item ? fieldToZod(field.item) : z.string());
+            break;
+        case 'object':
+            if (field.fields) {
+                const nested = {};
+                for (const [n, f] of Object.entries(field.fields)) {
+                    nested[n] = fieldToZod(f);
+                }
+                schema = z.object(nested);
+            }
+            else {
+                schema = z.record(z.unknown());
+            }
+            break;
+        case 'ref':
+            schema = z.string();
+            break;
+        default:
+            schema = z.unknown();
+            break;
+    }
+    if (field.nullable) {
+        schema = schema.nullable();
+    }
+    return schema;
+}
+// ---------------------------------------------------------------------------
+// Store write tool
+// ---------------------------------------------------------------------------
+/**
+ * Create a store write ToolDefinition for a single store.
+ */
+export function createStoreWriteTool(store, backend, appId) {
+    const entitySchema = buildStoreZodSchema(store);
+    return {
+        description: `Write a single ${store.entity.name} to the ${store.name} store. Creates or updates the document by key.`,
+        parameters: entitySchema,
+        readOnly: false,
+        metadata: { category: 'store' },
+        async execute(params, _ctx) {
+            const key = resolveKey(store.entity.key, params);
+            const result = await backend.put(appId, store.name, key, params, {});
+            return { stored: true, key, version: result.version };
+        },
+    };
+}
+// ---------------------------------------------------------------------------
+// Store batch tool
+// ---------------------------------------------------------------------------
+/**
+ * Create a store batch write ToolDefinition for a single store.
+ */
+export function createStoreBatchTool(store, backend, appId) {
+    const entitySchema = buildStoreZodSchema(store);
+    return {
+        description: `Write multiple ${store.entity.name}(s) to the ${store.name} store in one call. Each item is created or updated by key.`,
+        parameters: z.object({
+            items: z.array(entitySchema).min(1),
+        }),
+        readOnly: false,
+        metadata: { category: 'store' },
+        async execute(params, _ctx) {
+            let stored = 0;
+            let failed = 0;
+            const errors = [];
+            for (const item of params.items) {
+                try {
+                    const key = resolveKey(store.entity.key, item);
+                    await backend.put(appId, store.name, key, item, {});
+                    stored++;
+                }
+                catch (err) {
+                    failed++;
+                    errors.push(err instanceof Error ? err.message : String(err));
+                }
+            }
+            return {
+                stored,
+                failed,
+                total: params.items.length,
+                ...(errors.length > 0 ? { errors: errors.slice(0, 3) } : {}),
+            };
+        },
+    };
+}
+// ---------------------------------------------------------------------------
+// Store query tool
+// ---------------------------------------------------------------------------
+export const QUERY_STORE_TOOL_NAME = 'query_store';
+/**
+ * Create a store query ToolDefinition that reads from any store.
+ */
+export function createStoreQueryTool(stores, backend, appId) {
+    const storeNames = stores.map((s) => s.name);
+    return {
+        description: `Query documents from a data store. Use the "store" parameter to specify which store (${storeNames.join(', ')}). Pass "key" to fetch a single document by ID, or "filter" to search for documents matching field values. Returns the matching document(s) with all fields.`,
+        parameters: z.object({
+            store: (storeNames.length > 0
+                // eslint-disable-next-line @typescript-eslint/no-unsafe-type-assertion -- guarded by length > 0 check
+                ? z.enum(storeNames)
+                : z.string()).describe(`Name of the store to query (${storeNames.join(', ')})`),
+            key: z.string().optional().describe('Fetch a single document by its key/ID'),
+            filter: z.record(z.unknown()).optional().describe('Filter documents by field values, e.g. {"status": "active"}'),
+            sort: z.string().optional().describe('Sort field name'),
+            limit: z.number().optional().describe('Max number of documents to return'),
+        }),
+        readOnly: true,
+        metadata: { category: 'store' },
+        async execute(params, _ctx) {
+            if (params.key) {
+                const doc = await backend.get(appId, params.store, params.key);
+                if (!doc) {
+                    return { found: false, key: params.key };
+                }
+                return { found: true, ...doc };
+            }
+            const result = await backend.list(appId, params.store, {
+                filter: params.filter,
+                sort: params.sort,
+                limit: typeof params.limit === 'number' ? params.limit : 20,
+            });
+            return result;
+        },
+    };
+}
+// ---------------------------------------------------------------------------
+// Register all store tools
+// ---------------------------------------------------------------------------
+/**
+ * Register store tools on a ToolRegistry for all stores in a bundle.
+ */
+export function registerStoreTools(registry, stores, backend, appId) {
+    for (const store of stores) {
+        registry.register(storeToToolName(store.name), createStoreWriteTool(store, backend, appId));
+        registry.register(`${storeToToolName(store.name)}_batch`, createStoreBatchTool(store, backend, appId));
+    }
+    if (stores.length > 0) {
+        registry.register(QUERY_STORE_TOOL_NAME, createStoreQueryTool(stores, backend, appId));
+    }
+}
+//# sourceMappingURL=store-tools.js.map

package/dist/src/tools/store-tools.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"store-tools.js","sourceRoot":"","sources":["../../../src/tools/store-tools.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AAEH;;;;;;;;;;;GAWG;AAEH,OAAO,EAAC,CAAC,EAAC,MAAM,KAAK,CAAC;AAEtB,OAAO,EAAC,UAAU,EAAC,MAAM,cAAc,CAAC;AAGxC,8EAA8E;AAC9E,UAAU;AACV,8EAA8E;AAE9E;;;GAGG;AACH,MAAM,UAAU,eAAe,CAAC,SAAiB;IAC/C,OAAO,SAAS,SAAS,CAAC,OAAO,CAAC,IAAI,EAAE,GAAG,CAAC,EAAE,CAAC;AACjD,CAAC;AAED;;GAEG;AACH,SAAS,UAAU,CAAC,QAAgB,EAAE,OAAgC;IACpE,OAAO,QAAQ,CAAC,OAAO,CAAC,YAAY,EAAE,CAAC,MAAM,EAAE,KAAa,EAAE,EAAE;QAC9D,MAAM,KAAK,GAAG,OAAO,CAAC,KAAK,CAAC,CAAC;QAC7B,IAAI,KAAK,KAAK,SAAS,IAAI,KAAK,KAAK,IAAI,EAAE,CAAC;YAC1C,MAAM,IAAI,UAAU,CAAC,2BAA2B,KAAK,uBAAuB,QAAQ,GAAG,EAAE;gBACvF,KAAK,EAAE,EAAE;gBACT,SAAS,EAAE,aAAa;gBACxB,OAAO,EAAE,EAAC,QAAQ,EAAE,KAAK,EAAC;aAC3B,CAAC,CAAC;QACL,CAAC;QACD,OAAO,MAAM,CAAC,KAAK,CAAC,CAAC;IACvB,CAAC,CAAC,CAAC;AACL,CAAC;AAED;;GAEG;AACH,SAAS,mBAAmB,CAAC,KAAkB;IAC7C,MAAM,KAAK,GAAiC,EAAE,CAAC;IAE/C,KAAK,MAAM,CAAC,IAAI,EAAE,KAAK,CAAC,IAAI,MAAM,CAAC,OAAO,CAAC,KAAK,CAAC,MAAM,CAAC,MAAM,CAAC,EAAE,CAAC;QAChE,KAAK,CAAC,IAAI,CAAC,GAAG,UAAU,CAAC,KAAK,CAAC,CAAC;IAClC,CAAC;IAED,OAAO,CAAC,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC,WAAW,EAAE,CAAC;AACvC,CAAC;AAED,SAAS,UAAU,CAAC,KAA2B;IAC7C,IAAI,MAAoB,CAAC;IAEzB,QAAQ,KAAK,CAAC,IAAI,EAAE,CAAC;QACnB,KAAK,QAAQ;YACX,MAAM,GAAG,CAAC,CAAC,MAAM,EAAE,CAAC;YACpB,MAAM;QACR,KAAK,QAAQ,CAAC,CAAC,CAAC;YACd,IAAI,SAAS,GAAG,CAAC,CAAC,MAAM,EAAE,CAAC;YAC3B,IAAI,KAAK,CAAC,GAAG,KAAK,SAAS;gBAAE,SAAS,GAAG,SAAS,CAAC,GAAG,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC;YAClE,IAAI,KAAK,CAAC,GAAG,KAAK,SAAS;gBAAE,SAAS,GAAG,SAAS,CAAC,GAAG,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC;YAClE,MAAM,GAAG,SAAS,CAAC;YACnB,MAAM;QACR,CAAC;QACD,KAAK,SAAS;YACZ,MAAM,GAAG,CAAC,CAAC,OAAO,EAAE,CAAC;YACrB,MAAM;QACR,KAAK,UAAU;YACb,MAAM,GAAG,CAAC,CAAC,MAAM,EAAE,CAAC;YACpB,MAAM;QACR,KAAK,MAAM;YACT,MAAM,GAAG,KAAK,CAAC,MAAM,IAAI,KAAK,CAAC,MAAM,CAAC,MAAM,GAAG,CAAC;gBAC9C,sGAAsG;gBACtG,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,MAA+B,CAAC;gBAC/C,CAAC,CAAC,CAAC,CAAC,MAAM,EAAE,CAAC;YACf,MAAM;QACR,KAAK,OAAO;YACV,MAAM,GAAG,CAAC,CAAC,KAAK,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC,CAAC,UAAU,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,MAAM,EAAE,CAAC,CAAC;YACnE,MAAM;QACR,KAAK,QAAQ;YACX,IAAI,KAAK,CAAC,MAAM,EAAE,CAAC;gBACjB,MAAM,MAAM,GAAiC,EAAE,CAAC;gBAChD,KAAK,MAAM,CAAC,CAAC,EAAE,CAAC,CAAC,IAAI,MAAM,CAAC,OAAO,CAAC,KAAK,CAAC,MAAM,CAAC,EAAE,CAAC;oBAClD,MAAM,CAAC,CAAC,CAAC,GAAG,UAAU,CAAC,CAAC,CAAC,CAAC;gBAC5B,CAAC;gBACD,MAAM,GAAG,CAAC,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC;YAC5B,CAAC;iBAAM,CAAC;gBACN,MAAM,GAAG,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC,OAAO,EAAE,CAAC,CAAC;YACjC,CAAC;YACD,MAAM;QACR,KAAK,KAAK;YACR,MAAM,GAAG,CAAC,CAAC,MAAM,EAAE,CAAC;YACpB,MAAM;QACR;YACE,MAAM,GAAG,CAAC,CAAC,OAAO,EAAE,CAAC;YACrB,MAAM;IACV,CAAC;IAED,IAAI,KAAK,CAAC,QAAQ,EAAE,CAAC;QACnB,MAAM,GAAG,MAAM,CAAC,QAAQ,EAAE,CAAC;IAC7B,CAAC;IAED,OAAO,MAAM,CAAC;AAChB,CAAC;AAED,8EAA8E;AAC9E,mBAAmB;AACnB,8EAA8E;AAE9E;;GAEG;AACH,MAAM,UAAU,oBAAoB,CAClC,KAAkB,EAClB,OAAqB,EACrB,KAAa;IAEb,MAAM,YAAY,GAAG,mBAAmB,CAAC,KAAK,CAAC,CAAC;IAEhD,OAAO;QACL,WAAW,EAAE,kBAAkB,KAAK,CAAC,MAAM,CAAC,IAAI,WAAW,KAAK,CAAC,IAAI,iDAAiD;QACtH,UAAU,EAAE,YAAY;QACxB,QAAQ,EAAE,KAAK;QACf,QAAQ,EAAE,EAAC,QAAQ,EAAE,OAAO,EAAC;QAE7B,KAAK,CAAC,OAAO,CAAC,MAA+B,EAAE,IAAiB;YAC9D,MAAM,GAAG,GAAG,UAAU,CAAC,KAAK,CAAC,MAAM,CAAC,GAAG,EAAE,MAAM,CAAC,CAAC;YACjD,MAAM,MAAM,GAAG,MAAM,OAAO,CAAC,GAAG,CAAC,KAAK,EAAE,KAAK,CAAC,IAAI,EAAE,GAAG,EAAE,MAAM,EAAE,EAAE,CAAC,CAAC;YACrE,OAAO,EAAC,MAAM,EAAE,IAAI,EAAE,GAAG,EAAE,OAAO,EAAE,MAAM,CAAC,OAAO,EAAC,CAAC;QACtD,CAAC;KACF,CAAC;AACJ,CAAC;AAED,8EAA8E;AAC9E,mBAAmB;AACnB,8EAA8E;AAE9E;;GAEG;AACH,MAAM,UAAU,oBAAoB,CAClC,KAAkB,EAClB,OAAqB,EACrB,KAAa;IAEb,MAAM,YAAY,GAAG,mBAAmB,CAAC,KAAK,CAAC,CAAC;IAEhD,OAAO;QACL,WAAW,EAAE,kBAAkB,KAAK,CAAC,MAAM,CAAC,IAAI,cAAc,KAAK,CAAC,IAAI,6DAA6D;QACrI,UAAU,EAAE,CAAC,CAAC,MAAM,CAAC;YACnB,KAAK,EAAE,CAAC,CAAC,KAAK,CAAC,YAAY,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC;SACpC,CAAC;QACF,QAAQ,EAAE,KAAK;QACf,QAAQ,EAAE,EAAC,QAAQ,EAAE,OAAO,EAAC;QAE7B,KAAK,CAAC,OAAO,CAAC,MAA+C,EAAE,IAAiB;YAC9E,IAAI,MAAM,GAAG,CAAC,CAAC;YACf,IAAI,MAAM,GAAG,CAAC,CAAC;YACf,MAAM,MAAM,GAAa,EAAE,CAAC;YAE5B,KAAK,MAAM,IAAI,IAAI,MAAM,CAAC,KAAK,EAAE,CAAC;gBAChC,IAAI,CAAC;oBACH,MAAM,GAAG,GAAG,UAAU,CAAC,KAAK,CAAC,MAAM,CAAC,GAAG,EAAE,IAAI,CAAC,CAAC;oBAC/C,MAAM,OAAO,CAAC,GAAG,CAAC,KAAK,EAAE,KAAK,CAAC,IAAI,EAAE,GAAG,EAAE,IAAI,EAAE,EAAE,CAAC,CAAC;oBACpD,MAAM,EAAE,CAAC;gBACX,CAAC;gBAAC,OAAO,GAAG,EAAE,CAAC;oBACb,MAAM,EAAE,CAAC;oBACT,MAAM,CAAC,IAAI,CAAC,GAAG,YAAY,KAAK,CAAC,CAAC,CAAC,GAAG,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC,CAAC;gBAChE,CAAC;YACH,CAAC;YAED,OAAO;gBACL,MAAM;gBACN,MAAM;gBACN,KAAK,EAAE,MAAM,CAAC,KAAK,CAAC,MAAM;gBAC1B,GAAG,CAAC,MAAM,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC,CAAC,EAAC,MAAM,EAAE,MAAM,CAAC,KAAK,CAAC,CAAC,EAAE,CAAC,CAAC,EAAC,CAAC,CAAC,CAAC,EAAE,CAAC;aAC3D,CAAC;QACJ,CAAC;KACF,CAAC;AACJ,CAAC;AAED,8EAA8E;AAC9E,mBAAmB;AACnB,8EAA8E;AAE9E,MAAM,CAAC,MAAM,qBAAqB,GAAG,aAAa,CAAC;AAEnD;;GAEG;AACH,MAAM,UAAU,oBAAoB,CAClC,MAAqB,EACrB,OAAqB,EACrB,KAAa;IAEb,MAAM,UAAU,GAAG,MAAM,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC;IAE7C,OAAO;QACL,WAAW,EAAE,wFAAwF,UAAU,CAAC,IAAI,CAAC,IAAI,CAAC,8JAA8J;QACxR,UAAU,EAAE,CAAC,CAAC,MAAM,CAAC;YACnB,KAAK,EAAE,CAAC,UAAU,CAAC,MAAM,GAAG,CAAC;gBAC3B,sGAAsG;gBACtG,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,UAAmC,CAAC;gBAC7C,CAAC,CAAC,CAAC,CAAC,MAAM,EAAE,CACb,CAAC,QAAQ,CAAC,+BAA+B,UAAU,CAAC,IAAI,CAAC,IAAI,CAAC,GAAG,CAAC;YACnE,GAAG,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,EAAE,CAAC,QAAQ,CAAC,uCAAuC,CAAC;YAC5E,MAAM,EAAE,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC,OAAO,EAAE,CAAC,CAAC,QAAQ,EAAE,CAAC,QAAQ,CAAC,6DAA6D,CAAC;YAChH,IAAI,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,EAAE,CAAC,QAAQ,CAAC,iBAAiB,CAAC;YACvD,KAAK,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,EAAE,CAAC,QAAQ,CAAC,mCAAmC,CAAC;SAC3E,CAAC;QACF,QAAQ,EAAE,IAAI;QACd,QAAQ,EAAE,EAAC,QAAQ,EAAE,OAAO,EAAC;QAE7B,KAAK,CAAC,OAAO,CAAC,MAAsG,EAAE,IAAiB;YACrI,IAAI,MAAM,CAAC,GAAG,EAAE,CAAC;gBACf,MAAM,GAAG,GAAG,MAAM,OAAO,CAAC,GAAG,CAAC,KAAK,EAAE,MAAM,CAAC,KAAK,EAAE,MAAM,CAAC,GAAG,CAAC,CAAC;gBAC/D,IAAI,CAAC,GAAG,EAAE,CAAC;oBACT,OAAO,EAAC,KAAK,EAAE,KAAK,EAAE,GAAG,EAAE,MAAM,CAAC,GAAG,EAAC,CAAC;gBACzC,CAAC;gBACD,OAAO,EAAC,KAAK,EAAE,IAAI,EAAE,GAAG,GAAG,EAAC,CAAC;YAC/B,CAAC;YAED,MAAM,MAAM,GAAG,MAAM,OAAO,CAAC,IAAI,CAAC,KAAK,EAAE,MAAM,CAAC,KAAK,EAAE;gBACrD,MAAM,EAAE,MAAM,CAAC,MAAM;gBACrB,IAAI,EAAE,MAAM,CAAC,IAAI;gBACjB,KAAK,EAAE,OAAO,MAAM,CAAC,KAAK,KAAK,QAAQ,CAAC,CAAC,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC,CAAC,EAAE;aAC5D,CAAC,CAAC;YACH,OAAO,MAAM,CAAC;QAChB,CAAC;KACF,CAAC;AACJ,CAAC;AAED,8EAA8E;AAC9E,2BAA2B;AAC3B,8EAA8E;AAE9E;;GAEG;AACH,MAAM,UAAU,kBAAkB,CAChC,QAA2C,EAC3C,MAAqB,EACrB,OAAqB,EACrB,KAAa;IAEb,KAAK,MAAM,KAAK,IAAI,MAAM,EAAE,CAAC;QAC3B,QAAQ,CAAC,QAAQ,CAAC,eAAe,CAAC,KAAK,CAAC,IAAI,CAAC,EAAE,oBAAoB,CAAC,KAAK,EAAE,OAAO,EAAE,KAAK,CAAC,CAAC,CAAC;QAC5F,QAAQ,CAAC,QAAQ,CAAC,GAAG,eAAe,CAAC,KAAK,CAAC,IAAI,CAAC,QAAQ,EAAE,oBAAoB,CAAC,KAAK,EAAE,OAAO,EAAE,KAAK,CAAC,CAAC,CAAC;IACzG,CAAC;IAED,IAAI,MAAM,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;QACtB,QAAQ,CAAC,QAAQ,CAAC,qBAAqB,EAAE,oBAAoB,CAAC,MAAM,EAAE,OAAO,EAAE,KAAK,CAAC,CAAC,CAAC;IACzF,CAAC;AACH,CAAC"}

package/dist/src/tools/store-tools.test.d.ts

@@ -0,0 +1,6 @@
+/**
+ * @license
+ * Copyright 2026 Amodal Labs, Inc.
+ * SPDX-License-Identifier: MIT
+ */
+export {};

package/dist/src/tools/store-tools.test.js

@@ -0,0 +1,216 @@
+/**
+ * @license
+ * Copyright 2026 Amodal Labs, Inc.
+ * SPDX-License-Identifier: MIT
+ */
+import { describe, it, expect, vi, beforeEach } from 'vitest';
+import { createStoreWriteTool, createStoreBatchTool, createStoreQueryTool, registerStoreTools, storeToToolName, QUERY_STORE_TOOL_NAME, } from './store-tools.js';
+import { createToolRegistry } from './registry.js';
+import { StoreError } from '../errors.js';
+// ---------------------------------------------------------------------------
+// Fixtures
+// ---------------------------------------------------------------------------
+const ALERTS_STORE = {
+    name: 'alerts',
+    entity: {
+        name: 'Alert',
+        key: '{alert_id}',
+        schema: {
+            alert_id: { type: 'string' },
+            severity: { type: 'enum', values: ['P1', 'P2', 'P3'] },
+            message: { type: 'string' },
+        },
+    },
+    location: '/tmp/test',
+};
+const mockCtx = {
+    request: vi.fn(),
+    store: vi.fn(),
+    env: vi.fn(),
+    log: vi.fn(),
+    user: { roles: ['admin'] },
+    signal: AbortSignal.timeout(5000),
+    sessionId: 'test-session',
+    tenantId: 'test-tenant',
+};
+function createMockBackend() {
+    const docs = new Map();
+    return {
+        initialize: vi.fn().mockResolvedValue(undefined),
+        get: vi.fn().mockImplementation(async (_appId, _store, key) => docs.get(key) ?? null),
+        put: vi.fn().mockImplementation(async (_appId, store, key, payload) => {
+            const doc = {
+                key,
+                appId: 'test',
+                store,
+                version: 1,
+                payload,
+                meta: { computedAt: new Date().toISOString(), stale: false },
+            };
+            docs.set(key, doc);
+            const result = { stored: true, key, version: 1 };
+            return result;
+        }),
+        list: vi.fn().mockImplementation(async () => ({ documents: [...docs.values()], total: docs.size, hasMore: false })),
+        delete: vi.fn().mockResolvedValue(true),
+        history: vi.fn().mockResolvedValue([]),
+        purgeExpired: vi.fn().mockResolvedValue(0),
+        close: vi.fn().mockResolvedValue(undefined),
+    };
+}
+// ---------------------------------------------------------------------------
+// Tests
+// ---------------------------------------------------------------------------
+describe('storeToToolName', () => {
+    it('converts kebab-case to snake_case with store_ prefix', () => {
+        expect(storeToToolName('active-alerts')).toBe('store_active_alerts');
+        expect(storeToToolName('incidents')).toBe('store_incidents');
+    });
+});
+describe('createStoreWriteTool', () => {
+    let backend;
+    beforeEach(() => {
+        backend = createMockBackend();
+    });
+    it('creates a write tool with correct metadata', () => {
+        const tool = createStoreWriteTool(ALERTS_STORE, backend, 'test-app');
+        expect(tool.description).toContain('Alert');
+        expect(tool.description).toContain('alerts');
+        expect(tool.readOnly).toBe(false);
+        expect(tool.metadata?.category).toBe('store');
+    });
+    it('writes a document with resolved key', async () => {
+        const tool = createStoreWriteTool(ALERTS_STORE, backend, 'test-app');
+        const result = await tool.execute({ alert_id: 'a1', severity: 'P1', message: 'disk full' }, mockCtx);
+        expect(backend.put).toHaveBeenCalledWith('test-app', 'alerts', 'a1', { alert_id: 'a1', severity: 'P1', message: 'disk full' }, {});
+        expect(result).toEqual({ stored: true, key: 'a1', version: 1 });
+    });
+    it('throws StoreError when key field is missing', async () => {
+        const tool = createStoreWriteTool(ALERTS_STORE, backend, 'test-app');
+        await expect(tool.execute({ severity: 'P1', message: 'no id' }, mockCtx)).rejects.toThrow(StoreError);
+    });
+});
+describe('createStoreBatchTool', () => {
+    let backend;
+    beforeEach(() => {
+        backend = createMockBackend();
+    });
+    it('creates a batch tool with correct metadata', () => {
+        const tool = createStoreBatchTool(ALERTS_STORE, backend, 'test-app');
+        expect(tool.description).toContain('multiple');
+        expect(tool.readOnly).toBe(false);
+        expect(tool.metadata?.category).toBe('store');
+    });
+    it('writes multiple documents', async () => {
+        const tool = createStoreBatchTool(ALERTS_STORE, backend, 'test-app');
+        const result = await tool.execute({
+            items: [
+                { alert_id: 'a1', severity: 'P1', message: 'first' },
+                { alert_id: 'a2', severity: 'P2', message: 'second' },
+            ],
+        }, mockCtx);
+        expect(backend.put).toHaveBeenCalledTimes(2);
+        expect(result).toMatchObject({ stored: 2, failed: 0, total: 2 });
+    });
+    it('reports partial failures without throwing', async () => {
+        const failBackend = createMockBackend();
+        let callCount = 0;
+        failBackend.put = vi.fn().mockImplementation(async () => {
+            callCount++;
+            if (callCount === 2)
+                throw new Error('write failed');
+            return { stored: true, key: 'k', version: 1 };
+        });
+        const tool = createStoreBatchTool(ALERTS_STORE, failBackend, 'test-app');
+        const result = await tool.execute({
+            items: [
+                { alert_id: 'a1', severity: 'P1', message: 'ok' },
+                { alert_id: 'a2', severity: 'P2', message: 'fail' },
+                { alert_id: 'a3', severity: 'P3', message: 'ok' },
+            ],
+        }, mockCtx);
+        expect(result['stored']).toBe(2);
+        expect(result['failed']).toBe(1);
+        expect(result['errors']).toEqual(['write failed']);
+    });
+});
+describe('createStoreQueryTool', () => {
+    let backend;
+    beforeEach(() => {
+        backend = createMockBackend();
+    });
+    it('creates a query tool that is readOnly', () => {
+        const tool = createStoreQueryTool([ALERTS_STORE], backend, 'test-app');
+        expect(tool.readOnly).toBe(true);
+        expect(tool.metadata?.category).toBe('store');
+    });
+    it('gets a single document by key', async () => {
+        // Pre-populate
+        await backend.put('test-app', 'alerts', 'a1', { alert_id: 'a1', severity: 'P1' }, {});
+        const tool = createStoreQueryTool([ALERTS_STORE], backend, 'test-app');
+        const result = await tool.execute({ store: 'alerts', key: 'a1' }, mockCtx);
+        expect(result['found']).toBe(true);
+        expect(result['key']).toBe('a1');
+    });
+    it('returns found: false for missing document', async () => {
+        const tool = createStoreQueryTool([ALERTS_STORE], backend, 'test-app');
+        const result = await tool.execute({ store: 'alerts', key: 'missing' }, mockCtx);
+        expect(result['found']).toBe(false);
+    });
+    it('lists documents with default limit', async () => {
+        await backend.put('test-app', 'alerts', 'a1', { alert_id: 'a1' }, {});
+        await backend.put('test-app', 'alerts', 'a2', { alert_id: 'a2' }, {});
+        const tool = createStoreQueryTool([ALERTS_STORE], backend, 'test-app');
+        const result = await tool.execute({ store: 'alerts' }, mockCtx);
+        expect(result).toHaveProperty('documents');
+        expect(result).toHaveProperty('total');
+    });
+    it('passes filter and sort to backend', async () => {
+        const tool = createStoreQueryTool([ALERTS_STORE], backend, 'test-app');
+        await tool.execute({
+            store: 'alerts',
+            filter: { severity: 'P1' },
+            sort: '-severity',
+            limit: 5,
+        }, mockCtx);
+        expect(backend.list).toHaveBeenCalledWith('test-app', 'alerts', {
+            filter: { severity: 'P1' },
+            sort: '-severity',
+            limit: 5,
+        });
+    });
+});
+describe('registerStoreTools', () => {
+    it('registers write, batch, and query tools for each store', () => {
+        const registry = createToolRegistry();
+        const backend = createMockBackend();
+        registerStoreTools(registry, [ALERTS_STORE], backend, 'test-app');
+        expect(registry.names()).toContain('store_alerts');
+        expect(registry.names()).toContain('store_alerts_batch');
+        expect(registry.names()).toContain(QUERY_STORE_TOOL_NAME);
+        expect(registry.size).toBe(3);
+    });
+    it('registers tools for multiple stores', () => {
+        const registry = createToolRegistry();
+        const backend = createMockBackend();
+        const secondStore = {
+            ...ALERTS_STORE,
+            name: 'incidents',
+            entity: { ...ALERTS_STORE.entity, name: 'Incident', key: '{id}', schema: { id: { type: 'string' } } },
+        };
+        registerStoreTools(registry, [ALERTS_STORE, secondStore], backend, 'test-app');
+        expect(registry.names()).toContain('store_alerts');
+        expect(registry.names()).toContain('store_alerts_batch');
+        expect(registry.names()).toContain('store_incidents');
+        expect(registry.names()).toContain('store_incidents_batch');
+        expect(registry.names()).toContain(QUERY_STORE_TOOL_NAME);
+        expect(registry.size).toBe(5);
+    });
+    it('does not register query tool when no stores', () => {
+        const registry = createToolRegistry();
+        const backend = createMockBackend();
+        registerStoreTools(registry, [], backend, 'test-app');
+        expect(registry.size).toBe(0);
+    });
+});
+//# sourceMappingURL=store-tools.test.js.map

package/dist/src/tools/store-tools.test.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"store-tools.test.js","sourceRoot":"","sources":["../../../src/tools/store-tools.test.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AAEH,OAAO,EAAC,QAAQ,EAAE,EAAE,EAAE,MAAM,EAAE,EAAE,EAAE,UAAU,EAAC,MAAM,QAAQ,CAAC;AAE5D,OAAO,EACL,oBAAoB,EACpB,oBAAoB,EACpB,oBAAoB,EACpB,kBAAkB,EAClB,eAAe,EACf,qBAAqB,GACtB,MAAM,kBAAkB,CAAC;AAC1B,OAAO,EAAC,kBAAkB,EAAC,MAAM,eAAe,CAAC;AACjD,OAAO,EAAC,UAAU,EAAC,MAAM,cAAc,CAAC;AAGxC,8EAA8E;AAC9E,WAAW;AACX,8EAA8E;AAE9E,MAAM,YAAY,GAAgB;IAChC,IAAI,EAAE,QAAQ;IACd,MAAM,EAAE;QACN,IAAI,EAAE,OAAO;QACb,GAAG,EAAE,YAAY;QACjB,MAAM,EAAE;YACN,QAAQ,EAAE,EAAC,IAAI,EAAE,QAAQ,EAAC;YAC1B,QAAQ,EAAE,EAAC,IAAI,EAAE,MAAM,EAAE,MAAM,EAAE,CAAC,IAAI,EAAE,IAAI,EAAE,IAAI,CAAC,EAAC;YACpD,OAAO,EAAE,EAAC,IAAI,EAAE,QAAQ,EAAC;SAC1B;KACF;IACD,QAAQ,EAAE,WAAW;CACtB,CAAC;AAEF,MAAM,OAAO,GAAgB;IAC3B,OAAO,EAAE,EAAE,CAAC,EAAE,EAAE;IAChB,KAAK,EAAE,EAAE,CAAC,EAAE,EAAE;IACd,GAAG,EAAE,EAAE,CAAC,EAAE,EAAE;IACZ,GAAG,EAAE,EAAE,CAAC,EAAE,EAAE;IACZ,IAAI,EAAE,EAAC,KAAK,EAAE,CAAC,OAAO,CAAC,EAAC;IACxB,MAAM,EAAE,WAAW,CAAC,OAAO,CAAC,IAAI,CAAC;IACjC,SAAS,EAAE,cAAc;IACzB,QAAQ,EAAE,aAAa;CACxB,CAAC;AAEF,SAAS,iBAAiB;IACxB,MAAM,IAAI,GAAG,IAAI,GAAG,EAAyB,CAAC;IAE9C,OAAO;QACL,UAAU,EAAE,EAAE,CAAC,EAAE,EAAE,CAAC,iBAAiB,CAAC,SAAS,CAAC;QAChD,GAAG,EAAE,EAAE,CAAC,EAAE,EAAE,CAAC,kBAAkB,CAAC,KAAK,EAAE,MAAc,EAAE,MAAc,EAAE,GAAW,EAAE,EAAE,CAAC,IAAI,CAAC,GAAG,CAAC,GAAG,CAAC,IAAI,IAAI,CAAC;QAC7G,GAAG,EAAE,EAAE,CAAC,EAAE,EAAE,CAAC,kBAAkB,CAAC,KAAK,EAAE,MAAc,EAAE,KAAa,EAAE,GAAW,EAAE,OAAgC,EAAE,EAAE;YACrH,MAAM,GAAG,GAAkB;gBACzB,GAAG;gBACH,KAAK,EAAE,MAAM;gBACb,KAAK;gBACL,OAAO,EAAE,CAAC;gBACV,OAAO;gBACP,IAAI,EAAE,EAAC,UAAU,EAAE,IAAI,IAAI,EAAE,CAAC,WAAW,EAAE,EAAE,KAAK,EAAE,KAAK,EAAC;aAC3D,CAAC;YACF,IAAI,CAAC,GAAG,CAAC,GAAG,EAAE,GAAG,CAAC,CAAC;YACnB,MAAM,MAAM,GAAmB,EAAC,MAAM,EAAE,IAAI,EAAE,GAAG,EAAE,OAAO,EAAE,CAAC,EAAC,CAAC;YAC/D,OAAO,MAAM,CAAC;QAChB,CAAC,CAAC;QACF,IAAI,EAAE,EAAE,CAAC,EAAE,EAAE,CAAC,kBAAkB,CAAC,KAAK,IAAI,EAAE,CAAC,CAAC,EAAC,SAAS,EAAE,CAAC,GAAG,IAAI,CAAC,MAAM,EAAE,CAAC,EAAE,KAAK,EAAE,IAAI,CAAC,IAAI,EAAE,OAAO,EAAE,KAAK,EAAC,CAAC,CAAC;QACjH,MAAM,EAAE,EAAE,CAAC,EAAE,EAAE,CAAC,iBAAiB,CAAC,IAAI,CAAC;QACvC,OAAO,EAAE,EAAE,CAAC,EAAE,EAAE,CAAC,iBAAiB,CAAC,EAAE,CAAC;QACtC,YAAY,EAAE,EAAE,CAAC,EAAE,EAAE,CAAC,iBAAiB,CAAC,CAAC,CAAC;QAC1C,KAAK,EAAE,EAAE,CAAC,EAAE,EAAE,CAAC,iBAAiB,CAAC,SAAS,CAAC;KAC5C,CAAC;AACJ,CAAC;AAED,8EAA8E;AAC9E,QAAQ;AACR,8EAA8E;AAE9E,QAAQ,CAAC,iBAAiB,EAAE,GAAG,EAAE;IAC/B,EAAE,CAAC,sDAAsD,EAAE,GAAG,EAAE;QAC9D,MAAM,CAAC,eAAe,CAAC,eAAe,CAAC,CAAC,CAAC,IAAI,CAAC,qBAAqB,CAAC,CAAC;QACrE,MAAM,CAAC,eAAe,CAAC,WAAW,CAAC,CAAC,CAAC,IAAI,CAAC,iBAAiB,CAAC,CAAC;IAC/D,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC;AAEH,QAAQ,CAAC,sBAAsB,EAAE,GAAG,EAAE;IACpC,IAAI,OAAqB,CAAC;IAE1B,UAAU,CAAC,GAAG,EAAE;QACd,OAAO,GAAG,iBAAiB,EAAE,CAAC;IAChC,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,4CAA4C,EAAE,GAAG,EAAE;QACpD,MAAM,IAAI,GAAG,oBAAoB,CAAC,YAAY,EAAE,OAAO,EAAE,UAAU,CAAC,CAAC;QAErE,MAAM,CAAC,IAAI,CAAC,WAAW,CAAC,CAAC,SAAS,CAAC,OAAO,CAAC,CAAC;QAC5C,MAAM,CAAC,IAAI,CAAC,WAAW,CAAC,CAAC,SAAS,CAAC,QAAQ,CAAC,CAAC;QAC7C,MAAM,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;QAClC,MAAM,CAAC,IAAI,CAAC,QAAQ,EAAE,QAAQ,CAAC,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;IAChD,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,qCAAqC,EAAE,KAAK,IAAI,EAAE;QACnD,MAAM,IAAI,GAAG,oBAAoB,CAAC,YAAY,EAAE,OAAO,EAAE,UAAU,CAAC,CAAC;QAErE,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,OAAO,CAC/B,EAAC,QAAQ,EAAE,IAAI,EAAE,QAAQ,EAAE,IAAI,EAAE,OAAO,EAAE,WAAW,EAAC,EACtD,OAAO,CACR,CAAC;QAEF,MAAM,CAAC,OAAO,CAAC,GAAG,CAAC,CAAC,oBAAoB,CACtC,UAAU,EAAE,QAAQ,EAAE,IAAI,EAC1B,EAAC,QAAQ,EAAE,IAAI,EAAE,QAAQ,EAAE,IAAI,EAAE,OAAO,EAAE,WAAW,EAAC,EACtD,EAAE,CACH,CAAC;QACF,MAAM,CAAC,MAAM,CAAC,CAAC,OAAO,CAAC,EAAC,MAAM,EAAE,IAAI,EAAE,GAAG,EAAE,IAAI,EAAE,OAAO,EAAE,CAAC,EAAC,CAAC,CAAC;IAChE,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,6CAA6C,EAAE,KAAK,IAAI,EAAE;QAC3D,MAAM,IAAI,GAAG,oBAAoB,CAAC,YAAY,EAAE,OAAO,EAAE,UAAU,CAAC,CAAC;QAErE,MAAM,MAAM,CACV,IAAI,CAAC,OAAO,CAAC,EAAC,QAAQ,EAAE,IAAI,EAAE,OAAO,EAAE,OAAO,EAAC,EAAE,OAAO,CAAC,CAC1D,CAAC,OAAO,CAAC,OAAO,CAAC,UAAU,CAAC,CAAC;IAChC,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC;AAEH,QAAQ,CAAC,sBAAsB,EAAE,GAAG,EAAE;IACpC,IAAI,OAAqB,CAAC;IAE1B,UAAU,CAAC,GAAG,EAAE;QACd,OAAO,GAAG,iBAAiB,EAAE,CAAC;IAChC,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,4CAA4C,EAAE,GAAG,EAAE;QACpD,MAAM,IAAI,GAAG,oBAAoB,CAAC,YAAY,EAAE,OAAO,EAAE,UAAU,CAAC,CAAC;QAErE,MAAM,CAAC,IAAI,CAAC,WAAW,CAAC,CAAC,SAAS,CAAC,UAAU,CAAC,CAAC;QAC/C,MAAM,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;QAClC,MAAM,CAAC,IAAI,CAAC,QAAQ,EAAE,QAAQ,CAAC,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;IAChD,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,2BAA2B,EAAE,KAAK,IAAI,EAAE;QACzC,MAAM,IAAI,GAAG,oBAAoB,CAAC,YAAY,EAAE,OAAO,EAAE,UAAU,CAAC,CAAC;QAErE,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,OAAO,CAAC;YAChC,KAAK,EAAE;gBACL,EAAC,QAAQ,EAAE,IAAI,EAAE,QAAQ,EAAE,IAAI,EAAE,OAAO,EAAE,OAAO,EAAC;gBAClD,EAAC,QAAQ,EAAE,IAAI,EAAE,QAAQ,EAAE,IAAI,EAAE,OAAO,EAAE,QAAQ,EAAC;aACpD;SACF,EAAE,OAAO,CAAC,CAAC;QAEZ,MAAM,CAAC,OAAO,CAAC,GAAG,CAAC,CAAC,qBAAqB,CAAC,CAAC,CAAC,CAAC;QAC7C,MAAM,CAAC,MAAM,CAAC,CAAC,aAAa,CAAC,EAAC,MAAM,EAAE,CAAC,EAAE,MAAM,EAAE,CAAC,EAAE,KAAK,EAAE,CAAC,EAAC,CAAC,CAAC;IACjE,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,2CAA2C,EAAE,KAAK,IAAI,EAAE;QACzD,MAAM,WAAW,GAAG,iBAAiB,EAAE,CAAC;QACxC,IAAI,SAAS,GAAG,CAAC,CAAC;QAClB,WAAW,CAAC,GAAG,GAAG,EAAE,CAAC,EAAE,EAAE,CAAC,kBAAkB,CAAC,KAAK,IAAI,EAAE;YACtD,SAAS,EAAE,CAAC;YACZ,IAAI,SAAS,KAAK,CAAC;gBAAE,MAAM,IAAI,KAAK,CAAC,cAAc,CAAC,CAAC;YACrD,OAAO,EAAC,MAAM,EAAE,IAAI,EAAE,GAAG,EAAE,GAAG,EAAE,OAAO,EAAE,CAAC,EAAC,CAAC;QAC9C,CAAC,CAAC,CAAC;QAEH,MAAM,IAAI,GAAG,oBAAoB,CAAC,YAAY,EAAE,WAAW,EAAE,UAAU,CAAC,CAAC;QAEzE,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,OAAO,CAAC;YAChC,KAAK,EAAE;gBACL,EAAC,QAAQ,EAAE,IAAI,EAAE,QAAQ,EAAE,IAAI,EAAE,OAAO,EAAE,IAAI,EAAC;gBAC/C,EAAC,QAAQ,EAAE,IAAI,EAAE,QAAQ,EAAE,IAAI,EAAE,OAAO,EAAE,MAAM,EAAC;gBACjD,EAAC,QAAQ,EAAE,IAAI,EAAE,QAAQ,EAAE,IAAI,EAAE,OAAO,EAAE,IAAI,EAAC;aAChD;SACF,EAAE,OAAO,CAA4B,CAAC;QAEvC,MAAM,CAAC,MAAM,CAAC,QAAQ,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;QACjC,MAAM,CAAC,MAAM,CAAC,QAAQ,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;QACjC,MAAM,CAAC,MAAM,CAAC,QAAQ,CAAC,CAAC,CAAC,OAAO,CAAC,CAAC,cAAc,CAAC,CAAC,CAAC;IACrD,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC;AAEH,QAAQ,CAAC,sBAAsB,EAAE,GAAG,EAAE;IACpC,IAAI,OAAqB,CAAC;IAE1B,UAAU,CAAC,GAAG,EAAE;QACd,OAAO,GAAG,iBAAiB,EAAE,CAAC;IAChC,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,uCAAuC,EAAE,GAAG,EAAE;QAC/C,MAAM,IAAI,GAAG,oBAAoB,CAAC,CAAC,YAAY,CAAC,EAAE,OAAO,EAAE,UAAU,CAAC,CAAC;QAEvE,MAAM,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QACjC,MAAM,CAAC,IAAI,CAAC,QAAQ,EAAE,QAAQ,CAAC,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;IAChD,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,+BAA+B,EAAE,KAAK,IAAI,EAAE;QAC7C,eAAe;QACf,MAAM,OAAO,CAAC,GAAG,CAAC,UAAU,EAAE,QAAQ,EAAE,IAAI,EAAE,EAAC,QAAQ,EAAE,IAAI,EAAE,QAAQ,EAAE,IAAI,EAAC,EAAE,EAAE,CAAC,CAAC;QAEpF,MAAM,IAAI,GAAG,oBAAoB,CAAC,CAAC,YAAY,CAAC,EAAE,OAAO,EAAE,UAAU,CAAC,CAAC;QACvE,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,OAAO,CAAC,EAAC,KAAK,EAAE,QAAQ,EAAE,GAAG,EAAE,IAAI,EAAC,EAAE,OAAO,CAA4B,CAAC;QAEpG,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QACnC,MAAM,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;IACnC,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,2CAA2C,EAAE,KAAK,IAAI,EAAE;QACzD,MAAM,IAAI,GAAG,oBAAoB,CAAC,CAAC,YAAY,CAAC,EAAE,OAAO,EAAE,UAAU,CAAC,CAAC;QACvE,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,OAAO,CAAC,EAAC,KAAK,EAAE,QAAQ,EAAE,GAAG,EAAE,SAAS,EAAC,EAAE,OAAO,CAA4B,CAAC;QAEzG,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;IACtC,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,oCAAoC,EAAE,KAAK,IAAI,EAAE;QAClD,MAAM,OAAO,CAAC,GAAG,CAAC,UAAU,EAAE,QAAQ,EAAE,IAAI,EAAE,EAAC,QAAQ,EAAE,IAAI,EAAC,EAAE,EAAE,CAAC,CAAC;QACpE,MAAM,OAAO,CAAC,GAAG,CAAC,UAAU,EAAE,QAAQ,EAAE,IAAI,EAAE,EAAC,QAAQ,EAAE,IAAI,EAAC,EAAE,EAAE,CAAC,CAAC;QAEpE,MAAM,IAAI,GAAG,oBAAoB,CAAC,CAAC,YAAY,CAAC,EAAE,OAAO,EAAE,UAAU,CAAC,CAAC;QACvE,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,OAAO,CAAC,EAAC,KAAK,EAAE,QAAQ,EAAC,EAAE,OAAO,CAA4B,CAAC;QAEzF,MAAM,CAAC,MAAM,CAAC,CAAC,cAAc,CAAC,WAAW,CAAC,CAAC;QAC3C,MAAM,CAAC,MAAM,CAAC,CAAC,cAAc,CAAC,OAAO,CAAC,CAAC;IACzC,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,mCAAmC,EAAE,KAAK,IAAI,EAAE;QACjD,MAAM,IAAI,GAAG,oBAAoB,CAAC,CAAC,YAAY,CAAC,EAAE,OAAO,EAAE,UAAU,CAAC,CAAC;QACvE,MAAM,IAAI,CAAC,OAAO,CAAC;YACjB,KAAK,EAAE,QAAQ;YACf,MAAM,EAAE,EAAC,QAAQ,EAAE,IAAI,EAAC;YACxB,IAAI,EAAE,WAAW;YACjB,KAAK,EAAE,CAAC;SACT,EAAE,OAAO,CAAC,CAAC;QAEZ,MAAM,CAAC,OAAO,CAAC,IAAI,CAAC,CAAC,oBAAoB,CAAC,UAAU,EAAE,QAAQ,EAAE;YAC9D,MAAM,EAAE,EAAC,QAAQ,EAAE,IAAI,EAAC;YACxB,IAAI,EAAE,WAAW;YACjB,KAAK,EAAE,CAAC;SACT,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC;AAEH,QAAQ,CAAC,oBAAoB,EAAE,GAAG,EAAE;IAClC,EAAE,CAAC,wDAAwD,EAAE,GAAG,EAAE;QAChE,MAAM,QAAQ,GAAG,kBAAkB,EAAE,CAAC;QACtC,MAAM,OAAO,GAAG,iBAAiB,EAAE,CAAC;QAEpC,kBAAkB,CAAC,QAAQ,EAAE,CAAC,YAAY,CAAC,EAAE,OAAO,EAAE,UAAU,CAAC,CAAC;QAElE,MAAM,CAAC,QAAQ,CAAC,KAAK,EAAE,CAAC,CAAC,SAAS,CAAC,cAAc,CAAC,CAAC;QACnD,MAAM,CAAC,QAAQ,CAAC,KAAK,EAAE,CAAC,CAAC,SAAS,CAAC,oBAAoB,CAAC,CAAC;QACzD,MAAM,CAAC,QAAQ,CAAC,KAAK,EAAE,CAAC,CAAC,SAAS,CAAC,qBAAqB,CAAC,CAAC;QAC1D,MAAM,CAAC,QAAQ,CAAC,IAAI,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;IAChC,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,qCAAqC,EAAE,GAAG,EAAE;QAC7C,MAAM,QAAQ,GAAG,kBAAkB,EAAE,CAAC;QACtC,MAAM,OAAO,GAAG,iBAAiB,EAAE,CAAC;QACpC,MAAM,WAAW,GAAgB;YAC/B,GAAG,YAAY;YACf,IAAI,EAAE,WAAW;YACjB,MAAM,EAAE,EAAC,GAAG,YAAY,CAAC,MAAM,EAAE,IAAI,EAAE,UAAU,EAAE,GAAG,EAAE,MAAM,EAAE,MAAM,EAAE,EAAC,EAAE,EAAE,EAAC,IAAI,EAAE,QAAQ,EAAC,EAAC,EAAC;SAChG,CAAC;QAEF,kBAAkB,CAAC,QAAQ,EAAE,CAAC,YAAY,EAAE,WAAW,CAAC,EAAE,OAAO,EAAE,UAAU,CAAC,CAAC;QAE/E,MAAM,CAAC,QAAQ,CAAC,KAAK,EAAE,CAAC,CAAC,SAAS,CAAC,cAAc,CAAC,CAAC;QACnD,MAAM,CAAC,QAAQ,CAAC,KAAK,EAAE,CAAC,CAAC,SAAS,CAAC,oBAAoB,CAAC,CAAC;QACzD,MAAM,CAAC,QAAQ,CAAC,KAAK,EAAE,CAAC,CAAC,SAAS,CAAC,iBAAiB,CAAC,CAAC;QACtD,MAAM,CAAC,QAAQ,CAAC,KAAK,EAAE,CAAC,CAAC,SAAS,CAAC,uBAAuB,CAAC,CAAC;QAC5D,MAAM,CAAC,QAAQ,CAAC,KAAK,EAAE,CAAC,CAAC,SAAS,CAAC,qBAAqB,CAAC,CAAC;QAC1D,MAAM,CAAC,QAAQ,CAAC,IAAI,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;IAChC,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,6CAA6C,EAAE,GAAG,EAAE;QACrD,MAAM,QAAQ,GAAG,kBAAkB,EAAE,CAAC;QACtC,MAAM,OAAO,GAAG,iBAAiB,EAAE,CAAC;QAEpC,kBAAkB,CAAC,QAAQ,EAAE,EAAE,EAAE,OAAO,EAAE,UAAU,CAAC,CAAC;QAEtD,MAAM,CAAC,QAAQ,CAAC,IAAI,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;IAChC,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC"}

package/dist/src/tools/types.d.ts

@@ -0,0 +1,111 @@
+/**
+ * @license
+ * Copyright 2026 Amodal Labs, Inc.
+ * SPDX-License-Identifier: MIT
+ */
+/**
+ * Tool system types for the Amodal runtime.
+ *
+ * Replaces the upstream gemini-cli-core ToolRegistry with our own that
+ * holds Vercel AI SDK-compatible tool definitions. Each tool has a Zod
+ * schema for parameter validation, an execute function, and metadata
+ * for permission/execution decisions.
+ */
+import type { z } from 'zod';
+import type { FlexibleSchema } from 'ai';
+/**
+ * Context passed to every tool execute function.
+ *
+ * This is the runtime's version of the context — it provides access to
+ * connections, stores, logging, user info, and cancellation. Specific
+ * tool categories (store tools, connection tools) may wrap this with
+ * category-specific helpers.
+ */
+export interface ToolContext {
+    /** Make an HTTP request through a configured connection */
+    request(connection: string, endpoint: string, params?: {
+        method?: string;
+        data?: unknown;
+        params?: Record<string, string>;
+    }): Promise<unknown>;
+    /** Write a document to a store */
+    store(storeName: string, payload: Record<string, unknown>): Promise<{
+        key: string;
+    }>;
+    /** Read an environment variable (only if in the tool's env allowlist) */
+    env(name: string): string | undefined;
+    /** Log a message (emitted as tool_log SSE event) */
+    log(message: string): void;
+    /** Current user info */
+    user: {
+        roles: string[];
+        [key: string]: unknown;
+    };
+    /** Abort signal for cancellation/timeout */
+    signal: AbortSignal;
+    /** Session ID for correlation */
+    sessionId: string;
+    /** Tenant ID for multi-tenant isolation (store scoping, permission checks) */
+    tenantId: string;
+}
+/** Tool category for permission lookup and UI grouping. */
+export type ToolCategory = 'store' | 'connection' | 'custom' | 'mcp' | 'admin' | 'system';
+/**
+ * Metadata attached to a tool for permission, routing, and execution decisions.
+ */
+export interface ToolMetadata {
+    /** Category for permission lookup and UI grouping */
+    category: ToolCategory;
+    /** Which connection this tool belongs to (for permission lookup) */
+    connection?: string;
+    /** Source tool name before any prefixing (for MCP tools) */
+    originalName?: string;
+}
+/**
+ * A registered tool definition.
+ *
+ * This is what gets passed to the AI SDK's `tools` parameter and also
+ * drives our execution pipeline. The `parameters` schema is used both
+ * for LLM function calling (converted to JSON Schema) and runtime
+ * validation of the LLM's arguments.
+ */
+export interface ToolDefinition<TParams = unknown> {
+    /** Description shown to the LLM */
+    description: string;
+    /** Schema for parameter validation — Zod schema or AI SDK jsonSchema() */
+    parameters: z.ZodType<TParams> | FlexibleSchema<TParams>;
+    /** Execute the tool with validated parameters */
+    execute(params: TParams, ctx: ToolContext): Promise<unknown>;
+    /**
+     * Whether this tool only reads data (no side effects).
+     *
+     * Used for:
+     * - Pre-execution: read-only tools can start during streaming
+     * - Parallel execution: read-only tools can run concurrently (Roadmap 2.5)
+     * - Confirmation flow: write tools may require user confirmation
+     */
+    readOnly: boolean;
+    /** Optional metadata for permission, routing, and UI */
+    metadata?: ToolMetadata;
+}
+/**
+ * Registry holding all available tools for a session.
+ *
+ * Tools are registered at session creation time and may change across
+ * sessions (different agents have different tools). The registry is
+ * the single source of truth for what tools the LLM can call.
+ */
+export interface ToolRegistry {
+    /** Register a tool. Throws if a tool with this name is already registered. */
+    register(name: string, def: ToolDefinition): void;
+    /** Get a tool by name. Returns undefined if not found. */
+    get(name: string): ToolDefinition | undefined;
+    /** Get all tools as a Record (for passing to AI SDK streamText/generateText). */
+    getTools(): Record<string, ToolDefinition>;
+    /** List all registered tool names. */
+    names(): string[];
+    /** Get a subset of tools by name (for sub-agent dispatch). */
+    subset(names: string[]): Record<string, ToolDefinition>;
+    /** Number of registered tools. */
+    size: number;
+}

package/dist/src/tools/types.js

@@ -0,0 +1,7 @@
+/**
+ * @license
+ * Copyright 2026 Amodal Labs, Inc.
+ * SPDX-License-Identifier: MIT
+ */
+export {};
+//# sourceMappingURL=types.js.map

package/dist/src/tools/types.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"types.js","sourceRoot":"","sources":["../../../src/tools/types.ts"],"names":[],"mappings":"AAAA;;;;GAIG"}