Kims12 commited on
Commit
e095e1d
·
verified ·
1 Parent(s): ab94c40

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +13 -13
app.py CHANGED
@@ -179,29 +179,29 @@ def respond_o3mini_qna(
179
  top_p: float
180
  ):
181
  """
182
- o3-mini 모델을 이용해 번의 질문(question)에 대한 답변을 반환하는 함수.
 
183
  """
184
  openai_token = os.getenv("OPENAI_TOKEN")
185
  if not openai_token:
186
  return "OpenAI API 토큰이 필요합니다."
187
 
188
- openai.api_key = openai_token
189
-
190
- messages = [
191
- {"role": "system", "content": system_message},
192
- {"role": "user", "content": question}
193
- ]
194
-
195
  try:
196
- response = openai.ChatCompletion.create(
197
- model="o3-mini", # o3-mini 모델 사용
 
 
 
 
 
 
 
198
  messages=messages,
199
  max_tokens=max_tokens,
200
  temperature=temperature,
201
  top_p=top_p,
202
  )
203
- assistant_message = response.choices[0].message['content']
204
- return assistant_message
205
  except Exception as e:
206
  return f"오류가 발생했습니다: {str(e)}"
207
 
@@ -313,7 +313,7 @@ with gr.Blocks() as demo:
313
  #################
314
  # o3-mini 탭
315
  #################
316
- with gr.Tab("o1-mini"):
317
  o3mini_input1 = gr.Textbox(label="입력1", lines=1)
318
  o3mini_input2 = gr.Textbox(label="입력2", lines=1)
319
  o3mini_input3 = gr.Textbox(label="입력3", lines=1)
 
179
  top_p: float
180
  ):
181
  """
182
+ o3-mini 모델 대신, OpenAI의 새로운 API 클라이언트를 이용하여
183
+ "o1-preview" 모델로 한 번의 질문(question)에 대한 답변을 반환하는 함수.
184
  """
185
  openai_token = os.getenv("OPENAI_TOKEN")
186
  if not openai_token:
187
  return "OpenAI API 토큰이 필요합니다."
188
 
 
 
 
 
 
 
 
189
  try:
190
+ # OpenAI 클라이언트를 사용
191
+ from openai import OpenAI
192
+ client = OpenAI(api_key=openai_token)
193
+ messages = [
194
+ {"role": "system", "content": system_message},
195
+ {"role": "user", "content": question}
196
+ ]
197
+ completion = client.chat.completions.create(
198
+ model="o1-mini",
199
  messages=messages,
200
  max_tokens=max_tokens,
201
  temperature=temperature,
202
  top_p=top_p,
203
  )
204
+ return completion.choices[0].message.content
 
205
  except Exception as e:
206
  return f"오류가 발생했습니다: {str(e)}"
207
 
 
313
  #################
314
  # o3-mini 탭
315
  #################
316
+ with gr.Tab("o3-mini"):
317
  o3mini_input1 = gr.Textbox(label="입력1", lines=1)
318
  o3mini_input2 = gr.Textbox(label="입력2", lines=1)
319
  o3mini_input3 = gr.Textbox(label="입력3", lines=1)