-
Notifications
You must be signed in to change notification settings - Fork 1.9k
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 #628 from irisdv/feat/support_starkname
feat: support starkname
- Loading branch information
Showing
4 changed files
with
321 additions
and
12 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 |
|---|---|---|
| @@ -0,0 +1,189 @@ | ||
| // It should just transfer subdomain from the root domain owned by the agent's wallet to the recipient. | ||
|
|
||
| import { | ||
| ActionExample, | ||
| HandlerCallback, | ||
| IAgentRuntime, | ||
| Memory, | ||
| ModelClass, | ||
| State, | ||
| type Action, | ||
| composeContext, | ||
| generateObject, | ||
| Content, | ||
| elizaLogger, | ||
| } from "@ai16z/eliza"; | ||
| import { getStarknetAccount } from "../utils"; | ||
| import { validateStarknetConfig } from "../enviroment"; | ||
| import { getTransferSubdomainCall, isStarkDomain } from "../utils/starknetId"; | ||
|
|
||
| export interface SubdomainCreationContent extends Content { | ||
| recipient: string; | ||
| subdomain: string; | ||
| } | ||
|
|
||
| export function isSubdomainCreation( | ||
| content: SubdomainCreationContent | ||
| ): content is SubdomainCreationContent { | ||
| // Validate types | ||
| const validTypes = | ||
| typeof content.recipient === "string" && | ||
| typeof content.subdomain === "string"; | ||
| if (!validTypes) { | ||
| return false; | ||
| } | ||
|
|
||
| // Validate recipient (must be 32-bytes long with 0x prefix) | ||
| const validTokenAddress = | ||
| content.recipient.startsWith("0x") && content.recipient.length === 66; | ||
| if (!validTokenAddress) { | ||
| return false; | ||
| } | ||
|
|
||
| // Validate subdomain | ||
| const validStarkName = | ||
| isStarkDomain(content.subdomain) && | ||
| content.subdomain.split(".").length === 3; | ||
|
|
||
| if (!validStarkName) { | ||
| return false; | ||
| } | ||
| return true; | ||
| } | ||
|
|
||
| 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": "0x1234567890ABCDEF1234567890ABCDEF1234567890ABCDEF1234567890ABCDEF", | ||
| "subdomain": "subdomain.domain.stark", | ||
| } | ||
| \`\`\` | ||
| {{recentMessages}} | ||
| Given the recent messages, extract the following information about the requested subdomain creation: | ||
| - Subdomain to create | ||
| - Recipient wallet address | ||
| Respond with a JSON markdown block containing only the extracted values.`; | ||
|
|
||
| export default { | ||
| name: "CREATE_SUBDOMAIN", | ||
| similes: [ | ||
| "CREATE_SUBDOMAIN_ON_STARKNET", | ||
| "SUBDOMAIN_ON_STARKNET", | ||
| "SUBDOMAIN_CREATION", | ||
| "SEND_SUBDOMAIN_ON_STARKNET", | ||
| ], | ||
| validate: async (runtime: IAgentRuntime, _message: Memory) => { | ||
| await validateStarknetConfig(runtime); | ||
| return true; | ||
| }, | ||
| description: | ||
| "MUST use this action if the user requests create a subdomain, the request might be varied, but it will always be a subdomain creation.", | ||
| handler: async ( | ||
| runtime: IAgentRuntime, | ||
| message: Memory, | ||
| state: State, | ||
| _options: { [key: string]: unknown }, | ||
| callback?: HandlerCallback | ||
| ): Promise<boolean> => { | ||
| elizaLogger.log("Starting CREATE_SUBDOMAIN handler..."); | ||
|
|
||
| // Initialize or update state | ||
| if (!state) { | ||
| state = (await runtime.composeState(message)) as State; | ||
| } else { | ||
| state = await runtime.updateRecentMessageState(state); | ||
| } | ||
|
|
||
| // Compose transfer context | ||
| const transferContext = composeContext({ | ||
| state, | ||
| template: transferTemplate, | ||
| }); | ||
|
|
||
| // Generate transfer content | ||
| const content = await generateObject({ | ||
| runtime, | ||
| context: transferContext, | ||
| modelClass: ModelClass.MEDIUM, | ||
| }); | ||
|
|
||
| elizaLogger.debug("Transfer content:", content); | ||
|
|
||
| // Validate transfer content | ||
| if (!isSubdomainCreation(content)) { | ||
| elizaLogger.error("Invalid content for CREATE_SUBDOMAIN action."); | ||
| if (callback) { | ||
| callback({ | ||
| text: "Not enough information to create subdomain. Please respond with your domain and the subdomain to create.", | ||
| content: { error: "Invalid subdomain creation content" }, | ||
| }); | ||
| } | ||
| return false; | ||
| } | ||
|
|
||
| try { | ||
| const account = getStarknetAccount(runtime); | ||
|
|
||
| const transferCall = getTransferSubdomainCall( | ||
| account.address, | ||
| content.subdomain, | ||
| content.recipient | ||
| ); | ||
|
|
||
| elizaLogger.success( | ||
| "Transferring", | ||
| content.subdomain, | ||
| "to", | ||
| content.recipient | ||
| ); | ||
|
|
||
| const tx = await account.execute(transferCall); | ||
|
|
||
| elizaLogger.success( | ||
| "Transfer completed successfully! tx: " + tx.transaction_hash | ||
| ); | ||
| if (callback) { | ||
| callback({ | ||
| text: | ||
| "Transfer completed successfully! tx: " + | ||
| tx.transaction_hash, | ||
| content: {}, | ||
| }); | ||
| } | ||
|
|
||
| return true; | ||
| } catch (error) { | ||
| elizaLogger.error("Error during subdomain transfer:", error); | ||
| if (callback) { | ||
| callback({ | ||
| text: `Error transferring subdomain ${content.subdomain}: ${error.message}`, | ||
| content: { error: error.message }, | ||
| }); | ||
| } | ||
| return false; | ||
| } | ||
| }, | ||
|
|
||
| examples: [ | ||
| [ | ||
| { | ||
| user: "{{user1}}", | ||
| content: { | ||
| text: "Send me subdomain.domain.stark to 0x0124aeb495b947201f5fac96fd1138e326ad86195b98df6dec9009158a533b49", | ||
| }, | ||
| }, | ||
| { | ||
| user: "{{agent}}", | ||
| content: { | ||
| text: "I'll transfer subdomain.domain.stark to that address right away. Let me process that for you.", | ||
| }, | ||
| }, | ||
| ], | ||
| ] as ActionExample[][], | ||
| } as Action; |
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
Oops, something went wrong.