Final_Assignment / agent.py
Anoire's picture
a
7372fba
raw
history blame contribute delete
576 Bytes
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,
)