mirror of
https://github.com/microsoft/autogen.git
synced 2025-07-14 20:40:45 +00:00
29 lines
778 B
Python
29 lines
778 B
Python
|
from autogen_core.tools import FunctionTool
|
||
|
|
|
||
|
|
|
||
|
|
def calculator(a: float, b: float, operator: str) -> str:
|
||
|
|
try:
|
||
|
|
if operator == "+":
|
||
|
|
return str(a + b)
|
||
|
|
elif operator == "-":
|
||
|
|
return str(a - b)
|
||
|
|
elif operator == "*":
|
||
|
|
return str(a * b)
|
||
|
|
elif operator == "/":
|
||
|
|
if b == 0:
|
||
|
|
return "Error: Division by zero"
|
||
|
|
return str(a / b)
|
||
|
|
else:
|
||
|
|
return "Error: Invalid operator. Please use +, -, *, or /"
|
||
|
|
except Exception as e:
|
||
|
|
return f"Error: {str(e)}"
|
||
|
|
|
||
|
|
|
||
|
|
# Create calculator tool
|
||
|
|
calculator_tool = FunctionTool(
|
||
|
|
name="calculator",
|
||
|
|
description="A simple calculator that performs basic arithmetic operations",
|
||
|
|
func=calculator,
|
||
|
|
global_imports=[],
|
||
|
|
)
|