fix new api for taichu

2024-06-26 15:18:11 +00:00
parent 5e64a50898
commit 26e7677dc3
1 changed files with 2 additions and 1 deletions
--- 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: