[autofix.ci] apply automated fixes

anti-work · Jan 4, 2025 · 27fab9b · 27fab9b
1 parent f7c5c76
commit 27fab9b
Show file tree

Hide file tree

Showing 3 changed files with 30 additions and 18 deletions.
diff --git a/packages/shortest/src/ai/client.ts b/packages/shortest/src/ai/client.ts
@@ -36,7 +36,11 @@ export class AIClient {
       content: Anthropic.Beta.Messages.BetaContentBlockParam,
     ) => void,
     toolOutputCallback?: (name: string, input: any) => void,
-  ): Promise<{ finalResponse: any; tokenUsage: { input: number; output: number }; pendingCache: any }> {
+  ): Promise<{
+    finalResponse: any;
+    tokenUsage: { input: number; output: number };
+    pendingCache: any;
+  }> {
     const maxRetries = 3;
     let attempts = 0;
 
@@ -70,7 +74,12 @@ export class AIClient {
       content: Anthropic.Beta.Messages.BetaContentBlockParam,
     ) => void,
     _toolOutputCallback?: (name: string, input: any) => void,
-  ): Promise<{ messages: any; finalResponse: any; pendingCache: any; tokenUsage: { input: number; output: number } }> {
+  ): Promise<{
+    messages: any;
+    finalResponse: any;
+    pendingCache: any;
+    tokenUsage: { input: number; output: number };
+  }> {
     const messages: Anthropic.Beta.Messages.BetaMessageParam[] = [];
     // temp cache store
     const pendingCache: Partial<{ steps?: CacheStep[] }> = {};

diff --git a/packages/shortest/src/core/runner/index.ts b/packages/shortest/src/core/runner/index.ts
@@ -140,7 +140,11 @@ export class TestRunner {
     test: TestFunction,
     context: BrowserContext,
     config: { noCache: boolean } = { noCache: false },
-  ): Promise<{ result: "pass" | "fail"; reason: string; tokenUsage: { input: number; output: number } }> {
+  ): Promise<{
+    result: "pass" | "fail";
+    reason: string;
+    tokenUsage: { input: number; output: number };
+  }> {
     // If it's direct execution, skip AI
     if (test.directExecution) {
       try {
@@ -149,7 +153,7 @@ export class TestRunner {
         return {
           result: "pass" as const,
           reason: "Direct execution successful",
-          tokenUsage: { input: 0, output: 0  },
+          tokenUsage: { input: 0, output: 0 },
         };
       } catch (error) {
         return {
@@ -240,11 +244,11 @@ export class TestRunner {
                     : error instanceof Error
                       ? error.message
                       : String(error),
-                tokenUsage: { input: 0, output: 0  },
+                tokenUsage: { input: 0, output: 0 },
               };
             }
           }
-          return { ...result, tokenUsage: { input: 0, output: 0  } };
+          return { ...result, tokenUsage: { input: 0, output: 0 } };
         } catch {
           // delete stale cached test entry
           await this.cache.delete(test);
@@ -266,7 +270,7 @@ export class TestRunner {
         return {
           result: "fail" as const,
           reason: error instanceof Error ? error.message : String(error),
-          tokenUsage: { input: 0, output: 0  },
+          tokenUsage: { input: 0, output: 0 },
         };
       }
     }
@@ -280,7 +284,7 @@ export class TestRunner {
 
     // Parse AI result first
     const finalMessage = result.finalResponse.content.find(
-      (block:any) =>
+      (block: any) =>
         block.type === "text" &&
         (block as Anthropic.Beta.Messages.BetaTextBlock).text.includes(
           '"result":',
@@ -468,4 +472,4 @@ export class TestRunner {
       process.exit(1);
     }
   }
-}
+}
diff --git a/packages/shortest/src/utils/logger.ts b/packages/shortest/src/utils/logger.ts
@@ -24,7 +24,6 @@ export class Logger {
   private readonly COST_PER_1K_INPUT_TOKENS = 0.015;
   private readonly COST_PER_1K_OUTPUT_TOKENS = 0.075;
 
-
   startFile(file: string) {
     this.currentFile = file.split("/").pop() || file;
     console.log(pc.blue(`\n📄 ${pc.bold(this.currentFile)}`));
@@ -34,7 +33,7 @@ export class Logger {
     name: string | undefined,
     status: "passed" | "failed",
     error?: Error,
-    tokenUsage?: TokenMetrics
+    tokenUsage?: TokenMetrics,
   ) {
     const testName = name || "Unnamed Test";
     const symbol = status === "passed" ? "✓" : "✗";
@@ -48,8 +47,8 @@ export class Logger {
       console.log(
         pc.dim(
           `    ↳ ${totalTokens.toLocaleString()} tokens ` +
-          `(≈ $${cost.toFixed(2)})`
-        )
+            `(≈ $${cost.toFixed(2)})`,
+        ),
       );
     }
 
@@ -111,10 +110,10 @@ export class Logger {
     ).length;
     const passedTests = totalTests - failedTests;
 
-    const { totalInputTokens, totalOutputTokens, totalCost } = this.calculateTotalTokenUsage();
+    const { totalInputTokens, totalOutputTokens, totalCost } =
+      this.calculateTotalTokenUsage();
     const totalTokens = totalInputTokens + totalOutputTokens;
 
-
     console.log(pc.dim("⎯".repeat(50)));
 
     console.log(
@@ -131,11 +130,11 @@ export class Logger {
       pc.dim(new Date(this.startTime).toLocaleTimeString()),
     );
     console.log(
-      pc.bold(" Tokens    "), 
+      pc.bold(" Tokens    "),
       pc.dim(
         `${totalTokens.toLocaleString()} tokens ` +
-        `(≈ $${totalCost.toFixed(2)})`
-      )
+          `(≈ $${totalCost.toFixed(2)})`,
+      ),
     );
     console.log(pc.dim("\n" + "⎯".repeat(50)));
   }