Skip to content

Commit 23f283f

Browse files
committed
refactor: use await
1 parent 64a2016 commit 23f283f

File tree

1 file changed

+5
-7
lines changed

1 file changed

+5
-7
lines changed

templates/types/streaming/express/src/controllers/chat.controller.ts

Lines changed: 5 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -63,19 +63,17 @@ export const chat = async (req: Request, res: Response) => {
6363
data: vercelStreamData,
6464
callbacks: { onCompletion },
6565
});
66-
// TODO: move to LlamaIndexAdapter
67-
const reader = streamResponse.body?.getReader();
68-
function read() {
69-
reader?.read().then(({ done, value }: { done: boolean; value?: any }) => {
66+
if (streamResponse.body) {
67+
const reader = streamResponse.body.getReader();
68+
while (true) {
69+
const { done, value } = await reader.read();
7070
if (done) {
7171
res.end();
7272
return;
7373
}
7474
res.write(value);
75-
read();
76-
});
75+
}
7776
}
78-
read();
7977
} catch (error) {
8078
console.error("[LlamaIndex]", error);
8179
return res.status(500).json({

0 commit comments

Comments
 (0)