merged

app.py

@@ -14,6 +14,7 @@ import re
 # Model name for LLM calls. Can be overridden by setting the LLM_MODEL
 # environment variable. Falls back to a sensible default if unset.
 MODEL_NAME = os.getenv("LLM_MODEL", "models/gemini-flash-latest")
+GENAUI_API = os.getenv("GENAI_API", "gemini")
 
 # LLM_MODEL_NAME must be set in the environment
 
@@ -124,10 +125,8 @@ def llm_expand_query(query):
     presence_penalty=0
   )
 
-    return response.choices[0].message.content
-  else:
-     response = genai.GenerativeModel("gemini-1.5-flash").generate_content(prompt)
-     return response.text
+  return response.choices[0].message.content
+
 
 def llm_generate_answer(prompt):
   """ Generate a response from the LLM """
@@ -173,11 +172,8 @@ def llm_generate_answer(prompt):
     presence_penalty=0
   )
 
-    return response.choices[0].message.content
+  return response.choices[0].message.content
   
-  else:
-     response = genai.GenerativeModel("gemini-1.5-flash").generate_content(system_desc + "\n\n" + prompt)
-     return response.text
 
 def clean_refs(answer, results):
   """ Clean the references from the answer """