BBIT-Kai
19 lines
647 B
Python
19 lines
647 B
Python
# server.py
|
|
import logging
|
|
import math
|
|
import random
|
|
|
|
from fastmcp import FastMCP
|
|
|
|
logger = logging.getLogger("Calculator")
|
|
|
|
|
|
def register_calculator_tools(mcp: FastMCP):
|
|
# Add an addition tool
|
|
@mcp.tool()
|
|
def calculator(python_expression: str) -> dict:
|
|
"""For mathamatical calculation, always use this tool to calculate the result of a python expression. You can use 'math' or 'random' directly, without 'import'."""
|
|
result = eval(python_expression, {"math": math, "random": random})
|
|
logger.info(f"Calculating formula: {python_expression}, result: {result}")
|
|
return {"success": True, "result": result}
|