From 26e7677dc38d8be89e5a370b7305f8b1b2ede85d Mon Sep 17 00:00:00 2001
From: binary-husky <qingxu.fu@outlook.com>
Date: Wed, 26 Jun 2024 15:18:11 +0000
Subject: [PATCH] fix new api for taichu

---
 request_llms/com_taichu.py | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/request_llms/com_taichu.py b/request_llms/com_taichu.py
index 17250ddb..f8eb3981 100644
--- a/request_llms/com_taichu.py
+++ b/request_llms/com_taichu.py
@@ -43,7 +43,8 @@ class TaichuChatInit:
         if response.status_code == 200:
             response.encoding = 'utf-8'
             for line in response.iter_lines(decode_unicode=True):
-                delta = json.loads(line)['choices'][0]['text']
+                try: delta = json.loads(line)['data']['content']
+                except: delta = json.loads(line)['choices'][0]['text']
                 results += delta
                 yield delta, results
         else: