Anoire commited on
Commit
b6ca59d
·
1 Parent(s): c7f4838
Files changed (1) hide show
  1. agent.py +2 -2
agent.py CHANGED
@@ -1,7 +1,7 @@
1
  from smolagents import tool
2
  from smolagents import CodeAgent, VisitWebpageTool, DuckDuckGoSearchTool
3
  from smolagents import OpenAIServerModel
4
-
5
 
6
  with open('prompt.txt', 'r') as f:
7
  system_prompt = f.read()
@@ -9,7 +9,7 @@ system_prompt
9
 
10
  model = OpenAIServerModel(
11
  model_id="gemini-2.0-flash",
12
- api_key= Gemini_Key,
13
  # Google Gemini OpenAI-compatible API base URL
14
  api_base="https://generativelanguage.googleapis.com/v1beta/openai/",
15
  )
 
1
  from smolagents import tool
2
  from smolagents import CodeAgent, VisitWebpageTool, DuckDuckGoSearchTool
3
  from smolagents import OpenAIServerModel
4
+ import os
5
 
6
  with open('prompt.txt', 'r') as f:
7
  system_prompt = f.read()
 
9
 
10
  model = OpenAIServerModel(
11
  model_id="gemini-2.0-flash",
12
+ api_key= os.getenv("Gemini_Key"),
13
  # Google Gemini OpenAI-compatible API base URL
14
  api_base="https://generativelanguage.googleapis.com/v1beta/openai/",
15
  )