藍+85CD
901d593611
* refactor(providers)!: move to `@xsai-ext` scope * fix: lint errors * refactor(providers-core): split packages * refactor(shared-providers): rename file * refactor(providers-cloud/openai): allow custom base url * feat(providers-cloud): migrate providers * feat: new package `providers-local` * refactor(providers)!: remove old providers * refactor(docs/references): include ext package * fix: lint errors * chore(docs): clean deps
18 lines
438 B
TypeScript
18 lines
438 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: 'nomic-embed-text',
|
|
})
|
|
|
|
expect(embedding).toMatchSnapshot()
|
|
expect(usage.prompt_tokens).toBe(5)
|
|
expect(usage.total_tokens).toBe(5)
|
|
})
|
|
})
|