pendrag commited on
Commit
29c8c51
·
1 Parent(s): ed91a00

client error fixed

Browse files
Files changed (1) hide show
  1. app.py +5 -1
app.py CHANGED
@@ -224,8 +224,12 @@ def search(query, progress=gr.Progress()):
224
 
225
  if GENAI_API == "openai":
226
  client = OpenAI()
 
 
 
227
  else:
228
- genai.configure(api_key=os.getenv('GEMINI_API_KEY'))
 
229
 
230
  with gr.Blocks() as demo:
231
  gr.Markdown("# Feynbot on INSPIRE HEP Search")
 
224
 
225
  if GENAI_API == "openai":
226
  client = OpenAI()
227
+ elif GENAI_API.lower() in ("gemini", "google", "genai"):
228
+ genai.configure(api_key=os.getenv('GEMINI_API_KEY'))
229
+ client = genai
230
  else:
231
+ # Default to OpenAI client if GENAI_API is unrecognized.
232
+ client = OpenAI()
233
 
234
  with gr.Blocks() as demo:
235
  gr.Markdown("# Feynbot on INSPIRE HEP Search")