68 lines
2.8 KiB
JavaScript
68 lines
2.8 KiB
JavaScript
import { describe, expect, it, vi } from "vitest";
|
|
|
|
describe("OllamaLlmProvider chat capability detection", () => {
|
|
it("uses the configured model for the chat probe and keeps the chat path", async () => {
|
|
const originalBaseUrl = process.env.OLLAMA_BASE_URL;
|
|
const fetchSpy = vi.fn()
|
|
.mockResolvedValueOnce({ ok: true, body: { consume: vi.fn() } })
|
|
.mockResolvedValueOnce({
|
|
ok: true,
|
|
json: async () => ({ message: { content: "{}" } }),
|
|
});
|
|
vi.stubGlobal("fetch", fetchSpy);
|
|
process.env.OLLAMA_BASE_URL = "http://ollama.test";
|
|
|
|
try {
|
|
const { getProvider } = await import("@/lib/llm/provider.js");
|
|
const result = await getProvider().generateReconstruction("prompt", "configured-model");
|
|
|
|
expect(JSON.parse(fetchSpy.mock.calls[0][1].body)).toMatchObject({
|
|
model: "configured-model",
|
|
stream: false,
|
|
});
|
|
expect(fetchSpy.mock.calls[0][0]).toBe("http://ollama.test/api/chat");
|
|
expect(fetchSpy.mock.calls[1][0]).toBe("http://ollama.test/api/chat");
|
|
expect(fetchSpy.mock.calls[1][0]).not.toContain("/api/generate");
|
|
const chatRequest = JSON.parse(fetchSpy.mock.calls[1][1].body);
|
|
expect(chatRequest).toMatchObject({
|
|
model: "configured-model",
|
|
messages: [{ role: "user", content: "prompt" }],
|
|
stream: false,
|
|
});
|
|
expect(chatRequest.format).toBeTypeOf("object");
|
|
expect(chatRequest.format).not.toBe("json");
|
|
const formatText = JSON.stringify(chatRequest.format);
|
|
expect(formatText).toContain("relationship");
|
|
expect(formatText).toContain("evidenceType");
|
|
expect(result).toMatchObject({
|
|
response: {},
|
|
providerApiPath: "/api/chat",
|
|
});
|
|
} finally {
|
|
vi.unstubAllGlobals();
|
|
if (originalBaseUrl === undefined) delete process.env.OLLAMA_BASE_URL;
|
|
else process.env.OLLAMA_BASE_URL = originalBaseUrl;
|
|
}
|
|
});
|
|
|
|
it("reports the actually attempted generate path when generation fails", async () => {
|
|
const originalBaseUrl = process.env.OLLAMA_BASE_URL;
|
|
const fetchSpy = vi.fn()
|
|
.mockResolvedValueOnce({ ok: false, status: 501, body: { consume: vi.fn() } })
|
|
.mockResolvedValueOnce({ ok: false, status: 500, text: async () => "failure" });
|
|
vi.stubGlobal("fetch", fetchSpy);
|
|
process.env.OLLAMA_BASE_URL = "http://ollama.test";
|
|
|
|
try {
|
|
const { getProvider } = await import("@/lib/llm/provider.js");
|
|
await expect(
|
|
getProvider().generateReconstruction("prompt", "configured-model"),
|
|
).rejects.toMatchObject({ providerApiPath: "/api/generate" });
|
|
expect(fetchSpy.mock.calls[1][0]).toBe("http://ollama.test/api/generate");
|
|
} finally {
|
|
vi.unstubAllGlobals();
|
|
if (originalBaseUrl === undefined) delete process.env.OLLAMA_BASE_URL;
|
|
else process.env.OLLAMA_BASE_URL = originalBaseUrl;
|
|
}
|
|
});
|
|
}); |