Dify自定义工具开发与部署【2】

MCP模式

1. Python服务代码

import json

from mcp.server.fastmcp import FastMCP

mcp = FastMCP("weather_sse", port=8002, host="0.0.0.0")


@mcp.tool()
async def get_weather(city: str) -> str:
    """
     获取指定城市的当前天气信息。
    参数：
        - **city**: 城市名称
    返回：
        str: 天气信息。
    """
    return json.dumps({"status": "success", "data": {'weather': 'rainy', 'temperature': '20.5'}})


@mcp.tool()
async def get_air_quality(city: str):
    """
    获取指定城市的当前空气质量信息。

    - **city**: 城市名称
    """
    return json.dumps({"status": "success", "data": {'PM_2_5': '120', 'pm_10': '180'}})


if __name__ == "__main__":
    mcp.run(transport="sse")

2. 启动

E:\PythonProject\myPythonDemo\.venv\Scripts\python.exe E:\PythonProject\myPythonDemo\src\agent_integration\mcpDemo.py 
INFO:     Started server process [12988]
INFO:     Waiting for application startup.
INFO:     Application startup complete.
INFO:     Uvicorn running on http://0.0.0.0:8002 (Press CTRL+C to quit)

4. 准备MCP服务配置

{
    "server_name":
    {
        "transport": "sse",
        "url": "http://真实IP:8002/sse",
        "headers":
        {},
        "timeout": 60,
        "sse_read_timeout": 60
    }
}

5. 在Dify上安装这个插件，并将上述JSON转成字符串形式，点击保存
   
6. 配置工具
   
7. 实践（注意：这里提示词为空好像不行，随便写个提示词就行）
   

大功告成！！！