OpenWebUI/open-webui
1
0
Fork 0
mirror of https://github.com/open-webui/open-webui synced 2025-06-26 18:26:48 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

Merge pull request #2183 from cheahjs/fix/streaming-error

Browse Source 
feat: better handle openai errors, add error message to message
This commit is contained in:
Timothy Jaeryang Baek
2024-05-12 14:28:13 -10:00
committed by GitHub
parent ba13321728 3113825895
commit 88a94cc50d
3 changed files with 330 additions and 288 deletions
Download Patch File Download Diff File Expand all files Collapse all files

7
src/lib/apis/streaming/index.ts
View File

@@ -6,6 +6,8 @@ type TextStreamUpdate = {
value: string;
// eslint-disable-next-line @typescript-eslint/no-explicit-any
citations?: any;
// eslint-disable-next-line @typescript-eslint/no-explicit-any
error?: any;
};
// createOpenAITextStream takes a responseBody with a SSE response,
@@ -47,6 +49,11 @@ async function* openAIStreamToIterator(
const parsedData = JSON.parse(data);
console.log(parsedData);
if (parsedData.error) {
yield { done: true, value: '', error: parsedData.error };
break;
}
if (parsedData.citations) {
yield { done: false, value: '', citations: parsedData.citations };
continue;