From 703653e25b4b319e71878a6aef54c4ffc2fa15b0 Mon Sep 17 00:00:00 2001
From: mkm <727897186@qq.com>
Date: Tue, 12 Sep 2023 16:44:15 +0800
Subject: [PATCH] =?UTF-8?q?=E6=9B=B4=E6=96=B0?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 app/api/controller/XunFeiController.php | 13 ++++++++++---
 1 file changed, 10 insertions(+), 3 deletions(-)

diff --git a/app/api/controller/XunFeiController.php b/app/api/controller/XunFeiController.php
index 69b48c60f..96059e052 100644
--- a/app/api/controller/XunFeiController.php
+++ b/app/api/controller/XunFeiController.php
@@ -33,6 +33,7 @@ class XunFeiController extends BaseApiController
 
     public function chat()
     {
+        header('X-Accel-Buffering: no');
         $parmas=$this->request->param('content');
         if(empty($parmas)){
             return $this->success('success');
@@ -57,12 +58,17 @@ class XunFeiController extends BaseApiController
                     if($status != 2){
                         $content = $resp['payload']['choices']['text'][0]['content'];
                         $answer .= $content;
+                        print($answer);
+                        ob_flush(); // 刷新输出缓冲区
+                        flush(); // 刷新系统输出缓冲区
                     }else{
                         $content = $resp['payload']['choices']['text'][0]['content'];
                         $answer .= $content;
                         $total_tokens = $resp['payload']['usage']['text']['total_tokens'];
                         print("\n本次消耗token用量：\n");
                         print($total_tokens);
+                        ob_flush(); // 刷新输出缓冲区
+                        flush(); // 刷新系统输出缓冲区
                         break;
                     }
                 }else{
@@ -70,11 +76,12 @@ class XunFeiController extends BaseApiController
                     break;
                 }
             }
+            ob_flush(); // 刷新输出缓冲区
+            flush(); // 刷新系统输出缓冲区
+            return $this->success('success');
 
-            print("\n返回结果为：\n");
-            print($answer);
         } else {
-            echo "无法连接到 WebSocket 服务器";
+            return $this->fail('无法连接到 WebSocket 服务器');
         }
     }