d0zingcat/solana-agent-kit
1
0
Fork 0
mirror of https://github.com/d0zingcat/solana-agent-kit.git synced 2026-06-07 15:11:47 +00:00
Code Issues Packages Projects Releases Wiki Activity

fix: token address prompt

Browse Source
This commit is contained in:
aryan
2024-12-19 05:36:03 +05:30
parent 28a7952797
commit b078cc3b37
3 changed files with 68 additions and 49 deletions
Download Patch File Download Diff File Expand all files Collapse all files

30
src/langchain/index.ts
View File

@@ -572,39 +572,21 @@ export class SolanaTPSCalculatorTool extends Tool {
export class SolanaTokenDataTool extends Tool {
name = "solana_token_data";
description = `Get the token data for a given token mint address, token name or symbol.
description = `Get the token data for a given token mint address
Inputs: Either one of mintAddress, tokenName or symbol is required.
mintAddress: string, eg "So11111111111111111111111111111111111111112" (optional)
tokenName: string, eg "USD Coin" (optional)
symbol: string, eg "USDC" (optional)`;
Inputs: mintAddress is required.
mintAddress: string, eg "So11111111111111111111111111111111111111112" (required)`;
constructor(private solanaKit: SolanaAgentKit) {
super();
}
private validateInput(input: any): void {
if (!input.mintAddress && !input.tokenName && !input.symbol) {
throw new Error("Either mintAddress, tokenName or symbol is required");
}
if (
input.mintAddress &&
typeof input.mintAddress !== "string" &&
!PublicKey.isOnCurve(input.mintAddress)
) {
throw new Error("mintAddress must be a valid base58 string");
}
}
protected async _call(input: string): Promise<string> {
try {
const parsedInput = JSON.parse(input);
this.validateInput(parsedInput);
const parsedInput = input.trim();
const tokenData = await this.solanaKit.getTokenData(
parsedInput.tokenName,
parsedInput.symbol,
new PublicKey(parsedInput.mintAddress),
const tokenData = await this.solanaKit.getTokenDataByAddress(
parsedInput
);
return JSON.stringify({