|
1 | 1 | from api.prompt import Prompt |
2 | | - |
3 | 2 | import os |
4 | | -import openai |
| 3 | +from openai import OpenAI |
| 4 | +client = OpenAI() |
5 | 5 |
|
6 | | -openai.api_key = os.getenv("OPENAI_API_KEY") |
| 6 | +client.api_key = os.getenv("OPENAI_API_KEY") |
7 | 7 |
|
8 | 8 |
|
9 | 9 | class ChatGPT: |
10 | 10 | def __init__(self): |
11 | 11 | self.prompt = Prompt() |
12 | | - self.model = os.getenv("OPENAI_MODEL", default = "text-davinci-003") |
| 12 | + self.model = os.getenv("OPENAI_MODEL", default = "gpt-4-1106-preview") |
13 | 13 | self.temperature = float(os.getenv("OPENAI_TEMPERATURE", default = 0)) |
14 | | - self.frequency_penalty = float(os.getenv("OPENAI_FREQUENCY_PENALTY", default = 0)) |
15 | | - self.presence_penalty = float(os.getenv("OPENAI_PRESENCE_PENALTY", default = 0.6)) |
16 | | - self.max_tokens = int(os.getenv("OPENAI_MAX_TOKENS", default = 240)) |
| 14 | + self.max_tokens = int(os.getenv("OPENAI_MAX_TOKENS", default = 500)) |
17 | 15 |
|
18 | 16 | def get_response(self): |
19 | | - response = openai.Completion.create( |
| 17 | + response = client.chat.completions.create( |
20 | 18 | model=self.model, |
21 | | - prompt=self.prompt.generate_prompt(), |
22 | | - temperature=self.temperature, |
23 | | - frequency_penalty=self.frequency_penalty, |
24 | | - presence_penalty=self.presence_penalty, |
25 | | - max_tokens=self.max_tokens |
| 19 | + messages=self.prompt.generate_prompt(), |
26 | 20 | ) |
27 | | - return response['choices'][0]['text'].strip() |
| 21 | + return response.choices[0].message.content |
28 | 22 |
|
29 | 23 | def add_msg(self, text): |
30 | 24 | self.prompt.add_msg(text) |
0 commit comments