chore: format

Author: sestinj

extensions/cli/src/commands/chat.ts

@@ -174,7 +174,10 @@ async function handleAutoCompaction(
   // Custom callbacks for headless mode console output
   const callbacks = {
     onSystemMessage: (message: string) => {
-      if (message.includes("Auto-compacting") || message.includes("Approaching")) {
+      if (
+        message.includes("Auto-compacting") ||
+        message.includes("Approaching")
+      ) {
         if (!isHeadless) {
           console.info(chalk.yellow(`\n${message}`));
         } else if (format === "json") {

extensions/cli/src/compaction.infiniteLoop.test.ts

@@ -24,25 +24,29 @@ describe("compaction infinite loop prevention", () => {
     defaultCompletionOptions: {
       maxTokens: 1000,
       contextLength: 4000,
-    }
+    },
   } as ModelConfig;
 
   const mockLlmApi = {} as BaseLlmApi;
 
   it("should not loop infinitely when pruning doesn't reduce history size", async () => {
-    const { countChatHistoryTokens, getModelContextLimit } = await import("./util/tokenizer.js");
+    const { countChatHistoryTokens, getModelContextLimit } = await import(
+      "./util/tokenizer.js"
+    );
     const mockStreamResponse = vi.mocked(streamChatResponse);
     const mockCountTokens = vi.mocked(countChatHistoryTokens);
     const mockGetContextLimit = vi.mocked(getModelContextLimit);
 
     // Setup mocks
     mockGetContextLimit.mockReturnValue(4000);
     mockCountTokens.mockReturnValue(5000); // Always too big
-    mockStreamResponse.mockImplementation(async (history, model, api, controller, callbacks) => {
-      callbacks?.onContent?.("Summary");
-      callbacks?.onContentComplete?.();
-      return "Summary";
-    });
+    mockStreamResponse.mockImplementation(
+      async (history, model, api, controller, callbacks) => {
+        callbacks?.onContent?.("Summary");
+        callbacks?.onContentComplete?.();
+        return "Summary";
+      },
+    );
 
     // History that can't be pruned further (only system message)
     const history: ChatCompletionMessageParam[] = [
@@ -51,26 +55,30 @@ describe("compaction infinite loop prevention", () => {
 
     // This should not hang - it should break out of the loop
     const result = await compactChatHistory(history, mockModel, mockLlmApi);
-    
+
     // Should complete successfully even though token count is still too high
     expect(result.compactedHistory).toBeDefined();
     expect(mockCountTokens).toHaveBeenCalled();
   });
 
   it("should not loop infinitely with history ending in assistant message", async () => {
-    const { countChatHistoryTokens, getModelContextLimit } = await import("./util/tokenizer.js");
+    const { countChatHistoryTokens, getModelContextLimit } = await import(
+      "./util/tokenizer.js"
+    );
     const mockStreamResponse = vi.mocked(streamChatResponse);
     const mockCountTokens = vi.mocked(countChatHistoryTokens);
     const mockGetContextLimit = vi.mocked(getModelContextLimit);
 
     // Setup mocks
     mockGetContextLimit.mockReturnValue(4000);
     mockCountTokens.mockReturnValue(5000); // Always too big
-    mockStreamResponse.mockImplementation(async (history, model, api, controller, callbacks) => {
-      callbacks?.onContent?.("Summary");
-      callbacks?.onContentComplete?.();
-      return "Summary";
-    });
+    mockStreamResponse.mockImplementation(
+      async (history, model, api, controller, callbacks) => {
+        callbacks?.onContent?.("Summary");
+        callbacks?.onContentComplete?.();
+        return "Summary";
+      },
+    );
 
     // History that ends with assistant - pruning won't change it
     const history: ChatCompletionMessageParam[] = [
@@ -81,19 +89,21 @@ describe("compaction infinite loop prevention", () => {
 
     // This should not hang
     const result = await compactChatHistory(history, mockModel, mockLlmApi);
-    
+
     expect(result.compactedHistory).toBeDefined();
   });
 
   it("should successfully prune when pruning actually reduces size", async () => {
-    const { countChatHistoryTokens, getModelContextLimit } = await import("./util/tokenizer.js");
+    const { countChatHistoryTokens, getModelContextLimit } = await import(
+      "./util/tokenizer.js"
+    );
     const mockStreamResponse = vi.mocked(streamChatResponse);
     const mockCountTokens = vi.mocked(countChatHistoryTokens);
     const mockGetContextLimit = vi.mocked(getModelContextLimit);
 
     // Setup mocks
     mockGetContextLimit.mockReturnValue(4000);
-    
+
     // Mock token counting to show reduction after pruning
     let callCount = 0;
     mockCountTokens.mockImplementation(() => {
@@ -102,12 +112,14 @@ describe("compaction infinite loop prevention", () => {
       if (callCount === 2) return 3000; // After pruning, fits
       return 2000; // Subsequent calls
     });
-    
-    mockStreamResponse.mockImplementation(async (history, model, api, controller, callbacks) => {
-      callbacks?.onContent?.("Summary");
-      callbacks?.onContentComplete?.();
-      return "Summary";
-    });
+
+    mockStreamResponse.mockImplementation(
+      async (history, model, api, controller, callbacks) => {
+        callbacks?.onContent?.("Summary");
+        callbacks?.onContentComplete?.();
+        return "Summary";
+      },
+    );
 
     // History that can be successfully pruned
     const history: ChatCompletionMessageParam[] = [
@@ -118,9 +130,9 @@ describe("compaction infinite loop prevention", () => {
     ];
 
     const result = await compactChatHistory(history, mockModel, mockLlmApi);
-    
+
     expect(result.compactedHistory).toBeDefined();
     // The function will call countTokens multiple times during the process
     expect(mockCountTokens).toHaveBeenCalled();
   });
-});
+});

extensions/cli/src/compaction.pruneLastMessage.test.ts

@@ -53,16 +53,18 @@ describe("pruneLastMessage", () => {
     ];
 
     const result = pruneLastMessage(history);
-    expect(result).toEqual([
-      { role: "system", content: "System" },
-    ]);
+    expect(result).toEqual([{ role: "system", content: "System" }]);
   });
 
   it("should remove assistant+tool sequence when second-to-last is assistant with tool calls", () => {
     const history: ChatCompletionMessageParam[] = [
       { role: "system", content: "System" },
       { role: "user", content: "Run command" },
-      { role: "assistant", content: "Running...", tool_calls: [{ id: "1" } as any] },
+      {
+        role: "assistant",
+        content: "Running...",
+        tool_calls: [{ id: "1" } as any],
+      },
       { role: "tool", content: "Command output", tool_call_id: "1" },
     ];
 
@@ -95,16 +97,18 @@ describe("pruneLastMessage", () => {
     ];
 
     const result = pruneLastMessage(history);
-    expect(result).toEqual([
-      { role: "system", content: "System" },
-    ]);
+    expect(result).toEqual([{ role: "system", content: "System" }]);
   });
 
   it("should handle tool call sequences by removing user after tool", () => {
     const history: ChatCompletionMessageParam[] = [
       { role: "system", content: "System" },
       { role: "user", content: "Do something" },
-      { role: "assistant", content: "I'll help", tool_calls: [{ id: "1" } as any] },
+      {
+        role: "assistant",
+        content: "I'll help",
+        tool_calls: [{ id: "1" } as any],
+      },
       { role: "tool", content: "Tool result", tool_call_id: "1" },
       { role: "user", content: "Follow up question" },
     ];
@@ -113,7 +117,11 @@ describe("pruneLastMessage", () => {
     expect(result).toEqual([
       { role: "system", content: "System" },
       { role: "user", content: "Do something" },
-      { role: "assistant", content: "I'll help", tool_calls: [{ id: "1" } as any] },
+      {
+        role: "assistant",
+        content: "I'll help",
+        tool_calls: [{ id: "1" } as any],
+      },
       { role: "tool", content: "Tool result", tool_call_id: "1" },
     ]);
   });
@@ -134,7 +142,11 @@ describe("pruneLastMessage", () => {
       { role: "user", content: "Request 1" },
       { role: "assistant", content: "Response 1" },
       { role: "user", content: "Request 2" },
-      { role: "assistant", content: "Using tool", tool_calls: [{ id: "1" } as any] },
+      {
+        role: "assistant",
+        content: "Using tool",
+        tool_calls: [{ id: "1" } as any],
+      },
       { role: "tool", content: "Tool result", tool_call_id: "1" },
       { role: "assistant", content: "Final response" },
       { role: "user", content: "Follow up 1" },
@@ -147,7 +159,11 @@ describe("pruneLastMessage", () => {
       { role: "user", content: "Request 1" },
       { role: "assistant", content: "Response 1" },
       { role: "user", content: "Request 2" },
-      { role: "assistant", content: "Using tool", tool_calls: [{ id: "1" } as any] },
+      {
+        role: "assistant",
+        content: "Using tool",
+        tool_calls: [{ id: "1" } as any],
+      },
       { role: "tool", content: "Tool result", tool_call_id: "1" },
       { role: "assistant", content: "Final response" },
     ]);
@@ -166,4 +182,4 @@ describe("pruneLastMessage", () => {
       { role: "assistant", content: "Hi" },
     ]);
   });
-});
+});