You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
41 lines
1.1 KiB
41 lines
1.1 KiB
|
|
from fastmcp import FastMCP
|
|
from typing import Dict, Any
|
|
import time
|
|
|
|
# Initialize MCP server
|
|
mcp = FastMCP("Math-Mock-Server")
|
|
|
|
@mcp.tool()
|
|
def add(a: int, b: int) -> int:
|
|
"""Add two numbers together"""
|
|
try:
|
|
time.sleep(0.1) # Simulate processing time
|
|
return a + b
|
|
except Exception as e:
|
|
return {"error": f"Error adding numbers: {str(e)}"}
|
|
|
|
@mcp.tool()
|
|
def multiply(a: int, b: int) -> int:
|
|
"""Multiply two numbers together"""
|
|
try:
|
|
time.sleep(0.1) # Simulate processing time
|
|
return a * b
|
|
except Exception as e:
|
|
return {"error": f"Error multiplying numbers: {str(e)}"}
|
|
|
|
@mcp.tool()
|
|
def divide(a: int, b: int) -> float:
|
|
"""Divide two numbers"""
|
|
try:
|
|
if b == 0:
|
|
return {"error": "Cannot divide by zero"}
|
|
time.sleep(0.1) # Simulate processing time
|
|
return a / b
|
|
except Exception as e:
|
|
return {"error": f"Error dividing numbers: {str(e)}"}
|
|
|
|
if __name__ == "__main__":
|
|
print("Starting Mock Math Server on port 8000...")
|
|
mcp.run(transport="sse", host="0.0.0.0", port=8000)
|