init raydium create ammv4/clmm/cpmm pool func

package.json

@@ -14,6 +14,7 @@
   "author": "",
   "license": "ISC",
   "dependencies": {
+    "@raydium-io/raydium-sdk-v2": "0.1.95-alpha",
     "@bonfida/spl-name-service": "^3.0.7",
     "@langchain/core": "^0.3.18",
     "@langchain/groq": "^0.1.2",
@@ -27,6 +28,8 @@
     "@solana/spl-token": "^0.4.9",
     "@solana/web3.js": "^1.95.4",
     "bs58": "^6.0.0",
+    "bn.js": "^5.2.1",
+    "decimal.js": "^10.4.3",
     "dotenv": "^16.4.5",
     "form-data": "^4.0.1",
     "langchain": "^0.3.6",
@@ -34,6 +37,7 @@
     "typedoc": "^0.26.11"
   },
   "devDependencies": {
+    "@types/bn.js": "^5.1.5",
     "@types/node": "^22.9.0",
     "ts-node": "^10.9.2"
   }

src/agent/index.ts

@@ -12,9 +12,14 @@ import {
   launchPumpFunToken,
   lendAsset,
   getTPS,
+  raydiumCreateAmmV4,
+  raydiumCreateClmm,
+  raydiumCreateCpmm,
 } from "../tools";
 import { CollectionOptions, PumpFunTokenOptions } from "../types";
 import { DEFAULT_OPTIONS } from "../constants";
+import BN from "bn.js";
+import Decimal from "decimal.js";
 
 /**
  * Main class for interacting with Solana blockchain
@@ -111,4 +116,71 @@ export class SolanaAgentKit {
       options,
     );
   }
+
+  async raydiumCreateAmmV4(
+    marketId: PublicKey,
+
+    baseAmount: BN,
+    quoteAmount: BN,
+
+    startTime: BN,
+  ) {
+    return raydiumCreateAmmV4(
+      this,
+      marketId,
+
+      baseAmount,
+      quoteAmount,
+
+      startTime,
+    )
+  }
+
+  async raydiumCreateClmm(
+    mint1: PublicKey,
+    mint2: PublicKey,
+
+    configId: PublicKey,
+
+    initialPrice: Decimal,
+    startTime: BN,
+  ) {
+    return raydiumCreateClmm(
+      this,
+
+      mint1,
+      mint2,
+
+      configId,
+
+      initialPrice,
+      startTime,
+    )
+  }
+
+  async raydiumCreateCpmm(
+    mint1: PublicKey,
+    mint2: PublicKey,
+
+    configId: PublicKey,
+
+    mintAAmount: BN,
+    mintBAmount: BN,
+
+    startTime: BN,
+  ) {
+    return raydiumCreateCpmm(
+      this,
+
+      mint1,
+      mint2,
+
+      configId,
+
+      mintAAmount,
+      mintBAmount,
+
+      startTime,
+    )
+  }
 }

src/langchain/index.ts

@@ -1,8 +1,10 @@
+import { PublicKey } from "@solana/web3.js";
+import { BN } from "bn.js";
+import Decimal from "decimal.js";
 import { Tool } from "langchain/tools";
 import { SolanaAgentKit } from "../index";
-import { PublicKey } from "@solana/web3.js";
-import { toJSON } from "../utils/toJSON";
 import { create_image } from "../tools/create_image";
+import { toJSON } from "../utils/toJSON";
 
 export class SolanaBalanceTool extends Tool {
   name = "solana_balance";
@@ -568,6 +570,140 @@ export class SolanaTPSCalculatorTool extends Tool {
   }
 }
 
+export class SolanaRaydiumCreateAmmV4 extends Tool {
+  name = "raydium_create_ammV4";
+  description = `create raydium amm v4 pool
+
+  Inputs (input is a json string):
+  marketId: string (required)
+  baseAmount: number(int), eg: 111111 (required)
+  quoteAmount: number(int), eg: 111111 (required)
+  startTime: number(seconds), eg: now number or zero (required)
+  `;
+
+  constructor(private solanaKit: SolanaAgentKit) {
+    super();
+  }
+
+  async _call(input: string): Promise<string> {
+    try {
+      let inputFormat = JSON.parse(input)
+
+      const tx = await this.solanaKit.raydiumCreateAmmV4(
+        new PublicKey(inputFormat.marketId),
+        new BN(inputFormat.baseAmount),
+        new BN(inputFormat.quoteAmount),
+        new BN(inputFormat.startTime),
+      );
+
+      return JSON.stringify({
+        status: "success",
+        message: "Create raydium amm v4 pool successfully",
+        transaction: tx,
+      });
+    } catch (error: any) {
+      return JSON.stringify({
+        status: "error",
+        message: error.message,
+        code: error.code || "UNKNOWN_ERROR",
+      });
+    }
+  }
+}
+
+export class SolanaRaydiumCreateClmm extends Tool {
+  name = "raydium_create_clmm";
+  description = `create raydium clmm pool
+
+  Inputs (input is a json string):
+  mint1: string (required)
+  mint2: string (required)
+  configId: string (required)
+  initialPrice: number, eg: 123.12 (required)
+  startTime: number(seconds), eg: now number or zero (required)
+  `;
+
+  constructor(private solanaKit: SolanaAgentKit) {
+    super();
+  }
+
+  async _call(input: string): Promise<string> {
+    try {
+      let inputFormat = JSON.parse(input)
+
+      const tx = await this.solanaKit.raydiumCreateClmm(
+        new PublicKey(inputFormat.mint1),
+        new PublicKey(inputFormat.mint2),
+
+        new PublicKey(inputFormat.configId),
+
+        new Decimal(inputFormat.initialPrice),
+        new BN(inputFormat.startTime),
+      );
+
+      return JSON.stringify({
+        status: "success",
+        message: "Create raydium clmm pool successfully",
+        transaction: tx,
+      });
+    } catch (error: any) {
+      return JSON.stringify({
+        status: "error",
+        message: error.message,
+        code: error.code || "UNKNOWN_ERROR",
+      });
+    }
+  }
+}
+
+export class SolanaRaydiumCreateCpmm extends Tool {
+  name = "raydium_create_cpmm";
+  description = `create raydium cpmm pool
+
+  Inputs (input is a json string):
+  mint1: string (required)
+  mint2: string (required)
+  configId: string (required)
+  mintAAmount: number(int), eg: 1111 (required)
+  mintBAmount: number(int), eg: 2222 (required)
+  startTime: number(seconds), eg: now number or zero (required)
+  `;
+
+  constructor(private solanaKit: SolanaAgentKit) {
+    super();
+  }
+
+  async _call(input: string): Promise<string> {
+    try {
+      let inputFormat = JSON.parse(input)
+
+      const tx = await this.solanaKit.raydiumCreateCpmm(
+        new PublicKey(inputFormat.mint1),
+        new PublicKey(inputFormat.mint2),
+
+        new PublicKey(inputFormat.configId),
+
+        new BN(inputFormat.mintAAmount),
+        new BN(inputFormat.mintBAmount),
+
+        new BN(inputFormat.startTime),
+      );
+
+      return JSON.stringify({
+        status: "success",
+        message: "Create raydium cpmm pool successfully",
+        transaction: tx,
+      });
+    } catch (error: any) {
+      return JSON.stringify({
+        status: "error",
+        message: error.message,
+        code: error.code || "UNKNOWN_ERROR",
+      });
+    }
+  }
+}
+
 export function createSolanaTools(solanaKit: SolanaAgentKit) {
   return [
     new SolanaBalanceTool(solanaKit),
@@ -583,5 +719,8 @@ export function createSolanaTools(solanaKit: SolanaAgentKit) {
     new SolanaCreateImageTool(solanaKit),
     new SolanaLendAssetTool(solanaKit),
     new SolanaTPSCalculatorTool(solanaKit),
+    new SolanaRaydiumCreateAmmV4(solanaKit),
+    new SolanaRaydiumCreateClmm(solanaKit),
+    new SolanaRaydiumCreateCpmm(solanaKit),
   ];
 }

src/tools/index.ts

@@ -9,3 +9,6 @@ export * from "./register_domain";
 export * from "./launch_pumpfun_token";
 export * from "./lend";
 export * from "./get_tps";
+export * from "./raydium_create_ammV4";
+export * from "./raydium_create_clmm";
+export * from "./raydium_create_cpmm";

src/tools/raydium_create_ammV4.ts

@@ -0,0 +1,71 @@
+import { AMM_V4, FEE_DESTINATION_ID, MARKET_STATE_LAYOUT_V3, OPEN_BOOK_PROGRAM, Raydium, TxVersion } from "@raydium-io/raydium-sdk-v2";
+import { MintLayout, TOKEN_PROGRAM_ID } from "@solana/spl-token";
+import { PublicKey } from "@solana/web3.js";
+import BN from 'bn.js';
+import { SolanaAgentKit } from "../agent";
+
+export async function raydiumCreateAmmV4(
+  agent: SolanaAgentKit,
+
+  marketId: PublicKey,
+
+  baseAmount: BN,
+  quoteAmount: BN,
+
+  startTime: BN,
+): Promise<string> {
+
+  const raydium = await Raydium.load({
+    owner: agent.wallet,
+    connection: agent.connection,
+  })
+
+  const marketBufferInfo = await agent.connection.getAccountInfo(new PublicKey(marketId))
+  const { baseMint, quoteMint } = MARKET_STATE_LAYOUT_V3.decode(marketBufferInfo!.data)
+
+  const baseMintInfo = await agent.connection.getAccountInfo(baseMint)
+  const quoteMintInfo = await agent.connection.getAccountInfo(quoteMint)
+
+  if (
+    baseMintInfo?.owner.toString() !== TOKEN_PROGRAM_ID.toBase58() ||
+    quoteMintInfo?.owner.toString() !== TOKEN_PROGRAM_ID.toBase58()
+  ) {
+    throw new Error(
+      'amm pools with openbook market only support TOKEN_PROGRAM_ID mints, if you want to create pool with token-2022, please create cpmm pool instead'
+    )
+  }
+
+  if (baseAmount.mul(quoteAmount).lte(new BN(1).mul(new BN(10 ** MintLayout.decode(baseMintInfo.data).decimals)).pow(new BN(2)))) {
+    throw new Error('initial liquidity too low, try adding more baseAmount/quoteAmount')
+  }
+
+  const { execute } = await raydium.liquidity.createPoolV4({
+    programId: AMM_V4,
+    marketInfo: {
+      marketId,
+      programId: OPEN_BOOK_PROGRAM,
+    },
+    baseMintInfo: {
+      mint: baseMint,
+      decimals: MintLayout.decode(baseMintInfo.data).decimals,
+    },
+    quoteMintInfo: {
+      mint: quoteMint,
+      decimals: MintLayout.decode(quoteMintInfo.data).decimals,
+    },
+    baseAmount,
+    quoteAmount,
+
+    startTime,
+    ownerInfo: {
+      useSOLBalance: true,
+    },
+    associatedOnly: false,
+    txVersion: TxVersion.V0,
+    feeDestinationId: FEE_DESTINATION_ID,
+  })
+
+  const { txId } = await execute({ sendAndConfirm: true })
+
+  return txId
+}

src/tools/raydium_create_clmm.ts

@@ -0,0 +1,73 @@
+import { CLMM_PROGRAM_ID, Raydium, TxVersion } from '@raydium-io/raydium-sdk-v2'
+import { MintLayout } from '@solana/spl-token'
+import { PublicKey } from '@solana/web3.js'
+import BN from 'bn.js'
+import Decimal from 'decimal.js'
+import { SolanaAgentKit } from '../agent'
+
+export async function raydiumCreateClmm(
+  agent: SolanaAgentKit,
+
+  mint1: PublicKey,
+  mint2: PublicKey,
+
+  configId: PublicKey,
+
+  initialPrice: Decimal,
+  startTime: BN,
+): Promise<string> {
+
+  const raydium = await Raydium.load({
+    owner: agent.wallet,
+    connection: agent.connection,
+  })
+
+  const [mintInfo1, mintInfo2] = await agent.connection.getMultipleAccountsInfo([mint1, mint2])
+  if (mintInfo1 === null || mintInfo2 === null) throw Error('fetch mint info error')
+
+  const mintDecodeInfo1 = MintLayout.decode(mintInfo1.data)
+  const mintDecodeInfo2 = MintLayout.decode(mintInfo2.data)
+
+  const mintFormatInfo1 = {
+    chainId: 101,
+    address: mint1.toString(),
+    programId: mintInfo1.owner.toString(),
+    logoURI: '',
+    symbol: '',
+    name: '',
+    decimals: mintDecodeInfo1.decimals,
+    tags: [],
+    extensions: {}
+  }
+  const mintFormatInfo2 = {
+    chainId: 101,
+    address: mint2.toString(),
+    programId: mintInfo2.owner.toString(),
+    logoURI: '',
+    symbol: '',
+    name: '',
+    decimals: mintDecodeInfo2.decimals,
+    tags: [],
+    extensions: {}
+  }
+
+  const { execute } = await raydium.clmm.createPool({
+    programId: CLMM_PROGRAM_ID,
+    // programId: DEVNET_PROGRAM_ID.CLMM,
+    mint1: mintFormatInfo1,
+    mint2: mintFormatInfo2,
+    // @ts-ignore
+    ammConfig: { id: configId },
+    initialPrice,
+    startTime,
+    txVersion: TxVersion.V0,
+    // computeBudgetConfig: {
+    //   units: 600000,
+    //   microLamports: 46591500,
+    // },
+  })
+
+  const { txId } = await execute({ sendAndConfirm: true })
+
+  return txId
+}

src/tools/raydium_create_cpmm.ts

@@ -0,0 +1,84 @@
+import {
+  CREATE_CPMM_POOL_FEE_ACC,
+  CREATE_CPMM_POOL_PROGRAM,
+  Raydium,
+  TxVersion
+} from '@raydium-io/raydium-sdk-v2'
+import { MintLayout } from '@solana/spl-token'
+import { PublicKey } from '@solana/web3.js'
+import BN from 'bn.js'
+import { SolanaAgentKit } from '../agent'
+
+export async function raydiumCreateCpmm(
+  agent: SolanaAgentKit,
+
+  mintA: PublicKey,
+  mintB: PublicKey,
+
+  configId: PublicKey,
+
+  mintAAmount: BN,
+  mintBAmount: BN,
+
+  startTime: BN,
+): Promise<string> {
+
+  const raydium = await Raydium.load({
+    owner: agent.wallet,
+    connection: agent.connection,
+  })
+
+  const [mintInfoA, mintInfoB] = await agent.connection.getMultipleAccountsInfo([mintA, mintB])
+  if (mintInfoA === null || mintInfoB === null) throw Error('fetch mint info error')
+
+  const mintDecodeInfoA = MintLayout.decode(mintInfoA.data)
+  const mintDecodeInfoB = MintLayout.decode(mintInfoB.data)
+
+  const mintFormatInfoA = {
+    chainId: 101,
+    address: mintA.toString(),
+    programId: mintInfoA.owner.toString(),
+    logoURI: '',
+    symbol: '',
+    name: '',
+    decimals: mintDecodeInfoA.decimals,
+    tags: [],
+    extensions: {}
+  }
+  const mintFormatInfoB = {
+    chainId: 101,
+    address: mintB.toString(),
+    programId: mintInfoB.owner.toString(),
+    logoURI: '',
+    symbol: '',
+    name: '',
+    decimals: mintDecodeInfoB.decimals,
+    tags: [],
+    extensions: {}
+  }
+
+  const { execute, extInfo } = await raydium.cpmm.createPool({
+    programId: CREATE_CPMM_POOL_PROGRAM,
+    poolFeeAccount: CREATE_CPMM_POOL_FEE_ACC,
+    mintA: mintFormatInfoA,
+    mintB: mintFormatInfoB,
+    mintAAmount,
+    mintBAmount,
+    startTime,
+    //@ts-ignore
+    feeConfig: { id: configId.toString() },
+    associatedOnly: false,
+    ownerInfo: {
+      useSOLBalance: true,
+    },
+    txVersion: TxVersion.V0,
+    // computeBudgetConfig: {
+    //   units: 600000,
+    //   microLamports: 46591500,
+    // },
+  })
+
+  const { txId } = await execute({ sendAndConfirm: true })
+
+  return txId
+}