fix: fireworks errors are absorbed

empirical-run · Apr 30, 2024 · f478c5e · f478c5e
1 parent c7e5a4a
commit f478c5e
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/packages/ai/src/providers/fireworks/index.ts b/packages/ai/src/providers/fireworks/index.ts
@@ -77,7 +77,11 @@ const createChatCompletion: ICreateChatCompletion = async (body) => {
     executionDone();
     return { ...completion, latency };
   } catch (err) {
-    throw new AIError(AIErrorEnum.FAILED_CHAT_COMPLETION, "Unknown error");
+    if (err instanceof AIError) {
+      throw err;
+    } else {
+      throw new AIError(AIErrorEnum.FAILED_CHAT_COMPLETION, "Unknown error");
+    }
   }
 };