wendru18 commited on
Commit
68fa535
·
1 Parent(s): 147ee28

added check for openai errors

Browse files
Files changed (1) hide show
  1. app.py +11 -5
app.py CHANGED
@@ -186,12 +186,18 @@ def generate_answer(question, model, token_budget, temperature):
186
  {"role": "user", "content": message},
187
  ]
188
 
189
- response = openai.ChatCompletion.create(
190
- model=model,
191
- messages=messages,
192
- temperature=temperature
193
- )
 
 
 
 
 
194
 
 
195
  response_message = response["choices"][0]["message"]["content"]
196
 
197
  return response_message, references
 
186
  {"role": "user", "content": message},
187
  ]
188
 
189
+ try:
190
+
191
+ response = openai.ChatCompletion.create(
192
+ model=model,
193
+ messages=messages,
194
+ temperature=temperature
195
+ )
196
+
197
+ except:
198
+ return "An OpenAI error occured. Make sure you did not exceed your usage limit or you provided a valid API key.", ""
199
 
200
+
201
  response_message = response["choices"][0]["message"]["content"]
202
 
203
  return response_message, references