From 5580958d49e5aab48908000614e47ecb75ff4797 Mon Sep 17 00:00:00 2001
From: zhaoqingang <zhaoqg0118@163.com>
Date: 星期四, 28 十一月 2024 19:14:26 +0800
Subject: [PATCH] 智能数据问题优化

---
 app/service/basic.py |   15 ++++++++++-----
 1 files changed, 10 insertions(+), 5 deletions(-)

diff --git a/app/service/basic.py b/app/service/basic.py
index b3ad295..29bb02a 100644
--- a/app/service/basic.py
+++ b/app/service/basic.py
@@ -51,7 +51,7 @@
         url = f"{self.base_url}/exceltalk/upload/files"
         params = {'chat_id': chat_id, 'is_col': '0'}
 
-        async with httpx.AsyncClient() as client:
+        async with httpx.AsyncClient(timeout=300) as client:
             response = await client.post(
                 url,
                 files=files,
@@ -73,18 +73,23 @@
                         if decoded_line.startswith("data:"):
                             decoded_line = decoded_line[5:]
                         answer = json.loads(decoded_line)
+                        answer["type"] = "message"
                         yield answer
                     except GeneratorExit as e:
                         logger.error("------------except GeneratorExit as e:---------------------")
                         logger.error(e)
                         print(e)
                         yield {"message": "鍐呴儴閿欒", "type": "close"}
-                    finally:
-                        # 鍦ㄦ墍鏈夋暟鎹帴鏀跺畬姣曞悗杩斿洖close
-                        yield {"message": "", "type": "close"}
+                    # finally:
+                    #     # 鍦ㄦ墍鏈夋暟鎹帴鏀跺畬姣曞悗杩斿洖close
+                    #     yield {"message": "", "type": "close"}
 
                 else:
-                    yield f"Error: {response.status_code}"
+                    continue
+                    # yield f"Error: {response.status_code}"
+            else:
+            # 鍦ㄦ墍鏈夋暟鎹帴鏀跺畬姣曞悗杩斿洖close
+                yield {"message": "", "type": "close"}
 
     async def questions_talk(self, question, chat_id: str):
         logger.error("---------------questions_talk--------------------------")

--
Gitblit v1.8.0