36 lines
853 B
Python
36 lines
853 B
Python
|
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 :
|
|||
|
|
"""
|
|||
|
|
功能：
|
|||
|
|
使用账号密码，调用登陆api进行登陆。
|
|||
|
|
参数：
|
|||
|
|
username:用户的账号，或者是用户名称
|
|||
|
|
password:用户的密码
|
|||
|
|
"""
|
|||
|
|
payload={
|
|||
|
|
'username':username,
|
|||
|
|
'password':password
|
|||
|
|
}
|
|||
|
|
url = 'http://localhost:8000/api/v1/login/access-token'
|
|||
|
|
response = requests.post(url,data=payload)
|
|||
|
|
|
|||
|
|
return response.text
|
|||
|
|
|
|||
|
|
@mcp.tool()
|
|||
|
|
def register_function(username:str, password:str):
|
|||
|
|
"""
|
|||
|
|
功能：
|
|||
|
|
注册用户，调用注册api进行新用户注册。
|
|||
|
|
参数：
|
|||
|
|
username:用户的账号，或者是用户名称
|
|||
|
|
password:用户的密码
|
|||
|
|
"""
|
|||
|
|
|
|||
|
|
|
|||
|
|
|