|
|
@@ -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
|
|
|
|
