tensaikit
Version:
An autonomous DeFi AI Agent Kit on Katana enabling AI agents to plan and execute on-chain financial operations.
93 lines (89 loc) • 4.38 kB
JavaScript
;
var __decorate = (this && this.__decorate) || function (decorators, target, key, desc) {
var c = arguments.length, r = c < 3 ? target : desc === null ? desc = Object.getOwnPropertyDescriptor(target, key) : desc, d;
if (typeof Reflect === "object" && typeof Reflect.decorate === "function") r = Reflect.decorate(decorators, target, key, desc);
else for (var i = decorators.length - 1; i >= 0; i--) if (d = decorators[i]) r = (c < 3 ? d(r) : c > 3 ? d(target, key, r) : d(target, key)) || r;
return c > 3 && r && Object.defineProperty(target, key, r), r;
};
var __metadata = (this && this.__metadata) || function (k, v) {
if (typeof Reflect === "object" && typeof Reflect.metadata === "function") return Reflect.metadata(k, v);
};
var __importDefault = (this && this.__importDefault) || function (mod) {
return (mod && mod.__esModule) ? mod : { "default": mod };
};
Object.defineProperty(exports, "__esModule", { value: true });
exports.SushiSwapQuoteActions = void 0;
const zod_1 = __importDefault(require("zod"));
const actionProvider_1 = require("../../actionProvider");
const actionDecorator_1 = require("../../actionDecorator");
const walletProviders_1 = require("../../../walletProviders");
const errors_1 = require("../../../common/errors");
const schemas_1 = require("../schemas");
const utlis_1 = require("../utlis");
const logic_1 = require("../logic");
const utils_1 = require("../../../common/utils");
/**
* SushiSwapQuoteActions provides functionality to generate swap quotes from SushiSwap.
*
* This action group handles fetching and formatting quote data for token swaps on EVM-compatible chains.
* It interacts with the SushiSwap public quote API and prepares data required for downstream swap execution.
*
* Responsibilities:
* - Convert human-readable token amounts into base units using token metadata
* - Handle optional slippage settings
* - Fetch swap quote including expected output, route, and calldata
*
*/
class SushiSwapQuoteActions extends actionProvider_1.ActionProvider {
constructor() {
super("sushi_swap.quote", []);
this.supportsNetwork = (network) => network.protocolFamily === "evm";
}
/**
* Fetches a swap quote for a given token pair and amount from SushiSwap’s public API.
*
* @param walletProvider - An instance of EVM-compatible WalletProvider
* @param args - Object containing tokenIn, tokenOut, amount, and optional maxSlippage
* @returns A formatted string representing the swap quote
* @throws If token metadata is invalid or the API request fails
*/
async getSwapQuote(walletProvider, args) {
try {
const { tokenIn, tokenOut, amount, maxSlippage = utlis_1.DEFAULT_MAX_SLIPPAGE, } = args;
const chainId = walletProvider.getNetwork().chainId;
if (!chainId) {
throw (0, errors_1.createError)("Invalid or missing chain ID", errors_1.ErrorCode.INVALID_NETWORK);
}
const response = await (0, logic_1.fetchSwapQuote)(chainId, {
tokenIn,
tokenOut,
amount,
maxSlippage,
});
return (0, utils_1.wrapAndStringify)("sushi_swap.quote.get_token_price", response);
}
catch (error) {
throw (0, errors_1.handleError)("Failed to generate swap quote", error);
}
}
}
exports.SushiSwapQuoteActions = SushiSwapQuoteActions;
__decorate([
(0, actionDecorator_1.CreateAction)({
name: "get_swap_quote",
description: `
Generates a swap quote for a given token pair and amount on a specific chain.
Inputs:
- tokenIn: Address of the input token.
- tokenOut: Address of the output token.
- amount: Amount of the input token to swap. Example: 3.5 (will be converted to base units using token decimals)
- maxSlippage: Optional parameters for route planning. Maximum allowed slippage for the swap is 1, e.g., 0.005 for 0.5%. Default: 0.005 for 0.5%
Notes:
- Ensure all addresses are checksummed and valid.
`,
schema: schemas_1.GetSwapQuoteSchema,
}),
__metadata("design:type", Function),
__metadata("design:paramtypes", [walletProviders_1.EvmWalletProvider, void 0]),
__metadata("design:returntype", Promise)
], SushiSwapQuoteActions.prototype, "getSwapQuote", null);