From a1f58157a1af4ec2ad838655eb5dc823ee3b45f7 Mon Sep 17 00:00:00 2001
From: Aaron Delasy <aaron@thelang.io>
Date: Wed, 23 Oct 2024 14:51:20 +0300
Subject: [PATCH] Add 3 attempts to run a model

---
 models/index.ts | 28 ++++++++++++++++------------
 1 file changed, 16 insertions(+), 12 deletions(-)

diff --git a/models/index.ts b/models/index.ts
index 0d39c42..40efa63 100644
--- a/models/index.ts
+++ b/models/index.ts
@@ -6,6 +6,8 @@ import { gpt4o } from "./gpt-4o";
 import { mistralLarge2 } from "./mistral-large-2";
 import { perplexityLlama31 } from "./perplexity-llama-3.1";
 
+const MAX_RETRIES = 3;
+
 export type ModelHandler = (
   prompt: string,
   map: string[][],
@@ -19,12 +21,14 @@ export async function runModel(
   modelId: string,
   map: string[][],
   prompt: string,
+  retry = 1,
 ): Promise<{
   solution?: string[][];
   reasoning: string;
   error?: string;
 }> {
   let result;
+  let reasoning: string | null = null;
 
   try {
     switch (modelId) {
@@ -53,14 +57,13 @@ export async function runModel(
       }
     }
 
+    reasoning = result.reasoning;
+
     const originalMap = ZombieSurvival.cloneMap(map);
     const [playerRow, playerCol] = result.playerCoordinates;
 
     if (originalMap[playerRow][playerCol] !== " ") {
-      return {
-        reasoning: result.reasoning,
-        error: "Tried to place player in a non-empty space",
-      };
+      throw new Error("Tried to place player in a non-empty space");
     }
 
     originalMap[playerRow][playerCol] = "P";
@@ -69,10 +72,7 @@ export async function runModel(
       const [blockRow, blockCol] = block;
 
       if (originalMap[blockRow][blockCol] !== " ") {
-        return {
-          reasoning: result.reasoning,
-          error: "Tried to place block in a non-empty space",
-        };
+        throw new Error("Tried to place block in a non-empty space");
       }
 
       originalMap[blockRow][blockCol] = "B";
@@ -83,9 +83,13 @@ export async function runModel(
       reasoning: result.reasoning,
     };
   } catch (error) {
-    return {
-      reasoning: "Internal error",
-      error: errorMessage(error),
-    };
+    if (retry === MAX_RETRIES || reasoning === null) {
+      return {
+        reasoning: reasoning ?? "Internal error",
+        error: errorMessage(error),
+      };
+    }
+
+    return await runModel(modelId, map, prompt, retry + 1);
   }
 }