From 751738f7dec23cc960dced95188616ed7e2ec0b9 Mon Sep 17 00:00:00 2001
From: aryan <agarwalaryan139@gmail.com>
Date: Mon, 16 Dec 2024 02:22:02 +0530
Subject: [PATCH] fix: jup staking

---
 src/langchain/index.ts            | 18 ++----------------
 src/tools/deploy_token.ts         | 11 -----------
 src/tools/launch_pumpfun_token.ts | 18 ------------------
 src/tools/trade.ts                |  2 --
 4 files changed, 2 insertions(+), 47 deletions(-)

diff --git a/src/langchain/index.ts b/src/langchain/index.ts
index 4f1f876..a406c61 100644
--- a/src/langchain/index.ts
+++ b/src/langchain/index.ts
@@ -54,7 +54,6 @@ export class SolanaTransferTool extends Tool {
   protected async _call(input: string): Promise<string> {
     try {
       const parsedInput = JSON.parse(input);
-      console.log(parsedInput);
 
       const recipient = new PublicKey(parsedInput.to);
       const mintAddress = parsedInput.mint
@@ -304,7 +303,6 @@ export class SolanaTradeTool extends Tool {
         outputToken: parsedInput.outputMint,
       });
     } catch (error: any) {
-      console.log(error);
       return JSON.stringify({
         status: "error",
         message: error.message,
@@ -425,7 +423,6 @@ export class SolanaPumpfunTokenLaunchTool extends Tool {
   }
 
   private validateInput(input: any): void {
-    console.log(input);
     if (!input.tokenName || typeof input.tokenName !== "string") {
       throw new Error("tokenName is required and must be a string");
     }
@@ -521,7 +518,7 @@ export class SolanaCreateImageTool extends Tool {
 
 export class SolanaStakeTool extends Tool {
   name = "solana_stake";
-  description = `This tool can be used to stake your SOL (Solana)
+  description = `This tool can be used to stake your SOL (Solana), also called as SOL staking or liquid staking.
 
   Inputs ( input is a JSON string ):
   amount: number, eg 1 or 0.01 (required)`;
@@ -530,19 +527,9 @@ export class SolanaStakeTool extends Tool {
     super();
   }
 
-  private validateInput(input: any): void {
-    if (
-      input.amount !== undefined &&
-      (typeof input.amount !== "number" || input.amount <= 0)
-    ) {
-      throw new Error("amount must be a positive number when provided");
-    }
-  }
-
   protected async _call(input: string): Promise<string> {
     try {
-      const parsedInput = toJSON(input);
-      this.validateInput(parsedInput);
+      const parsedInput = JSON.parse(input) || Number(input);
 
       const tx = await this.solanaKit.stake(parsedInput.amount);
 
@@ -553,7 +540,6 @@ export class SolanaStakeTool extends Tool {
         amount: parsedInput.amount,
       });
     } catch (error: any) {
-      console.log(error);
       return JSON.stringify({
         status: "error",
         message: error.message,
diff --git a/src/tools/deploy_token.ts b/src/tools/deploy_token.ts
index 65d537a..4b0b497 100644
--- a/src/tools/deploy_token.ts
+++ b/src/tools/deploy_token.ts
@@ -27,10 +27,6 @@ export async function deploy_token(
     );
 
     const mint = Keypair.generate();
-
-    console.log("Mint address: ", mint.publicKey.toString());
-    console.log("Agent address: ", agent.wallet_address.toString());
-
     let account_create_ix = SystemProgram.createAccount({
       fromPubkey: agent.wallet_address,
       newAccountPubkey: mint.publicKey,
@@ -51,13 +47,6 @@ export async function deploy_token(
 
     let hash = await sendTx(agent, tx, [mint]);
 
-    console.log("Transaction hash: ", hash);
-
-    console.log(
-      "Token deployed successfully. Mint address: ",
-      mint.publicKey.toString()
-    );
-
     return {
       mint: mint.publicKey,
     };
diff --git a/src/tools/launch_pumpfun_token.ts b/src/tools/launch_pumpfun_token.ts
index 42c9ee0..397a177 100644
--- a/src/tools/launch_pumpfun_token.ts
+++ b/src/tools/launch_pumpfun_token.ts
@@ -38,7 +38,6 @@ async function uploadMetadata(
     finalFormData.append('file', files.file);
   }
 
-  console.log("Final form data:", finalFormData);
 
   const metadataResponse = await fetch("https://pump.fun/api/ipfs", {
     method: "POST",
@@ -46,7 +45,6 @@ async function uploadMetadata(
   });
 
   if (!metadataResponse.ok) {
-    console.log("Metadata response:", await metadataResponse.json());
     throw new Error(`Metadata upload failed: ${metadataResponse.statusText}`);
   }
 
@@ -152,30 +150,14 @@ export async function launchPumpFunToken(
   options?: PumpFunTokenOptions
 ) {
   try {
-    // TBD : Remove clgs after approval
-    console.log("Starting token launch process...");
 
-    // Generate mint keypair
     const mintKeypair = Keypair.generate();
-    console.log("Mint public key:", mintKeypair.publicKey.toBase58());
-
-    // Upload metadata
-    console.log("Uploading metadata to IPFS...");
     const metadataResponse = await uploadMetadata(tokenName, tokenTicker, description, imageUrl, options);
-    console.log("Metadata response:", metadataResponse);
-
-    // Create token transaction
-    console.log("Creating token transaction...");
     const response = await createTokenTransaction(agent, mintKeypair, metadataResponse, options);
-
     const transactionData = await response.arrayBuffer();
     const tx = VersionedTransaction.deserialize(new Uint8Array(transactionData));
-
-    // Send transaction with proper blockhash handling
-    console.log("Sending transaction...");
     const signature = await signAndSendTransaction(agent, tx, mintKeypair);
 
-    console.log("Token launch successful!");
     return {
       signature,
       mint: mintKeypair.publicKey.toBase58(),
diff --git a/src/tools/trade.ts b/src/tools/trade.ts
index d31055d..b17bc41 100644
--- a/src/tools/trade.ts
+++ b/src/tools/trade.ts
@@ -19,8 +19,6 @@ export async function trade(
   slippageBps: number = DEFAULT_OPTIONS.SLIPPAGE_BPS,
 ): Promise<string> {
   try {
-    // Get quote for the swap
-    console.log(inputMint.toString(), outputMint.toString(), inputAmount, slippageBps);
     const quoteResponse = await (
       await fetch(
         `${JUP_API}/quote?` +