fix: move cache-fetch inside the retry-block.

wix-incubator · Oct 18, 2024 · 24c6d76 · 24c6d76
1 parent ec93a46
commit 24c6d76
Showing 1 changed file with 19 additions and 17 deletions.
diff --git a/src/actions/StepPerformer.ts b/src/actions/StepPerformer.ts
@@ -66,27 +66,29 @@ export class StepPerformer {
 
         const cacheKey = this.generateCacheKey(step, previous, viewHierarchy);
 
-        if (this.cache.has(cacheKey)) {
-            const cachedCode = this.cache.get(cacheKey);
-            return this.codeEvaluator.evaluate(cachedCode, this.context);
-        }
-
-        const prompt = this.promptCreator.createPrompt(
-            step,
-            viewHierarchy,
-            isSnapshotImageAttached,
-            previous,
-        );
-
         let code: string | undefined = undefined;
 
         try {
-            const promptResult = await this.promptHandler.runPrompt(prompt, snapshot);
-            code = extractCodeBlock(promptResult);
+            if (this.cache.has(cacheKey)) {
+                code = this.cache.get(cacheKey);
+            } else {
+                const prompt = this.promptCreator.createPrompt(
+                    step,
+                    viewHierarchy,
+                    isSnapshotImageAttached,
+                    previous,
+                );
+
+                const promptResult = await this.promptHandler.runPrompt(prompt, snapshot);
+                code = extractCodeBlock(promptResult);
 
-            // Cache the result
-            this.cache.set(cacheKey, code);
-            this.saveCacheToFile();
+                this.cache.set(cacheKey, code);
+                this.saveCacheToFile();
+            }
+
+            if (!code) {
+                throw new Error('Failed to generate code from intent');
+            }
 
             return await this.codeEvaluator.evaluate(code, this.context);
         } catch (error) {