@lantos1618/better-ui 0.7.0 → 0.7.1

package/dist/{ThemeProvider-CGxHnF2T.d.ts → ThemeProvider-BESbs5p6.d.ts}

@@ -1,7 +1,7 @@
 import * as react_jsx_runtime from 'react/jsx-runtime';
 import React from 'react';
 import { UIMessage, ChatStatus } from 'ai';
-import { T as Tool } from './tool-DGpwM6G6.js';
+import { T as Tool } from './tool-yZTixiN2.js';
 import { P as PersistenceAdapter, T as Thread$1 } from './types-CAOfGUPH.js';
 
 interface ToolStateEntry {

package/dist/{ThemeProvider-A7krFsCJ.d.mts → ThemeProvider-Dx3STQtp.d.mts}

@@ -1,7 +1,7 @@
 import * as react_jsx_runtime from 'react/jsx-runtime';
 import React from 'react';
 import { UIMessage, ChatStatus } from 'ai';
-import { T as Tool } from './tool-DGpwM6G6.mjs';
+import { T as Tool } from './tool-yZTixiN2.mjs';
 import { P as PersistenceAdapter, T as Thread$1 } from './types-CAOfGUPH.mjs';
 
 interface ToolStateEntry {

package/dist/agui/index.d.mts

@@ -1,4 +1,4 @@
-import { T as Tool, g as ToolContext } from '../tool-DGpwM6G6.mjs';
+import { T as Tool, g as ToolContext } from '../tool-yZTixiN2.mjs';
 import 'zod';
 import 'react';
 

package/dist/agui/index.d.ts

@@ -1,4 +1,4 @@
-import { T as Tool, g as ToolContext } from '../tool-DGpwM6G6.js';
+import { T as Tool, g as ToolContext } from '../tool-yZTixiN2.js';
 import 'zod';
 import 'react';
 

package/dist/components/index.d.mts

@@ -1,7 +1,7 @@
-import { f as ToolStateStore, g as ToolStateEntry } from '../ThemeProvider-A7krFsCJ.mjs';
-export { h as Chat, C as ChatProps, i as ChatProvider, a as ChatProviderProps, j as Composer, b as ComposerProps, k as Message, M as MessageProps, l as ThemeProvider, T as ThemeProviderProps, m as Thread, c as ThreadProps, d as ToolPartInfo, n as ToolResult, e as ToolResultProps, o as createToolStateStore, u as useChatContext, p as useToolState } from '../ThemeProvider-A7krFsCJ.mjs';
+import { f as ToolStateStore, g as ToolStateEntry } from '../ThemeProvider-Dx3STQtp.mjs';
+export { h as Chat, C as ChatProps, i as ChatProvider, a as ChatProviderProps, j as Composer, b as ComposerProps, k as Message, M as MessageProps, l as ThemeProvider, T as ThemeProviderProps, m as Thread, c as ThreadProps, d as ToolPartInfo, n as ToolResult, e as ToolResultProps, o as createToolStateStore, u as useChatContext, p as useToolState } from '../ThemeProvider-Dx3STQtp.mjs';
 import * as react_jsx_runtime from 'react/jsx-runtime';
-import { T as Tool } from '../tool-DGpwM6G6.mjs';
+import { T as Tool } from '../tool-yZTixiN2.mjs';
 import React from 'react';
 import 'ai';
 import '../types-CAOfGUPH.mjs';

package/dist/components/index.d.ts

@@ -1,7 +1,7 @@
-import { f as ToolStateStore, g as ToolStateEntry } from '../ThemeProvider-CGxHnF2T.js';
-export { h as Chat, C as ChatProps, i as ChatProvider, a as ChatProviderProps, j as Composer, b as ComposerProps, k as Message, M as MessageProps, l as ThemeProvider, T as ThemeProviderProps, m as Thread, c as ThreadProps, d as ToolPartInfo, n as ToolResult, e as ToolResultProps, o as createToolStateStore, u as useChatContext, p as useToolState } from '../ThemeProvider-CGxHnF2T.js';
+import { f as ToolStateStore, g as ToolStateEntry } from '../ThemeProvider-BESbs5p6.js';
+export { h as Chat, C as ChatProps, i as ChatProvider, a as ChatProviderProps, j as Composer, b as ComposerProps, k as Message, M as MessageProps, l as ThemeProvider, T as ThemeProviderProps, m as Thread, c as ThreadProps, d as ToolPartInfo, n as ToolResult, e as ToolResultProps, o as createToolStateStore, u as useChatContext, p as useToolState } from '../ThemeProvider-BESbs5p6.js';
 import * as react_jsx_runtime from 'react/jsx-runtime';
-import { T as Tool } from '../tool-DGpwM6G6.js';
+import { T as Tool } from '../tool-yZTixiN2.js';
 import React from 'react';
 import 'ai';
 import '../types-CAOfGUPH.js';

package/dist/index.d.mts

@@ -1,6 +1,6 @@
-export { C as CacheConfig, a as ClientFetchConfig, b as ClientHandler, S as ServerHandler, c as StreamCallback, d as StreamHandler, T as Tool, e as ToolBuilder, f as ToolConfig, g as ToolContext, V as ViewComponent, h as ViewState, t as tool } from './tool-DGpwM6G6.mjs';
+export { C as CacheConfig, a as ClientFetchConfig, b as ClientHandler, S as ServerHandler, c as StreamCallback, d as StreamHandler, T as Tool, e as ToolBuilder, f as ToolConfig, g as ToolContext, V as ViewComponent, h as ViewState, t as tool } from './tool-yZTixiN2.mjs';
 import { LanguageModelV2 } from '@ai-sdk/provider';
-export { C as ChatProps, a as ChatProviderProps, b as ComposerProps, M as MessageProps, T as ThemeProviderProps, c as ThreadProps, d as ToolPartInfo, e as ToolResultProps } from './ThemeProvider-A7krFsCJ.mjs';
+export { C as ChatProps, a as ChatProviderProps, b as ComposerProps, M as MessageProps, T as ThemeProviderProps, c as ThreadProps, d as ToolPartInfo, e as ToolResultProps } from './ThemeProvider-Dx3STQtp.mjs';
 import 'zod';
 import 'react';
 import 'react/jsx-runtime';

package/dist/index.d.ts

@@ -1,6 +1,6 @@
-export { C as CacheConfig, a as ClientFetchConfig, b as ClientHandler, S as ServerHandler, c as StreamCallback, d as StreamHandler, T as Tool, e as ToolBuilder, f as ToolConfig, g as ToolContext, V as ViewComponent, h as ViewState, t as tool } from './tool-DGpwM6G6.js';
+export { C as CacheConfig, a as ClientFetchConfig, b as ClientHandler, S as ServerHandler, c as StreamCallback, d as StreamHandler, T as Tool, e as ToolBuilder, f as ToolConfig, g as ToolContext, V as ViewComponent, h as ViewState, t as tool } from './tool-yZTixiN2.js';
 import { LanguageModelV2 } from '@ai-sdk/provider';
-export { C as ChatProps, a as ChatProviderProps, b as ComposerProps, M as MessageProps, T as ThemeProviderProps, c as ThreadProps, d as ToolPartInfo, e as ToolResultProps } from './ThemeProvider-CGxHnF2T.js';
+export { C as ChatProps, a as ChatProviderProps, b as ComposerProps, M as MessageProps, T as ThemeProviderProps, c as ThreadProps, d as ToolPartInfo, e as ToolResultProps } from './ThemeProvider-BESbs5p6.js';
 import 'zod';
 import 'react';
 import 'react/jsx-runtime';

package/dist/mcp/index.d.mts

@@ -1,4 +1,4 @@
-import { T as Tool, g as ToolContext } from '../tool-DGpwM6G6.mjs';
+import { T as Tool, g as ToolContext } from '../tool-yZTixiN2.mjs';
 import { z } from 'zod';
 import 'react';
 
@@ -152,6 +152,8 @@ interface JsonSchema {
     additionalProperties?: boolean | JsonSchema;
     [key: string]: unknown;
 }
-declare function zodToJsonSchema(schema: z.ZodType): JsonSchema;
+declare function zodToJsonSchema(schema: z.ZodType | {
+    [key: string]: any;
+}): JsonSchema;
 
 export { MCPServer, type MCPServerConfig, createMCPServer, zodToJsonSchema };

package/dist/mcp/index.d.ts

@@ -1,4 +1,4 @@
-import { T as Tool, g as ToolContext } from '../tool-DGpwM6G6.js';
+import { T as Tool, g as ToolContext } from '../tool-yZTixiN2.js';
 import { z } from 'zod';
 import 'react';
 
@@ -152,6 +152,8 @@ interface JsonSchema {
     additionalProperties?: boolean | JsonSchema;
     [key: string]: unknown;
 }
-declare function zodToJsonSchema(schema: z.ZodType): JsonSchema;
+declare function zodToJsonSchema(schema: z.ZodType | {
+    [key: string]: any;
+}): JsonSchema;
 
 export { MCPServer, type MCPServerConfig, createMCPServer, zodToJsonSchema };

package/dist/react/index.d.mts

@@ -1,4 +1,4 @@
-import { g as ToolContext, T as Tool } from '../tool-DGpwM6G6.mjs';
+import { g as ToolContext, T as Tool } from '../tool-yZTixiN2.mjs';
 import 'zod';
 import 'react';
 

package/dist/react/index.d.ts

@@ -1,4 +1,4 @@
-import { g as ToolContext, T as Tool } from '../tool-DGpwM6G6.js';
+import { g as ToolContext, T as Tool } from '../tool-yZTixiN2.js';
 import 'zod';
 import 'react';
 

package/dist/{tool-DGpwM6G6.d.mts → tool-yZTixiN2.d.mts}

@@ -8,6 +8,16 @@ import React, { ReactElement } from 'react';
  * with Better UI's unique view integration.
  */
 
+/**
+ * Schema type that works with both Zod 3 and Zod 4.
+ * Zod 4 removed internal properties (_parse, _type, etc.) from ZodType,
+ * so we use a structural type instead of z.ZodType directly.
+ */
+type ZodSchema<T = any> = z.ZodType<T> | {
+    parse: (data: unknown) => T;
+    safeParse: (data: unknown) => any;
+    [key: string]: any;
+};
 /** Behavioral hints for tools */
 interface ToolHints {
     /** Tool performs destructive/irreversible actions (auto-implies requiresConfirmation) */
@@ -63,8 +73,8 @@ interface ClientFetchConfig {
 interface ToolConfig<TInput, TOutput> {
     name: string;
     description?: string;
-    input: z.ZodType<TInput>;
-    output?: z.ZodType<TOutput>;
+    input: ZodSchema<TInput>;
+    output?: ZodSchema<TOutput>;
     tags?: string[];
     cache?: CacheConfig<TInput>;
     /** Configure auto-fetch behavior for client-side execution */
@@ -99,8 +109,8 @@ type ViewComponent<TOutput, TInput = unknown> = (data: TOutput, state?: ViewStat
 declare class Tool<TInput = any, TOutput = any> {
     readonly name: string;
     readonly description?: string;
-    readonly inputSchema: z.ZodType<TInput>;
-    readonly outputSchema?: z.ZodType<TOutput>;
+    readonly inputSchema: ZodSchema<TInput>;
+    readonly outputSchema?: ZodSchema<TOutput>;
     readonly tags: string[];
     readonly cacheConfig?: CacheConfig<TInput>;
     readonly clientFetchConfig?: ClientFetchConfig;
@@ -233,11 +243,11 @@ declare class Tool<TInput = any, TOutput = any> {
      */
     toAITool(): {
         description: string;
-        inputSchema: z.ZodType<TInput, z.ZodTypeDef, TInput>;
+        inputSchema: ZodSchema<TInput>;
         execute?: undefined;
     } | {
         description: string;
-        inputSchema: z.ZodType<TInput, z.ZodTypeDef, TInput>;
+        inputSchema: ZodSchema<TInput>;
         execute: (input: TInput) => Promise<TOutput>;
     };
 }
@@ -294,11 +304,11 @@ declare class ToolBuilder<TInput = any, TOutput = any> {
      * 2. The return type correctly reflects the new generic parameter
      * 3. TypeScript doesn't support "this type mutation" in fluent builders
      */
-    input<T>(schema: z.ZodType<T>): ToolBuilder<T, TOutput>;
+    input<T>(schema: ZodSchema<T>): ToolBuilder<T, TOutput>;
     /**
      * Define output schema - enables type inference for results
      */
-    output<O>(schema: z.ZodType<O>): ToolBuilder<TInput, O>;
+    output<O>(schema: ZodSchema<O>): ToolBuilder<TInput, O>;
     tags(...tags: string[]): this;
     cache(config: CacheConfig<TInput>): this;
     /** Configure auto-fetch endpoint for client-side execution */
@@ -349,11 +359,11 @@ declare class ToolBuilder<TInput = any, TOutput = any> {
     };
     toAITool(): {
         description: string;
-        inputSchema: z.ZodType<TInput, z.ZodTypeDef, TInput>;
+        inputSchema: ZodSchema<TInput>;
         execute?: undefined;
     } | {
         description: string;
-        inputSchema: z.ZodType<TInput, z.ZodTypeDef, TInput>;
+        inputSchema: ZodSchema<TInput>;
         execute: (input: TInput) => Promise<TOutput>;
     };
 }

package/dist/{tool-DGpwM6G6.d.ts → tool-yZTixiN2.d.ts}

@@ -8,6 +8,16 @@ import React, { ReactElement } from 'react';
  * with Better UI's unique view integration.
  */
 
+/**
+ * Schema type that works with both Zod 3 and Zod 4.
+ * Zod 4 removed internal properties (_parse, _type, etc.) from ZodType,
+ * so we use a structural type instead of z.ZodType directly.
+ */
+type ZodSchema<T = any> = z.ZodType<T> | {
+    parse: (data: unknown) => T;
+    safeParse: (data: unknown) => any;
+    [key: string]: any;
+};
 /** Behavioral hints for tools */
 interface ToolHints {
     /** Tool performs destructive/irreversible actions (auto-implies requiresConfirmation) */
@@ -63,8 +73,8 @@ interface ClientFetchConfig {
 interface ToolConfig<TInput, TOutput> {
     name: string;
     description?: string;
-    input: z.ZodType<TInput>;
-    output?: z.ZodType<TOutput>;
+    input: ZodSchema<TInput>;
+    output?: ZodSchema<TOutput>;
     tags?: string[];
     cache?: CacheConfig<TInput>;
     /** Configure auto-fetch behavior for client-side execution */
@@ -99,8 +109,8 @@ type ViewComponent<TOutput, TInput = unknown> = (data: TOutput, state?: ViewStat
 declare class Tool<TInput = any, TOutput = any> {
     readonly name: string;
     readonly description?: string;
-    readonly inputSchema: z.ZodType<TInput>;
-    readonly outputSchema?: z.ZodType<TOutput>;
+    readonly inputSchema: ZodSchema<TInput>;
+    readonly outputSchema?: ZodSchema<TOutput>;
     readonly tags: string[];
     readonly cacheConfig?: CacheConfig<TInput>;
     readonly clientFetchConfig?: ClientFetchConfig;
@@ -233,11 +243,11 @@ declare class Tool<TInput = any, TOutput = any> {
      */
     toAITool(): {
         description: string;
-        inputSchema: z.ZodType<TInput, z.ZodTypeDef, TInput>;
+        inputSchema: ZodSchema<TInput>;
         execute?: undefined;
     } | {
         description: string;
-        inputSchema: z.ZodType<TInput, z.ZodTypeDef, TInput>;
+        inputSchema: ZodSchema<TInput>;
         execute: (input: TInput) => Promise<TOutput>;
     };
 }
@@ -294,11 +304,11 @@ declare class ToolBuilder<TInput = any, TOutput = any> {
      * 2. The return type correctly reflects the new generic parameter
      * 3. TypeScript doesn't support "this type mutation" in fluent builders
      */
-    input<T>(schema: z.ZodType<T>): ToolBuilder<T, TOutput>;
+    input<T>(schema: ZodSchema<T>): ToolBuilder<T, TOutput>;
     /**
      * Define output schema - enables type inference for results
      */
-    output<O>(schema: z.ZodType<O>): ToolBuilder<TInput, O>;
+    output<O>(schema: ZodSchema<O>): ToolBuilder<TInput, O>;
     tags(...tags: string[]): this;
     cache(config: CacheConfig<TInput>): this;
     /** Configure auto-fetch endpoint for client-side execution */
@@ -349,11 +359,11 @@ declare class ToolBuilder<TInput = any, TOutput = any> {
     };
     toAITool(): {
         description: string;
-        inputSchema: z.ZodType<TInput, z.ZodTypeDef, TInput>;
+        inputSchema: ZodSchema<TInput>;
         execute?: undefined;
     } | {
         description: string;
-        inputSchema: z.ZodType<TInput, z.ZodTypeDef, TInput>;
+        inputSchema: ZodSchema<TInput>;
         execute: (input: TInput) => Promise<TOutput>;
     };
 }

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@lantos1618/better-ui",
-  "version": "0.7.0",
+  "version": "0.7.1",
   "description": "A minimal, type-safe AI-first UI framework for building tools",
   "main": "dist/index.js",
   "module": "dist/index.mjs",
@@ -81,7 +81,7 @@
     "react-markdown": "^10.1.0",
     "remark-gfm": "^4.0.1",
     "shiki": "^4.0.2",
-    "zod": "^3.25.76"
+    "zod": "^3.25.76 || ^4.0.0"
   },
   "peerDependencies": {
     "@ai-sdk/anthropic": "^3.0.0",