Administrator
发布于 2025-06-05 / 25 阅读
0
0

MaxKB API 项目

简介

MaxKB API 项目提供了一个与AI助手进行交互的接口,用户可以通过该接口发送问题并获取AI助手的回答。

功能

  1. 获取AI助手的ID。
  2. 打开与AI助手的聊天会话。
  3. 发送消息到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

评论