OMGeeky/gpt-pilot
1
0
Fork 0
mirror of https://github.com/OMGeeky/gpt-pilot.git synced 2026-01-05 19:00:22 +01:00
Code Issues Packages Projects Releases Wiki Activity

retry request to openai if there was exception

Browse Source
This commit is contained in:
LeonOstrez
2023-08-14 23:44:57 +02:00
parent 257b94274f
commit 29ceedf700
1 changed files with 23 additions and 14 deletions
Download Patch File Download Diff File Expand all files Collapse all files

37
euclid/utils/llm_connection.py
View File

@@ -129,6 +129,28 @@ def count_lines_based_on_width(content, width):
return lines_required
def retry_on_exception(func):
def wrapper(*args, **kwargs):
while True:
try:
return func(*args, **kwargs)
except Exception as e:
print(colored(f'There was a problem with request to openai API:', 'red'))
print(str(e))
user_message = questionary.text(
"Do you want to try make the same request again? If yes, just press ENTER. Otherwise, type 'no'.",
style=questionary.Style([
('question', 'fg:red'),
('answer', 'fg:orange')
])).ask()
if user_message != '':
return {}
return wrapper
@retry_on_exception
def stream_gpt_completion(data, req_type):
terminal_width = os.get_terminal_size().columns
lines_printed = 2
@@ -159,21 +181,8 @@ def stream_gpt_completion(data, req_type):
logger.info(f'Response status code: {response.status_code}')
if response.status_code != 200:
print(colored(f'There was a problem with request to openai API:', 'red'))
print(response.text)
user_message = questionary.text("Do you want to try make same request again? If yes, just press ENTER.",
style=questionary.Style([
('question', 'fg:red'),
('answer', 'fg:orange')
])).ask()
lines_printed += count_lines_based_on_width(response.text, terminal_width) + 1
if user_message == '':
delete_last_n_lines(lines_printed)
return stream_gpt_completion(data, req_type)
logger.debug(f'problem with request: {response.text}')
return return_result({}, lines_printed)
raise Exception(f"API responded with status code: {response.status_code}. Response text: {response.text}")
gpt_response = ''
function_calls = {'name': '', 'arguments': ''}