From 38831fcf149bc17686f4eb2a7ec7ba9622b21611 Mon Sep 17 00:00:00 2001
From: Joaquim Verges <joaquim.verges@gmail.com>
Date: Fri, 5 Sep 2025 10:15:54 +1200
Subject: [PATCH] [AI SDK Provider] Update sign swap action type with token
 fields

---
 .changeset/quiet-llamas-chew.md          |  5 +++++
 packages/ai-sdk-provider/src/provider.ts |  2 ++
 packages/ai-sdk-provider/src/tools.ts    | 11 +++++++++++
 3 files changed, 18 insertions(+)
 create mode 100644 .changeset/quiet-llamas-chew.md

diff --git a/.changeset/quiet-llamas-chew.md b/.changeset/quiet-llamas-chew.md
new file mode 100644
index 00000000000..a487968ae41
--- /dev/null
+++ b/.changeset/quiet-llamas-chew.md
@@ -0,0 +1,5 @@
+---
+"@thirdweb-dev/ai-sdk-provider": patch
+---
+
+Update sign swap action type
diff --git a/packages/ai-sdk-provider/src/provider.ts b/packages/ai-sdk-provider/src/provider.ts
index 5b7599d9f9c..96e916ed2f6 100644
--- a/packages/ai-sdk-provider/src/provider.ts
+++ b/packages/ai-sdk-provider/src/provider.ts
@@ -453,6 +453,8 @@ class ThirdwebLanguageModel implements LanguageModelV2 {
                                 action: parsed.data.action,
                                 intent: parsed.data.intent,
                                 transaction: parsed.data.transaction,
+                                from_token: parsed.data.from_token,
+                                to_token: parsed.data.to_token,
                               } satisfies SignSwapInput;
                               break;
 
diff --git a/packages/ai-sdk-provider/src/tools.ts b/packages/ai-sdk-provider/src/tools.ts
index adc10fd44a8..542741629c4 100644
--- a/packages/ai-sdk-provider/src/tools.ts
+++ b/packages/ai-sdk-provider/src/tools.ts
@@ -21,10 +21,21 @@ const AgentActionSignSwapDataIntent = z.object({
   max_steps: z.number(),
 });
 
+const AgentActionSignSwapDataToken = z.object({
+  address: z.string(),
+  chain_id: z.number(),
+  amount: z.string(),
+  symbol: z.string(),
+  decimals: z.number(),
+  price: z.number().nullable(),
+});
+
 const AgentActionSignSwapData = z.object({
   action: z.string(),
   intent: AgentActionSignSwapDataIntent,
   transaction: AgentActionSignTransactionData,
+  from_token: AgentActionSignSwapDataToken,
+  to_token: AgentActionSignSwapDataToken,
 });
 
 const AgentActionTransactionOutputData = z.object({