藍+85CD
24432f881c
* refactor(test): use `granite4:350m-h` instead of `granite3:2b`, `qwen3:0.6b` * refactor(test): add 1b-h model, update test snapshot * refactor: simplify embed test * chore: revert changes * fix: test * fix: test * fix: try another seed * fix: test
19 lines
544 B
TypeScript
19 lines
544 B
TypeScript
import { describe, expect, it } from 'vitest'
|
|
|
|
import { embedMany } from '../src'
|
|
|
|
describe('@xsai/embed', () => {
|
|
it('embedMany', async () => {
|
|
const { embeddings, usage } = await embedMany({
|
|
baseURL: 'http://localhost:11434/v1/',
|
|
input: ['why is the sky blue?', 'why is the grass green?'],
|
|
model: 'all-minilm',
|
|
})
|
|
|
|
expect(embeddings.length).toBe(2)
|
|
embeddings.forEach(embedding => expect(embedding.length).toBe(384))
|
|
expect(usage.prompt_tokens).toBe(16)
|
|
expect(usage.total_tokens).toBe(16)
|
|
})
|
|
})
|