mirror of
https://github.com/Routstr/routstrd.git
synced 2026-07-30 15:46:14 +00:00
98 lines
2.8 KiB
TypeScript
98 lines
2.8 KiB
TypeScript
import { describe, expect, it } from "bun:test";
|
|
import { parse } from "yaml";
|
|
import { mergeHermesConfig } from "../../src/integrations/hermes";
|
|
|
|
const ROUTSTR = {
|
|
baseUrl: "http://localhost:8008/v1",
|
|
apiKey: "routstr-key",
|
|
defaultModel: "deepseek-v4-flash",
|
|
};
|
|
|
|
describe("mergeHermesConfig", () => {
|
|
it("preserves the user's model and unrelated custom providers", () => {
|
|
const existing = `model:
|
|
default: user-model
|
|
provider: custom
|
|
base_url: https://user.example/v1
|
|
api_key: user-key
|
|
|
|
custom_providers:
|
|
- name: User Provider
|
|
base_url: https://user.example/v1
|
|
api_key: user-key
|
|
model: user-model
|
|
|
|
other_setting: true
|
|
`;
|
|
|
|
const merged = parse(mergeHermesConfig(existing, ROUTSTR));
|
|
|
|
expect(merged.model).toEqual({
|
|
default: "user-model",
|
|
provider: "custom",
|
|
base_url: "https://user.example/v1",
|
|
api_key: "user-key",
|
|
});
|
|
expect(merged.other_setting).toBe(true);
|
|
expect(merged.custom_providers).toContainEqual({
|
|
name: "User Provider",
|
|
base_url: "https://user.example/v1",
|
|
api_key: "user-key",
|
|
model: "user-model",
|
|
});
|
|
expect(merged.custom_providers).toContainEqual({
|
|
name: "Routstr (localhost:8008)",
|
|
base_url: ROUTSTR.baseUrl,
|
|
api_key: ROUTSTR.apiKey,
|
|
model: ROUTSTR.defaultModel,
|
|
});
|
|
});
|
|
|
|
it("does not add a model configuration to an existing Hermes file", () => {
|
|
const merged = parse(mergeHermesConfig("other_setting: true\n", ROUTSTR));
|
|
|
|
expect(merged.model).toBeUndefined();
|
|
expect(merged.other_setting).toBe(true);
|
|
});
|
|
|
|
it("creates a Routstr default when creating a new Hermes file", () => {
|
|
const merged = parse(mergeHermesConfig("", ROUTSTR));
|
|
|
|
expect(merged.model).toEqual({
|
|
default: ROUTSTR.defaultModel,
|
|
provider: "custom",
|
|
base_url: ROUTSTR.baseUrl,
|
|
api_key: ROUTSTR.apiKey,
|
|
});
|
|
});
|
|
|
|
it("keeps an existing Routstr provider unchanged without duplicating it", () => {
|
|
const existing = `custom_providers:
|
|
- name: Routstr (old-host:8008)
|
|
base_url: http://old-host:8008/v1
|
|
api_key: old-key
|
|
model: old-model
|
|
`;
|
|
|
|
const once = mergeHermesConfig(existing, ROUTSTR);
|
|
const twice = mergeHermesConfig(once, ROUTSTR);
|
|
const merged = parse(twice);
|
|
const routstrProviders = merged.custom_providers.filter(
|
|
(provider: { name?: string }) => provider.name?.startsWith("Routstr ("),
|
|
);
|
|
|
|
expect(routstrProviders).toHaveLength(1);
|
|
expect(routstrProviders[0]).toEqual({
|
|
name: "Routstr (old-host:8008)",
|
|
base_url: "http://old-host:8008/v1",
|
|
api_key: "old-key",
|
|
model: "old-model",
|
|
});
|
|
expect(twice).toBe(once);
|
|
});
|
|
|
|
it("rejects malformed YAML instead of replacing it", () => {
|
|
expect(() => mergeHermesConfig("model: [unterminated", ROUTSTR)).toThrow();
|
|
});
|
|
});
|