apunkt/trustgraph
1
0
Fork 0
mirror of https://github.com/trustgraph-ai/trustgraph.git synced 2026-07-01 17:39:39 +02:00
Code Issues Projects Releases Packages Wiki Activity Actions 3

Simplify text completion generator boundary

Browse source
This commit is contained in:
elpresidank 2026-06-02 02:58:19 -05:00
parent bfb5494552
commit b3aaa9a4af
3 changed files with 56 additions and 8 deletions
Download patch file Download diff file Expand all files Collapse all files

12
ts/packages/flow/src/model/text-completion/common.ts
View file

@ -86,16 +86,16 @@ export const toAsyncGenerator = (
const iterator = iterable[Symbol.asyncIterator]();
let generator: AsyncGenerator<LlmChunk>;
generator = {
next: (value?: unknown) => iterator.next(value as never),
next: (value?: unknown) => iterator.next(value),
return: (value?: unknown) =>
iterator.return === undefined
? Promise.resolve({ done: true, value: value as LlmChunk })
: iterator.return(value as never) as Promise<IteratorResult<LlmChunk>>,
? Promise.resolve({ done: true, value })
: iterator.return(value),
throw: (error?: unknown) =>
iterator.throw === undefined
? Effect.runPromise(Effect.fail(mapError(error))) as Promise<IteratorResult<LlmChunk>>
: iterator.throw(error) as Promise<IteratorResult<LlmChunk>>,
? Promise.reject(mapError(error))
: iterator.throw(error),
[Symbol.asyncIterator]: () => generator,
} as AsyncGenerator<LlmChunk>;
};
return generator;
};