llamasrock commited on
Commit
b7bb69f
·
verified ·
1 Parent(s): 99e1c43

Update app.py

Browse files

added api base url for gemini smolagent client

Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -25,8 +25,9 @@ SERPER_API_KEY = os.getenv("SERPER_API_KEY")
25
  class BasicAgent:
26
  def __init__(self):
27
  self.llm_model = LiteLLMModel(
28
- model_id="gemini/gemini-2.0-flash", # you can see other model names here: https://cloud.google.com/vertex-ai/generative-ai/docs/learn/models. It is important to prefix the name with "gemini/"
29
  api_key=GEMINI_API_KEY,
 
30
  max_tokens=8192
31
  )
32
  # self.google_search_tool = Tool(google_search = GoogleSearch())
 
25
  class BasicAgent:
26
  def __init__(self):
27
  self.llm_model = LiteLLMModel(
28
+ model_id="gemini/gemini-2.5-flash", # you can see other model names here: https://cloud.google.com/vertex-ai/generative-ai/docs/learn/models. It is important to prefix the name with "gemini/"
29
  api_key=GEMINI_API_KEY,
30
+ api_base = 'https://generativelanguage.googleapis.com',
31
  max_tokens=8192
32
  )
33
  # self.google_search_tool = Tool(google_search = GoogleSearch())