2025-08-28 17:32:41 +08:00
|
|
|
|
from mcp.server.fastmcp import FastMCP
|
|
|
|
|
|
import requests
|
|
|
|
|
|
|
|
|
|
|
|
mcp = FastMCP(name='FastMCP',host='0.0.0.0',port=8000)
|
|
|
|
|
|
|
|
|
|
|
|
@mcp.tool()
|
|
|
|
|
|
def login_function(username:str, password:str) -> dict :
|
|
|
|
|
|
"""
|
|
|
|
|
|
功能:
|
2025-08-29 17:30:00 +08:00
|
|
|
|
接受账号密码,调用登陆api,发送调用结果,并展示调用输出。
|
2025-08-28 17:32:41 +08:00
|
|
|
|
参数:
|
|
|
|
|
|
username:用户的账号,或者是用户名称
|
|
|
|
|
|
password:用户的密码
|
|
|
|
|
|
"""
|
|
|
|
|
|
payload={
|
|
|
|
|
|
'username':username,
|
|
|
|
|
|
'password':password
|
|
|
|
|
|
}
|
2025-08-29 17:30:00 +08:00
|
|
|
|
url = 'http://192.168.50.11:8000/api/v1/login/access-token'
|
2025-08-28 17:32:41 +08:00
|
|
|
|
response = requests.post(url,data=payload)
|
|
|
|
|
|
|
|
|
|
|
|
return response.text
|
|
|
|
|
|
|
|
|
|
|
|
@mcp.tool()
|
|
|
|
|
|
def register_function(username:str, password:str):
|
|
|
|
|
|
"""
|
|
|
|
|
|
功能:
|
2025-08-29 17:30:00 +08:00
|
|
|
|
注册用户,调用注册api进行新用户注册。+
|
2025-08-28 17:32:41 +08:00
|
|
|
|
参数:
|
|
|
|
|
|
username:用户的账号,或者是用户名称
|
|
|
|
|
|
password:用户的密码
|
|
|
|
|
|
"""
|
2025-08-29 17:30:00 +08:00
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
if __name__=='__main__':
|
|
|
|
|
|
mcp.run(transport='streamable-http')
|
|
|
|
|
|
|
2025-08-28 17:32:41 +08:00
|
|
|
|
|
|
|
|
|
|
|