import wikipediaapi

def get_studio_albums(query):
	# Use the Wikipedia API to search for the artist's studio albums
	wiki = wikipediaapi.Wikipedia(
						user_agent="FinalAssignmentAgent (raff-huggingface@usa.net)",
						language="en",
						# content_type="text",  # "summary" or "text"
						extract_format=wikipediaapi.ExtractFormat.WIKI, # "WIKI" or "HTML"
	)
	page = wiki.page(query)
	if page.exists():
		return page.text
	return "No information found."

artist_name = "Mercedes Sosa"
# studio_albums = get_studio_albums(artist_name)

# print(f"The artist {artist_name} published {studio_albums} studio albums between 2000 and 2009.")

from smolagents import CodeAgent, ToolCallingAgent, TransformersModel, VisitWebpageTool, PythonInterpreterTool, WebSearchTool, WikipediaSearchTool, FinalAnswerTool, Tool, tool # InferenceClientModel, GoogleSearchTool (usa SERPAPI_API_KEY), DuckDuckGoSearchTool
model_id = "HuggingFaceTB/SmolLM2-1.7B-Instruct"
model = TransformersModel(model_id=model_id, trust_remote_code=True, max_new_tokens=4096)
agent = CodeAgent(
	model=model,
	tools=[
		WebSearchTool(),
		VisitWebpageTool(),
		PythonInterpreterTool(),
		FinalAnswerTool(),
	],
)
print(agent.prompt_templates)