modality-kit 0.11.2 → 0.12.0

package/dist/types/schemas/schemas_symbol.d.ts

@@ -91,12 +91,12 @@ declare const fileEntrySchema: z.ZodObject<{
     type: z.ZodEnum<["file", "directory"]>;
     lastModified: z.ZodNullable<z.ZodOptional<z.ZodNumber>>;
 }, "strip", z.ZodTypeAny, {
-    type: "file" | "directory";
     path: string;
+    type: "file" | "directory";
     lastModified?: number | null | undefined;
 }, {
-    type: "file" | "directory";
     path: string;
+    type: "file" | "directory";
     lastModified?: number | null | undefined;
 }>;
 export type FileEntryType = z.infer<typeof fileEntrySchema>;

package/dist/types/schemas/schemas_tool_config.d.ts

@@ -1,19 +1,22 @@
 /**
  * TypeScript interfaces for tool configuration objects
  */
-export type AIToolExecutor = (...args: any[]) => Promise<string>;
+import { z } from "zod";
 /**
  * Tool interface for AI SDK compatibility
  */
-export interface AITool {
+export interface AITool<T extends z.ZodSchema = z.ZodSchema> {
     name?: string;
     annotations?: any;
     description: string;
-    inputSchema: any;
-    execute: AIToolExecutor;
+    inputSchema: T;
+    execute: (args: z.infer<T>) => Promise<string>;
 }
 /**
- * Type for a collection of AI tools
- * @template T - The key type for the tools record, defaults to string
+ * Type for a collection of AI tools with preserved schema types
+ * @template T - Record mapping tool names to their inputSchema types
+ * @example AITools<{getUserById: z.object({id: z.string()}), createUser: z.object({name: z.string()})}>
  */
-export type AITools<T extends string | number | symbol = string> = Record<T, AITool>;
+export type AITools<T extends Record<string, z.ZodSchema> = Record<string, z.ZodSchema>> = {
+    [K in keyof T]: AITool<T[K]>;
+};

package/dist/types/util_mcp_tools_converter.d.ts

@@ -1,9 +1,11 @@
 import type { FastMCP } from "fastmcp";
 import type { AITools } from "./schemas/schemas_tool_config.ts";
+import type { z } from "zod";
 /**
  * Setup function that optionally registers AITools with MCP server
- * @param aiTools - The AITools object to optionally register
+ * Automatically infers and preserves schema types from the input
+ * @param aiTools - The AITools object with schema mapping
  * @param mcpServer - Optional MCP server to register tools with
- * @returns The AITools object
+ * @returns The same AITools object with preserved types
  */
-export declare const setupAITools: (aiTools: AITools, mcpServer?: FastMCP) => AITools;
+export declare const setupAITools: <T extends Record<string, z.ZodSchema>>(aiTools: AITools<T>, mcpServer?: FastMCP) => AITools<T>;

package/package.json

@@ -1,5 +1,5 @@
 {
-  "version": "0.11.2",
+  "version": "0.12.0",
   "name": "modality-kit",
   "repository": {
     "type": "git",
@@ -24,7 +24,7 @@
   "scripts": {
     "update-compile-sh": "yo reshow:compile-sh",
     "build:clean": "find ./dist -name '*.*' | xargs rm -rf",
-    "build:types": "bunx tsc -p ./",
+    "build:types": "bun tsc -p ./",
     "build:src": "bun build src/index.ts --outdir dist",
     "build": "bun run build:clean && bun run build:src && bun run build:types",
     "dev": "bunx concurrently 'bunx --watch tsc  -p ./' 'bun build:src -- --watch --sourcemap=inline'",