简介
MaxKB API 项目提供了一个与AI助手进行交互的接口,用户可以通过该接口发送问题并获取AI助手的回答。
功能
- 获取AI助手的ID。
- 打开与AI助手的聊天会话。
- 发送消息到AI助手并获取回复。
使用方法
安装依赖
确保你已经安装了requests库。如果没有安装,可以使用以下命令进行安装:
pip install requests
python main.py
使用的 API 接口中对于请求接口的描述如下
此处使用接口{host}/api/application/profile获取应用ID
def get_ai_id():
headers = {
"AUTHORIZATION": key
}
url = f"{host}/api/application/profile"
r = requests.get(url=url, headers=headers)
if r.status_code == 200:
data = r.json()
return data["data"]["id"]
else:
print(f"Error: {r.status_code}")
return None
获取id之后使用获取的应用id获取会话{host}/api/application/{ai_id}/chat/open的接口
def get_chat_id(ai_id):
headers = {
"AUTHORIZATION": key
}
url = f"{host}/api/application/{ai_id}/chat/open"
r = requests.get(url=url, headers=headers)
if r.status_code == 200:
data = r.json()
return data["data"]
else:
print(f"Error: {r.status_code}")
return None
使用获取的会话ID发送message,获取内容。
def send_message(chat_id, message):
headers = {
"Content-Type": "application/json",
"AUTHORIZATION": key
}
url = f"{host}/api/application/chat_message/{chat_id}"
data = {
"message": message,
"re_chat": "false",
"stream": "false"
}
r = requests.post(url=url, headers=headers, json=data)
if r.status_code == 200:
data = r.json()
return data["data"]["content"]
else:
print(f"Error: {r.status_code}")
return None