add retry import

elizaOS · Feb 11, 2025 · 388bd43 · 388bd43
1 parent c934cde
commit 388bd43
Show file tree

Hide file tree

Showing 2 changed files with 47 additions and 4 deletions.
diff --git a/packages/agent/src/index.ts b/packages/agent/src/index.ts
@@ -16,7 +16,8 @@ import {
     parseBooleanFromText,
     settings,
     stringToUuid,
-    validateCharacterConfig
+    validateCharacterConfig,
+    importWithRetry
 } from "@elizaos/core";
 import fs from "node:fs";
 import net from "node:net";
@@ -342,13 +343,19 @@ function initializeCache(
             );
     }
 }
+
 
 async function findDatabaseAdapter(runtime: IAgentRuntime) {
   const { adapters } = runtime;
   let adapter: Adapter | undefined;
   // if not found, default to sqlite
   if (adapters.length === 0) {
-    const sqliteAdapterPlugin = await import('@elizaos-plugins/sqlite');
+    const sqliteAdapterPlugin = await importWithRetry<any>(
+        '@elizaos-plugins/sqlite',
+        3,
+        1000
+    );
+    // const sqliteAdapterPlugin = await import('@elizaos-plugins/sqlite');
     const sqliteAdapterPluginDefault = sqliteAdapterPlugin.default;
     adapter = sqliteAdapterPluginDefault.adapters[0];
     if (!adapter) {

diff --git a/packages/core/src/import.ts b/packages/core/src/import.ts
@@ -1,13 +1,45 @@
 import logger from "./logger";
 
+/**
+ * Attempts to import a module with retry logic.
+ * @param modulePath - The module path to import.
+ * @param retries - Number of retry attempts (default: 3).
+ * @param delay - Delay between retries in milliseconds (default: 1000).
+ * @returns The imported module.
+ * @throws An error if all attempts fail.
+ */
+export async function importWithRetry<T = any>(
+  modulePath: string,
+  retries = 3,
+  delay = 1000
+): Promise<T> {
+  for (let attempt = 1; attempt <= retries; attempt++) {
+    try {
+      return await import(modulePath);
+    } catch (error) {
+      if (attempt === retries) {
+        throw new Error(
+          `Failed to import ${modulePath} after ${retries} attempts: ${error}`
+        );
+      }
+      await new Promise((resolve) => setTimeout(resolve, delay));
+    }
+  }
+  throw new Error("Unexpected error in importWithRetry");
+}
+
 const registrations = new Map<string, any>();
 
 export const dynamicImport = async (specifier: string) => {
   const module = registrations.get(specifier);
   if (module !== undefined) {
     return module;
   } else {
-    return await import(specifier);
+    return await importWithRetry<any>(
+        specifier,
+        3,
+        1000
+    );
   }
 };
 
@@ -21,7 +53,11 @@ export async function handlePluginImporting(plugins: string[]) {
       const importedPlugins = await Promise.all(
           plugins.map(async (plugin) => {
               try {
-                  const importedPlugin = await import(plugin);
+                  const importedPlugin = await importWithRetry<any>(
+                      plugin,
+                      3,
+                      1000
+                  );
                   const functionName =
                       `${plugin
                           .replace("@elizaos/plugin-", "")