Skip to content

Commit c154eaa

Browse files
authored
move redundant part to init
1 parent 300aad2 commit c154eaa

1 file changed

Lines changed: 10 additions & 5 deletions

File tree

api/chatgpt.py

Lines changed: 10 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -8,15 +8,20 @@
88
class ChatGPT:
99
def __init__(self):
1010
self.prompt = Prompt()
11+
self.model = os.getenv("OPENAI_MODEL",default="text-davinci-003")
12+
self.temperature = int(os.getenv("OPENAI_TEMPERATURE",default="0"))
13+
self.frequency_penalty = int(os.getenv("OPENAI_FREQUENCY_PENALTY",default="0"))
14+
self.presence_penalty = float(os.getenv("OPENAI_PRESENCE_PENALTY",default="0.6"))
15+
self.max_tokens = int(os.getenv("OPENAI_MAX_TOKENS",default="240"))
1116

1217
def get_response(self):
1318
response = openai.Completion.create(
14-
model = os.getenv("OPENAI_MODEL",default="text-davinci-003"),
19+
model = self.model,
1520
prompt = self.prompt.generate_prompt(),
16-
temperature = int(os.getenv("OPENAI_TEMPERATURE",default="0")),
17-
frequency_penalty = int(os.getenv("OPENAI_FREQUENCY_PENALTY",default="0")),
18-
presence_penalty = float(os.getenv("OPENAI_PRESENCE_PENALTY",default="0.6")),
19-
max_tokens = int(os.getenv("OPENAI_MAX_TOKENS",default="240"))
21+
temperature = self.temperature,
22+
frequency_penalty = self.frequency_penalty,
23+
presence_penalty = self.presence_penalty,
24+
max_tokens = self.max_tokens
2025
)
2126
return response['choices'][0]['text'].strip()
2227

0 commit comments

Comments
 (0)