Update app.py

app.py

@@ -99,6 +99,7 @@ PRUNING_PHASE_COEFFS = {
 
 # Tool definitions
 
+@tool
 def get_moon_info(date_time: str) -> dict:
     """
     Returns Moon's Zodiac position, phase, and fertility and pruning indices for the given date/time.
@@ -190,6 +191,7 @@ def get_moon_info(date_time: str) -> dict:
     except Exception as e:
         raise ValueError(f"Error in get_moon_info: {str(e)}")
 
+@tool
 def get_current_time_in_timezone(timezone: str) -> str:
     """
     Returns the current local time in the specified timezone with description.
@@ -206,6 +208,7 @@ def get_current_time_in_timezone(timezone: str) -> str:
     except Exception as e:
         return f"Error: {str(e)}"
 
+@tool
 def get_current_time_raw(timezone: str) -> str:
     """
     Returns current local time in specified timezone as ISO 8601 string.
@@ -273,7 +276,7 @@ image_generation_tool = load_tool("agents-course/text-to-image", trust_remote_co
 # Initialize the agent without the 'memory' parameter
 agent = CodeAgent(
     model=model,
-    tools=[final_answer, get_moon_info, get_current_time_in_timezone, get_current_time_raw, image_generation_tool],
+    tools=[final_answer, get_moon_info, get_current_time_in_timezone, get_current_time_raw],
     max_steps=10,
     verbosity_level=1,
     prompt_templates=prompt_templates,