"...composable_kernel.git" did not exist on "0619ebf70bc6d0bd8b44cb41b5a662ddfc4def56"
Unverified Commit bbdedc6c authored by Tong Gao's avatar Tong Gao Committed by GitHub
Browse files

[Enhancement] Optimize OpenAI models (#128)

* [Feature] Enhance OpenAI API, add example config for GPT evaluation
parent d17a5b94
import json
import os
import time
from concurrent.futures import ThreadPoolExecutor
from threading import Lock
from typing import Dict, List, Optional, Union
......@@ -188,10 +189,14 @@ class OpenAI(BaseAPIModel):
except requests.JSONDecodeError:
self.logger.error('JsonDecode error, got',
str(raw_response.content))
continue
try:
return response['choices'][0]['message']['content'].strip()
except KeyError:
if 'error' in response:
if response['error']['code'] == 'rate_limit_exceeded':
time.sleep(1)
continue
self.logger.error('Find error message in response: ',
str(response['error']))
max_num_retries += 1
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment