feat: make twitter retry limit a env

AbdelStark · Dec 19, 2024 · fe3afef · fe3afef
1 parent 81d0273
commit fe3afef
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 5 deletions.
diff --git a/agent/src/index.ts b/agent/src/index.ts
@@ -573,7 +573,7 @@ function initializeDbCache(character: Character, db: IDatabaseCacheAdapter) {
 
 async function startAgent(
     character: Character,
-    directClient
+    directClient: DirectClient
 ): Promise<AgentRuntime> {
     let db: IDatabaseAdapter & IDatabaseCacheAdapter;
     try {
@@ -648,9 +648,9 @@ const startAgents = async () => {
     }
 
     // upload some agent functionality into directClient
-    directClient.startAgent = async character => {
-      // wrap it so we don't have to inject directClient later
-      return startAgent(character, directClient)
+    directClient.startAgent = async (character: Character) => {
+        // wrap it so we don't have to inject directClient later
+        return startAgent(character, directClient);
     };
     directClient.start(serverPort);
 

diff --git a/packages/client-twitter/src/base.ts b/packages/client-twitter/src/base.ts
@@ -157,6 +157,10 @@ export class ClientBase extends EventEmitter {
         const username = this.runtime.getSetting("TWITTER_USERNAME");
         const password = this.runtime.getSetting("TWITTER_PASSWORD");
         const email = this.runtime.getSetting("TWITTER_EMAIL");
+        let retries = parseInt(
+            this.runtime.getSetting("TWITTER_RETRY_LIMIT") || "5",
+            10
+        );
         const twitter2faSecret =
             this.runtime.getSetting("TWITTER_2FA_SECRET") || undefined;
         const cookies = this.runtime.getSetting("TWITTER_COOKIES");
@@ -180,7 +184,6 @@ export class ClientBase extends EventEmitter {
         }
 
         elizaLogger.log("Waiting for Twitter login");
-        let retries = 5; // Optional: Set a retry limit
         while (retries > 0) {
             const cookies = await this.twitterClient.getCookies();
             if ((await this.twitterClient.isLoggedIn()) && !!cookies) {