Kims12 commited on
Commit
46444e4
·
verified ·
1 Parent(s): 7770ee0

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +5 -5
app.py CHANGED
@@ -175,13 +175,13 @@ def respond_o1mini_qna(
175
  question: str,
176
  system_message: str,
177
  max_tokens: int,
178
- temperature: float
179
  ):
180
  """
181
  o1-mini 모델을 이용해 한 번의 질문(question)에 대한 답변을 반환하는 함수.
182
  o1-mini에서는 'system' 역할 메시지를 지원하지 않으므로,
183
  system_message와 question을 하나의 'user' 메시지로 합쳐 전달합니다.
184
- 또한, o1-mini에서는 'max_tokens' 대신 'max_completion_tokens' 파라미터를 사용합니다.
185
  """
186
  openai_token = os.getenv("OPENAI_TOKEN")
187
  if not openai_token:
@@ -189,7 +189,7 @@ def respond_o1mini_qna(
189
 
190
  openai.api_key = openai_token
191
 
192
- # system_message와 question을 하나로 합쳐서 전달
193
  combined_message = f"{system_message}\n\n{question}"
194
  messages = [
195
  {"role": "user", "content": combined_message}
@@ -199,8 +199,8 @@ def respond_o1mini_qna(
199
  response = openai.ChatCompletion.create(
200
  model="o1-mini",
201
  messages=messages,
202
- max_completion_tokens=max_tokens, # 수정된 부분
203
- temperature=temperature,
204
  )
205
  assistant_message = response.choices[0].message['content']
206
  return assistant_message
 
175
  question: str,
176
  system_message: str,
177
  max_tokens: int,
178
+ temperature: float # 입력값은 있으나 사용하지 않습니다.
179
  ):
180
  """
181
  o1-mini 모델을 이용해 한 번의 질문(question)에 대한 답변을 반환하는 함수.
182
  o1-mini에서는 'system' 역할 메시지를 지원하지 않으므로,
183
  system_message와 question을 하나의 'user' 메시지로 합쳐 전달합니다.
184
+ 또한, 온도(temperature)는 고정값 1만 지원하므로 해당 값으로 호출합니다.
185
  """
186
  openai_token = os.getenv("OPENAI_TOKEN")
187
  if not openai_token:
 
189
 
190
  openai.api_key = openai_token
191
 
192
+ # system_message와 question을 하나의 메시지로 합침
193
  combined_message = f"{system_message}\n\n{question}"
194
  messages = [
195
  {"role": "user", "content": combined_message}
 
199
  response = openai.ChatCompletion.create(
200
  model="o1-mini",
201
  messages=messages,
202
+ max_completion_tokens=max_tokens,
203
+ temperature=1, # 고정된 값 1 사용
204
  )
205
  assistant_message = response.choices[0].message['content']
206
  return assistant_message