mirrors/puter
1
0
Fork 0
mirror of https://github.com/HeyPuter/puter synced 2024-11-14 22:06:00 +00:00
Code Issues Packages Projects Releases Wiki Activity

fix(puterai): close streams

Browse Source
This commit is contained in:
KernelDeimos 2024-08-23 19:42:53 -04:00
parent 5a616f67dd
commit eb18550f41
6 changed files with 12 additions and 10 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
src/backend/src/modules/puterai/ClaudeService.js
View File

@ -90,6 +90,7 @@ class ClaudeService extends BaseService {
});
stream.write(str + '\n');
}
stream.end();
})();
return retval;

1
src/backend/src/modules/puterai/GroqAIService.js
View File

@ -54,6 +54,7 @@ class GroqAIService extends BaseService {
});
stream.write(str + '\n');
}
stream.end();
})();
return retval;
}

1
src/backend/src/modules/puterai/MistralAIService.js
View File

@ -55,6 +55,7 @@ class MistralAIService extends BaseService {
});
stream.write(str + '\n');
}
stream.end();
})();
return retval;
}

1
src/backend/src/modules/puterai/OpenAICompletionService.js
View File

@ -232,6 +232,7 @@ class OpenAICompletionService extends BaseService {
});
stream.write(str + '\n');
}
stream.end();
})();
return retval;
}

1
src/backend/src/modules/puterai/TogetherAIService.js
View File

@ -59,6 +59,7 @@ class TogetherAIService extends BaseService {
});
stream.write(str + '\n');
}
stream.end();
})();
return retval;
}

17
src/puter-js/src/lib/utils.js
View File

@ -279,15 +279,12 @@ async function driverCall_(
// ===============================================
let is_stream = false;
let got_headers = false;
let signal_stream_update = null;
let lastLength = 0;
let response_complete = false;
const parts_received = [];
xhr.onreadystatechange = () => {
if ( got_headers ) return;
got_headers = true;
if ( xhr.readyState >= 2 ) {
if ( xhr.readyState === 2 ) {
if ( xhr.getResponseHeader("Content-Type") !==
'application/x-ndjson'
) return;
@ -311,6 +308,12 @@ async function driverCall_(
return resolve_func(Stream());
}
if ( xhr.readyState === 4 ) {
response_complete = true;
if ( is_stream ) {
signal_stream_update?.();
}
}
};
xhr.onprogress = function() {
@ -323,19 +326,13 @@ async function driverCall_(
signal_stream_update();
};
xhr.addEventListener('load', () => {
response_complete = true;
});
// ========================
// === END OF STREAMING ===
// ========================
// load: success or error
xhr.addEventListener('load', async function(response){
response_complete = true;
if ( is_stream ) {
signal_stream_update?.();
return;
}
const resp = await parseResponse(response.target);