1 files changed, 35 insertions, 0 deletions
diff --git a/packages/core/src/tools/registry.ts b/packages/core/src/tools/registry.ts
new file mode 100644
index 0000000..4699c93
--- /dev/null
+++ b/packages/core/src/tools/registry.ts
@@ -0,0 +1,35 @@
+import { tool } from "ai";
+import { z } from "zod";
+import type { ToolDefinition } from "../types/index.js";
+
+export function createToolRegistry(tools: ToolDefinition[]) {
+	const toolMap = new Map<string, ToolDefinition>(tools.map((t) => [t.name, t]));
+
+	return {
+		getTools(): ToolDefinition[] {
+			return [...toolMap.values()];
+		},
+
+		getTool(name: string): ToolDefinition | undefined {
+			return toolMap.get(name);
+		},
+
+		getAISDKTools() {
+			const result: Record<string, ReturnType<typeof tool>> = {};
+			for (const [name, def] of toolMap) {
+				const schema = def.parameters;
+				const t = tool({
+					description: def.description,
+					parameters: schema instanceof z.ZodObject ? schema : z.object({}),
+					execute: async (args) => {
+						return def.execute(args as Record<string, unknown>);
+					},
+				});
+				// The AI SDK tool() overloads cause type narrowing issues when
+				// execute is provided. The runtime value is correct.
+				result[name] = t as unknown as ReturnType<typeof tool>;
+			}
+			return result;
+		},
+	};
+}