Timothy J. Baek 6 months ago
parent
commit
fc71f441c4

+ 2 - 3
backend/open_webui/utils/response.py

@@ -26,7 +26,6 @@ async def convert_streaming_response_ollama_to_openai(ollama_streaming_response)
         )
         )
 
 
         line = f"data: {json.dumps(data)}\n\n"
         line = f"data: {json.dumps(data)}\n\n"
-        if done:
-            line += "data: [DONE]\n\n"
-
         yield line
         yield line
+
+    yield "data: [DONE]\n\n"

+ 1 - 2
src/lib/components/playground/Completions.svelte

@@ -78,8 +78,7 @@
 
 
 					for (const line of lines) {
 					for (const line of lines) {
 						if (line !== '') {
 						if (line !== '') {
-							if (line === 'data: [DONE]') {
-								// responseMessage.done = true;
+							if (line.includes('[DONE]')) {
 								console.log('done');
 								console.log('done');
 							} else {
 							} else {
 								let data = JSON.parse(line.replace(/^data: /, ''));
 								let data = JSON.parse(line.replace(/^data: /, ''));