81 lines
2.9 KiB
Python
81 lines
2.9 KiB
Python
from mcp.server.fastmcp import FastMCP
|
||
|
||
from conf import settings
|
||
from create_logger import logger
|
||
|
||
# 创建FastMCP实例
|
||
order_mcp = FastMCP(name="OrderTools",
|
||
instructions="票务预定工具，通过调用API完成火车票、飞机票和演唱会票的预定。",
|
||
log_level="ERROR",
|
||
host="127.0.0.1", port=8003)
|
||
|
||
|
||
@order_mcp.tool(
|
||
name="order_train",
|
||
description="根据时间、车次、座位类型、数量预定火车票"
|
||
)
|
||
def order_train(departure_date: str, train_number: str, seat_type: str, number: int) -> str:
|
||
'''
|
||
Args:
|
||
departure_date (str): 出发日期，如 '2025-10-30'
|
||
train_number (str): 火车车次，如 'G346'
|
||
seat_type (str): 座位类型，如 '二等座'
|
||
number (int): 订购张数
|
||
'''
|
||
logger.info(f"正在订购火车票: {departure_date}, {train_number}, {seat_type}, {number}")
|
||
logger.info(f"恭喜，火车票预定成功！")
|
||
return "恭喜，火车票预定成功！"
|
||
|
||
@order_mcp.tool(
|
||
name="order_flight",
|
||
description="根据时间、班次、座位类型、数量预定飞机票"
|
||
)
|
||
def order_flight(departure_date: str, flight_number: str, seat_type: str, number: int) -> str:
|
||
'''
|
||
Args:
|
||
departure_date (str): 出发日期，如 '2025-10-30'
|
||
flight_number (str): 飞机班次，如 'CA6557'
|
||
seat_type (str): 座位类型，如 '经济舱'
|
||
number (int): 订购张数
|
||
'''
|
||
logger.info(f"正在订购飞机票: {departure_date}, {flight_number}, {seat_type}, {number}")
|
||
logger.info(f"恭喜，飞机票预定成功！")
|
||
return "恭喜，飞机票预定成功！"
|
||
|
||
|
||
@order_mcp.tool(
|
||
name="order_concert",
|
||
description="根据时间、明星、场地、座位类型、数量预定演出票"
|
||
)
|
||
def order_concert(start_date: str, aritist: str, venue: str, seat_type: str, number: int) -> str:
|
||
'''
|
||
Args:
|
||
start_date (str): 开始日期，如 '2025-10-30'
|
||
aritist (str): 明星，如 '刀郎'
|
||
venue (str): 场地，如 '上海体育馆'
|
||
seat_type (str): 座位类型，如 '看台'
|
||
number (int): 订购张数
|
||
'''
|
||
logger.info(f"正在订购演出票: {start_date}, {aritist}, {venue}, {seat_type}, {number}")
|
||
logger.info(f"恭喜，演出票预定成功！")
|
||
return "恭喜，演出票预定成功！"
|
||
|
||
|
||
# 创建票务预定MCP服务器
|
||
def create_order_mcp_server():
|
||
# 打印服务器信息
|
||
logger.info("=== 票务预定MCP服务器信息 ===")
|
||
logger.info(f"名称: {order_mcp.name}")
|
||
logger.info(f"描述: {order_mcp.instructions}")
|
||
|
||
# 运行服务器
|
||
try:
|
||
print("服务器已启动，请访问 http://127.0.0.1:8003/mcp")
|
||
order_mcp.run(transport="streamable-http") # 使用 streamable-http 传输方式
|
||
except Exception as e:
|
||
print(f"服务器启动失败: {e}")
|
||
|
||
|
||
if __name__ == "__main__":
|
||
# 调用创建服务器函数
|
||
create_order_mcp_server() |