Explorar o código

fix function signature in node create

Bruce MacDonald hai 1 ano
pai
achega
503224e0ba
Modificáronse 1 ficheiros con 4 adicións e 2 borrados
  1. 4 2
      src/index.ts

+ 4 - 2
src/index.ts

@@ -1,4 +1,6 @@
 import * as utils from './utils.js'
+import { AbortableAsyncIterator } from './utils.js'
+
 import fs, { createReadStream, promises } from 'fs'
 import { dirname, join, resolve } from 'path'
 import { createHash } from 'crypto'
@@ -138,12 +140,12 @@ export class Ollama extends OllamaBrowser {
 
   create(
     request: CreateRequest & { stream: true },
-  ): Promise<AsyncGenerator<ProgressResponse>>
+  ): Promise<AbortableAsyncIterator<ProgressResponse>>
   create(request: CreateRequest & { stream?: false }): Promise<ProgressResponse>
 
   async create(
     request: CreateRequest,
-  ): Promise<ProgressResponse | AsyncGenerator<ProgressResponse>> {
+  ): Promise<ProgressResponse | AbortableAsyncIterator<ProgressResponse>> {
     let modelfileContent = ''
     if (request.path) {
       modelfileContent = await promises.readFile(request.path, { encoding: 'utf8' })