Update app.py

app.py

@@ -8,7 +8,11 @@ import yaml
 
 @tool
 def is_prime(num: int) -> str:
-    """A tool that determines if an input number is prime and returns a string result."""
+    """A tool that determines if an input number is prime and returns a string result.
+    Args:
+        num: A int that is to be checked for being prime.
+    """
+    
     try:
         prime = check_prime(num)
         return f"The number {num} is {'Prime!!!' if prime else 'not Prime :('}"
@@ -17,7 +21,10 @@ def is_prime(num: int) -> str:
 
 @tool
 def get_current_time_in_timezone(timezone: str) -> str:
-    """A tool that fetches the current local time in a specified timezone."""
+    """A tool that fetches the current local time in a specified timezone.
+    Args:
+        timezone: A string representing a valid timezone (e.g., 'America/New_York').
+    """
     try:
         tz = pytz.timezone(timezone)
         local_time = datetime.datetime.now(tz).strftime("%Y-%m-%d %H:%M:%S")