Move mm agent to examples

2026-05-13 23:16:55 +00:00 · 2025-01-06 12:35:52 -04:00
parent 57ce14f8b1
commit f03f5dd3ff
6 changed files with 6460 additions and 139 deletions
--- a/examples/market-making-agent/.env.example
+++ b/examples/market-making-agent/.env.example
@@ -0,0 +1,3 @@
+OPENAI_API_KEY=
+RPC_URL=
+SOLANA_PRIVATE_KEY=
--- a/examples/market-making-agent/README.md
+++ b/examples/market-making-agent/README.md
@@ -0,0 +1,50 @@
+# AI Guided Market Making Agent
+
+This agent showcases an ai guided market maker on Manifest, Solana's CLOB DEX. The agent guides the user to setup basic two-sided quotes on Manifest markets.
+[Reference](https://github.com/CKS-Systems/manifest)
+
+## Key Features
+
+- **Automated Quoting**: The agent automatically refreshes quotes on an interval.
+- **Reducing Complexity**: Designed to abstract away parameters for setting up market making.
+- **Random Model**: The market making introduces randomness to prevent front running or other negative botting behavoirs.
+
+
+## Example
+=== Market Maker Configuration ===
+
+Enter the market ID: 2Uj8277fkaVBtTU6Wp2GPRbQC86SkSdgQ2mp1Q5N2LHc
+Enter the base token symbol (e.g., SEND): SEND
+Enter the quote token symbol (e.g., USDC): USDC
+
+=== Quote Parameters (applies to both buy and sell sides) ===
+Enter number of quotes to place on each side: 4
+Enter minimum quote depth (% distance from mid price): 0.1
+Enter maximum quote depth (% distance from mid price): 2
+
+=== Token Allowances ===
+Enter total SEND allowance: 2
+Enter total USDC allowance: 3
+
+Enter update interval in seconds: 20
+
+=== Configuration Summary ===
+{
+  "marketId": "2Uj8277fkaVBtTU6Wp2GPRbQC86SkSdgQ2mp1Q5N2LHc",
+  "baseToken": "SEND",
+  "quoteToken": "USDC",
+  "quoteParams": {
+    "number": 4,
+    "minDepth": 0.1,
+    "maxDepth": 2
+  },
+  "allowance": {
+    "base": 2,
+    "quote": 3
+  },
+  "intervalSeconds": 20
+}
+
+Is this configuration correct? (yes/no): yes
+
+Starting market maker mode for SEND/USDC...
--- a/examples/market-making-agent/index.ts
+++ b/examples/market-making-agent/index.ts
@@ -0,0 +1,389 @@
+import { HumanMessage } from "@langchain/core/messages";
+import { MemorySaver } from "@langchain/langgraph";
+import { createReactAgent } from "@langchain/langgraph/prebuilt";
+import { ChatOpenAI } from "@langchain/openai";
+import * as dotenv from "dotenv";
+import * as fs from "fs";
+import * as readline from "readline";
+import { createSolanaTools, SolanaAgentKit } from "../../src";
+
+dotenv.config();
+
+function validateEnvironment(): void {
+  const missingVars: string[] = [];
+  const requiredVars = ["OPENAI_API_KEY", "RPC_URL", "SOLANA_PRIVATE_KEY"];
+
+  requiredVars.forEach((varName) => {
+    if (!process.env[varName]) {
+      missingVars.push(varName);
+    }
+  });
+
+  if (missingVars.length > 0) {
+    console.error("Error: Required environment variables are not set");
+    missingVars.forEach((varName) => {
+      console.error(`${varName}=your_${varName.toLowerCase()}_here`);
+    });
+    process.exit(1);
+  }
+}
+
+validateEnvironment();
+
+const WALLET_DATA_FILE = "wallet_data.txt";
+
+async function initializeAgent() {
+  try {
+    const llm = new ChatOpenAI({
+      modelName: "gpt-4o-mini",
+      temperature: 0.3,
+    });
+
+    let walletDataStr: string | null = null;
+
+    if (fs.existsSync(WALLET_DATA_FILE)) {
+      try {
+        walletDataStr = fs.readFileSync(WALLET_DATA_FILE, "utf8");
+      } catch (error) {
+        console.error("Error reading wallet data:", error);
+      }
+    }
+
+    const solanaAgent = new SolanaAgentKit(
+      process.env.SOLANA_PRIVATE_KEY!,
+      process.env.RPC_URL!,
+      {
+        OPENAI_API_KEY: process.env.OPENAI_API_KEY!,
+      },
+    );
+
+    const tools = createSolanaTools(solanaAgent);
+
+    const memory = new MemorySaver();
+    const config = { configurable: { thread_id: "Solana Agent Kit!" } };
+
+    const agent = createReactAgent({
+      llm,
+      tools,
+      checkpointSaver: memory,
+      messageModifier: `
+        You are a helpful agent that can interact onchain using the Solana Agent Kit. You are
+        empowered to interact onchain using your tools. If you ever need funds, you can request them from the
+        faucet. If not, you can provide your wallet details and request funds from the user. If there is a 5XX
+        (internal) HTTP error code, ask the user to try again later. If someone asks you to do something you
+        can't do with your currently available tools, you must say so, and encourage them to implement it
+        themselves using the Solana Agent Kit, recommend they go to https://www.solanaagentkit.xyz for more information. Be
+        concise and helpful with your responses. Refrain from restating your tools' descriptions unless it is explicitly requested.
+      `,
+    });
+
+    if (walletDataStr) {
+      fs.writeFileSync(WALLET_DATA_FILE, walletDataStr);
+    }
+
+    return { agent, config };
+  } catch (error) {
+    console.error("Failed to initialize agent:", error);
+    throw error;
+  }
+}
+
+async function runAutonomousMode(agent: any, config: any, interval = 10) {
+  console.log("Starting autonomous mode...");
+
+  while (true) {
+    try {
+      const thought =
+        "Be creative and do something interesting on the blockchain. " +
+        "Choose an action or set of actions and execute it that highlights your abilities.";
+
+      const stream = await agent.stream(
+        { messages: [new HumanMessage(thought)] },
+        config,
+      );
+
+      for await (const chunk of stream) {
+        if ("agent" in chunk) {
+          console.log(chunk.agent.messages[0].content);
+        } else if ("tools" in chunk) {
+          console.log(chunk.tools.messages[0].content);
+        }
+        console.log("-------------------");
+      }
+
+      await new Promise((resolve) => setTimeout(resolve, interval * 1000));
+    } catch (error) {
+      if (error instanceof Error) {
+        console.error("Error:", error.message);
+      }
+      process.exit(1);
+    }
+  }
+}
+
+async function runChatMode(agent: any, config: any) {
+  console.log("Starting chat mode... Type 'exit' to end.");
+
+  const rl = readline.createInterface({
+    input: process.stdin,
+    output: process.stdout,
+  });
+
+  const question = (prompt: string): Promise<string> =>
+    new Promise((resolve) => rl.question(prompt, resolve));
+
+  try {
+    while (true) {
+      const userInput = await question("\nPrompt: ");
+
+      if (userInput.toLowerCase() === "exit") {
+        break;
+      }
+
+      const stream = await agent.stream(
+        { messages: [new HumanMessage(userInput)] },
+        config,
+      );
+
+      for await (const chunk of stream) {
+        if ("agent" in chunk) {
+          console.log(chunk.agent.messages[0].content);
+        } else if ("tools" in chunk) {
+          console.log(chunk.tools.messages[0].content);
+        }
+        console.log("-------------------");
+      }
+    }
+  } catch (error) {
+    if (error instanceof Error) {
+      console.error("Error:", error.message);
+    }
+    process.exit(1);
+  } finally {
+    rl.close();
+  }
+}
+
+interface MarketMakerConfig {
+  marketId: string;
+  baseToken: string;
+  quoteToken: string;
+  quoteParams: {
+    number: number; // Number of quotes on each side
+    minDepth: number; // Minimum distance from mid (%)
+    maxDepth: number; // Maximum distance from mid (%)
+  };
+  allowance: {
+    base: number;
+    quote: number;
+  };
+  intervalSeconds: number;
+}
+
+function createReadlineInterface() {
+  return readline.createInterface({
+    input: process.stdin,
+    output: process.stdout,
+  });
+}
+
+async function askQuestion(
+  rl: readline.Interface,
+  question: string,
+): Promise<string> {
+  return new Promise((resolve) => {
+    rl.question(question, (answer) => {
+      resolve(answer);
+    });
+  });
+}
+
+async function configureMarketMaker(): Promise<MarketMakerConfig> {
+  const rl = createReadlineInterface();
+
+  try {
+    console.log("\n=== Market Maker Configuration ===\n");
+
+    // Basic market information
+    const marketId = await askQuestion(rl, "Enter the market ID: ");
+    const baseToken = await askQuestion(
+      rl,
+      "Enter the base token symbol (e.g., SEND): ",
+    );
+    const quoteToken = await askQuestion(
+      rl,
+      "Enter the quote token symbol (e.g., USDC): ",
+    );
+
+    // Quote parameters
+    console.log(
+      "\n=== Quote Parameters (applies to both buy and sell sides) ===",
+    );
+    const quoteNumber = parseInt(
+      await askQuestion(rl, "Enter number of quotes to place on each side: "),
+    );
+    const minDepth = parseFloat(
+      await askQuestion(
+        rl,
+        "Enter minimum quote depth (% distance from mid price): ",
+      ),
+    );
+    const maxDepth = parseFloat(
+      await askQuestion(
+        rl,
+        "Enter maximum quote depth (% distance from mid price): ",
+      ),
+    );
+
+    // Token allowances
+    console.log("\n=== Token Allowances ===");
+    const baseAllowance = parseFloat(
+      await askQuestion(rl, `Enter total ${baseToken} allowance: `),
+    );
+    const quoteAllowance = parseFloat(
+      await askQuestion(rl, `Enter total ${quoteToken} allowance: `),
+    );
+
+    // Update interval
+    const interval = parseInt(
+      await askQuestion(rl, "\nEnter update interval in seconds: "),
+    );
+
+    const config: MarketMakerConfig = {
+      marketId,
+      baseToken,
+      quoteToken,
+      quoteParams: {
+        number: quoteNumber,
+        minDepth: minDepth,
+        maxDepth: maxDepth,
+      },
+      allowance: {
+        base: baseAllowance,
+        quote: quoteAllowance,
+      },
+      intervalSeconds: interval,
+    };
+
+    // Display summary
+    console.log("\n=== Configuration Summary ===");
+    console.log(JSON.stringify(config, null, 2));
+
+    const confirm = await askQuestion(
+      rl,
+      "\nIs this configuration correct? (yes/no): ",
+    );
+    if (confirm.toLowerCase() !== "yes") {
+      throw new Error("Configuration cancelled by user");
+    }
+
+    return config;
+  } finally {
+    rl.close();
+  }
+}
+
+async function runMarketMakerMode(agent: any, config: any) {
+  try {
+    const marketMakerConfig = await configureMarketMaker();
+    console.log(
+      `\nStarting market maker mode for ${marketMakerConfig.baseToken}/${marketMakerConfig.quoteToken}...`,
+    );
+
+    while (true) {
+      try {
+        const thought = `You are an on-chain Solana market maker for the ${marketMakerConfig.baseToken}/${marketMakerConfig.quoteToken} Manifest market, ${marketMakerConfig.marketId}. 
+        Find the ${marketMakerConfig.baseToken}/${marketMakerConfig.quoteToken} live price by checking Jupiter. 
+        Use solana_batch_order to provide ${marketMakerConfig.quoteParams.number} buys at different prices between -${marketMakerConfig.quoteParams.minDepth}% to -${marketMakerConfig.quoteParams.maxDepth}% and ${marketMakerConfig.quoteParams.number} sells at different prices between +${marketMakerConfig.quoteParams.minDepth}% to +${marketMakerConfig.quoteParams.maxDepth}% with increasing quantities further from the live price. 
+        You have an allowance of ${marketMakerConfig.allowance.base} ${marketMakerConfig.baseToken} and ${marketMakerConfig.allowance.quote} ${marketMakerConfig.quoteToken}.
+        Important! Only send 1 transaction, buy and sells can be combined into a single solana_batch_order.`;
+
+        const stream = await agent.stream(
+          { messages: [new HumanMessage(thought)] },
+          config,
+        );
+
+        for await (const chunk of stream) {
+          if ("agent" in chunk) {
+            console.log(chunk.agent.messages[0].content);
+          } else if ("tools" in chunk) {
+            console.log(chunk.tools.messages[0].content);
+          }
+          console.log("-------------------");
+        }
+
+        await new Promise((resolve) =>
+          setTimeout(resolve, marketMakerConfig.intervalSeconds * 1000),
+        );
+      } catch (error) {
+        if (error instanceof Error) {
+          console.error("Error:", error);
+        }
+        process.exit(1);
+      }
+    }
+  } catch (error) {
+    console.error("Configuration error:", error);
+    process.exit(1);
+  }
+}
+
+async function chooseMode(): Promise<"chat" | "auto" | "mm"> {
+  const rl = readline.createInterface({
+    input: process.stdin,
+    output: process.stdout,
+  });
+
+  const question = (prompt: string): Promise<string> =>
+    new Promise((resolve) => rl.question(prompt, resolve));
+
+  while (true) {
+    console.log("\nAvailable modes:");
+    console.log("1. chat    - Interactive chat mode");
+    console.log("2. auto    - Autonomous action mode");
+    console.log("3. mm      - AI guided market making");
+
+    const choice = (await question("\nChoose a mode (enter number or name): "))
+      .toLowerCase()
+      .trim();
+
+    rl.close();
+
+    if (choice === "1" || choice === "chat") {
+      return "chat";
+    } else if (choice === "2" || choice === "auto") {
+      return "auto";
+    } else if (choice === "3" || choice === "mm") {
+      return "mm";
+    }
+    console.log("Invalid choice. Please try again.");
+  }
+}
+
+async function main() {
+  try {
+    console.log("Starting Agent...");
+    const { agent, config } = await initializeAgent();
+    const mode = await chooseMode();
+
+    if (mode === "chat") {
+      await runChatMode(agent, config);
+    } else if (mode === "auto") {
+      await runAutonomousMode(agent, config);
+    } else {
+      await runMarketMakerMode(agent, config);
+    }
+  } catch (error) {
+    if (error instanceof Error) {
+      console.error("Error:", error.message);
+    }
+    process.exit(1);
+  }
+}
+
+if (require.main === module) {
+  main().catch((error) => {
+    console.error("Fatal error:", error);
+    process.exit(1);
+  });
+}
--- a/examples/market-making-agent/package.json
+++ b/examples/market-making-agent/package.json
@@ -0,0 +1,16 @@
+{
+  "name": "market-making-agent",
+  "version": "1.0.0",
+  "description": "",
+  "main": "index.js",
+  "scripts": {
+    "main": "ts-node index.ts"
+  },
+  "keywords": [],
+  "author": "",
+  "license": "ISC",
+  "dependencies": {
+    "@langchain/langgraph-checkpoint-postgres": "^0.0.2",
+    "solana-agent-kit": "^1.3.6"
+  }
+}
--- a/examples/market-making-agent/pnpm-lock.yaml
+++ b/examples/market-making-agent/pnpm-lock.yaml
--- a/test/index.ts
+++ b/test/index.ts
@@ -165,139 +165,7 @@ async function runChatMode(agent: any, config: any) {
  }
 }

-interface MarketMakerConfig {
-  marketId: string;
-  baseToken: string;
-  quoteToken: string;
-  quoteParams: {
-    number: number;        // Number of quotes on each side
-    minDepth: number;      // Minimum distance from mid (%)
-    maxDepth: number;      // Maximum distance from mid (%)
-  };
-  allowance: {
-    base: number;
-    quote: number;
-  };
-  intervalSeconds: number;
-}
-
-function createReadlineInterface() {
-  return readline.createInterface({
-    input: process.stdin,
-    output: process.stdout
-  });
-}
-
-async function askQuestion(rl: readline.Interface, question: string): Promise<string> {
-  return new Promise((resolve) => {
-    rl.question(question, (answer) => {
-      resolve(answer);
-    });
-  });
-}
-
-async function configureMarketMaker(): Promise<MarketMakerConfig> {
-  const rl = createReadlineInterface();
-
-  try {
-    console.log("\n=== Market Maker Configuration ===\n");
-
-    // Basic market information
-    const marketId = await askQuestion(rl, "Enter the market ID: ");
-    const baseToken = await askQuestion(rl, "Enter the base token symbol (e.g., SEND): ");
-    const quoteToken = await askQuestion(rl, "Enter the quote token symbol (e.g., USDC): ");
-
-    // Quote parameters
-    console.log("\n=== Quote Parameters (applies to both buy and sell sides) ===");
-    const quoteNumber = parseInt(await askQuestion(rl, "Enter number of quotes to place on each side: "));
-    const minDepth = parseFloat(await askQuestion(rl, "Enter minimum quote depth (% distance from mid price): "));
-    const maxDepth = parseFloat(await askQuestion(rl, "Enter maximum quote depth (% distance from mid price): "));
-
-    // Token allowances
-    console.log("\n=== Token Allowances ===");
-    const baseAllowance = parseFloat(await askQuestion(rl, `Enter total ${baseToken} allowance: `));
-    const quoteAllowance = parseFloat(await askQuestion(rl, `Enter total ${quoteToken} allowance: `));
-
-    // Update interval
-    const interval = parseInt(await askQuestion(rl, "\nEnter update interval in seconds: "));
-
-    const config: MarketMakerConfig = {
-      marketId,
-      baseToken,
-      quoteToken,
-      quoteParams: {
-        number: quoteNumber,
-        minDepth: minDepth,
-        maxDepth: maxDepth
-      },
-      allowance: {
-        base: baseAllowance,
-        quote: quoteAllowance
-      },
-      intervalSeconds: interval
-    };
-
-    // Display summary
-    console.log("\n=== Configuration Summary ===");
-    console.log(JSON.stringify(config, null, 2));
-    
-    const confirm = await askQuestion(rl, "\nIs this configuration correct? (yes/no): ");
-    if (confirm.toLowerCase() !== 'yes') {
-      throw new Error("Configuration cancelled by user");
-    }
-
-    return config;
-
-  } finally {
-    rl.close();
-  }
-}
-
-async function runMarketMakerMode(agent: any, config: any) {
-  try {
-    const marketMakerConfig = await configureMarketMaker();
-    console.log(`\nStarting market maker mode for ${marketMakerConfig.baseToken}/${marketMakerConfig.quoteToken}...`);
-
-    while (true) {
-      try {
-        const thought =          
-        `You are an on-chain Solana market maker for the ${marketMakerConfig.baseToken}/${marketMakerConfig.quoteToken} Manifest market, ${marketMakerConfig.marketId}. 
-        Find the ${marketMakerConfig.baseToken}/${marketMakerConfig.quoteToken} live price by checking Jupiter. 
-        Use solana_batch_order to provide ${marketMakerConfig.quoteParams.number} buys at different prices between -${marketMakerConfig.quoteParams.minDepth}% to -${marketMakerConfig.quoteParams.maxDepth}% and ${marketMakerConfig.quoteParams.number} sells at different prices between +${marketMakerConfig.quoteParams.minDepth}% to +${marketMakerConfig.quoteParams.maxDepth}% with increasing quantities further from the live price. 
-        You have an allowance of ${marketMakerConfig.allowance.base} ${marketMakerConfig.baseToken} and ${marketMakerConfig.allowance.quote} ${marketMakerConfig.quoteToken}.
-        Important! Only send 1 transaction, buy and sells can be combined into a single solana_batch_order.`;
-
-        const stream = await agent.stream(
-          { messages: [new HumanMessage(thought)] },
-          config,
-        );
-
-        for await (const chunk of stream) {
-          if ("agent" in chunk) {
-            console.log(chunk.agent.messages[0].content);
-          } else if ("tools" in chunk) {
-            console.log(chunk.tools.messages[0].content);
-          }
-          console.log("-------------------");
-        }
-
-        await new Promise((resolve) => 
-          setTimeout(resolve, marketMakerConfig.intervalSeconds * 1000)
-        );
-      } catch (error) {
-        if (error instanceof Error) {
-          console.error("Error:", error);
-        }
-        process.exit(1);
-      }
-    }
-  } catch (error) {
-    console.error("Configuration error:", error);
-    process.exit(1);
-  }
-}
-
-async function chooseMode(): Promise<"chat" | "auto" | "mm"> {
+async function chooseMode(): Promise<"chat" | "auto"> {
  const rl = readline.createInterface({
    input: process.stdin,
    output: process.stdout,
@@ -310,7 +178,6 @@ async function chooseMode(): Promise<"chat" | "auto" | "mm"> {
    console.log("\nAvailable modes:");
    console.log("1. chat    - Interactive chat mode");
    console.log("2. auto    - Autonomous action mode");
-    console.log("3. mm      - AI guided market making")

    const choice = (await question("\nChoose a mode (enter number or name): "))
      .toLowerCase()
@@ -322,8 +189,6 @@ async function chooseMode(): Promise<"chat" | "auto" | "mm"> {
      return "chat";
    } else if (choice === "2" || choice === "auto") {
      return "auto";
-    } else if (choice === "3" || choice === "mm") {
-      return "mm";
    }
    console.log("Invalid choice. Please try again.");
  }
@@ -337,10 +202,8 @@ async function main() {

    if (mode === "chat") {
      await runChatMode(agent, config);
-    } else if( mode === "auto") {
-      await runAutonomousMode(agent, config);
    } else {
-      await runMarketMakerMode(agent, config);
+      await runAutonomousMode(agent, config);
    }
  } catch (error) {
    if (error instanceof Error) {