-
Notifications
You must be signed in to change notification settings - Fork 2.8k
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
Merge branch 'jnaulty/add-sui-plugin' of https://github.com/jnaulty/e…
…liza into develop
- Loading branch information
Showing
16 changed files
with
859 additions
and
13 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
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
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
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
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
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,6 @@ | ||
| * | ||
|
|
||
| !dist/** | ||
| !package.json | ||
| !readme.md | ||
| !tsup.config.ts |
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,3 @@ | ||
| import eslintGlobalConfig from "../../eslint.config.mjs"; | ||
|
|
||
| export default [...eslintGlobalConfig]; |
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,26 @@ | ||
| { | ||
| "name": "@ai16z/plugin-sui", | ||
| "version": "0.1.5-alpha.5", | ||
| "main": "dist/index.js", | ||
| "type": "module", | ||
| "types": "dist/index.d.ts", | ||
| "dependencies": { | ||
| "@ai16z/eliza": "workspace:*", | ||
| "@ai16z/plugin-trustdb": "workspace:*", | ||
| "@mysten/sui": "^1.16.0", | ||
| "bignumber": "1.1.0", | ||
| "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,214 @@ | ||
| import { | ||
| ActionExample, | ||
| Content, | ||
| HandlerCallback, | ||
| IAgentRuntime, | ||
| Memory, | ||
| ModelClass, | ||
| State, | ||
| composeContext, | ||
| elizaLogger, | ||
| generateObject, | ||
| type Action, | ||
| } from "@ai16z/eliza"; | ||
| import { z } from "zod"; | ||
|
|
||
| import { SuiClient, getFullnodeUrl } from "@mysten/sui/client"; | ||
| import { Ed25519Keypair } from "@mysten/sui/keypairs/ed25519"; | ||
| import { Transaction } from "@mysten/sui/transactions"; | ||
| import { SUI_DECIMALS } from "@mysten/sui/utils"; | ||
|
|
||
| import { walletProvider } from "../providers/wallet"; | ||
|
|
||
| type SuiNetwork = "mainnet" | "testnet" | "devnet" | "localnet"; | ||
|
|
||
| export interface TransferContent extends Content { | ||
| recipient: string; | ||
| amount: string | number; | ||
| } | ||
|
|
||
| function isTransferContent(content: Content): content is TransferContent { | ||
| console.log("Content for transfer", content); | ||
| return ( | ||
| typeof content.recipient === "string" && | ||
| (typeof content.amount === "string" || | ||
| typeof content.amount === "number") | ||
| ); | ||
| } | ||
|
|
||
| const transferTemplate = `Respond with a JSON markdown block containing only the extracted values. Use null for any values that cannot be determined. | ||
| Example response: | ||
| \`\`\`json | ||
| { | ||
| "recipient": "0xaa000b3651bd1e57554ebd7308ca70df7c8c0e8e09d67123cc15c8a8a79342b3", | ||
| "amount": "1" | ||
| } | ||
| \`\`\` | ||
| {{recentMessages}} | ||
| Given the recent messages, extract the following information about the requested token transfer: | ||
| - Recipient wallet address | ||
| - Amount to transfer | ||
| Respond with a JSON markdown block containing only the extracted values.`; | ||
|
|
||
| export default { | ||
| name: "SEND_TOKEN", | ||
| similes: [ | ||
| "TRANSFER_TOKEN", | ||
| "TRANSFER_TOKENS", | ||
| "SEND_TOKENS", | ||
| "SEND_SUI", | ||
| "PAY", | ||
| ], | ||
| validate: async (runtime: IAgentRuntime, message: Memory) => { | ||
| console.log("Validating sui transfer from user:", message.userId); | ||
| //add custom validate logic here | ||
| /* | ||
| const adminIds = runtime.getSetting("ADMIN_USER_IDS")?.split(",") || []; | ||
| //console.log("Admin IDs from settings:", adminIds); | ||
| const isAdmin = adminIds.includes(message.userId); | ||
| if (isAdmin) { | ||
| //console.log(`Authorized transfer from user: ${message.userId}`); | ||
| return true; | ||
| } | ||
| else | ||
| { | ||
| //console.log(`Unauthorized transfer attempt from user: ${message.userId}`); | ||
| return false; | ||
| } | ||
| */ | ||
| return true; | ||
| }, | ||
| description: "Transfer tokens from the agent's wallet to another address", | ||
| handler: async ( | ||
| runtime: IAgentRuntime, | ||
| message: Memory, | ||
| state: State, | ||
| _options: { [key: string]: unknown }, | ||
| callback?: HandlerCallback | ||
| ): Promise<boolean> => { | ||
| elizaLogger.log("Starting SEND_TOKEN handler..."); | ||
|
|
||
| const walletInfo = await walletProvider.get(runtime, message, state); | ||
| state.walletInfo = walletInfo; | ||
|
|
||
| // Initialize or update state | ||
| if (!state) { | ||
| state = (await runtime.composeState(message)) as State; | ||
| } else { | ||
| state = await runtime.updateRecentMessageState(state); | ||
| } | ||
|
|
||
| // Define the schema for the expected output | ||
| const transferSchema = z.object({ | ||
| recipient: z.string(), | ||
| amount: z.union([z.string(), z.number()]), | ||
| }); | ||
|
|
||
| // Compose transfer context | ||
| const transferContext = composeContext({ | ||
| state, | ||
| template: transferTemplate, | ||
| }); | ||
|
|
||
| // Generate transfer content with the schema | ||
| const content = await generateObject({ | ||
| runtime, | ||
| context: transferContext, | ||
| schema: transferSchema, | ||
| modelClass: ModelClass.SMALL, | ||
| }); | ||
|
|
||
| const transferContent = content.object as TransferContent; | ||
|
|
||
| // Validate transfer content | ||
| if (!isTransferContent(transferContent)) { | ||
| console.error("Invalid content for TRANSFER_TOKEN action."); | ||
| if (callback) { | ||
| callback({ | ||
| text: "Unable to process transfer request. Invalid content provided.", | ||
| content: { error: "Invalid transfer content" }, | ||
| }); | ||
| } | ||
| return false; | ||
| } | ||
|
|
||
| try { | ||
| const privateKey = runtime.getSetting("SUI_PRIVATE_KEY"); | ||
| const suiAccount = Ed25519Keypair.deriveKeypair(privateKey); | ||
| const network = runtime.getSetting("SUI_NETWORK"); | ||
| const suiClient = new SuiClient({ | ||
| url: getFullnodeUrl(network as SuiNetwork), | ||
| }); | ||
|
|
||
| const adjustedAmount = BigInt( | ||
| Number(transferContent.amount) * Math.pow(10, SUI_DECIMALS) | ||
| ); | ||
| console.log( | ||
| `Transferring: ${transferContent.amount} tokens (${adjustedAmount} base units)` | ||
| ); | ||
| const tx = new Transaction(); | ||
| const [coin] = tx.splitCoins(tx.gas, [adjustedAmount]); | ||
| tx.transferObjects([coin], transferContent.recipient); | ||
| const executedTransaction = | ||
| await suiClient.signAndExecuteTransaction({ | ||
| signer: suiAccount, | ||
| transaction: tx, | ||
| }); | ||
|
|
||
| console.log("Transfer successful:", executedTransaction.digest); | ||
|
|
||
| if (callback) { | ||
| callback({ | ||
| text: `Successfully transferred ${transferContent.amount} SUI to ${transferContent.recipient}, Transaction: ${executedTransaction.digest}`, | ||
| content: { | ||
| success: true, | ||
| hash: executedTransaction.digest, | ||
| amount: transferContent.amount, | ||
| recipient: transferContent.recipient, | ||
| }, | ||
| }); | ||
| } | ||
|
|
||
| return true; | ||
| } catch (error) { | ||
| console.error("Error during token transfer:", error); | ||
| if (callback) { | ||
| callback({ | ||
| text: `Error transferring tokens: ${error.message}`, | ||
| content: { error: error.message }, | ||
| }); | ||
| } | ||
| return false; | ||
| } | ||
| }, | ||
|
|
||
| examples: [ | ||
| [ | ||
| { | ||
| user: "{{user1}}", | ||
| content: { | ||
| text: "Send 1 SUI tokens to 0x4f2e63be8e7fe287836e29cde6f3d5cbc96eefd0c0e3f3747668faa2ae7324b0", | ||
| }, | ||
| }, | ||
| { | ||
| user: "{{user2}}", | ||
| content: { | ||
| text: "I'll send 1 SUI tokens now...", | ||
| action: "SEND_TOKEN", | ||
| }, | ||
| }, | ||
| { | ||
| user: "{{user2}}", | ||
| content: { | ||
| text: "Successfully sent 1 SUI tokens to 0x4f2e63be8e7fe287836e29cde6f3d5cbc96eefd0c0e3f3747668faa2ae7324b0, Transaction: 0x39a8c432d9bdad993a33cc1faf2e9b58fb7dd940c0425f1d6db3997e4b4b05c0", | ||
| }, | ||
| }, | ||
| ], | ||
| ] as ActionExample[][], | ||
| } as Action; |
Oops, something went wrong.