cubukcum commited on
Commit
e5cabf3
·
verified ·
1 Parent(s): ee0a948

Update agent.py

Browse files
Files changed (1) hide show
  1. agent.py +4 -0
agent.py CHANGED
@@ -1,5 +1,9 @@
1
  import os
2
  from smolagents import CodeAgent, DuckDuckGoSearchTool, tool, InferenceClientModel, OpenAIServerModel
 
 
 
 
3
 
4
  @tool
5
  def add(a: int, b: int) -> int:
 
1
  import os
2
  from smolagents import CodeAgent, DuckDuckGoSearchTool, tool, InferenceClientModel, OpenAIServerModel
3
+ from dotenv import load_dotenv
4
+ load_dotenv()
5
+
6
+ GEMINI_API_KEY = os.getenv("GEMINI_API_KEY")
7
 
8
  @tool
9
  def add(a: int, b: int) -> int: