File tree Expand file tree Collapse file tree
Expand file tree Collapse file tree Original file line number Diff line number Diff line change 88class 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
You can’t perform that action at this time.
0 commit comments