f04087b389
Migrated all interfaces to get and return strings. Added tests and evals for all functions (except the random generation) Math functions are now organized into different math categories --------- Co-authored-by: Nate Barbettini <nate@arcade-ai.com>
40 lines
991 B
Python
40 lines
991 B
Python
import math
|
|
from decimal import Decimal
|
|
from typing import Annotated
|
|
|
|
from arcade.sdk import tool
|
|
|
|
|
|
@tool
|
|
def abs_val(
|
|
a: Annotated[str, "The number as a string"],
|
|
) -> Annotated[str, "The absolute value of the number as a string"]:
|
|
"""
|
|
Calculate the absolute value of a number
|
|
"""
|
|
# Use Decimal for arbitrary precision
|
|
return str(abs(Decimal(a)))
|
|
|
|
|
|
@tool
|
|
def factorial(
|
|
a: Annotated[str, "The non-negative integer to compute the factorial for as a string"],
|
|
) -> Annotated[str, "The factorial of the number as a string"]:
|
|
"""
|
|
Compute the factorial of a non-negative integer
|
|
Returns "1" for "0"
|
|
"""
|
|
return str(math.factorial(int(a)))
|
|
|
|
|
|
@tool
|
|
def sqrt(
|
|
a: Annotated[str, "The number to square root as a string"],
|
|
) -> Annotated[str, "The square root of the number as a string"]:
|
|
"""
|
|
Get the square root of a number
|
|
If
|
|
"""
|
|
# Use Decimal for arbitrary precision
|
|
a_decimal = Decimal(a)
|
|
return str(a_decimal.sqrt())
|