-
Notifications
You must be signed in to change notification settings - Fork 4k
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
Merge pull request #3012 from v1xingyue/plugin-sui-swap
feat: use Aggregator swap sui tokens
- Loading branch information
Showing
8 changed files
with
572 additions
and
77 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -1,38 +1,40 @@ | ||
| { | ||
| "name": "@elizaos/plugin-sui", | ||
| "version": "0.1.9", | ||
| "type": "module", | ||
| "main": "dist/index.js", | ||
| "module": "dist/index.js", | ||
| "types": "dist/index.d.ts", | ||
| "exports": { | ||
| "./package.json": "./package.json", | ||
| ".": { | ||
| "import": { | ||
| "@elizaos/source": "./src/index.ts", | ||
| "types": "./dist/index.d.ts", | ||
| "default": "./dist/index.js" | ||
| } | ||
| } | ||
| }, | ||
| "files": [ | ||
| "dist" | ||
| ], | ||
| "dependencies": { | ||
| "@elizaos/core": "workspace:*", | ||
| "@mysten/sui": "^1.16.0", | ||
| "axios": "^1.7.9", | ||
| "bignumber.js": "9.1.2", | ||
| "node-cache": "5.1.2", | ||
| "tsup": "8.3.5", | ||
| "vitest": "2.1.4" | ||
| }, | ||
| "scripts": { | ||
| "build": "tsup --format esm --dts", | ||
| "test": "vitest run" | ||
| }, | ||
| "peerDependencies": { | ||
| "form-data": "4.0.1", | ||
| "whatwg-url": "7.1.0" | ||
| } | ||
| "name": "@elizaos/plugin-sui", | ||
| "version": "0.1.8-alpha.1", | ||
| "type": "module", | ||
| "main": "dist/index.js", | ||
| "module": "dist/index.js", | ||
| "types": "dist/index.d.ts", | ||
| "exports": { | ||
| "./package.json": "./package.json", | ||
| ".": { | ||
| "import": { | ||
| "@elizaos/source": "./src/index.ts", | ||
| "types": "./dist/index.d.ts", | ||
| "default": "./dist/index.js" | ||
| } | ||
| } | ||
| }, | ||
| "files": [ | ||
| "dist" | ||
| ], | ||
| "dependencies": { | ||
| "@cetusprotocol/aggregator-sdk": "^0.3.21", | ||
| "@elizaos/core": "workspace:*", | ||
| "@mysten/sui": "^1.16.0", | ||
| "axios": "^1.7.9", | ||
| "bignumber.js": "9.1.2", | ||
| "node-cache": "5.1.2", | ||
| "tsup": "8.3.5", | ||
| "vitest": "2.1.4" | ||
| }, | ||
| "scripts": { | ||
| "build": "tsup --format esm --dts", | ||
| "lint": "eslint . --fix", | ||
| "test": "vitest run" | ||
| }, | ||
| "peerDependencies": { | ||
| "form-data": "4.0.1", | ||
| "whatwg-url": "7.1.0" | ||
| } | ||
| } |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,244 @@ | ||
| import { | ||
| ActionExample, | ||
| Content, | ||
| HandlerCallback, | ||
| IAgentRuntime, | ||
| Memory, | ||
| ModelClass, | ||
| ServiceType, | ||
| // ServiceType, | ||
| State, | ||
| composeContext, | ||
| elizaLogger, | ||
| generateObject, | ||
| type Action, | ||
| } from "@elizaos/core"; | ||
| import { SuiService } from "../services/sui"; | ||
| import { z } from "zod"; | ||
|
|
||
| export interface SwapPayload extends Content { | ||
| from_token: string; | ||
| destination_token: string; | ||
| amount: string | number; | ||
| } | ||
|
|
||
| function isSwapContent(content: Content): content is SwapPayload { | ||
| console.log("Content for transfer", content); | ||
| return ( | ||
| typeof content.from_token === "string" && | ||
| typeof content.destination_token === "string" && | ||
| (typeof content.amount === "string" || | ||
| typeof content.amount === "number") | ||
| ); | ||
| } | ||
|
|
||
| const swapTemplate = `Respond with a JSON markdown block containing only the extracted values. Use null for any values that cannot be determined. | ||
| Example response: | ||
| \`\`\`json | ||
| { | ||
| "from_token": "sui", | ||
| "destination_token": "usdc", | ||
| "amount": "1" | ||
| } | ||
| \`\`\` | ||
| {{recentMessages}} | ||
| Given the recent messages, extract the following information about the requested token swap: | ||
| - Source Token you want to swap from | ||
| - Destination token you want to swap to | ||
| - Source Token Amount to swap | ||
| Respond with a JSON markdown block containing only the extracted values.`; | ||
|
|
||
| export default { | ||
| name: "SWAP_TOKEN", | ||
| similes: ["SWAP_TOKENS", "SWAP_SUI"], | ||
| validate: async (runtime: IAgentRuntime, message: Memory) => { | ||
| console.log("Validating sui transfer from user:", message.userId); | ||
| return true; | ||
| }, | ||
| description: "Swap from any token in the agent's wallet to another token", | ||
| handler: async ( | ||
| runtime: IAgentRuntime, | ||
| message: Memory, | ||
| state: State, | ||
| _options: { [key: string]: unknown }, | ||
| callback?: HandlerCallback | ||
| ): Promise<boolean> => { | ||
| elizaLogger.log("Starting SWAP_TOKEN handler..."); | ||
|
|
||
| const service = runtime.getService<SuiService>( | ||
| ServiceType.TRANSCRIPTION | ||
| ); | ||
|
|
||
| if (!state) { | ||
| // Initialize or update state | ||
| state = (await runtime.composeState(message)) as State; | ||
| } else { | ||
| state = await runtime.updateRecentMessageState(state); | ||
| } | ||
|
|
||
| // Define the schema for the expected output | ||
| const swapSchema = z.object({ | ||
| from_token: z.string(), | ||
| destination_token: z.string(), | ||
| amount: z.union([z.string(), z.number()]), | ||
| }); | ||
|
|
||
| // Compose transfer context | ||
| const swapContext = composeContext({ | ||
| state, | ||
| template: swapTemplate, | ||
| }); | ||
|
|
||
| // Generate transfer content with the schema | ||
| const content = await generateObject({ | ||
| runtime, | ||
| context: swapContext, | ||
| schema: swapSchema, | ||
| modelClass: ModelClass.SMALL, | ||
| }); | ||
|
|
||
| console.log("Generated content:", content); | ||
| const swapContent = content.object as SwapPayload; | ||
| elizaLogger.info("Swap content:", swapContent); | ||
|
|
||
| if (service.getNetwork() == "mainnet") { | ||
| // Validate transfer content | ||
| if (!isSwapContent(swapContent)) { | ||
| console.error("Invalid content for SWAP_TOKEN action."); | ||
| if (callback) { | ||
| callback({ | ||
| text: "Unable to process swap request. Invalid content provided.", | ||
| content: { error: "Invalid transfer content" }, | ||
| }); | ||
| } | ||
| return false; | ||
| } | ||
|
|
||
| const destinationToken = await service.getTokenMetadata( | ||
| swapContent.destination_token | ||
| ); | ||
|
|
||
| elizaLogger.log("Destination token:", destinationToken); | ||
|
|
||
| const fromToken = await service.getTokenMetadata( | ||
| swapContent.from_token | ||
| ); | ||
|
|
||
| elizaLogger.log("From token:", fromToken); | ||
|
|
||
| // one action only can call one callback to save new message. | ||
| // runtime.processActions | ||
| if (destinationToken && fromToken) { | ||
| try { | ||
| const swapAmount = service.getAmount( | ||
| swapContent.amount, | ||
| fromToken | ||
| ); | ||
|
|
||
| elizaLogger.info("Swap amount:", swapAmount); | ||
|
|
||
| elizaLogger.info( | ||
| "Destination token address:", | ||
| destinationToken.tokenAddress | ||
| ); | ||
|
|
||
| elizaLogger.info( | ||
| "From token address:", | ||
| fromToken.tokenAddress | ||
| ); | ||
|
|
||
| elizaLogger.info("Swap amount:", swapAmount); | ||
|
|
||
| const result = await service.swapToken( | ||
| fromToken.symbol, | ||
| swapAmount.toString(), | ||
| 0, | ||
| destinationToken.symbol | ||
| ); | ||
|
|
||
| if (result.success) { | ||
| callback({ | ||
| text: `Successfully swapped ${swapContent.amount} ${swapContent.from_token} to ${swapContent.destination_token}, Transaction: ${service.getTransactionLink( | ||
| result.tx | ||
| )}`, | ||
| content: swapContent, | ||
| }); | ||
| } | ||
| } catch (error) { | ||
| elizaLogger.error("Error swapping token:", error); | ||
| callback({ | ||
| text: `Failed to swap ${error}, swapContent : ${JSON.stringify( | ||
| swapContent | ||
| )}`, | ||
| content: { error: "Failed to swap token" }, | ||
| }); | ||
| } | ||
| } else { | ||
| callback({ | ||
| text: `destination token: ${swapContent.destination_token} or from token: ${swapContent.from_token} not found`, | ||
| content: { error: "Destination token not found" }, | ||
| }); | ||
| } | ||
| } else { | ||
| callback({ | ||
| text: | ||
| "Sorry, I can only swap on the mainnet, parsed params : " + | ||
| JSON.stringify(swapContent, null, 2), | ||
| content: { error: "Unsupported network" }, | ||
| }); | ||
| return false; | ||
| } | ||
|
|
||
| return true; | ||
| }, | ||
|
|
||
| examples: [ | ||
| [ | ||
| { | ||
| user: "{{user1}}", | ||
| content: { | ||
| text: "Swap 1 SUI to USDC", | ||
| }, | ||
| }, | ||
| { | ||
| user: "{{user2}}", | ||
| content: { | ||
| text: "I'll help you swap 1 SUI to USDC now...", | ||
| action: "SWAP_TOKEN", | ||
| }, | ||
| }, | ||
| { | ||
| user: "{{user2}}", | ||
| content: { | ||
| text: "Successfully swapped 1 SUI to USDC, Transaction: 0x39a8c432d9bdad993a33cc1faf2e9b58fb7dd940c0425f1d6db3997e4b4b05c0", | ||
| }, | ||
| }, | ||
| ], | ||
| [ | ||
| { | ||
| user: "{{user1}}", | ||
| content: { | ||
| text: "Swap 1 USDC to SUI", | ||
| }, | ||
| }, | ||
| { | ||
| user: "{{user2}}", | ||
| content: { | ||
| text: "I'll help you swap 1 SUI to USDC now...", | ||
| action: "SWAP_TOKEN", | ||
| }, | ||
| }, | ||
| { | ||
| user: "{{user2}}", | ||
| content: { | ||
| text: "Successfully swapped 1 SUI to USDC, Transaction: 0x39a8c432d9bdad993a33cc1faf2e9b58fb7dd940c0425f1d6db3997e4b4b05c0", | ||
| }, | ||
| }, | ||
| ], | ||
| ] as ActionExample[][], | ||
| } as Action; |
Oops, something went wrong.