Przeglądaj źródła

fix: errors after stream starts

KernelDeimos 3 miesięcy temu
rodzic
commit
759c32ac72
1 zmienionych plików z 11 dodań i 1 usunięć
  1. 11 1
      src/backend/src/modules/puterai/AIChatService.js

+ 11 - 1
src/backend/src/modules/puterai/AIChatService.js

@@ -553,7 +553,17 @@ class AIChatService extends BaseService {
                             stream,
                         });
 
-                        ret.result.value.init_chat_stream({ chatStream });
+                        (async () => {
+                            try {
+                                await ret.result.value.init_chat_stream({ chatStream });
+                            } catch (e) {
+                                stream.write(JSON.stringify({
+                                    type: 'error',
+                                    message: e.message,
+                                }) + '\n');
+                                stream.end();
+                            }
+                        })();
 
                         return retval;
                     }