aljaz/gpt-engineer
1
0
Fork 0
mirror of https://github.com/aljazceru/gpt-engineer.git synced 2025-12-18 13:15:00 +01:00
Code Issues Packages Projects Releases Wiki Activity

revert to gpt-3.5-turbo when gpt-4 is not available

Browse Source
This commit is contained in:
jnichols0
2023-06-15 01:10:42 -04:00
parent e7e3292116
commit 046089d38c
1 changed files with 9 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

10
ai.py
View File

@@ -6,6 +6,14 @@ class AI:
def __init__(self, **kwargs): def __init__(self, **kwargs):
self.kwargs = 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): def start(self, system, user):
messages = [ messages = [
{"role": "system", "content": system}, {"role": "system", "content": system},
@@ -36,4 +44,4 @@ class AI:
msg = delta.get('content', '') msg = delta.get('content', '')
print(msg, end="") print(msg, end="")
chat.append(msg) chat.append(msg)
return messages + [{"role": "assistant", "content": "".join(chat)}] return messages + [{"role": "assistant", "content": "".join(chat)}]