Merge branch 'main' into readme-fix

src/agent/index.ts

@@ -26,6 +26,7 @@ import {
   sendCompressedAirdrop,
   createOrcaSingleSidedWhirlpool,
   FEE_TIERS,
+  pythFetchPrice,
 } from "../tools";
 import {
   CollectionDeployment,
@@ -270,6 +271,10 @@ export class SolanaAgentKit {
 
       lotSize,
       tickSize,
-    );
+    )
+  }
+
+  async pythFetchPrice(priceFeedID: string) {
+    return pythFetchPrice(this, priceFeedID);
   }
 }

src/langchain/index.ts

@@ -1,7 +1,7 @@
 import { PublicKey } from "@solana/web3.js";
 import Decimal from "decimal.js";
 import { Tool } from "langchain/tools";
-import { SolanaAgentKit } from "../index";
+import { PythFetchPriceResponse, SolanaAgentKit } from "../index";
 import { create_image } from "../tools/create_image";
 import { fetchPrice } from "../tools/fetch_price";
 import { BN } from "@coral-xyz/anchor";
@@ -980,6 +980,38 @@ export class SolanaOpenbookCreateMarket extends Tool {
   }
 }
 
+export class SolanaPythFetchPrice extends Tool {
+  name = "solana_pyth_fetch_price";
+  description = `Fetch the price of a given price feed from Pyth's Hermes service
+
+  Inputs:
+  priceFeedID: string, the price feed ID, e.g., "0xe62df6c8b4a85fe1a67db44dc12de5db330f7ac66b72dc658afedf0f4a415b43" for BTC/USD`;
+
+  constructor(private solanaKit: SolanaAgentKit) {
+    super();
+  }
+
+  async _call(input: string): Promise<string> {
+    try {
+      const price = await this.solanaKit.pythFetchPrice(input);
+      let response: PythFetchPriceResponse = {
+        status: "success",
+        priceFeedID: input,
+        price: price,
+      };
+      return JSON.stringify(response);
+    } catch (error: any) {
+      let response: PythFetchPriceResponse = {
+        status: "error",
+        priceFeedID: input,
+        message: error.message,
+        code: error.code || "UNKNOWN_ERROR",
+      };
+      return JSON.stringify(response);
+    }
+  }
+}
+
 export function createSolanaTools(solanaKit: SolanaAgentKit) {
   return [
     new SolanaBalanceTool(solanaKit),
@@ -1007,5 +1039,7 @@ export function createSolanaTools(solanaKit: SolanaAgentKit) {
     new SolanaRaydiumCreateCpmm(solanaKit),
     new SolanaOpenbookCreateMarket(solanaKit),
     new SolanaCreateSingleSidedWhirlpoolTool(solanaKit),
+    new SolanaPythFetchPrice(solanaKit),
   ];
 }
+

src/tools/deploy_token.ts

@@ -4,6 +4,7 @@ import { createUmi } from "@metaplex-foundation/umi-bundle-defaults";
 import { generateSigner, keypairIdentity } from "@metaplex-foundation/umi";
 import { createFungible, mintV1, TokenStandard } from "@metaplex-foundation/mpl-token-metadata";
 import { fromWeb3JsKeypair, fromWeb3JsPublicKey, toWeb3JsPublicKey } from "@metaplex-foundation/umi-web3js-adapters";
+import {mplToolbox} from "@metaplex-foundation/mpl-toolbox"
 
 /**
  * Deploy a new SPL token
@@ -25,7 +26,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/index.ts

@@ -15,8 +15,9 @@ export * from "./get_token_data";
 export * from "./stake_with_jup";
 export * from "./fetch_price";
 export * from "./send_compressed_airdrop";
-
-export * from "./create_orca_single_sided_whirlpool";export * from "./raydium_create_ammV4";
+export * from "./create_orca_single_sided_whirlpool";
+export * from "./raydium_create_ammV4";
 export * from "./raydium_create_clmm";
 export * from "./raydium_create_cpmm";
-export * from "./openbook_create_market";
+export * from "./openbook_create_market";
+export * from "./pyth_fetch_price";

src/tools/pyth_fetch_price.ts

@@ -0,0 +1,48 @@
+import { PublicKey } from "@solana/web3.js";
+import { SolanaAgentKit } from "../index";
+import { Tool } from "langchain/tools";
+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(
+  agent: SolanaAgentKit,
+  priceFeedID: 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}`);
+  }
+}
+
+

src/types/index.ts

@@ -77,3 +77,11 @@ export interface FetchPriceResponse {
   message?: string;
   code?: string;
 }
+
+export interface PythFetchPriceResponse {
+  status: "success" | "error";
+  priceFeedID: string;
+  price?: string;
+  message?: string;
+  code?: string;
+}