藍+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
18 lines
431 B
TypeScript
18 lines
431 B
TypeScript
import { describe, expect, it } from 'vitest'
|
|
|
|
import { embed } from '../src'
|
|
|
|
describe('@xsai/embed', () => {
|
|
it('embed', async () => {
|
|
const { embedding, usage } = await embed({
|
|
baseURL: 'http://localhost:11434/v1/',
|
|
input: 'sunny day at the beach',
|
|
model: 'all-minilm',
|
|
})
|
|
|
|
expect(embedding.length).toBe(384)
|
|
expect(usage.prompt_tokens).toBe(7)
|
|
expect(usage.total_tokens).toBe(7)
|
|
})
|
|
})
|