diff --git a/ai.py b/ai.py index dc332cf..7ed0a3f 100644 --- a/ai.py +++ b/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)}] \ No newline at end of file + return messages + [{"role": "assistant", "content": "".join(chat)}]