yxmiler commited on
Commit
9d46f45
·
verified ·
1 Parent(s): ad23443

Update index.js

Browse files
Files changed (1) hide show
  1. index.js +5 -1
index.js CHANGED
@@ -570,7 +570,11 @@ async function handleResponse(response, model, res, isStream) {
570
  let buffer = '';
571
  let fullResponse = '';
572
  const dataPromises = [];
573
-
 
 
 
 
574
  return new Promise((resolve, reject) => {
575
  stream.on('data', async (chunk) => {
576
  buffer += chunk.toString();
 
570
  let buffer = '';
571
  let fullResponse = '';
572
  const dataPromises = [];
573
+ if(isStream){
574
+ res.setHeader('Content-Type', 'text/event-stream');
575
+ res.setHeader('Cache-Control', 'no-cache');
576
+ res.setHeader('Connection', 'keep-alive');
577
+ }
578
  return new Promise((resolve, reject) => {
579
  stream.on('data', async (chunk) => {
580
  buffer += chunk.toString();