@ai-sdk/provider-utils 4.0.6 → 4.0.8

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@ai-sdk/provider-utils",
-  "version": "4.0.6",
+  "version": "4.0.8",
   "license": "Apache-2.0",
   "sideEffects": false,
   "main": "./dist/index.js",
@@ -31,7 +31,7 @@
   "dependencies": {
     "@standard-schema/spec": "^1.1.0",
     "eventsource-parser": "^3.0.6",
-    "@ai-sdk/provider": "3.0.3"
+    "@ai-sdk/provider": "3.0.4"
   },
   "devDependencies": {
     "@types/node": "20.17.24",

package/src/types/tool.test-d.ts

@@ -31,6 +31,41 @@ describe('tool type', () => {
       expectTypeOf(aTool.execute).not.toEqualTypeOf<Function>();
       expectTypeOf(aTool.inputSchema).toEqualTypeOf<FlexibleSchema<T>>();
     });
+
+    it('should infer input type correctly when inputExamples are present with optional/default zod schema', () => {
+      const inputSchema = z.object({
+        location: z.string(),
+        unit: z.enum(['celsius', 'fahrenheit']).optional().default('celsius'),
+      });
+
+      tool({
+        description: 'Get the weather for a location',
+        inputSchema,
+        inputExamples: [
+          { input: { location: 'San Francisco', unit: 'celsius' } },
+        ],
+        execute: async input => {
+          expectTypeOf(input).toEqualTypeOf<z.infer<typeof inputSchema>>();
+          return { temperature: 20, unit: input.unit };
+        },
+      });
+    });
+
+    it('should infer input type correctly when inputExamples are present with refine zod schema', () => {
+      const inputSchema = z.object({
+        code: z.string().refine(val => val.length === 3),
+      });
+
+      tool({
+        description: 'Get code details',
+        inputSchema,
+        inputExamples: [{ input: { code: 'ABC' } }],
+        execute: async input => {
+          expectTypeOf(input).toEqualTypeOf<z.infer<typeof inputSchema>>();
+          return { valid: true };
+        },
+      });
+    });
   });
 
   describe('output type', () => {

package/src/types/tool.ts

@@ -141,7 +141,7 @@ functionality that can be fully encapsulated in the provider.
    * An optional list of input examples that show the language
    * model what the input should look like.
    */
-  inputExamples?: Array<{ input: INPUT }>;
+  inputExamples?: Array<{ input: NoInfer<INPUT> }>;
 
   /**
    * Whether the tool needs approval before it can be executed.