ce4a9b28a9
* Add new tool to the arithmetic toolkit for summation of a range. * Add ability to attach debugger to cli. Use `.vscode/launch.json`'s "Debug arcade dev" to do so. * Fix issue in cli's main that used the incorrect url.
75 lines
1.6 KiB
Python
75 lines
1.6 KiB
Python
import math
|
|
from typing import Annotated
|
|
|
|
from arcade.sdk import tool
|
|
|
|
|
|
@tool
|
|
def add(
|
|
a: Annotated[int, "The first number"], b: Annotated[int, "The second number"]
|
|
) -> Annotated[int, "The sum of the two numbers"]:
|
|
"""
|
|
Add two numbers together
|
|
"""
|
|
return a + b
|
|
|
|
|
|
@tool
|
|
def subtract(
|
|
a: Annotated[int, "The first number"], b: Annotated[int, "The second number"]
|
|
) -> Annotated[int, "The difference of the two numbers"]:
|
|
"""
|
|
Subtract two numbers
|
|
"""
|
|
return a - b
|
|
|
|
|
|
@tool
|
|
def multiply(
|
|
a: Annotated[int, "The first number"], b: Annotated[int, "The second number"]
|
|
) -> Annotated[int, "The product of the two numbers"]:
|
|
"""
|
|
Multiply two numbers together
|
|
"""
|
|
return a * b
|
|
|
|
|
|
@tool
|
|
def divide(
|
|
a: Annotated[int, "The first number"], b: Annotated[int, "The second number"]
|
|
) -> Annotated[float, "The quotient of the two numbers"]:
|
|
"""
|
|
Divide two numbers
|
|
"""
|
|
return a / b
|
|
|
|
|
|
@tool
|
|
def sqrt(
|
|
a: Annotated[int, "The number to square root"],
|
|
) -> Annotated[float, "The square root of the number"]:
|
|
"""
|
|
Get the square root of a number
|
|
"""
|
|
return math.sqrt(a)
|
|
|
|
|
|
@tool
|
|
def sum_list(
|
|
numbers: Annotated[list[float], "The list of numbers"],
|
|
) -> Annotated[float, "The sum of the numbers in the list"]:
|
|
"""
|
|
Sum all numbers in a list
|
|
"""
|
|
return sum(numbers)
|
|
|
|
|
|
@tool
|
|
def sum_range(
|
|
start: Annotated[int, "The start of the range to sum"],
|
|
end: Annotated[int, "The end of the range to sum"],
|
|
) -> Annotated[int, "The sum of the numbers in the list"]:
|
|
"""
|
|
Sum all numbers from start through end
|
|
"""
|
|
return sum([i for i in range(start, end + 1)])
|