[AI SDK Provider] Fix type inference for tools (#7944)

Author: joaquim-verges

.changeset/cool-jeans-end.md

@@ -0,0 +1,5 @@
+---
+"@thirdweb-dev/ai-sdk-provider": patch
+---
+
+Proper type inference

apps/portal/src/app/ai/chat/ai-sdk/page.mdx

@@ -1,6 +1,6 @@
 # Usage with the Vercel AI SDK
 
-The `@thirdweb-dev/ai-sdk-provider` A thin provider that plugs thirdweb into the [Vercel AI SDK](https://ai-sdk.dev/docs/introduction).
+The `@thirdweb-dev/ai-sdk-provider` is a lightweight provider that lets you use thirdweb AI with the [Vercel AI SDK](https://ai-sdk.dev/docs/introduction).
 
 It standardizes message parts, exposes wallet-aware tools (`sign_transaction`, `sign_swap`), and ships types/utilities so you can build chat UIs that execute on-chain actions from AI responses.
 
@@ -67,7 +67,6 @@ export async function POST(req: Request) {
     },
   });
 }
-
 ```
 
 #### Client side (React, using `useChat`)

apps/portal/src/app/ai/chat/page.mdx

@@ -14,16 +14,50 @@ The thirdweb API exposes a standard OpenAI-compatible chat completion API that a
 - Search the web
 - And more!
 
-You can use the API with the HTTP API directly, or with any OpenAI-compatible client library.
+You can use the API with the HTTP API directly, the AI SDK or with any OpenAI-compatible client library.
 
 <Tabs defaultValue="api">
   <TabsList>
     <TabsTrigger value="api">HTTP API</TabsTrigger>
+    <TabsTrigger value="ai-sdk">Vercel AI SDK</TabsTrigger>
     <TabsTrigger value="openai">OpenAI Client</TabsTrigger>
   </TabsList>
   <TabsContent value="api">
     <OpenApiEndpoint path="/ai/chat" method="POST" />
   </TabsContent>
+
+  <TabsContent value="ai-sdk">
+    ### Usage with the Vercel AI SDK
+
+    The `@thirdweb-dev/ai-sdk-provider` is a lightweight provider that lets you use thirdweb AI with the [Vercel AI SDK](https://ai-sdk.dev/docs/introduction).
+
+    ```ts
+    import { streamText } from "ai";
+    import { createThirdwebAI } from "@thirdweb-dev/ai-sdk-provider";
+
+    const thirdwebAI = createThirdwebAI({
+      secretKey: "<your-project-secret-key>",
+    });
+
+    const result = streamText({
+      model: thirdwebAI.chat({
+        context: {
+          chain_ids: [8453], // optional chain ids
+          from: "0x...", // optional wallet address
+          auto_execute_transactions: true, // defaults to false
+        },
+      }),
+      tools: thirdwebAI.tools(),
+      messages: [
+        {
+          role: "user",
+          content: "Swap 0.01 ETH to USDC",
+        },
+      ],
+    });
+  ```
+  </TabsContent>
+
   <TabsContent value="openai">
 
 ### Using the OpenAI Client
@@ -41,7 +75,7 @@ chat_completion = client.chat.completions.create(
 	model="t0",
 	messages=[{"role": "user", "content": "Transfer 10 USDC to vitalik.eth"}],
 	stream=False,
-    extra_body={ "context": { "from": "0x...", "chain_ids": [8453] }}
+  extra_body={ "context": { "from": "0x...", "chain_ids": [8453] }}
 )
 
 print(chat_completion)
@@ -52,4 +86,6 @@ print(chat_completion)
 ### Going further
 
 - [Handle streaming responses](/ai/chat/streaming)
+- [Transaction Execution](/ai/chat/execution)
+- [Use the Vercel AI SDK](/ai/chat/ai-sdk)
 - [Full API Reference](https://api.thirdweb.com/reference#tag/ai/post/ai/chat)

packages/ai-sdk-provider/src/tools.ts

@@ -48,36 +48,36 @@ export type MonitorTransactionOutput = z.infer<
   typeof AgentActionMonitorTransactionOutputData
 >;
 
-export function createTools(_config: ThirdwebConfig): Record<string, Tool> {
+export type SignTransactionInput = z.infer<
+  typeof AgentActionSignTransactionData
+>;
+export type SignSwapInput = z.infer<typeof AgentActionSignSwapData>;
+export type MonitorTransactionInput = z.infer<
+  typeof AgentActionMonitorTransactionData
+>;
+
+export function createTools(_config: ThirdwebConfig) {
   return {
     sign_transaction: {
       id: "thirdweb.sign_transaction" as const,
       name: "sign_transaction",
       description: "Sign a transaction",
       inputSchema: AgentActionSignTransactionData,
       outputSchema: AgentActionTransactionOutputData,
-    } as const,
+    } satisfies Tool<SignTransactionInput, TransactionOutput>,
     sign_swap: {
       id: "thirdweb.sign_swap" as const,
       name: "sign_swap",
       description: "Sign a swap transaction",
       inputSchema: AgentActionSignSwapData,
       outputSchema: AgentActionTransactionOutputData,
-    } as const,
+    } satisfies Tool<SignSwapInput, TransactionOutput>,
     monitor_transaction: {
       id: "thirdweb.monitor_transaction" as const,
       name: "monitor_transaction",
       description: "Monitor a transaction",
       inputSchema: AgentActionMonitorTransactionData,
       outputSchema: AgentActionMonitorTransactionOutputData,
-    } as const,
+    } satisfies Tool<MonitorTransactionInput, MonitorTransactionOutput>,
   } as const;
 }
-
-export type SignTransactionInput = z.infer<
-  typeof AgentActionSignTransactionData
->;
-export type SignSwapInput = z.infer<typeof AgentActionSignSwapData>;
-export type MonitorTransactionInput = z.infer<
-  typeof AgentActionMonitorTransactionData
->;