fix : able to send transaction with priority fee

src/tools/send_transaction_with_priority.ts

@@ -8,69 +8,17 @@ import {
   LAMPORTS_PER_SOL,
 } from "@solana/web3.js";
 import bs58 from "bs58";
-import { PriorityFeeTransaction } from "../types";
+import { PriorityFeeTransaction, PriorityFeeResponse } from "../types";
 
 /**
- * Fetches an estimated priority fee (in microLamports) from Helius.
+ * Sends a transaction with an estimated priority fee using the provided SolanaAgentKit.
  *
  * @param agent         An instance of SolanaAgentKit containing connection, wallet, etc.
- * @param priorityLevel The priority level (e.g. "Min", "Low", "Medium", "High", "VeryHigh", or "UnsafeMax").
- * @param transaction   The (unsigned or partially-signed) Transaction you want to estimate fees for.
- * @returns             The numeric priority fee estimate in microLamports.
- */
-export async function getPriorityFeeEstimate(
-  agent: SolanaAgentKit,
-  priorityLevel: string,
-  transaction: Transaction,
-): Promise<number> {
-  const apiKey = agent.config.HELIUS_API_KEY;
-  if (!apiKey) {
-    throw new Error(
-      "HELIUS_API_KEY not found in agent.config or environment variables",
-    );
-  }
-
-  const url = `https://api.helius.xyz/v0/transactions/?api-key=${apiKey}`;
-
-  const payload = {
-    jsonrpc: "2.0",
-    id: "1",
-    method: "getPriorityFeeEstimate",
-    params: [
-      {
-        transaction: bs58.encode(transaction.serialize()),
-        options: { priorityLevel },
-      },
-    ],
-  };
-  const response = await fetch(url, {
-    method: "POST",
-    headers: { "Content-Type": "application/json" },
-    body: JSON.stringify(payload),
-  });
-
-  if (!response.ok) {
-    throw new Error(`Helius request failed with status ${response.status}`);
-  }
-
-  const data = await response.json();
-  if (!data.result || data.result.priorityFeeEstimate === undefined) {
-    throw new Error(`Invalid response from Helius: ${JSON.stringify(data)}`);
-  }
-
-  return data.result.priorityFeeEstimate;
-}
-
-/**
- * Sends a transaction with an optional priority fee using the provided SolanaAgentKit.
- *
- * @param agent         An instance of SolanaAgentKit containing connection, wallet, etc.
- * @param priorityLevel The priority level ("NONE", "Min", "Low", "Medium", "High", "VeryHigh", or "UnsafeMax").
+ * @param priorityLevel The priority level (e.g., "Min", "Low", "Medium", "High", "VeryHigh", or "UnsafeMax").
  * @param amount        The amount of SOL to send (in SOL, not lamports).
  * @param to            The recipient's PublicKey.
- * @returns             The transaction signature (string) once confirmed.
+ * @returns             The transaction signature (string) once confirmed along with the fee used.
  */
-
 export async function sendTransactionWithPriorityFee(
   agent: SolanaAgentKit,
   priorityLevel: string,
@@ -78,34 +26,53 @@ export async function sendTransactionWithPriorityFee(
   to: PublicKey,
 ): Promise<PriorityFeeTransaction> {
   const transaction = new Transaction();
+  const { blockhash, lastValidBlockHeight } =
+    await agent.connection.getLatestBlockhash();
+  transaction.recentBlockhash = blockhash;
+  transaction.lastValidBlockHeight = lastValidBlockHeight;
+  transaction.feePayer = agent.wallet_address;
 
   const transferIx = SystemProgram.transfer({
     fromPubkey: agent.wallet_address,
     toPubkey: to,
     lamports: amount * LAMPORTS_PER_SOL,
   });
+
   transaction.add(transferIx);
-
-  let feeEstimate = 0;
-  if (priorityLevel !== "NONE") {
-    feeEstimate = await getPriorityFeeEstimate(
-      agent,
-      priorityLevel,
-      transaction,
-    );
-    if (feeEstimate > 0) {
-      const computePriceIx = ComputeBudgetProgram.setComputeUnitPrice({
-        microLamports: feeEstimate,
-      });
-      transaction.add(computePriceIx);
-    }
-  }
-
-  transaction.recentBlockhash = (
-    await agent.connection.getLatestBlockhash()
-  ).blockhash;
   transaction.sign(agent.wallet);
 
+  const response = await fetch(
+    `https://mainnet.helius-rpc.com/?api-key=${agent.config.HELIUS_API_KEY}`,
+    {
+      method: "POST",
+      headers: { "Content-Type": "application/json" },
+      body: JSON.stringify({
+        jsonrpc: "2.0",
+        id: "1",
+        method: "getPriorityFeeEstimate",
+        params: [
+          {
+            transaction: bs58.encode(transaction.serialize()),
+            options: { priorityLevel: priorityLevel },
+          },
+        ],
+      } as PriorityFeeResponse),
+    },
+  );
+
+  const data = await response.json();
+  if (data.error) {
+    throw new Error("Error fetching priority fee:");
+  }
+  const feeEstimate: number = data.result.priorityFeeEstimate;
+
+  // Set the priority fee if applicable
+  const computePriceIx = ComputeBudgetProgram.setComputeUnitPrice({
+    microLamports: feeEstimate,
+  });
+  transaction.add(computePriceIx);
+
+  // Send the transaction and confirm
   const txSignature = await sendAndConfirmTransaction(
     agent.connection,
     transaction,