@curvefi/llamalend-api 1.0.42 → 1.1.1

package/lib/llamalend.js

@@ -488,6 +488,7 @@ class Llamalend {
             this.setContract(this.constants.ALIASES['one_way_factory'], OneWayLendingFactoryABI);
             this.setContract(this.constants.ALIASES['gauge_controller'], GaugeControllerABI);
             this.setContract(this.constants.ALIASES['leverage_zap'], LeverageZapABI);
+            this.setContract(this.constants.ALIASES['leverage_zap_v2'], LeverageZapABI);
             if (this.chainId === 1) {
                 this.setContract(this.constants.ALIASES.minter, MinterABI);
                 this.setContract(this.constants.ALIASES.gauge_factory, GaugeFactoryMainnetABI);

package/lib/utils.d.ts

@@ -58,3 +58,4 @@ export declare const getLsdApy: ((name: "wstETH" | "sfrxETH") => Promise<{
     apyMean30d: number;
 }>>;
 export declare const calculateFutureLeverage: (currentCollateral: number | string, totalDepositFromUser: number | string, collateral: number | string, operation: "add" | "remove") => string;
+export declare const buildCalldataForLeverageZapV2: (routerAddress: string, exchangeCalldata: string) => string;

package/lib/utils.js

@@ -474,3 +474,8 @@ export const calculateFutureLeverage = (currentCollateral, totalDepositFromUser,
         return currentCollateralBN.minus(collateralBN).div(totalDepositBN.minus(collateralBN)).toString();
     }
 };
+export const buildCalldataForLeverageZapV2 = (routerAddress, exchangeCalldata) => {
+    const cleanCalldata = exchangeCalldata.startsWith('0x') ? exchangeCalldata.slice(2) : exchangeCalldata;
+    const abiCoder = ethers.AbiCoder.defaultAbiCoder();
+    return abiCoder.encode(['address', 'bytes'], [routerAddress, '0x' + cleanCalldata]);
+};

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@curvefi/llamalend-api",
-  "version": "1.0.42",
+  "version": "1.1.1",
   "description": "JavaScript library for Curve Lending",
   "main": "lib/index.js",
   "author": "Macket",
@@ -16,6 +16,7 @@
   "scripts": {
     "build": "rm -rf lib && tsc --project tsconfig.build.json",
     "lint": "eslint src --ext .ts",
+    "lint:fix": "eslint src --ext .ts --fix",
     "watch": "tsc --watch",
     "watch:lib": "rm -rf lib && tsc --watch --project tsconfig.build.json"
   },

package/src/cache/index.ts

@@ -1,5 +1,5 @@
 class Cache {
-    // eslint-disable-next-line no-use-before-define
+     
     private static instance: Cache;
     readonly cache: Map<string, any>;
 

package/src/constants/aliases.ts

@@ -9,6 +9,7 @@ export const ALIASES_ETHEREUM = lowerCaseValues({
     "gauge_factory": "0xabC000d88f23Bb45525E447528DBF656A9D55bf5",
     // "leverage_zap": "0x3294514B78Df4Bb90132567fcf8E5e99f390B687", // 1inch
     "leverage_zap": "0xC5898606BdB494a994578453B92e7910a90aA873", // odos
+    "leverage_zap_v2": "0x0000000000000000000000000000000000000000",
     "leverage_markets_start_id": "9",
     "crvUSD": "0xf939E0A03FB07F59A73314E73794Be0E57ac1b4E",
     "st_crvUSD": "0x0655977FEb2f289A4aB78af67BAB0d17aAb84367",
@@ -20,6 +21,7 @@ export const ALIASES_POLYGON = lowerCaseValues({
     "gauge_controller": "0x2F50D538606Fa9EDD2B11E2446BEb18C9D5846bB",
     "gauge_factory": "0xabC000d88f23Bb45525E447528DBF656A9D55bf5",
     "leverage_zap": "0x0000000000000000000000000000000000000000",
+    "leverage_zap_v2": "0x0000000000000000000000000000000000000000",
 });
 
 export const ALIASES_FANTOM = lowerCaseValues({
@@ -28,6 +30,7 @@ export const ALIASES_FANTOM = lowerCaseValues({
     "gauge_controller": "0x2F50D538606Fa9EDD2B11E2446BEb18C9D5846bB",
     "gauge_factory": "0xabC000d88f23Bb45525E447528DBF656A9D55bf5",
     "leverage_zap": "0x0000000000000000000000000000000000000000",
+    "leverage_zap_v2": "0x0000000000000000000000000000000000000000",
 });
 
 export const ALIASES_AVALANCHE = lowerCaseValues({
@@ -36,6 +39,7 @@ export const ALIASES_AVALANCHE = lowerCaseValues({
     "gauge_controller": "0x2F50D538606Fa9EDD2B11E2446BEb18C9D5846bB",
     "gauge_factory": "0xabC000d88f23Bb45525E447528DBF656A9D55bf5",
     "leverage_zap": "0x0000000000000000000000000000000000000000",
+    "leverage_zap_v2": "0x0000000000000000000000000000000000000000",
 });
 
 export const ALIASES_ARBITRUM = lowerCaseValues({
@@ -45,6 +49,7 @@ export const ALIASES_ARBITRUM = lowerCaseValues({
     "gauge_factory": "0xabC000d88f23Bb45525E447528DBF656A9D55bf5",
     // "leverage_zap": "0x61C404B60ee9c5fB09F70F9A645DD38fE5b3A956", // 1inch
     "leverage_zap": "0xb7b240CFa985306563A301bC417Bc9715059a117", // odos
+    "leverage_zap_v2": "0x5b07Db9a85992c877b9fBeA6DCC4F79292577640",
     "leverage_markets_start_id": "9",
 });
 
@@ -55,6 +60,7 @@ export const ALIASES_OPTIMISM = lowerCaseValues({
     "gauge_factory_old": "0xabC000d88f23Bb45525E447528DBF656A9D55bf5",
     "gauge_factory": "0x871fBD4E01012e2E8457346059e8C189d664DbA4",
     "leverage_zap": "0x273e44B9a1841857d9360e8792bB59f9e1FfE9Da",
+    "leverage_zap_v2": "0x0000000000000000000000000000000000000000",
     "leverage_markets_start_id": "0",
     "gas_oracle": '0xc0d3C0d3C0d3c0D3C0D3C0d3C0d3C0D3C0D3000f',
     "gas_oracle_blob": '0x420000000000000000000000000000000000000f',
@@ -66,6 +72,7 @@ export const ALIASES_XDAI = lowerCaseValues({
     "gauge_controller": "0x2F50D538606Fa9EDD2B11E2446BEb18C9D5846bB",
     "gauge_factory": "0xabC000d88f23Bb45525E447528DBF656A9D55bf5",
     "leverage_zap": "0x0000000000000000000000000000000000000000",
+    "leverage_zap_v2": "0x0000000000000000000000000000000000000000",
 });
 
 export const ALIASES_MOONBEAM = lowerCaseValues({
@@ -74,6 +81,7 @@ export const ALIASES_MOONBEAM = lowerCaseValues({
     "gauge_controller": "0x2F50D538606Fa9EDD2B11E2446BEb18C9D5846bB",
     "gauge_factory": "0xabC000d88f23Bb45525E447528DBF656A9D55bf5",
     "leverage_zap": "0x0000000000000000000000000000000000000000",
+    "leverage_zap_v2": "0x0000000000000000000000000000000000000000",
 });
 
 export const ALIASES_AURORA = lowerCaseValues({
@@ -89,6 +97,7 @@ export const ALIASES_KAVA = lowerCaseValues({
     "gauge_controller": "0x2F50D538606Fa9EDD2B11E2446BEb18C9D5846bB",
     "gauge_factory": "0xabC000d88f23Bb45525E447528DBF656A9D55bf5",
     "leverage_zap": "0x0000000000000000000000000000000000000000",
+    "leverage_zap_v2": "0x0000000000000000000000000000000000000000",
 });
 
 export const ALIASES_CELO = lowerCaseValues({
@@ -97,6 +106,7 @@ export const ALIASES_CELO = lowerCaseValues({
     "gauge_controller": "0x2F50D538606Fa9EDD2B11E2446BEb18C9D5846bB",
     "gauge_factory": "0xabC000d88f23Bb45525E447528DBF656A9D55bf5",
     "leverage_zap": "0x0000000000000000000000000000000000000000",
+    "leverage_zap_v2": "0x0000000000000000000000000000000000000000",
 });
 
 export const ALIASES_ZKSYNC = lowerCaseValues({
@@ -105,6 +115,7 @@ export const ALIASES_ZKSYNC = lowerCaseValues({
     "gauge_controller": "0x2F50D538606Fa9EDD2B11E2446BEb18C9D5846bB", // <--- TODO CHANGE
     "gauge_factory": "0x0000000000000000000000000000000000000000",
     "leverage_zap": "0x0000000000000000000000000000000000000000",
+    "leverage_zap_v2": "0x0000000000000000000000000000000000000000",
 });
 
 export const ALIASES_BASE = lowerCaseValues({
@@ -113,6 +124,7 @@ export const ALIASES_BASE = lowerCaseValues({
     "gauge_controller": "0x2F50D538606Fa9EDD2B11E2446BEb18C9D5846bB", // <--- TODO CHANGE
     "gauge_factory": "0xabC000d88f23Bb45525E447528DBF656A9D55bf5",
     "leverage_zap": "0x0000000000000000000000000000000000000000",
+    "leverage_zap_v2": "0x0000000000000000000000000000000000000000",
 });
 
 export const ALIASES_BSC = lowerCaseValues({
@@ -121,6 +133,7 @@ export const ALIASES_BSC = lowerCaseValues({
     "gauge_controller": "0x0000000000000000000000000000000000000000", // <--- TODO CHANGE
     "gauge_factory": "0xabC000d88f23Bb45525E447528DBF656A9D55bf5",
     "leverage_zap": "0x0000000000000000000000000000000000000000",
+    "leverage_zap_v2": "0x0000000000000000000000000000000000000000",
 });
 
 export const ALIASES_FRAXTAL = lowerCaseValues({
@@ -130,6 +143,7 @@ export const ALIASES_FRAXTAL = lowerCaseValues({
     "gauge_factory_old": "0xeF672bD94913CB6f1d2812a6e18c1fFdEd8eFf5c",
     "gauge_factory": "0x0b8d6b6cefc7aa1c2852442e518443b1b22e1c52",
     "leverage_zap": "0x37c5ab57AF7100Bdc9B668d766e193CCbF6614FD", // odos
+    "leverage_zap_v2": "0x0000000000000000000000000000000000000000",
     "leverage_markets_start_id": "0",
 });
 
@@ -139,5 +153,6 @@ export const ALIASES_SONIC = lowerCaseValues({
     "gauge_controller": "0x2F50D538606Fa9EDD2B11E2446BEb18C9D5846bB",
     "gauge_factory": "0xf3A431008396df8A8b2DF492C913706BDB0874ef",
     "leverage_zap": "0x5552b631e2aD801fAa129Aacf4B701071cC9D1f7", // odos
+    "leverage_zap_v2": "0x0000000000000000000000000000000000000000",
     "leverage_markets_start_id": "0",
 });

package/src/external-api.ts

@@ -118,9 +118,9 @@ export const _getUserCollateral = memoize(
 )
 
 export const _getUserCollateralForce = async (
-    network: INetworkName, 
-    controller: string, 
-    user: string, 
+    network: INetworkName,
+    controller: string,
+    user: string,
     newTx: string
 ): Promise<void> => {
     await fetch(`https://prices.curve.finance/v1/lending/collateral_events/${network}/${controller}/${user}?new_tx=${newTx}`);
@@ -177,7 +177,7 @@ export async function _getQuoteOdos(this: Llamalend, fromToken: string, toToken:
     if (ethers.getAddress(fromToken) == "0xEeeeeEeeeEeEeeEeEeEeeEEEeeeeEeeeeeeeEEeE") fromToken = "0x0000000000000000000000000000000000000000";
     if (ethers.getAddress(toToken) == "0xEeeeeEeeeEeEeeEeEeEeeEEEeeeeEeeeeeeeEEeE") toToken = "0x0000000000000000000000000000000000000000";
 
-    const url = `https://prices.curve.finance/odos/quote?chain_id=${this.chainId}&from_address=${ethers.getAddress(fromToken)}` +
+    const url = `https://prices.curve.finance/odos/v3/quote?chain_id=${this.chainId}&from_address=${ethers.getAddress(fromToken)}` +
         `&to_address=${ethers.getAddress(toToken)}&amount=${_amount.toString()}&slippage=${slippage}&pathVizImage=${pathVizImage}` +
         `&caller_address=${ethers.getAddress(this.constants.ALIASES.leverage_zap)}&blacklist=${ethers.getAddress(blacklist)}`;
 
@@ -234,4 +234,4 @@ async function fetchJson(url: string): Promise<any> {
 async function fetchData(url: string) {
     const {data} = await fetchJson(url);
     return data;
-}
+}

package/src/interfaces.ts

@@ -230,3 +230,22 @@ export interface ILlamma {
 export interface IResponseApi {
     data: any
 }
+
+export interface IQuote {
+    outAmount: string,
+    priceImpact: number
+}
+
+export interface ILeverageMetrics {
+    priceImpact: number,
+    bands: [number, number],
+    prices: string[],
+    health: string,
+}
+
+export type GetExpectedFn = (
+    fromToken: string,
+    toToken: string,
+    amountIn: bigint,
+    blacklist: string | string[],
+) => Promise<IQuote>;

package/src/lendMarkets/LendMarketTemplate.ts

@@ -27,6 +27,8 @@ import {IDict, TGas, TAmount, IReward, IQuoteOdos, IOneWayMarket, IPartialFrac}
 import { _getExpectedOdos, _getQuoteOdos, _assembleTxOdos, _getUserCollateral, _getUserCollateralForce, _getMarketsData } from "../external-api.js";
 import ERC20Abi from '../constants/abis/ERC20.json' with {type: 'json'};
 import {cacheKey, cacheStats} from "../cache/index.js";
+import {ILeverageZapV2} from "./interfaces/leverageZapV2.js";
+import {LeverageZapV2Module} from "./modules/leverageZapV2.js";
 
 
 const DAY = 86400;
@@ -237,6 +239,7 @@ export class LendMarketTemplate {
             repay: (stateCollateral: TAmount, userCollateral: TAmount, userBorrowed: TAmount, slippage?: number) => Promise<number>,
         }
     };
+    leverageZapV2: ILeverageZapV2;
 
     constructor(id: string, marketData: IOneWayMarket, llamalend: Llamalend) {
         this.llamalend = llamalend;
@@ -387,6 +390,51 @@ export class LendMarketTemplate {
             },
         }
 
+        const leverageZapV2 = new LeverageZapV2Module(this);
+
+        this.leverageZapV2 = {
+            hasLeverage: leverageZapV2.hasLeverage.bind(leverageZapV2),
+
+            maxLeverage: leverageZapV2.maxLeverage.bind(leverageZapV2),
+
+            createLoanMaxRecv: leverageZapV2.leverageCreateLoanMaxRecv.bind(leverageZapV2),
+            createLoanMaxRecvAllRanges: leverageZapV2.leverageCreateLoanMaxRecvAllRanges.bind(leverageZapV2),
+            createLoanExpectedCollateral: leverageZapV2.leverageCreateLoanExpectedCollateral.bind(leverageZapV2),
+            createLoanMaxRange: leverageZapV2.leverageCreateLoanMaxRange.bind(leverageZapV2),
+            createLoanBandsAllRanges: leverageZapV2.leverageCreateLoanBandsAllRanges.bind(leverageZapV2),
+            createLoanPricesAllRanges: leverageZapV2.leverageCreateLoanPricesAllRanges.bind(leverageZapV2),
+            createLoanIsApproved: leverageZapV2.leverageCreateLoanIsApproved.bind(leverageZapV2),
+            createLoanApprove: leverageZapV2.leverageCreateLoanApprove.bind(leverageZapV2),
+            createLoanExpectedMetrics: leverageZapV2.leverageCreateLoanExpectedMetrics.bind(leverageZapV2),
+            createLoan: leverageZapV2.leverageCreateLoan.bind(leverageZapV2),
+
+            borrowMoreMaxRecv: leverageZapV2.leverageBorrowMoreMaxRecv.bind(leverageZapV2),
+            borrowMoreExpectedCollateral: leverageZapV2.leverageBorrowMoreExpectedCollateral.bind(leverageZapV2),
+            borrowMoreIsApproved: leverageZapV2.leverageCreateLoanIsApproved.bind(leverageZapV2),
+            borrowMoreApprove: leverageZapV2.leverageCreateLoanApprove.bind(leverageZapV2),
+            borrowMoreExpectedMetrics: leverageZapV2.leverageBorrowMoreExpectedMetrics.bind(leverageZapV2),
+            borrowMore: leverageZapV2.leverageBorrowMore.bind(leverageZapV2),
+
+            repayExpectedBorrowed: leverageZapV2.leverageRepayExpectedBorrowed.bind(leverageZapV2),
+            repayIsFull: leverageZapV2.leverageRepayIsFull.bind(leverageZapV2),
+            repayIsAvailable: leverageZapV2.leverageRepayIsAvailable.bind(leverageZapV2),
+            repayExpectedMetrics: leverageZapV2.leverageRepayExpectedMetrics.bind(leverageZapV2),
+            repayIsApproved: leverageZapV2.leverageRepayIsApproved.bind(leverageZapV2),
+            repayApprove: leverageZapV2.leverageRepayApprove.bind(leverageZapV2),
+            repay: leverageZapV2.leverageRepay.bind(leverageZapV2),
+
+            estimateGas: {
+                createLoanApprove: leverageZapV2.leverageCreateLoanApproveEstimateGas.bind(leverageZapV2),
+                createLoan: leverageZapV2.leverageCreateLoanEstimateGas.bind(leverageZapV2),
+
+                borrowMoreApprove: leverageZapV2.leverageCreateLoanApproveEstimateGas.bind(leverageZapV2),
+                borrowMore: leverageZapV2.leverageBorrowMoreEstimateGas.bind(leverageZapV2),
+
+                repayApprove: leverageZapV2.leverageRepayApproveEstimateGas.bind(leverageZapV2),
+                repay: leverageZapV2.leverageRepayEstimateGas.bind(leverageZapV2),
+            },
+        }
+
     }
 
     private _getMarketId = (): number => Number(this.id.split("-").slice(-1)[0]);
@@ -807,8 +855,7 @@ export class LendMarketTemplate {
         }
 
         return tokens.map((token, i) => ({ token, symbol: tokenInfo[i * 2] as string, decimals: Number(tokenInfo[(i * 2) + 1]) }));
-    },
-    {
+    }, {
         promise: true,
         maxAge: 30 * 60 * 1000, // 30m
     });
@@ -942,8 +989,7 @@ export class LendMarketTemplate {
             .map((_x) => formatUnits(_x * BigInt(100)));
 
         return { fee, admin_fee, liquidation_discount, loan_discount, base_price, A }
-    },
-    {
+    }, {
         promise: true,
         maxAge: 5 * 60 * 1000, // 5m
     });
@@ -1404,7 +1450,7 @@ export class LendMarketTemplate {
 
     // ---------------- CREATE LOAN ----------------
 
-    private _checkRange(range: number): void {
+    public _checkRange(range: number): void {
         if (range < this.minBands) throw Error(`range must be >= ${this.minBands}`);
         if (range > this.maxBands) throw Error(`range must be <= ${this.maxBands}`);
     }
@@ -1460,7 +1506,7 @@ export class LendMarketTemplate {
         return await this.llamalend.multicallProvider.all(calls) as bigint[];
     }
 
-    private async _getPrices(_n2: bigint, _n1: bigint): Promise<string[]> {
+    public async _getPrices(_n2: bigint, _n1: bigint): Promise<string[]> {
         const contract = this.llamalend.contracts[this.addresses.amm].multicallContract;
         return (await this.llamalend.multicallProvider.all([
             contract.p_oracle_down(_n2),
@@ -1468,7 +1514,7 @@ export class LendMarketTemplate {
         ]) as bigint[]).map((_p) => formatUnits(_p));
     }
 
-    private async _calcPrices(_n2: bigint, _n1: bigint): Promise<[string, string]> {
+    public async _calcPrices(_n2: bigint, _n1: bigint): Promise<[string, string]> {
         return [await this.calcTickPrice(Number(_n2) + 1), await this.calcTickPrice(Number(_n1))];
     }
 
@@ -2078,13 +2124,13 @@ export class LendMarketTemplate {
     public async calcPartialFrac(amount: TAmount, address = ""): Promise<IPartialFrac> {
         address = _getAddress.call(this.llamalend, address);
         const tokensToLiquidate = await this.tokensToLiquidate(address);
-        
+
         const amountBN = BN(amount);
         const tokensToLiquidateBN = BN(tokensToLiquidate);
-        
+
         if (amountBN.gt(tokensToLiquidateBN)) throw Error("Amount cannot be greater than total tokens to liquidate");
         if (amountBN.lte(0)) throw Error("Amount must be greater than 0");
-        
+
         // Calculate frac = amount / tokensToLiquidate * 10**18
         // 100% = 10**18
         const fracDecimalBN = amountBN.div(tokensToLiquidateBN);
@@ -2136,33 +2182,33 @@ export class LendMarketTemplate {
         if (slippage > 100) throw Error("Slippage must be <= 100");
         if (Number(currentDebt) === 0) throw Error(`Loan for ${address} does not exist`);
         if (Number(borrowed) === 0) throw Error(`User ${address} is not in liquidation mode`);
-        
+
         const frac = partialFrac.frac;
         const fracBN = BN(partialFrac.fracDecimal);
-        
+
         const borrowedBN = BN(borrowed);
         const expectedBorrowedBN = borrowedBN.times(fracBN);
         const minAmountBN = expectedBorrowedBN.times(100 - slippage).div(100);
         const _minAmount = fromBN(minAmountBN);
-        
+
         const contract = this.llamalend.contracts[this.addresses.controller].contract;
         const gas = (await contract.liquidate_extended.estimateGas(
-            address, 
-            _minAmount, 
-            frac, 
+            address,
+            _minAmount,
+            frac,
             this.llamalend.constants.ZERO_ADDRESS,
             [],
             this.llamalend.constantOptions
         ));
-        
+
         if (estimateGas) return smartNumber(gas);
 
         await this.llamalend.updateFeeData();
         const gasLimit = _mulBy1_3(DIGas(gas));
         return (await contract.liquidate_extended(
-            address, 
-            _minAmount, 
-            frac, 
+            address,
+            _minAmount,
+            frac,
             this.llamalend.constants.ZERO_ADDRESS,
             [],
             { ...this.llamalend.options, gasLimit }
@@ -2376,14 +2422,14 @@ export class LendMarketTemplate {
         const userEffectiveCollateralBN = BN(userCollateral).plus(BN(userBorrowed).div(pAvgBN as BigNumber));
 
         const res: IDict<{
-            maxDebt: string,
-            maxTotalCollateral: string,
-            userCollateral: string,
-            collateralFromUserBorrowed: string,
-            collateralFromMaxDebt: string,
-            maxLeverage: string,
-            avgPrice: string,
-        }> = {};
+                maxDebt: string,
+                maxTotalCollateral: string,
+                userCollateral: string,
+                collateralFromUserBorrowed: string,
+                collateralFromMaxDebt: string,
+                maxLeverage: string,
+                avgPrice: string,
+            }> = {};
         for (let N = this.minBands; N <= this.maxBands; N++) {
             const j = N - this.minBands;
             res[N] = {
@@ -2783,7 +2829,7 @@ export class LendMarketTemplate {
         const _userBorrowed = parseUnits(userBorrowed, this.borrowed_token.decimals);
         await this._setSwapDataToCache(this.addresses.borrowed_token, this.addresses.collateral_token, _dDebt + _userBorrowed, slippage);
         const { _totalCollateral, _userCollateral, _collateralFromUserBorrowed, _collateralFromDebt, avgPrice } =
-                await this._leverageExpectedCollateral(userCollateral, userBorrowed, dDebt, address);
+            await this._leverageExpectedCollateral(userCollateral, userBorrowed, dDebt, address);
         return {
             totalCollateral: formatUnits(_totalCollateral, this.collateral_token.decimals),
             userCollateral: formatUnits(_userCollateral, this.collateral_token.decimals),
@@ -3072,7 +3118,6 @@ export class LendMarketTemplate {
             calldata = await _assembleTxOdos.call(this.llamalend, swapData.pathId as string);
         }
 
-        console.log('params', [0, parseUnits(this._getMarketId(), 0), _userCollateral, _userBorrowed], calldata)
         const contract = this.llamalend.contracts[this.addresses.controller].contract;
         const gas = await contract.repay_extended.estimateGas(
             this.llamalend.constants.ALIASES.leverage_zap,
@@ -3189,7 +3234,7 @@ export class LendMarketTemplate {
     public async forceUpdateUserState(newTx: string, userAddress?: string): Promise<void> {
         const address = userAddress || this.llamalend.signerAddress;
         if (!address) throw Error("Need to connect wallet or pass address into args");
-        
+
         await _getUserCollateralForce(
             this.llamalend.constants.NETWORK_NAME,
             this.addresses.controller,
@@ -3197,4 +3242,8 @@ export class LendMarketTemplate {
             newTx
         );
     }
-}
+
+    public getLlamalend(): Llamalend {
+        return this.llamalend;
+    }
+}