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

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -2
app.py CHANGED
@@ -181,6 +181,7 @@ def respond_o1mini_qna(
181
  o1-mini 모델을 이용해 한 번의 질문(question)에 대한 답변을 반환하는 함수.
182
  o1-mini에서는 'system' 역할 메시지를 지원하지 않으므로,
183
  system_message와 question을 하나의 'user' 메시지로 합쳐 전달합니다.
 
184
  """
185
  openai_token = os.getenv("OPENAI_TOKEN")
186
  if not openai_token:
@@ -188,7 +189,7 @@ def respond_o1mini_qna(
188
 
189
  openai.api_key = openai_token
190
 
191
- # system_message와 question을 하나의 메시지로 합침
192
  combined_message = f"{system_message}\n\n{question}"
193
  messages = [
194
  {"role": "user", "content": combined_message}
@@ -198,7 +199,7 @@ def respond_o1mini_qna(
198
  response = openai.ChatCompletion.create(
199
  model="o1-mini",
200
  messages=messages,
201
- max_tokens=max_tokens,
202
  temperature=temperature,
203
  )
204
  assistant_message = response.choices[0].message['content']
 
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
 
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=temperature,
204
  )
205
  assistant_message = response.choices[0].message['content']