Skip to content

Commit fa8b93f

Browse files
committed
Use local embedders to avoid using open ai
1 parent a5dc5cf commit fa8b93f

File tree

1 file changed

+11
-21
lines changed

1 file changed

+11
-21
lines changed

tests/embedders.test.ts

Lines changed: 11 additions & 21 deletions
Original file line numberDiff line numberDiff line change
@@ -240,7 +240,6 @@ describe.each([{ permission: "Master" }, { permission: "Admin" }])(
240240

241241
test(`${permission} key: Update embedders with composite embedder`, async () => {
242242
const adminKey = await getKey("Admin");
243-
const embedderName = "composite_embedder";
244243

245244
// first enable the network endpoint.
246245
await fetch(`${HOST}/experimental-features`, {
@@ -254,19 +253,21 @@ describe.each([{ permission: "Master" }, { permission: "Admin" }])(
254253

255254
const client = await getClient(permission);
256255
const embedders = {
257-
[embedderName]: {
256+
default: {
258257
source: "composite",
259258
searchEmbedder: {
260-
source: "openAi",
261-
model: "text-embedding-3-small",
262-
dimensions: 1536,
259+
source: "huggingFace",
260+
model:
261+
"sentence-transformers/paraphrase-multilingual-MiniLM-L12-v2",
262+
pooling: "useModel",
263263
},
264264
indexingEmbedder: {
265-
source: "openAi",
266-
model: "text-embedding-3-small",
265+
source: "huggingFace",
266+
model:
267+
"sentence-transformers/paraphrase-multilingual-MiniLM-L12-v2",
267268
documentTemplate: "{{doc.title}}",
268-
documentTemplateMaxBytes: 400,
269-
dimensions: 1536,
269+
pooling: "useModel",
270+
documentTemplateMaxBytes: 500,
270271
},
271272
},
272273
} satisfies Embedders;
@@ -279,18 +280,7 @@ describe.each([{ permission: "Master" }, { permission: "Admin" }])(
279280

280281
const processedTask = await client.getTask(task.taskUid);
281282
expect(processedTask.status).toEqual("succeeded");
282-
283-
expect(response).toEqual({
284-
[embedderName]: {
285-
source: "composite",
286-
searchEmbedder: {
287-
...embedders[embedderName].searchEmbedder,
288-
},
289-
indexingEmbedder: {
290-
...embedders[embedderName].indexingEmbedder,
291-
},
292-
},
293-
});
283+
expect(response).toEqual(embedders);
294284
});
295285

296286
test(`${permission} key: Reset embedders`, async () => {

0 commit comments

Comments
 (0)