Rathapoom commited on
Commit
102fc7c
·
verified ·
1 Parent(s): ad1a715

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -78,8 +78,8 @@ def generate_questions_with_retry(knowledge_material, question_type, cognitive_l
78
  retries = 0
79
  while retries < max_retries:
80
  try:
81
- response = openai.ChatCompletion.create(
82
- model="gpt-4o-mini", # You can switch to gpt-4 if available in your plan
83
  messages=[
84
  {"role": "system", "content": "You are a helpful assistant for generating exam questions."},
85
  {"role": "user", "content": prompt}
@@ -87,7 +87,7 @@ def generate_questions_with_retry(knowledge_material, question_type, cognitive_l
87
  max_tokens=1000,
88
  temperature=0.7
89
  )
90
- return response.choices[0].message["content"]
91
  except openai.error.APIConnectionError:
92
  retries += 1
93
  time.sleep(2) # Wait for 2 seconds before retrying
 
78
  retries = 0
79
  while retries < max_retries:
80
  try:
81
+ response = client.chat.completions.create(
82
+ model="gpt-4o-mini",
83
  messages=[
84
  {"role": "system", "content": "You are a helpful assistant for generating exam questions."},
85
  {"role": "user", "content": prompt}
 
87
  max_tokens=1000,
88
  temperature=0.7
89
  )
90
+ return response.choices[0].message.content
91
  except openai.error.APIConnectionError:
92
  retries += 1
93
  time.sleep(2) # Wait for 2 seconds before retrying