modality-kit 0.11.1 → 0.11.3

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,18 +1,21 @@
 /**
  * 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
+ * Type for a collection of AI tools with preserved schema types
+ * Each tool maintains its specific inputSchema type relationship
  */
-export type AITools = Record<string, AITool>;
+export type AITools<T extends Record<string, z.ZodSchema> = Record<string, z.ZodSchema>> = {
+    [K in keyof T]: AITool<T[K]>;
+};

package/package.json

@@ -1,5 +1,5 @@
 {
-  "version": "0.11.1",
+  "version": "0.11.3",
   "name": "modality-kit",
   "repository": {
     "type": "git",