pi-llama-cpp 0.11.0 → 0.12.0

package/tests/settingsStore.test.ts

@@ -1,11 +1,6 @@
-import { getAgentDir } from "@earendil-works/pi-coding-agent";
 import { beforeEach, describe, expect, it, vi } from "vitest";
 import { SettingsStore } from "../src/utils/settingsStore";
 
-vi.mock("@earendil-works/pi-coding-agent", () => ({
-  getAgentDir: vi.fn().mockReturnValue("/fake/agent/dir"),
-}));
-
 const mockReadFile = vi.hoisted(() => vi.fn());
 const mockWriteFile = vi.hoisted(() => vi.fn());
 const mockRename = vi.hoisted(() => vi.fn());
@@ -16,14 +11,11 @@ vi.mock("node:fs/promises", () => ({
   rename: mockRename,
 }));
 
-const mockGetAgentDir = vi.mocked(getAgentDir);
-
 const SETTINGS_PATH = "/fake/agent/dir/settings.json";
 
 describe("SettingsStore.read", () => {
   beforeEach(() => {
     vi.clearAllMocks();
-    mockGetAgentDir.mockReturnValue("/fake/agent/dir");
   });
 
   it("should return {} when the file is missing (ENOENT)", async () => {
@@ -53,21 +45,11 @@ describe("SettingsStore.read", () => {
 
     await expect(store.read()).rejects.toMatchObject({ code: "EACCES" });
   });
-
-  it("should default the path to <agentDir>/settings.json", async () => {
-    mockReadFile.mockResolvedValue('{"a":1}');
-
-    const store = new SettingsStore();
-    await store.read();
-
-    expect(mockReadFile).toHaveBeenCalledWith(SETTINGS_PATH, "utf-8");
-  });
 });
 
 describe("SettingsStore.updateKey — file content", () => {
   beforeEach(() => {
     vi.clearAllMocks();
-    mockGetAgentDir.mockReturnValue("/fake/agent/dir");
     mockWriteFile.mockResolvedValue(undefined);
     mockRename.mockResolvedValue(undefined);
   });
@@ -122,7 +104,6 @@ describe("SettingsStore.updateKey — file content", () => {
 describe("SettingsStore.updateKey — write path", () => {
   beforeEach(() => {
     vi.clearAllMocks();
-    mockGetAgentDir.mockReturnValue("/fake/agent/dir");
     mockWriteFile.mockResolvedValue(undefined);
     mockRename.mockResolvedValue(undefined);
   });

package/tests/singleModel.test.ts

@@ -47,6 +47,38 @@ describe("SingleModel capabilities", () => {
 
     expect(capabilities).toEqual(["text"]);
   });
+
+  it("should fall back to the models endpoint when auth fails", async () => {
+    mockRpc
+      .mockRejectedValueOnce(new Error("401 Unauthorized")) // /props
+      .mockRejectedValueOnce(new Error("401 Unauthorized")) // /v1/models in BaseModel's catch
+      .mockResolvedValueOnce({
+        object: "list",
+        data: [],
+        models: [{ capabilities: ["multimodal"] }],
+      }); // /v1/models retry in SingleModel's catch
+
+    const model = createModel();
+    const capabilities = await model.getCapabilities();
+
+    expect(capabilities).toEqual(["text", "image"]);
+  });
+
+  it("should fall back to text-only when the models endpoint reports no multimodal", async () => {
+    mockRpc
+      .mockRejectedValueOnce(new Error("401 Unauthorized")) // /props
+      .mockRejectedValueOnce(new Error("401 Unauthorized")) // /v1/models in BaseModel's catch
+      .mockResolvedValueOnce({
+        object: "list",
+        data: [],
+        models: [{ capabilities: ["text"] }],
+      }); // /v1/models retry in SingleModel's catch
+
+    const model = createModel();
+    const capabilities = await model.getCapabilities();
+
+    expect(capabilities).toEqual(["text"]);
+  });
 });
 
 describe("SingleModel getStatus", () => {

package/tests/sseManager.test.ts

@@ -19,12 +19,12 @@ const createManager = (
     "",
   );
 
-describe("SSEManager timeouts", () => {
-  it("should expose the timeouts of its server", () => {
+describe("SSEManager timeouts", async () => {
+  it("should expose the timeouts of its server", async () => {
     const manager = createManager(5678, 1234);
 
-    expect(manager.pollingTimeout).toBe(5678);
-    expect(manager.serverTimeout).toBe(1234);
+    expect(await manager.getPollingTimeout()).toBe(5678);
+    expect(await manager.getServerTimeout()).toBe(1234);
   });
 });