APRG commited on
Commit
0b4f31e
·
verified ·
1 Parent(s): f29033e

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -47,7 +47,7 @@ class BasicAgent:
47
  REQUEST:
48
  """
49
  llm = LiteLLMModel(
50
- model_id="gemini-2.0-flash", api_key=os.getenv("GEMINI_API_KEY")
51
  )
52
  summary = llm.run(prompt+question)
53
  return summary
 
47
  REQUEST:
48
  """
49
  llm = LiteLLMModel(
50
+ model_id="gemini-2.0-flash", api_key=os.getenv("GOOGLE_API_KEY")
51
  )
52
  summary = llm.run(prompt+question)
53
  return summary