feat: add vercel ai sdk (#101)

Author: CocDap

packages/common/src/chains/chains.ts

@@ -11,7 +11,7 @@ import {
   westendAssetHubChain,
   westendChain,
   westendPeopleChain
-} from "./supported-chains"
+} from "./supportedChains"
 
 export type ChainIdRelay = "polkadot" | "west" | "paseo" | "kusama"
 export type ChainIdAssetHub =

packages/common/src/chains/index.ts

@@ -1,2 +1,2 @@
 export * from "./chains"
-export * from "./supported-chains"
+export * from "./supportedChains"

…es/common/src/chains/supported-chains.ts …ges/common/src/chains/supportedChains.tspackages/common/src/chains/supported-chains.ts renamed to packages/common/src/chains/supportedChains.ts packages/common/src/chains/supported-chains.ts renamed to packages/common/src/chains/supportedChains.ts

@@ -62,5 +62,8 @@
     "@langchain/ollama": "^0.2.2",
     "langchain": "^0.1.21",
     "@agentic/langchain": "^7.6.9"
+  },
+  "peerDependencies": {
+    "ai": "^4.1.16"
   }
 }

packages/sdk/package.json

@@ -1,2 +1,3 @@
 export * from "./api"
 export * from "./langchain"
+export * from "./vercel"

packages/sdk/src/index.ts

@@ -0,0 +1,30 @@
+import type { Action } from "@polkadot-agent-kit/llm"
+import type { Tool } from "ai"
+import { tool, type ToolSet } from "ai"
+import type { z } from "zod"
+
+import type { PolkadotAgentKit } from "./api"
+
+/**
+ * Get Vercel AI SDK tools from an PolkadotAgentKit instance
+ *
+ * @param polkadotAgentKit - The PolkadotAgentKit instance
+ * @returns An object containing Vercel AI SDK tools keyed by tool name
+ */
+export function getVercelAITools(polkadotAgentKit: PolkadotAgentKit): ToolSet {
+  const actions: Action[] = polkadotAgentKit.getActions()
+  const toolSet: Record<string, Tool> = {}
+
+  for (const action of actions) {
+    toolSet[action.name] = tool({
+      description: action.description,
+      parameters: action.schema,
+      execute: async (args: z.infer<typeof action.schema>) => {
+        const result = await action.invoke(args)
+        return result
+      }
+    })
+  }
+
+  return toolSet
+}

packages/sdk/src/vercel.ts

@@ -1,5 +1,6 @@
 import { describe, it, expect, vi, beforeAll, afterAll } from 'vitest';
 import { PolkadotAgentKit } from '../../src/api';
+import { getVercelAITools } from '../../src/vercel';
 import type { AgentConfig } from '@polkadot-agent-kit/common';
 
 const mockBalanceResult = { balance: '100.00', symbol: 'WND', chain: 'westend' };
@@ -58,6 +59,20 @@ vi.mock('../../src/api', () => {
         mintVdotTool: vi.fn(() => ({
           call: vi.fn(async (input: any) => mockMintVdotResult)
         })),
+        getActions: vi.fn(() => [
+          {
+            name: 'check_balance',
+            description: 'Check balance of the wallet address on a specific chain',
+            schema: {} as any,
+            invoke: vi.fn(async (args: any) => JSON.stringify(mockBalanceResult))
+          },
+          {
+            name: 'transfer_native',
+            description: 'Transfer native tokens to an address',
+            schema: {} as any,
+            invoke: vi.fn(async (args: any) => JSON.stringify(mockTransferResult))
+          }
+        ]),
         disconnect: vi.fn().mockResolvedValue(undefined),
         config: instanceConfig,
       };
@@ -282,5 +297,27 @@ describe('PolkadotAgentKit E2E', () => {
       assert(result);
     });
   });
+
+  describe('getVercelAITools', () => {
+    it('should return a toolset with correct structure', () => {
+      const toolSet = getVercelAITools(agent);
+
+      expect(toolSet).toBeDefined();
+      expect(typeof toolSet).toBe('object');
+      expect(toolSet.check_balance).toBeDefined();
+      expect(toolSet.transfer_native).toBeDefined();
+
+      // Check that the tools have the expected structure
+      expect(typeof toolSet.check_balance.execute).toBe('function');
+      expect(typeof toolSet.transfer_native.execute).toBe('function');
+    });
+
+    it('should have tools with correct descriptions', () => {
+      const toolSet = getVercelAITools(agent);
+
+      expect(toolSet.check_balance.description).toBe('Check balance of the wallet address on a specific chain');
+      expect(toolSet.transfer_native.description).toBe('Transfer native tokens to an address');
+    });
+  });
 });