move types and created actions

src/actions/flashCloseTrade.ts

@@ -0,0 +1,68 @@
+import { Action } from "../types/action";
+import { SolanaAgentKit } from "../agent";
+import { z } from "zod";
+import { flashCloseTrade } from "../tools";
+
+const flashCloseTradeAction: Action = {
+  name: "FLASH_CLOSE_TRADE",
+  similes: [
+    "close trade",
+    "close leveraged trade",
+    "exit position",
+    "close position",
+    "exit trade",
+    "close long",
+    "close short",
+    "take profit",
+    "stop loss",
+  ],
+  description:
+    "Close an existing leveraged trading position on Flash.Trade protocol",
+  examples: [
+    [
+      {
+        input: {
+          token: "SOL",
+          side: "long",
+        },
+        output: {
+          status: "success",
+          signature: "4xKpN2...",
+          message: "Successfully closed long position on SOL",
+        },
+        explanation: "Close an existing long position on SOL",
+      },
+    ],
+  ],
+  schema: z.object({
+    token: z
+      .string()
+      .describe("Token symbol of the position to close (e.g. SOL, ETH)"),
+    side: z
+      .enum(["long", "short"])
+      .describe("Position side to close - long or short"),
+  }),
+  handler: async (agent: SolanaAgentKit, input: Record<string, any>) => {
+    try {
+      const params = {
+        token: input.token as string,
+        side: input.side as "long" | "short",
+      };
+
+      const response = await flashCloseTrade(agent, params);
+
+      return {
+        status: "success",
+        signature: response,
+        message: `Successfully closed ${params.side} position on ${params.token}`,
+      };
+    } catch (error: any) {
+      return {
+        status: "error",
+        message: `Flash trade close failed: ${error.message}`,
+      };
+    }
+  },
+};
+
+export default flashCloseTradeAction;

src/actions/flashOpenTrade.ts

@@ -0,0 +1,78 @@
+import { Action } from "../types/action";
+import { SolanaAgentKit } from "../agent";
+import { z } from "zod";
+import { flashOpenTrade } from "../tools";
+
+const flashOpenTradeAction: Action = {
+  name: "FLASH_OPEN_TRADE",
+  similes: [
+    "open trade",
+    "open leveraged trade",
+    "start trading position",
+    "open position",
+    "long position",
+    "short position",
+    "leverage trade",
+    "margin trade",
+  ],
+  description: "Open a leveraged trading position on Flash.Trade protocol",
+  examples: [
+    [
+      {
+        input: {
+          token: "SOL",
+          side: "long",
+          collateralUsd: 100,
+          leverage: 5,
+        },
+        output: {
+          status: "success",
+          signature: "4xKpN2...",
+          message:
+            "Successfully opened 5x long position on SOL with $100 collateral",
+        },
+        explanation:
+          "Open a 5x leveraged long position on SOL using $100 as collateral",
+      },
+    ],
+  ],
+  schema: z.object({
+    token: z.string().describe("Token symbol to trade (e.g. SOL, ETH)"),
+    side: z
+      .enum(["long", "short"])
+      .describe("Trading direction - long or short"),
+    collateralUsd: z
+      .number()
+      .positive()
+      .describe("Amount of collateral in USD"),
+    leverage: z
+      .number()
+      .positive()
+      .describe("Leverage multiplier for the trade"),
+  }),
+  handler: async (agent: SolanaAgentKit, input: Record<string, any>) => {
+    try {
+      const params = {
+        token: input.token as string,
+        side: input.side as "long" | "short",
+        collateralUsd: input.collateralUsd as number,
+        leverage: input.leverage as number,
+      };
+
+      const response = await flashOpenTrade(agent, params);
+
+      return {
+        status: "success",
+        signature: response,
+        message: `Successfully opened ${params.leverage}x ${params.side} position on ${params.token} with $${params.collateralUsd} collateral`,
+      };
+    } catch (error: any) {
+      return {
+        status: "error",
+        message: `Flash trade failed: ${error.message}`,
+      };
+    }
+  },
+};
+
+export default flashOpenTradeAction;

src/agent/index.ts

@@ -60,8 +60,6 @@ import {
   fetchTokenDetailedReport,
   fetchPythPrice,
   fetchPythPriceFeedID,
-  FlashTradeParams,
-  FlashCloseTradeParams,
   flashOpenTrade,
   flashCloseTrade,
 } from "../tools";
@@ -74,6 +72,8 @@ import {
   PumpfunLaunchResponse,
   PumpFunTokenOptions,
   OrderParams,
+  FlashTradeParams,
+  FlashCloseTradeParams,
 } from "../types";
 
 /**

src/tools/flash_close_trade.ts

@@ -16,11 +16,7 @@ import {
   fetchOraclePrice,
   createPerpClient,
 } from "../utils/flashUtils";
-
-export interface FlashCloseTradeParams {
-  token: string;
-  side: "long" | "short";
-}
+import { FlashCloseTradeParams } from "../types";
 
 /**
  * Closes an existing position on Flash.Trade
@@ -64,7 +60,10 @@ export async function flashCloseTrade(
     ]);
 
     // Initialize pool configuration and perpClient
-    const poolConfig = PoolConfig.fromIdsByName(marketData.pool, "mainnet-beta");
+    const poolConfig = PoolConfig.fromIdsByName(
+      marketData.pool,
+      "mainnet-beta",
+    );
     const perpClient = createPerpClient(agent.connection, agent.wallet);
 
     // Calculate price after slippage

src/tools/flash_open_trade.ts

@@ -19,13 +19,7 @@ import {
   fetchOraclePrice,
   createPerpClient,
 } from "../utils/flashUtils";
-
-export interface FlashTradeParams {
-  token: string;
-  side: "long" | "short";
-  collateralUsd: number;
-  leverage: number;
-}
+import { FlashTradeParams } from "../types";
 
 /**
  * Opens a new position on Flash.Trade
@@ -77,7 +71,10 @@ export async function flashOpenTrade(
     ]);
 
     // Initialize pool configuration and perpClient
-    const poolConfig = PoolConfig.fromIdsByName(marketData.pool, "mainnet-beta");
+    const poolConfig = PoolConfig.fromIdsByName(
+      marketData.pool,
+      "mainnet-beta",
+    );
     const perpClient = createPerpClient(agent.connection, agent.wallet);
 
     // Calculate position parameters

src/types/index.ts

@@ -224,3 +224,15 @@ export interface BatchOrderPattern {
   numberOfOrders?: number;
   individualQuantity?: number;
 }
+
+export interface FlashTradeParams {
+  token: string;
+  side: "long" | "short";
+  collateralUsd: number;
+  leverage: number;
+}
+
+export interface FlashCloseTradeParams {
+  token: string;
+  side: "long" | "short";
+}