|
@@ -1492,7 +1492,8 @@ export async function generateEmbeddings(
|
|
|
modelArg: string,
|
|
modelArg: string,
|
|
|
): Promise<{ embedding: number[]; model: string } | null> => {
|
|
): Promise<{ embedding: number[]; model: string } | null> => {
|
|
|
if (provider) {
|
|
if (provider) {
|
|
|
- const r = await provider.embed(text, { model: modelArg, signal: session.signal });
|
|
|
|
|
|
|
+ const sig = provider.kind === 'local' ? session.signal : undefined;
|
|
|
|
|
+ const r = await provider.embed(text, { model: modelArg, signal: sig });
|
|
|
return r ? { embedding: r.embedding, model: r.model } : null;
|
|
return r ? { embedding: r.embedding, model: r.model } : null;
|
|
|
}
|
|
}
|
|
|
return session.embed(text, { model: modelArg });
|
|
return session.embed(text, { model: modelArg });
|
|
@@ -1502,7 +1503,8 @@ export async function generateEmbeddings(
|
|
|
modelArg: string,
|
|
modelArg: string,
|
|
|
): Promise<({ embedding: number[]; model: string } | null)[]> => {
|
|
): Promise<({ embedding: number[]; model: string } | null)[]> => {
|
|
|
if (provider) {
|
|
if (provider) {
|
|
|
- const r = await provider.embedBatch(texts, { model: modelArg, signal: session.signal });
|
|
|
|
|
|
|
+ const sig = provider.kind === 'local' ? session.signal : undefined;
|
|
|
|
|
+ const r = await provider.embedBatch(texts, { model: modelArg, signal: sig });
|
|
|
return r.map((x) => (x ? { embedding: x.embedding, model: x.model } : null));
|
|
return r.map((x) => (x ? { embedding: x.embedding, model: x.model } : null));
|
|
|
}
|
|
}
|
|
|
return session.embedBatch(texts, { model: modelArg });
|
|
return session.embedBatch(texts, { model: modelArg });
|