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: