mirror of
https://github.com/aljazceru/gpt-engineer.git
synced 2025-12-18 13:15:00 +01:00
revert to gpt-3.5-turbo when gpt-4 is not available
This commit is contained in:
10
ai.py
10
ai.py
@@ -6,6 +6,14 @@ class AI:
|
||||
def __init__(self, **kwargs):
|
||||
self.kwargs = kwargs
|
||||
|
||||
try:
|
||||
openai.Model.retrieve("gpt-4")
|
||||
except openai.error.InvalidRequestError:
|
||||
print("Model gpt-4 not available for provided api key reverting "
|
||||
"to gpt-3.5.turbo. Sign up for the gpt-4 wait list here: "
|
||||
"https://openai.com/waitlist/gpt-4-api")
|
||||
self.kwargs['model'] = "gpt-3.5-turbo"
|
||||
|
||||
def start(self, system, user):
|
||||
messages = [
|
||||
{"role": "system", "content": system},
|
||||
@@ -36,4 +44,4 @@ class AI:
|
||||
msg = delta.get('content', '')
|
||||
print(msg, end="")
|
||||
chat.append(msg)
|
||||
return messages + [{"role": "assistant", "content": "".join(chat)}]
|
||||
return messages + [{"role": "assistant", "content": "".join(chat)}]
|
||||
|
||||
Reference in New Issue
Block a user