Update agent.py

agent.py

@@ -4,7 +4,7 @@ from smolagents import CodeAgent, DuckDuckGoSearchTool, InferenceClientModel
 model = InferenceClientModel(model_id="meta-llama/Llama-2-7b-chat-hf")  # Or another model
 
 # Define custom tools if needed (optional)
-from smolagents.tool import tool
+from smolagents import tool
 
 @tool
 def add(a: int, b: int) -> int: