d0zingcat/solana-agent-kit
1
0
Fork 0
mirror of https://github.com/d0zingcat/solana-agent-kit.git synced 2026-05-24 23:26:44 +00:00
Code Issues Packages Projects Releases Wiki Activity

feat: test added

Browse Source
This commit is contained in:
aryan
2024-11-18 00:38:00 +07:00
parent 8d355378e0
commit 655c131b99
4 changed files with 71 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

16
src/langchain/index.ts
View File

@@ -166,6 +166,19 @@ export class SolanaRegisterDomainTool extends Tool {
}
}
export class SolanaGetWalletAddressTool extends Tool {
name = "solana_get_wallet_address";
description = "Get the wallet address of the agent";
constructor(private solanaKit: SolanaAgentKit) {
super();
}
async _call(_input: string): Promise<string> {
return this.solanaKit.wallet_address.toString();
}
}
// Updated createSolanaTools function
export function createSolanaTools(solanaKit: SolanaAgentKit) {
return [
@@ -177,5 +190,6 @@ export function createSolanaTools(solanaKit: SolanaAgentKit) {
new SolanaTradeTool(solanaKit),
new SolanaRequestFundsTool(solanaKit),
new SolanaRegisterDomainTool(solanaKit),
new SolanaGetWalletAddressTool(solanaKit),
];
}
}