@drift-labs/sdk 2.85.0-beta.1 → 2.85.0-beta.2

package/VERSION

@@ -1 +1 @@
-2.85.0-beta.1
+2.85.0-beta.2

package/lib/driftClient.js

@@ -2544,7 +2544,7 @@ class DriftClient {
             };
             if (shouldUseSimulationComputeUnits || shouldExitIfSimulationFails) {
                 const placeAndTakeTxToSim = (await this.buildTransaction(placeAndTakeIxs, txParams, undefined, undefined, true, recentBlockHash));
-                const simulationResult = await txParamProcessor_1.TransactionParamProcessor.getTxSimComputeUnits(placeAndTakeTxToSim, this.connection, (_a = txParams.computeUnitsBufferMultiplier) !== null && _a !== void 0 ? _a : 1.2);
+                const simulationResult = await txParamProcessor_1.TransactionParamProcessor.getTxSimComputeUnits(placeAndTakeTxToSim, this.connection, (_a = txParams.computeUnitsBufferMultiplier) !== null && _a !== void 0 ? _a : 1.2, txParams.lowerBoundCu);
                 if (shouldExitIfSimulationFails && !simulationResult.success) {
                     earlyExitFailedPlaceAndTakeSim = true;
                     return;

package/lib/tx/txParamProcessor.d.ts

@@ -8,7 +8,8 @@ type TransactionBuildingProps = {
  */
 export declare class TransactionParamProcessor {
     private static getComputeUnitsFromSim;
-    static getTxSimComputeUnits(tx: VersionedTransaction, connection: Connection, bufferMultiplier: number): Promise<{
+    static getTxSimComputeUnits(tx: VersionedTransaction, connection: Connection, bufferMultiplier: number, // Making this a mandatory param to force the user to remember that simulated CU's can be inaccurate and a buffer should be applied
+    lowerBoundCu?: number): Promise<{
         success: boolean;
         computeUnits: number;
     }>;

package/lib/tx/txParamProcessor.js

@@ -15,8 +15,8 @@ class TransactionParamProcessor {
         }
         return undefined;
     }
-    static async getTxSimComputeUnits(tx, connection, bufferMultiplier // Making this a mandatory param to force the user to remember that simulated CU's can be inaccurate and a buffer should be applied
-    ) {
+    static async getTxSimComputeUnits(tx, connection, bufferMultiplier, // Making this a mandatory param to force the user to remember that simulated CU's can be inaccurate and a buffer should be applied
+    lowerBoundCu) {
         var _a, _b, _c;
         try {
             if (TEST_SIMS_ALWAYS_FAIL)
@@ -29,7 +29,11 @@ class TransactionParamProcessor {
             }
             const computeUnits = await this.getComputeUnitsFromSim(simTxResult);
             // Apply the buffer, but round down to the MAX_COMPUTE_UNITS, and round up to the nearest whole number
-            const bufferedComputeUnits = Math.ceil(Math.min(computeUnits * bufferMultiplier, MAX_COMPUTE_UNITS));
+            let bufferedComputeUnits = Math.ceil(Math.min(computeUnits * bufferMultiplier, MAX_COMPUTE_UNITS));
+            // If a lower bound CU is passed then enforce it
+            if (lowerBoundCu) {
+                bufferedComputeUnits = Math.max(bufferedComputeUnits, Math.min(lowerBoundCu, MAX_COMPUTE_UNITS));
+            }
             return {
                 success: true,
                 computeUnits: bufferedComputeUnits,

package/lib/types.d.ts

@@ -1085,6 +1085,7 @@ export type ProcessingTxParams = {
     computeUnitsBufferMultiplier?: number;
     useSimulatedComputeUnitsForCUPriceCalculation?: boolean;
     getCUPriceFromComputeUnits?: (computeUnits: number) => number;
+    lowerBoundCu?: number;
 };
 export type TxParams = BaseTxParams & ProcessingTxParams;
 export declare class SwapReduceOnly {

package/package.json

@@ -1,6 +1,6 @@
 {
 	"name": "@drift-labs/sdk",
-	"version": "2.85.0-beta.1",
+	"version": "2.85.0-beta.2",
 	"main": "lib/index.js",
 	"types": "lib/index.d.ts",
 	"author": "crispheaney",

package/src/driftClient.ts

@@ -4664,7 +4664,8 @@ export class DriftClient {
 					await TransactionParamProcessor.getTxSimComputeUnits(
 						placeAndTakeTxToSim,
 						this.connection,
-						txParams.computeUnitsBufferMultiplier ?? 1.2
+						txParams.computeUnitsBufferMultiplier ?? 1.2,
+						txParams.lowerBoundCu
 					);
 
 				if (shouldExitIfSimulationFails && !simulationResult.success) {

package/src/tx/txParamProcessor.ts

@@ -32,7 +32,8 @@ export class TransactionParamProcessor {
 	public static async getTxSimComputeUnits(
 		tx: VersionedTransaction,
 		connection: Connection,
-		bufferMultiplier: number // Making this a mandatory param to force the user to remember that simulated CU's can be inaccurate and a buffer should be applied
+		bufferMultiplier: number, // Making this a mandatory param to force the user to remember that simulated CU's can be inaccurate and a buffer should be applied
+		lowerBoundCu?: number
 	): Promise<{ success: boolean; computeUnits: number }> {
 		try {
 			if (TEST_SIMS_ALWAYS_FAIL)
@@ -49,10 +50,18 @@ export class TransactionParamProcessor {
 			const computeUnits = await this.getComputeUnitsFromSim(simTxResult);
 
 			// Apply the buffer, but round down to the MAX_COMPUTE_UNITS, and round up to the nearest whole number
-			const bufferedComputeUnits = Math.ceil(
+			let bufferedComputeUnits = Math.ceil(
 				Math.min(computeUnits * bufferMultiplier, MAX_COMPUTE_UNITS)
 			);
 
+			// If a lower bound CU is passed then enforce it
+			if (lowerBoundCu) {
+				bufferedComputeUnits = Math.max(
+					bufferedComputeUnits,
+					Math.min(lowerBoundCu, MAX_COMPUTE_UNITS)
+				);
+			}
+
 			return {
 				success: true,
 				computeUnits: bufferedComputeUnits,

package/src/types.ts

@@ -1042,6 +1042,7 @@ export type ProcessingTxParams = {
 	computeUnitsBufferMultiplier?: number;
 	useSimulatedComputeUnitsForCUPriceCalculation?: boolean;
 	getCUPriceFromComputeUnits?: (computeUnits: number) => number;
+	lowerBoundCu?: number;
 };
 
 export type TxParams = BaseTxParams & ProcessingTxParams;