|
|
import requests |
|
|
import os |
|
|
from dotenv import load_dotenv |
|
|
from smolagents import CodeAgent, WikipediaSearchTool, VisitWebpageTool, DuckDuckGoSearchTool, InferenceClientModel |
|
|
|
|
|
load_dotenv() |
|
|
|
|
|
print(os.environ) |
|
|
|
|
|
class TheAgent: |
|
|
def __init__(self): |
|
|
headers = { |
|
|
"User-Agent": "Mozilla/5.0 (Macintosh; Intel Mac OS X 10_15_7) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/138.0.0.0 Safari/537.36", |
|
|
"Accept-Language": "en-US,en;q=0.9,en;q=0.8" |
|
|
} |
|
|
|
|
|
search_tool = DuckDuckGoSearchTool(rate_limit = 5, headers=headers) |
|
|
wiki_tool = WikipediaSearchTool() |
|
|
page_tool = VisitWebpageTool() |
|
|
|
|
|
model_id = "Qwen/Qwen3-235B-A22B-Instruct-2507" |
|
|
|
|
|
|
|
|
self.agent = CodeAgent( |
|
|
tools=[wiki_tool, page_tool, search_tool], |
|
|
model=InferenceClientModel(model_id=model_id), |
|
|
) |
|
|
|
|
|
print("BasicAgent initialized.") |
|
|
|
|
|
def __call__(self, question: str): |
|
|
try: |
|
|
result = self.agent.run(question) |
|
|
return str(result) |
|
|
except requests.exceptions.HTTPError as e: |
|
|
print(f"Error: {e.response.text}") |
|
|
|
|
|
if __name__ == "__main__": |
|
|
agent = TheAgent() |
|
|
question = "Who nominated the only Featured Article on English Wikipedia about a dinosaur that was promoted in November 2016?" |
|
|
question = "How many studio albums were published by Mercedes Sosa between 2000 and 2009 (included)? You can use the latest 2022 version of english wikipedia." |
|
|
answer = agent(question) |
|
|
print(f"Answer: {answer}") |
|
|
|
