Merge branch 'main' into no-log-kp

Author: michaelessiet

.env.example

@@ -6,3 +6,6 @@ JUPITER_FEE_BPS=
 FLASH_PRIVILEGE= referral | nft | none
 FLEXLEND_API_KEY=
 HELIUS_API_KEY=
+ALLORA_API_KEY=
+ALLORA_API_URL=
+ALLORA_NETWORK= testnet | mainnet

README.md

@@ -508,6 +508,29 @@ const signature = await agent.voltrWithdrawStrategy(
 const asset = await agent.getAsset("41Y8C4oxk4zgJT1KXyQr35UhZcfsp5mP86Z2G7UUzojU")
 ```
 
+### Get a price inference from Allora
+
+Get the price for a given token and timeframe from Allora's API
+
+```typescript
+const sol5mPrice = await agent.getPriceInference("SOL", "5m");
+console.log("5m price inference of SOL/USD:", sol5mPrice);
+```
+
+### List all topics from Allora
+
+```typescript
+const topics = await agent.getAllTopics();
+console.log("Allora topics:", topics);
+```
+
+### Get an inference for an specific topic from Allora
+
+```typescript
+const inference = await agent.getInferenceByTopicId(42);
+console.log("Allora inference for topic 42:", inference);
+```
+
 ## Examples
 
 ### LangGraph Multi-Agent System

package.json

@@ -31,6 +31,7 @@
   "dependencies": {
     "@3land/listings-sdk": "^0.0.7",
     "@ai-sdk/openai": "^1.0.11",
+    "@alloralabs/allora-sdk": "^0.1.0",
     "@bonfida/spl-name-service": "^3.0.7",
     "@cks-systems/manifest-sdk": "0.1.59",
     "@coral-xyz/anchor": "0.29",

pnpm-lock.yaml

@@ -0,0 +1,42 @@
+import { Action } from "../../types/action";
+import { SolanaAgentKit } from "../../agent";
+import { z } from "zod";
+import { getAllTopics } from "../../tools";
+
+const getAllTopicsAction: Action = {
+  name: "ALLORA_GET_ALL_TOPICS",
+  similes: [
+    "get all topics",
+    "get all inference topics",
+    "get allora topics",
+    "get all allora topics",
+    "get allora inference topics",
+    "get all allora inference topics",
+  ],
+  description: "Get all topics from Allora's API",
+  examples: [
+    [
+      {
+        input: {},
+        output: {
+          status: "success",
+          message: "Topics fetched successfully",
+          topics:
+            '[{"topic_id":5,"topic_name":"SOL 10min Prediction","description":null,"epoch_length":120,"ground_truth_lag":120,"loss_method":"mse","worker_submission_window":12,"worker_count":26859,"reputer_count":2735,"total_staked_allo":1.0200000004892,"total_emissions_allo":12.610109325686093,"is_active":true,"is_endorsed":false,"forge_competition_id":null,"forge_competition_start_date":null,"forge_competition_end_date":null,"updated_at":"2025-01-21T17:21:21.321Z"}]',
+        },
+        explanation: "Get all topics from Allora's API",
+      },
+    ],
+  ],
+  schema: z.object({}),
+  handler: async (agent: SolanaAgentKit) => {
+    const topics = await getAllTopics(agent);
+    return {
+      status: "success",
+      message: "Topics fetched successfully",
+      topics: topics,
+    };
+  },
+};
+
+export default getAllTopicsAction;

src/actions/allora/getAllTopics.ts

@@ -0,0 +1,50 @@
+import { Action } from "../../types/action";
+import { SolanaAgentKit } from "../../agent";
+import { z } from "zod";
+import { getInferenceByTopicId } from "../../tools";
+
+const getInferenceByTopicIdAction: Action = {
+  name: "ALLORA_GET_INFERENCE_BY_TOPIC_ID",
+  similes: ["get allora inference for topic 42", "get inference for topic 42"],
+  description: "Get the inference for a given topic ID from Allora's API",
+  examples: [
+    [
+      {
+        input: {
+          topicId: "42",
+        },
+        output: {
+          status: "success",
+          message: "Inference fetched successfully",
+          inference: "The inference for topic 42 is 100",
+        },
+        explanation: "Get the inference for topic 42",
+      },
+    ],
+  ],
+  schema: z.object({
+    topicId: z
+      .string()
+      .min(1)
+      .describe("The topic ID to get the inference for"),
+  }),
+  handler: async (agent: SolanaAgentKit, input: Record<string, any>) => {
+    try {
+      const { topicId } = input;
+
+      const inference = await getInferenceByTopicId(agent, topicId);
+      return {
+        status: "success",
+        message: "Inference fetched successfully",
+        inference: `The inference for topic ${topicId} is ${inference}`,
+      };
+    } catch (error: any) {
+      return {
+        status: "error",
+        message: `Failed to fetch inference from Allora: ${error.message}`,
+      };
+    }
+  },
+};
+
+export default getInferenceByTopicIdAction;

src/actions/allora/getInferenceByTopicId.ts

@@ -0,0 +1,63 @@
+import { Action } from "../../types/action";
+import { SolanaAgentKit } from "../../agent";
+import { z } from "zod";
+import { getPriceInference } from "../../tools";
+
+const getPriceInferenceAction: Action = {
+  name: "ALLORA_GET_PRICE_INFERENCE",
+  similes: [
+    "get price inference for SOL in 10m",
+    "get price forecast for SOL in 10m",
+    "get allora price inference for SOL in 10m",
+    "get allora price forecast for SOL in 10m",
+  ],
+  description:
+    "Get the price inference for a given token and timeframe from Allora's API",
+  examples: [
+    [
+      {
+        input: {
+          tokenSymbol: "SOL",
+          timeframe: "10m",
+        },
+        output: {
+          status: "success",
+          message: "Price inference fetched successfully",
+          priceInference:
+            "The price of SOL is expected to be 100 in 10 minutes",
+        },
+        explanation:
+          "Get the price inference for SOL/USD price feed for the next 10 minutes",
+      },
+    ],
+  ],
+  schema: z.object({
+    tokenSymbol: z
+      .string()
+      .min(1)
+      .describe("The token symbol to get the price inference for"),
+    timeframe: z
+      .string()
+      .min(1)
+      .describe("The timeframe to get the price inference for"),
+  }),
+  handler: async (agent: SolanaAgentKit, input: any) => {
+    try {
+      const { tokenSymbol, timeframe } = input;
+
+      const price = await getPriceInference(agent, tokenSymbol, timeframe);
+      return {
+        status: "success",
+        message: "Price inference fetched successfully",
+        priceInference: `The price of ${tokenSymbol} is expected to be ${price} in ${timeframe}`,
+      };
+    } catch (error: any) {
+      return {
+        status: "error",
+        message: `Failed to fetch price inference from Allora: ${error.message}`,
+      };
+    }
+  },
+};
+
+export default getPriceInferenceAction;

src/actions/allora/getPriceInference.ts

@@ -76,7 +76,9 @@ import getAssetAction from "./metaplex/getAsset";
 import getAssetsByAuthorityAction from "./metaplex/getAssetsByAuthority";
 import getAssetsByCreatorAction from "./metaplex/getAssetsByCreator";
 import getInfoAction from "./agent/get_info";
-
+import getPriceInferenceAction from "./allora/getPriceInference";
+import getAllTopicsAction from "./allora/getAllTopics";
+import getInferenceByTopicIdAction from "./allora/getInferenceByTopicId";
 export const ACTIONS = {
   GET_INFO_ACTION: getInfoAction,
   WALLET_ADDRESS_ACTION: getWalletAddressAction,
@@ -158,6 +160,9 @@ export const ACTIONS = {
   GET_ASSET_ACTION: getAssetAction,
   GET_ASSETS_BY_AUTHORITY_ACTION: getAssetsByAuthorityAction,
   GET_ASSETS_BY_CREATOR_ACTION: getAssetsByCreatorAction,
+  GET_PRICE_INFERENCE_ACTION: getPriceInferenceAction,
+  GET_ALL_TOPICS_ACTION: getAllTopicsAction,
+  GET_INFERENCE_BY_TOPIC_ID_ACTION: getInferenceByTopicIdAction,
 };
 
 export type { Action, ActionExample, Handler } from "../types/action";

src/actions/index.ts

@@ -117,6 +117,9 @@ import {
   get_asset,
   get_assets_by_authority,
   get_assets_by_creator,
+  getPriceInference,
+  getAllTopics,
+  getInferenceByTopicId,
 } from "../tools";
 import {
   Config,
@@ -141,6 +144,7 @@ import {
   GetAssetsByCreatorRpcInput,
   SearchAssetsRpcInput,
 } from "@metaplex-foundation/digital-asset-standard-api";
+import { AlloraInference, AlloraTopic } from "@alloralabs/allora-sdk";
 
 /**
  * Main class for interacting with Solana blockchain
@@ -1029,4 +1033,16 @@ export class SolanaAgentKit {
   ): Promise<DasApiAssetList> {
     return get_assets_by_creator(this, params);
   }
+  async getPriceInference(
+    tokenSymbol: string,
+    timeframe: string,
+  ): Promise<string> {
+    return getPriceInference(this, tokenSymbol, timeframe);
+  }
+  async getAllTopics(): Promise<AlloraTopic[]> {
+    return getAllTopics(this);
+  }
+  async getInferenceByTopicId(topicId: number): Promise<AlloraInference> {
+    return getInferenceByTopicId(this, topicId);
+  }
 }

src/agent/index.ts

@@ -0,0 +1,35 @@
+import { Tool } from "langchain/tools";
+import { SolanaAgentKit } from "../../agent";
+import { AlloraGetAllTopicsResponse } from "../../index";
+
+export class SolanaAlloraGetAllTopics extends Tool {
+  name = "solana_allora_get_all_topics";
+  description = `Get all topics from Allora's API
+  
+    Inputs: None`;
+
+  constructor(private solanaKit: SolanaAgentKit) {
+    super();
+  }
+
+  async _call(_: string): Promise<string> {
+    try {
+      const topics = await this.solanaKit.getAllTopics();
+
+      const response: AlloraGetAllTopicsResponse = {
+        status: "success",
+        message: "Topics fetched successfully",
+        topics,
+      };
+
+      return JSON.stringify(response);
+    } catch (error: any) {
+      const response: AlloraGetAllTopicsResponse = {
+        status: "error",
+        message: error.message,
+        code: error.code || "UNKNOWN_ERROR",
+      };
+      return JSON.stringify(response);
+    }
+  }
+}

src/langchain/allora/get_all_topics.ts

@@ -0,0 +1,38 @@
+import { Tool } from "langchain/tools";
+import { SolanaAgentKit } from "../../agent";
+import { AlloraGetInferenceByTopicIdResponse } from "../../types";
+
+export class SolanaAlloraGetInferenceByTopicId extends Tool {
+  name = "solana_allora_get_inference_by_topic_id";
+  description = `Get the inference for a given topic ID from Allora's API
+    Inputs:
+    topicId: number as a string, e.g., "42"`;
+
+  constructor(private solanaKit: SolanaAgentKit) {
+    super();
+  }
+
+  async _call(input: string): Promise<string> {
+    try {
+      const topicId = Number(input);
+
+      const inference = await this.solanaKit.getInferenceByTopicId(topicId);
+
+      const response: AlloraGetInferenceByTopicIdResponse = {
+        status: "success",
+        message: "Inference fetched successfully",
+        topicId,
+        inference,
+      };
+
+      return JSON.stringify(response);
+    } catch (error: any) {
+      const response: AlloraGetInferenceByTopicIdResponse = {
+        status: "error",
+        message: error.message,
+        code: error.code || "UNKNOWN_ERROR",
+      };
+      return JSON.stringify(response);
+    }
+  }
+}