File size: 576 Bytes
b4e3c13
 
 
b6ca59d
b4e3c13
 
 
 
 
 
 
7372fba
b4e3c13
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
from smolagents import tool
from smolagents import CodeAgent, VisitWebpageTool, DuckDuckGoSearchTool
from smolagents import OpenAIServerModel
import os

with open('prompt.txt', 'r') as f:
    system_prompt = f.read()
system_prompt

model = OpenAIServerModel(
    model_id="gemini-2.0-flash",
    api_key = os.environ["Gemini_Key"],
    # Google Gemini OpenAI-compatible API base URL
    api_base="https://generativelanguage.googleapis.com/v1beta/openai/",
)
    
agent = CodeAgent(
    model=model,
    tools=[ VisitWebpageTool(), DuckDuckGoSearchTool()],
    max_steps=20,
)