Update chatgpt.py

Author: willismax

api/chatgpt.py

@@ -1,30 +1,25 @@
 from api.prompt import Prompt
 
 import os
-import openai
+from openai import OpenAI
+client = OpenAI()
 
-openai.api_key = os.getenv("OPENAI_API_KEY")
+client.api_key = os.getenv("OPENAI_API_KEY")
 
 
 class ChatGPT:
     def __init__(self):
         self.prompt = Prompt()
-        self.model = os.getenv("OPENAI_MODEL", default = "text-davinci-003")
+        self.model = os.getenv("OPENAI_MODEL", default = "gpt-4-1106-preview")
         self.temperature = float(os.getenv("OPENAI_TEMPERATURE", default = 0))
-        self.frequency_penalty = float(os.getenv("OPENAI_FREQUENCY_PENALTY", default = 0))
-        self.presence_penalty = float(os.getenv("OPENAI_PRESENCE_PENALTY", default = 0.6))
-        self.max_tokens = int(os.getenv("OPENAI_MAX_TOKENS", default = 240))
+        self.max_tokens = int(os.getenv("OPENAI_MAX_TOKENS", default = 500))
 
     def get_response(self):
-        response = openai.Completion.create(
+        response = client.chat.completions.create(
             model=self.model,
-            prompt=self.prompt.generate_prompt(),
-            temperature=self.temperature,
-            frequency_penalty=self.frequency_penalty,
-            presence_penalty=self.presence_penalty,
-            max_tokens=self.max_tokens
+            messages=self.prompt.generate_prompt(),
         )
-        return response['choices'][0]['text'].strip()
+        return response.choices[0].message.content
 
     def add_msg(self, text):
         self.prompt.add_msg(text)