CC/gpt_academic
1
0
Fork 0
Code Issues Pull Requests Actions 1 Packages Projects Releases Wiki Activity

fix new api for taichu

Browse Source
This commit is contained in:
binary-husky
2024-06-26 15:18:11 +00:00
parent 5e64a50898
commit 26e7677dc3
1 changed files with 2 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
request_llms/com_taichu.py
View File

@@ -43,7 +43,8 @@ class TaichuChatInit:
if response.status_code == 200: if response.status_code == 200:
response.encoding = 'utf-8' response.encoding = 'utf-8'
for line in response.iter_lines(decode_unicode=True): 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 results += delta
yield delta, results yield delta, results
else: else: