fixes

src/tools/deploy_token.ts

@@ -12,6 +12,7 @@ import {
   fromWeb3JsPublicKey,
   toWeb3JsPublicKey,
 } from "@metaplex-foundation/umi-web3js-adapters";
+import { mplToolbox } from "@metaplex-foundation/mpl-toolbox";
 
 /**
  * Deploy a new SPL token
@@ -33,7 +34,7 @@ export async function deploy_token(
 ): Promise<{ mint: PublicKey }> {
   try {
     // Create UMI instance from agent
-    const umi = createUmi(agent.connection.rpcEndpoint);
+    const umi = createUmi(agent.connection.rpcEndpoint).use(mplToolbox());
     umi.use(keypairIdentity(fromWeb3JsKeypair(agent.wallet)));
 
     // Create new token mint

src/tools/get_balance.ts

@@ -10,7 +10,7 @@ import { SolanaAgentKit } from "../index";
 export async function get_balance(
   agent: SolanaAgentKit,
   token_address?: PublicKey,
-): Promise<number | null> {
+): Promise<number> {
   if (!token_address)
     return (
       (await agent.connection.getBalance(agent.wallet_address)) /
@@ -19,5 +19,5 @@ export async function get_balance(
 
   const token_account =
     await agent.connection.getTokenAccountBalance(token_address);
-  return token_account.value.uiAmount;
+  return token_account.value.uiAmount || 0;
 }

src/tools/index.ts

@@ -20,3 +20,4 @@ export * from "./raydium_create_ammV4";
 export * from "./raydium_create_clmm";
 export * from "./raydium_create_cpmm";
 export * from "./openbook_create_market";
+export * from "./pyth_fetch_price";

src/tools/pyth_fetch_price.ts

@@ -0,0 +1,40 @@
+import { PriceServiceConnection } from "@pythnetwork/price-service-client";
+import BN from "bn.js";
+
+/**
+ * Fetch the price of a given price feed from Pyth
+ * @param agent SolanaAgentKit instance
+ * @param priceFeedID Price feed ID
+ * @returns Latest price value from feed
+ *
+ * You can find priceFeedIDs here: https://www.pyth.network/developers/price-feed-ids#stable
+ */
+export async function pythFetchPrice(priceFeedID: string): Promise<string> {
+  // get Hermes service URL from https://docs.pyth.network/price-feeds/api-instances-and-providers/hermes
+  const stableHermesServiceUrl: string = "https://hermes.pyth.network";
+  const connection = new PriceServiceConnection(stableHermesServiceUrl);
+  const feeds = [priceFeedID];
+
+  try {
+    const currentPrice = await connection.getLatestPriceFeeds(feeds);
+
+    if (currentPrice === undefined) {
+      throw new Error("Price data not available for the given token.");
+    }
+
+    if (currentPrice.length === 0) {
+      throw new Error("Price data not available for the given token.");
+    }
+
+    // get price and exponent from price feed
+    let price = new BN(currentPrice[0].getPriceUnchecked().price);
+    let exponent = new BN(currentPrice[0].getPriceUnchecked().expo);
+
+    // convert to scaled price
+    let scaledPrice = price.div(new BN(10).pow(exponent));
+
+    return scaledPrice.toString();
+  } catch (error: any) {
+    throw new Error(`Fetching price from Pyth failed: ${error.message}`);
+  }
+}