akrstova commited on
Commit
33dc421
·
1 Parent(s): 2a6057d

Add Gemini API key

Browse files
Files changed (1) hide show
  1. agent.py +1 -0
agent.py CHANGED
@@ -19,6 +19,7 @@ def build_graph():
19
  max_tokens=None,
20
  timeout=None,
21
  max_retries=2,
 
22
  )
23
  tools = [add, subtract, multiply, divide, modulus, power, sqrt, web_search, search_wikipedia]
24
 
 
19
  max_tokens=None,
20
  timeout=None,
21
  max_retries=2,
22
+ google_api_key=os.getenv("GOOGLE_API_KEY") # Get API key from environment variable
23
  )
24
  tools = [add, subtract, multiply, divide, modulus, power, sqrt, web_search, search_wikipedia]
25