|
1 |
| -import { |
2 |
| - Client, |
3 |
| - IAgentRuntime |
4 |
| -} from "@ai16z/eliza/src/types.ts"; |
5 |
| - |
| 1 | +import { Client, IAgentRuntime } from "@ai16z/eliza/src/types.ts"; |
| 2 | +import { TrustScoreManager } from "@ai16z/plugin-solana/src/providers/trustScoreProvider.ts"; |
| 3 | +import { TokenProvider } from "@ai16z/plugin-solana/src/providers/token.ts"; |
| 4 | +import { WalletProvider } from "@ai16z/plugin-solana/src/providers/wallet.ts"; |
| 5 | +import { TrustScoreDatabase } from "@ai16z/plugin-solana/src/adapters/trustScoreDatabase.ts"; |
| 6 | +import { Connection, PublicKey } from "@solana/web3.js"; |
6 | 7 |
|
7 | 8 | export class AutoClient {
|
8 | 9 | interval: NodeJS.Timeout;
|
9 | 10 | runtime: IAgentRuntime;
|
| 11 | + trustScoreProvider: TrustScoreManager; |
| 12 | + walletProvider: WalletProvider; |
10 | 13 |
|
11 | 14 | constructor(runtime: IAgentRuntime) {
|
12 | 15 | this.runtime = runtime;
|
13 |
| - // start a loop that runs every x seconds |
14 |
| - this.interval = setInterval(() => { |
15 |
| - this.makeTrades(); |
16 |
| - }, 60 * 60 * 1000); // 1 hour in milliseconds |
17 | 16 |
|
| 17 | + const trustScoreDb = new TrustScoreDatabase(runtime.databaseAdapter.db); |
| 18 | + this.trustScoreProvider = new TrustScoreManager(null, trustScoreDb); |
| 19 | + this.walletProvider = new WalletProvider( |
| 20 | + new Connection(runtime.getSetting("RPC_URL")), |
| 21 | + new PublicKey(runtime.getSetting("WALLET_PUBLIC_KEY")) |
| 22 | + ); |
| 23 | + |
| 24 | + // start a loop that runs every x seconds |
| 25 | + this.interval = setInterval( |
| 26 | + async () => { |
| 27 | + await this.makeTrades(); |
| 28 | + }, |
| 29 | + 60 * 60 * 1000 |
| 30 | + ); // 1 hour in milliseconds |
18 | 31 | }
|
19 | 32 |
|
20 |
| - makeTrades() { |
| 33 | + async makeTrades() { |
21 | 34 | console.log("Running auto loop");
|
22 | 35 |
|
23 | 36 | // malibu todos
|
| 37 | + const startDate = new Date(new Date().getTime() - 60 * 60 * 1000); |
| 38 | + const endDate = new Date(); |
| 39 | + // get recommendations from the last hour (or whatever time period we want) in order desc by trust score |
| 40 | + const recommendations = |
| 41 | + await this.trustScoreProvider.getRecommendations( |
| 42 | + startDate, |
| 43 | + endDate |
| 44 | + ); |
24 | 45 | // get high trust recommendations
|
| 46 | + const highTrustRecommendations = recommendations.filter( |
| 47 | + (r) => r.averageTrustScore > 0.7 |
| 48 | + ); |
25 | 49 |
|
26 | 50 | // get information for all tokens which were recommended
|
| 51 | + const tokenInfos = highTrustRecommendations.map( |
| 52 | + async (highTrustRecommendation) => { |
| 53 | + const tokenProvider = new TokenProvider( |
| 54 | + highTrustRecommendation.tokenAddress, |
| 55 | + this.walletProvider |
| 56 | + ); |
| 57 | + const tokenInfo = await tokenProvider.getProcessedTokenData(); |
| 58 | + const shouldTrade = await tokenProvider.shouldTradeToken(); |
| 59 | + return { tokenInfo, shouldTrade }; |
| 60 | + } |
| 61 | + ); |
| 62 | + |
27 | 63 | // get any additional information we might need
|
28 | 64 | // make sure we're looking at the right tokens and data
|
29 | 65 |
|
|
0 commit comments