Definisci i prompt dell'agente

app.py

@@ -6,6 +6,7 @@ import inspect
 import pandas as pd
 
 from smolagents import CodeAgent, TransformersModel, VisitWebpageTool, PythonInterpreterTool, WebSearchTool, WikipediaSearchTool, FinalAnswerTool, Tool, tool # InferenceClientModel, GoogleSearchTool (usa SERPAPI_API_KEY), DuckDuckGoSearchTool
+from smolagents.agents import PromptTemplates, EMPTY_PROMPT_TEMPLATES
 
 # (Keep Constants as is)
 # --- Constants ---
@@ -27,12 +28,25 @@ def invert_sentence(sentence: str) -> str:
 class FirstAgent:
 	### First Agent is the first attempt to develop an agent for the course. ###
 	def __init__(self):
+		""" Initializes the FirstAgent with a TransformersModel and a CodeAgent. """
 		model_id = "HuggingFaceTB/SmolLM2-1.7B-Instruct"
 		model = TransformersModel(model_id=model_id)
 		
 		# Inizializza l'agente
+		prompt_templates = EMPTY_PROMPT_TEMPLATES
+		prompt_templates["system_prompt"] = """You are an intelligent agent that answers questions and uses tools to help users.
+		Think step by step and use tools when needed.
+		If you don't know the answer, say 'I don't know'.
+		If you need to search the web, use the WebSearchTool.
+		If you need to visit a webpage, use the VisitWebpageTool.
+		If you need to run Python code, use the PythonInterpreterTool.
+		If you need to generate code to extract information, generate the code and use the PythonInterpreterTool to execute the code and extract the informations.
+		If you find information in Wikipedia, use the WikipediaSearchTool.
+		Verify your final answer and the format of the final answer before returning it."""
+
 		self.agent = CodeAgent(
 			model=model,
+			prompt_templates=prompt_templates,
 			tools=[
 				WebSearchTool(),
 				PythonInterpreterTool(),
@@ -43,11 +57,12 @@ class FirstAgent:
 		print("FirstAgent inizializzato.")
 	
 	def __call__(self, question: str) -> str:
+		""" Runs the agent with the provided question and returns the answer. """
 		print(f"Agent ricevuto domanda (primi 50 char): {question[:50]}...")
 		try:
 			answer = self.agent.run(question)
 			print(f"Agent restituisce risposta: {str(answer)[:100]}...")
-			return str(answer)
+			return answer
 		except Exception as e:
 			print(f"Errore nell'agente: {e}")
 			return f"Errore nell'agente: {str(e)}"