Spaces:
Running
Running
File size: 3,131 Bytes
9509f5b | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 | import { describe, expect, it } from "vitest";
import type { ServerConfig } from "./config";
import {
applyServerConfig,
defaultSettings,
getInferenceTypes,
} from "./settings";
const mockConfig: ServerConfig = {
accessKeysEnabled: false,
accessKeyTimeoutHours: 0,
wllamaDefaultModelId: "littlelamb-290m",
internalApiEnabled: false,
internalApiName: "Internal API",
defaultInferenceType: "browser",
};
describe("Settings Module", () => {
it("should have correct default settings values", () => {
expect(defaultSettings.showEnableAiResponsePrompt).toBe(true);
expect(defaultSettings.enableAiResponse).toBe(false);
expect(defaultSettings.enableImageSearch).toBe(true);
expect(defaultSettings.searchResultsLimit).toBe(15);
expect(defaultSettings.inferenceType).toBeDefined();
});
it("should include core inference types", () => {
const values = getInferenceTypes(mockConfig).map((i) => i.value);
expect(values).toContain("browser");
expect(values).toContain("openai");
expect(values).toContain("horde");
});
it("should include internal API when enabled in config", () => {
const configWithInternalApi = {
...mockConfig,
internalApiEnabled: true,
internalApiName: "Custom LLM",
};
const values = getInferenceTypes(configWithInternalApi).map((i) => i.value);
expect(values).toContain("internal");
expect(
getInferenceTypes(configWithInternalApi).find(
(i) => i.value === "internal",
)?.label,
).toBe("Custom LLM");
});
it("should exclude internal API when disabled in config", () => {
const values = getInferenceTypes(mockConfig).map((i) => i.value);
expect(values).not.toContain("internal");
});
it("should apply server config defaults to a profile that was never saved", () => {
const config = {
...mockConfig,
wllamaDefaultModelId: "custom-model",
defaultInferenceType: "internal",
};
const applied = applyServerConfig(defaultSettings, config, false);
expect(applied.wllamaModelId).toBe("custom-model");
expect(applied.inferenceType).toBe("internal");
});
it("should preserve stored settings that differ from the server defaults", () => {
const config = {
...mockConfig,
wllamaDefaultModelId: "different-model",
defaultInferenceType: "openai",
};
const userSettings = {
...defaultSettings,
wllamaModelId: "user-picked-model",
inferenceType: "horde",
};
const applied = applyServerConfig(userSettings, config, true);
expect(applied.wllamaModelId).toBe("user-picked-model");
expect(applied.inferenceType).toBe("horde");
});
it("should preserve a stored setting that happens to equal a shipped default", () => {
const config = {
...mockConfig,
wllamaDefaultModelId: "admin-preferred-model",
defaultInferenceType: "internal",
};
const applied = applyServerConfig(defaultSettings, config, true);
expect(applied.wllamaModelId).toBe(defaultSettings.wllamaModelId);
expect(applied.inferenceType).toBe(defaultSettings.inferenceType);
});
});
|