reverting register basename input schema

coinbase · Dec 3, 2024 · bb6de6d · bb6de6d
1 parent d661818
commit bb6de6d
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/cdp-agentkit-core/src/actions/cdp/register_basename.ts b/cdp-agentkit-core/src/actions/cdp/register_basename.ts
@@ -98,7 +98,7 @@ const REGISTRAR_ABI = [
 export const RegisterBasenameInput = z
   .object({
     basename: z.string().describe("The Basename to assign to the agent"),
-    amount: z.custom<Amount>().default(0.002).describe("The amount of the from asset to trade"),
+    amount: z.string().default("0.002").describe("The amount of the from asset to trade"),
   })
   .strip()
   .describe("Instructions for registering a Basename");
@@ -175,7 +175,7 @@ export async function registerBasename(
       method: "register",
       args: registerArgs,
       abi: REGISTRAR_ABI,
-      amount: args.amount,
+      amount: new Decimal(args.amount),
       assetId: "eth",
     });
 

diff --git a/cdp-agentkit-core/src/tests/register_basename_test.ts b/cdp-agentkit-core/src/tests/register_basename_test.ts
@@ -2,7 +2,7 @@ import { Coinbase, Wallet } from "@coinbase/coinbase-sdk";
 
 import { registerBasename, RegisterBasenameInput } from "../actions/cdp/register_basename";
 
-const MOCK_AMMOUNT = 0.123;
+const MOCK_AMMOUNT = "0.123";
 const MOCK_BASENAME = "test-basename";
 
 describe("Register Basename Input", () => {