Final_Assignment / agent.py
Anoire's picture
a
b6ca59d
raw
history blame
574 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.getenv("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,
)