Spaces:
Runtime error
Runtime error
File size: 1,177 Bytes
1fa2efd 57ee912 1fa2efd 57ee912 1fa2efd |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 |
from smolagents import HfApiModel, CodeAgent, DuckDuckGoSearchTool, tool
@tool
def add(int: a, int: b) -> int:
"""
This tool adds two integers together and returns an integer.
args:
a: Integer 1
b: Integer 2
"""
return a + b
@tool
def subtract(int: a, int: b) -> int:
"""
This tool subtracts two integers and returns an integer.
args:
a: Integer 1
b: Integer 2
"""
return a - b
@tool
def multiply(int: a, int: b) -> int:
"""
This tool multiplies two integers together and returns an integer.
args:
a: Integer 1
b: Integer 2
"""
return a * b
@tool
def divide(int: a, int: b) -> float:
"""
This tool divides two integers together and returns a float.
args:
a: Integer 1
b: Integer 2
"""
if b == 0:
raise ValueError("Cannot divide by zero.")
return a* (1.0) / b
tools=[
add,
subtract,
multiply,
divide
]
def create_agent():
manager_agent = CodeAgent(
model = HfApiModel(),
)
return manager_agent
|