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.
34 lines
786 B
34 lines
786 B
|
|
from fastmcp import FastMCP
|
|
from litellm import LiteLLM
|
|
|
|
# Initialize MCP server for math operations
|
|
mcp = FastMCP("Math-Server")
|
|
|
|
@mcp.tool()
|
|
def add(a: float, b: float) -> float:
|
|
"""Add two numbers together"""
|
|
return a + b
|
|
|
|
@mcp.tool()
|
|
def subtract(a: float, b: float) -> float:
|
|
"""Subtract b from a"""
|
|
return a - b
|
|
|
|
@mcp.tool()
|
|
def multiply(a: float, b: float) -> float:
|
|
"""Multiply two numbers together"""
|
|
return a * b
|
|
|
|
@mcp.tool()
|
|
def divide(a: float, b: float) -> float:
|
|
"""Divide a by b"""
|
|
if b == 0:
|
|
return {"error": "Cannot divide by zero"}
|
|
return a / b
|
|
|
|
if __name__ == "__main__":
|
|
print("Starting Math Server on port 6274...")
|
|
llm = LiteLLM(model_name="gpt-4o-mini")
|
|
mcp.run(transport="sse", host="0.0.0.0", port=6274)
|