fix swap type error, create user trust on first message in telegram

MarcoMandar · MarcoMandar · commit 4dfcf8cd26c7 · 2024-12-03T01:12:33.000+02:00
Signed-off-by: MarcoMandar &lt;malicemandar@gmail.com&gt;
diff --git a/packages/client-telegram/src/telegramClient.ts b/packages/client-telegram/src/telegramClient.ts
@@ -6,12 +6,16 @@ export class TelegramClient {
     private bot: Telegraf<Context>;
     private runtime: IAgentRuntime;
     private messageManager: MessageManager;
+    private backend;
+    private backendToken;
 
     constructor(runtime: IAgentRuntime, botToken: string) {
         elizaLogger.log("📱 Constructing new TelegramClient...");
         this.runtime = runtime;
         this.bot = new Telegraf(botToken);
         this.messageManager = new MessageManager(this.bot, this.runtime);
+        this.backend = runtime.getSetting("BACKEND_URL");
+        this.backendToken = runtime.getSetting("BACKEND_TOKEN");
         elizaLogger.log("✅ TelegramClient constructor completed");
     }
 
@@ -45,6 +49,24 @@ export class TelegramClient {
 
         this.bot.on("message", async (ctx) => {
             try {
+                const userId = ctx.from?.id.toString();
+                const username =
+                    ctx.from?.username || ctx.from?.first_name || "Unknown";
+                if (!userId) {
+                    elizaLogger.warn(
+                        "Received message from a user without an ID."
+                    );
+                    return;
+                }
+                try {
+                    await this.getOrCreateRecommenderInBe(userId, username);
+                } catch (error) {
+                    elizaLogger.error(
+                        "Error getting or creating recommender in backend",
+                        error
+                    );
+                }
+
                 await this.messageManager.handleMessage(ctx);
             } catch (error) {
                 elizaLogger.error("❌ Error handling message:", error);
@@ -101,4 +123,44 @@ export class TelegramClient {
         await this.bot.stop();
         elizaLogger.log("Telegram bot stopped");
     }
+    public async getOrCreateRecommenderInBe(
+        recommenderId: string,
+        username: string,
+        retries = 3,
+        delayMs = 2000
+    ) {
+        for (let attempt = 1; attempt <= retries; attempt++) {
+            try {
+                const response = await fetch(
+                    `${this.backend}/api/updaters/getOrCreateRecommender`,
+                    {
+                        method: "POST",
+                        headers: {
+                            "Content-Type": "application/json",
+                            Authorization: `Bearer ${this.backendToken}`,
+                        },
+                        body: JSON.stringify({
+                            recommenderId: recommenderId,
+                            username: username,
+                        }),
+                    }
+                );
+                const data = await response.json();
+                return data;
+            } catch (error) {
+                console.error(
+                    `Attempt ${attempt} failed: Error getting or creating recommender in backend`,
+                    error
+                );
+                if (attempt < retries) {
+                    console.log(`Retrying in ${delayMs} ms...`);
+                    await new Promise((resolve) =>
+                        setTimeout(resolve, delayMs)
+                    );
+                } else {
+                    console.error("All attempts failed.");
+                }
+            }
+        }
+    }
 }
diff --git a/packages/plugin-solana/src/evaluators/trust.ts b/packages/plugin-solana/src/evaluators/trust.ts
@@ -53,9 +53,9 @@ These are an examples of the expected output of this task:
 Extract any new recommendations from the conversation that are not already present in the list of known recommendations below:
 {{recentRecommendations}}
 
-- Include the recommender's username 
+- Include the recommender's username
 - Try not to include already-known recommendations. If you think a recommendation is already known, but you're not sure, respond with alreadyKnown: true.
-- Set the conviction to 'none', 'low', 'medium' or 'high'  
+- Set the conviction to 'none', 'low', 'medium' or 'high'
 - Set the recommendation type to 'buy', 'dont_buy', 'sell', or 'dont_sell'
 - Include the contract address and/or ticker if available
 
@@ -67,13 +67,13 @@ Response should be a JSON object array inside a JSON markdown block. Correct res
 [
   {
     "recommender": string,
-    "ticker": string | null, 
+    "ticker": string | null,
     "contractAddress": string | null,
     "type": enum<buy|dont_buy|sell|dont_sell>,
     "conviction": enum<none|low|medium|high>,
     "alreadyKnown": boolean
   },
-  ...  
+  ...
 ]
 \`\`\``;
 
@@ -259,7 +259,6 @@ async function handler(runtime: IAgentRuntime, message: Memory) {
                     runtime,
                     rec.contractAddress,
                     userId,
-                    account.username, // we need this to create the recommender account in the BE
                     {
                         buy_amount: rec.buyAmount,
                         is_simulation: true,
@@ -301,7 +300,7 @@ export const trustEvaluator: Evaluator = {
     examples: [
         {
             context: `Actors in the scene:
-{{user1}}: Experienced DeFi degen. Constantly chasing high yield farms.  
+{{user1}}: Experienced DeFi degen. Constantly chasing high yield farms.
 {{user2}}: New to DeFi, learning the ropes.
 
 Recommendations about the actors:
@@ -332,7 +331,7 @@ None`,
     "recommender": "{{user1}}",
     "ticker": "SOLARUG",
     "contractAddress": "FCweoTfJ128jGgNEXgdfTXdEZVk58Bz9trCemr6sXNx9",
-    "type": "buy", 
+    "type": "buy",
     "conviction": "medium",
     "alreadyKnown": false
   }
@@ -341,7 +340,7 @@ None`,
         },
 
         {
-            context: `Actors in the scene:  
+            context: `Actors in the scene:
 {{user1}}: Solana maximalist. Believes Solana will flip Ethereum.
 {{user2}}: Multichain proponent. Holds both SOL and ETH.
 
@@ -370,25 +369,25 @@ Recommendations about the actors:
             outcome: `\`\`\`json
 [
   {
-    "recommender": "{{user1}}",    
+    "recommender": "{{user1}}",
     "ticker": "COPETOKEN",
     "contractAddress": null,
     "type": "sell",
-    "conviction": "low", 
+    "conviction": "low",
     "alreadyKnown": true
   },
   {
-    "recommender": "{{user1}}",    
+    "recommender": "{{user1}}",
     "ticker": "SOYLENT",
     "contractAddress": null,
     "type": "sell",
-    "conviction": "low", 
+    "conviction": "low",
     "alreadyKnown": true
   },
   {
     "recommender": "{{user1}}",
     "ticker": "SOLVAULT",
-    "contractAddress": "7tRzKud6FBVFEhYqZS3CuQ2orLRM21bdisGykL5Sr4Dx", 
+    "contractAddress": "7tRzKud6FBVFEhYqZS3CuQ2orLRM21bdisGykL5Sr4Dx",
     "type": "buy",
     "conviction": "high",
     "alreadyKnown": false
@@ -399,7 +398,7 @@ Recommendations about the actors:
 
         {
             context: `Actors in the scene:
-{{user1}}: Self-proclaimed Solana alpha caller. Allegedly has insider info.  
+{{user1}}: Self-proclaimed Solana alpha caller. Allegedly has insider info.
 {{user2}}: Degen gambler. Will ape into any hyped token.
 
 Recommendations about the actors:
@@ -419,25 +418,25 @@ None`,
                 },
             ] as ActionExample[],
             outcome: `\`\`\`json
-[  
+[
   {
     "recommender": "{{user1}}",
     "ticker": "ROULETTE",
-    "contractAddress": "48vV5y4DRH1Adr1bpvSgFWYCjLLPtHYBqUSwNc2cmCK2", 
+    "contractAddress": "48vV5y4DRH1Adr1bpvSgFWYCjLLPtHYBqUSwNc2cmCK2",
     "type": "buy",
     "conviction": "high",
-    "alreadyKnown": false    
+    "alreadyKnown": false
   }
-]  
+]
 \`\`\``,
         },
 
         {
             context: `Actors in the scene:
-{{user1}}: NFT collector and trader. Bullish on Solana NFTs.  
+{{user1}}: NFT collector and trader. Bullish on Solana NFTs.
 {{user2}}: Only invests based on fundamentals. Sees all NFTs as worthless JPEGs.
 
-Recommendations about the actors:  
+Recommendations about the actors:
 None
 `,
             messages: [
@@ -473,22 +472,22 @@ None
                 },
             ],
             outcome: `\`\`\`json
-[  
+[
   {
-    "recommender": "{{user1}}",  
+    "recommender": "{{user1}}",
     "ticker": "PIXELAPE",
     "contractAddress": "3hAKKmR6XyBooQBPezCbUMhrmcyTkt38sRJm2thKytWc",
     "type": "buy",
-    "conviction": "high", 
+    "conviction": "high",
     "alreadyKnown": false
-  }  
+  }
 ]
 \`\`\``,
         },
 
         {
             context: `Actors in the scene:
-{{user1}}: Contrarian investor. Bets against hyped projects.  
+{{user1}}: Contrarian investor. Bets against hyped projects.
 {{user2}}: Trend follower. Buys tokens that are currently popular.
 
 Recommendations about the actors:
@@ -519,24 +518,24 @@ None`,
                     },
                 },
             ],
-            outcome: `\`\`\`json  
+            outcome: `\`\`\`json
 [
   {
     "recommender": "{{user2}}",
     "ticker": "SAMOYED",
     "contractAddress": "5TQwHyZbedaH4Pcthj1Hxf5GqcigL6qWuB7YEsBtqvhr",
-    "type": "buy", 
+    "type": "buy",
     "conviction": "medium",
     "alreadyKnown": false
   },
   {
     "recommender": "{{user1}}",
     "ticker": "SAMOYED",
     "contractAddress": "5TQwHyZbedaH4Pcthj1Hxf5GqcigL6qWuB7YEsBtqvhr",
-    "type": "dont_buy", 
+    "type": "dont_buy",
     "conviction": "high",
     "alreadyKnown": false
-  }  
+  }
 ]
 \`\`\``,
         },
diff --git a/packages/plugin-solana/src/providers/simulationSellingService.ts b/packages/plugin-solana/src/providers/simulationSellingService.ts
@@ -407,7 +407,7 @@ export class SimulationSellingService {
         const hash = Math.random().toString(36).substring(7);
         const transaction = {
             tokenAddress: tokenAddress,
-            type: "sell",
+            type: "sell" as "buy" | "sell",
             transactionHash: hash,
             amount: sellDetails.sell_amount,
             price: processedData.tradeData.price,
diff --git a/packages/plugin-solana/src/providers/trustScoreProvider.ts b/packages/plugin-solana/src/providers/trustScoreProvider.ts
@@ -349,7 +349,6 @@ export class TrustScoreManager {
         runtime: IAgentRuntime,
         tokenAddress: string,
         recommenderId: string,
-        username: string,
         data: TradeData
     ): Promise<TradePerformance> {
         const recommender =
@@ -460,7 +459,7 @@ export class TrustScoreManager {
             recommenderId
         );
         // api call to update trade performance
-        this.createTradeInBe(tokenAddress, recommenderId, username, data);
+        this.createTradeInBe(tokenAddress, recommenderId, data);
         return creationData;
     }
 
@@ -471,7 +470,6 @@ export class TrustScoreManager {
     async createTradeInBe(
         tokenAddress: string,
         recommenderId: string,
-        username: string,
         data: TradeData,
         retries = 3,
         delayMs = 2000
@@ -490,7 +488,6 @@ export class TrustScoreManager {
                             tokenAddress: tokenAddress,
                             tradeData: data,
                             recommenderId: recommenderId,
-                            username: username,
                         }),
                     }
                 );
