Refactor TestRunner: Move AI processing logic to a more appropriate l… (

#248) …ocation Fixes #247
anti-work · Jan 2, 2025 · 007ecf8 · 007ecf8
1 parent 02f111f
commit 007ecf8
Showing 1 changed file with 7 additions and 7 deletions.
diff --git a/packages/shortest/src/core/runner/index.ts b/packages/shortest/src/core/runner/index.ts
@@ -255,13 +255,6 @@ export class TestRunner {
       }
     }
 
-    // Execute test with enhanced prompt
-    const result = await aiClient.processAction(prompt, browserTool);
-
-    if (!result) {
-      throw new Error("AI processing failed: no result returned");
-    }
-
     // Execute before function if present
     if (test.beforeFn) {
       try {
@@ -274,6 +267,13 @@ export class TestRunner {
       }
     }
 
+    // Execute test with enhanced prompt
+    const result = await aiClient.processAction(prompt, browserTool);
+
+    if (!result) {
+      throw new Error("AI processing failed: no result returned");
+    }
+
     // Parse AI result first
     const finalMessage = result.finalResponse.content.find(
       (block) =>