mirror of
https://github.com/openclaw/openclaw.git
synced 2026-05-08 09:11:26 +00:00
test(memory): dedupe voyage embedding provider test setup
This commit is contained in:
@@ -19,6 +19,28 @@ const createFetchMock = () => {
|
|||||||
return withFetchPreconnect(fetchMock);
|
return withFetchPreconnect(fetchMock);
|
||||||
};
|
};
|
||||||
|
|
||||||
|
function mockVoyageApiKey() {
|
||||||
|
vi.mocked(authModule.resolveApiKeyForProvider).mockResolvedValue({
|
||||||
|
apiKey: "voyage-key-123",
|
||||||
|
mode: "api-key",
|
||||||
|
source: "test",
|
||||||
|
});
|
||||||
|
}
|
||||||
|
|
||||||
|
async function createDefaultVoyageProvider(
|
||||||
|
model: string,
|
||||||
|
fetchMock: ReturnType<typeof createFetchMock>,
|
||||||
|
) {
|
||||||
|
vi.stubGlobal("fetch", fetchMock);
|
||||||
|
mockVoyageApiKey();
|
||||||
|
return createVoyageEmbeddingProvider({
|
||||||
|
config: {} as never,
|
||||||
|
provider: "voyage",
|
||||||
|
model,
|
||||||
|
fallback: "none",
|
||||||
|
});
|
||||||
|
}
|
||||||
|
|
||||||
describe("voyage embedding provider", () => {
|
describe("voyage embedding provider", () => {
|
||||||
afterEach(() => {
|
afterEach(() => {
|
||||||
vi.resetAllMocks();
|
vi.resetAllMocks();
|
||||||
@@ -27,20 +49,7 @@ describe("voyage embedding provider", () => {
|
|||||||
|
|
||||||
it("configures client with correct defaults and headers", async () => {
|
it("configures client with correct defaults and headers", async () => {
|
||||||
const fetchMock = createFetchMock();
|
const fetchMock = createFetchMock();
|
||||||
vi.stubGlobal("fetch", fetchMock);
|
const result = await createDefaultVoyageProvider("voyage-4-large", fetchMock);
|
||||||
|
|
||||||
vi.mocked(authModule.resolveApiKeyForProvider).mockResolvedValue({
|
|
||||||
apiKey: "voyage-key-123",
|
|
||||||
mode: "api-key",
|
|
||||||
source: "test",
|
|
||||||
});
|
|
||||||
|
|
||||||
const result = await createVoyageEmbeddingProvider({
|
|
||||||
config: {} as never,
|
|
||||||
provider: "voyage",
|
|
||||||
model: "voyage-4-large",
|
|
||||||
fallback: "none",
|
|
||||||
});
|
|
||||||
|
|
||||||
await result.provider.embedQuery("test query");
|
await result.provider.embedQuery("test query");
|
||||||
|
|
||||||
@@ -105,20 +114,7 @@ describe("voyage embedding provider", () => {
|
|||||||
),
|
),
|
||||||
),
|
),
|
||||||
);
|
);
|
||||||
vi.stubGlobal("fetch", fetchMock);
|
const result = await createDefaultVoyageProvider("voyage-4-large", fetchMock);
|
||||||
|
|
||||||
vi.mocked(authModule.resolveApiKeyForProvider).mockResolvedValue({
|
|
||||||
apiKey: "voyage-key-123",
|
|
||||||
mode: "api-key",
|
|
||||||
source: "test",
|
|
||||||
});
|
|
||||||
|
|
||||||
const result = await createVoyageEmbeddingProvider({
|
|
||||||
config: {} as never,
|
|
||||||
provider: "voyage",
|
|
||||||
model: "voyage-4-large",
|
|
||||||
fallback: "none",
|
|
||||||
});
|
|
||||||
|
|
||||||
await result.provider.embedBatch(["doc1", "doc2"]);
|
await result.provider.embedBatch(["doc1", "doc2"]);
|
||||||
|
|
||||||
|
|||||||
Reference in New Issue
Block a user