@@ -39,6 +39,8 @@ def request_gpt(prompt):
# result_content = response.json()['choices'][0]['message']['content']
# log_.info(f"response.text: {response.text}")
res_data = json.loads(response.text)
+ if res_data['code'] != 0:
+ raise ConnectionResetError
result_content = res_data['data']['choices'][0]['message']['content']
return result_content