fix(model): fix test

moeru-ai · Jan 9, 2025 · 604301a · 604301a
1 parent 986f6f2
commit 604301a
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/packages/model/test/index.test.ts b/packages/model/test/index.test.ts
@@ -6,7 +6,7 @@ import { listModels, retrieveModel } from '../src'
 describe('@xsai/model', () => {
   it('listModels', async () => {
     const models = await listModels({
-      ...ollama.models(),
+      ...ollama.model(),
     })
 
     const llama = models.find(({ id }) => id === 'llama3.2:latest')!
@@ -18,7 +18,7 @@ describe('@xsai/model', () => {
 
   it('retrieveModel', async () => {
     const llama = await retrieveModel({
-      ...ollama.models(),
+      ...ollama.model(),
       model: 'llama3.2:latest',
     })