(texts: string[], log: Logger)
| 24 | } |
| 25 | |
| 26 | export async function embedLocal(texts: string[], log: Logger): Promise<number[][]> { |
| 27 | const ext = await getExtractor(log); |
| 28 | const results: number[][] = []; |
| 29 | |
| 30 | for (const text of texts) { |
| 31 | const output = await ext(text, { pooling: "mean", normalize: true }); |
| 32 | results.push(Array.from(output.data as Float32Array).slice(0, DEFAULTS.localEmbeddingDimensions)); |
| 33 | } |
| 34 | |
| 35 | return results; |
| 36 | } |
no test coverage detected