ORDERS = {
    "A1024": {"item": "机械键盘", "status": "运输中", "location": "杭州转运中心"},
    "A1025": {"item": "显示器", "status": "已签收", "location": "上海，本人签收"},
    "B3301": {"item": "咖啡豆", "status": "待揽收", "location": "仓库"},
}
 
 
def track_order(order_id: str) -> str:
    order = ORDERS.get(order_id.strip().upper())
    if order is None:
        return f"没有单号 {order_id}"
    return f"{order_id}：{order['item']}，{order['status']}，现在在{order['location']}"
 
 
def list_open_orders() -> str:
    open_orders = [oid for oid, order in ORDERS.items() if order["status"] != "已签收"]
    if not open_orders:
        return "没有未完成的订单。"
    lines = [track_order(oid) for oid in open_orders]
    return "未完成的订单：\n" + "\n".join(lines)
 

"""快递查询 MCP 服务。
 
要点：
- 传输用 stdio。宿主往 stdin 发 JSON-RPC，本进程从 stdout 回。
  stdout 只能走协议，日志写 stderr，否则会把响应冲乱。
- 启动后宿主先 initialize，再 tools/list。@mcp.tool 登记的函数就是工具：
  函数名是工具名，文档字符串和参数注解会变成模型看到的说明和参数格式。
- instructions 是整台服务器的使用说明，initialize 时交给宿主。
- 真正执行发生在 tools/call。这里只返回结果，不自己调用模型。
"""
 
import logging
import sys
 
from mcp.server.mcpserver import MCPServer
 
import app
 
# 协议占着 stdout，日志必须走 stderr。
logging.basicConfig(stream=sys.stderr, level=logging.INFO, format="%(message)s")
 
# name 是宿主里看到的服务器名。instructions 约束模型怎么用这些工具。
mcp = MCPServer(
    "ParcelTracker",
    instructions="订单信息只能用工具查。没查到就说没有，不要编单号。",
)
 
 
# 装饰器把函数登记进 tools/list。文档字符串是给模型看的说明。
@mcp.tool()
def track_order(order_id: str) -> str:
    """按单号查询一件快递现在到哪了。
 
    Args:
        order_id: 运单号，例如 A1024
    """
    logging.info("track %s", order_id)
    return app.track_order(order_id)
 
 
@mcp.tool()
def list_open_orders() -> str:
    """列出还没签收的订单。"""
    return app.list_open_orders()
 
 
if __name__ == "__main__":
    # 阻塞在 stdio 上，直到宿主关掉管道。不要往 stdout 打印别的内容。
    mcp.run(transport="stdio")
 

"mcp": {
    "test": {
        "type": "local",
        "command": [
            "python",
            "/home/Brant/test/chat/server.py"
        ],
        "enabled": true
    }
},