Python调用AnythingLLM API使用流输出的实现

import requests
import json

def ask_anythingllm(question, workspace_name, api_key):
    url = f"http://ip:port/api/v1/workspace/{workspace_name}/stream-chat" #注意api中有一个不带stream的接口，该接口会一次性返回
    headers = {
        "Authorization": f"Bearer {api_key}",
        "Content-Type": "application/json",
        "accept": "text/event-stream"
    }
    data = {
        "message": question,
        "mode": "query",  # 可选chat/query模式
        'max_tokens': 1024,  # 控制生成文本的长度
        "stream":True
    }
    with requests.post(url, headers=headers, json=data, stream=True) as response:
        if response.status_code == 200:
            # 逐行读取流式响应
            for chunk in response.iter_lines():
                if chunk:  # 确保数据块不为空
                    print(chunk.decode("utf-8"))

# 示例调用
api_key = "WQ59GRH-1JC4M3R-MS2NN3X-VBQCY7H"  # 替换成你自己的apikey
workspace = "8ceb3fb1-4e75-40fe-87db-570d5a689113"
question = "三字经讲的是什么，用50个字概括"
ask_anythingllm(question, workspace, api_key)

import requests

headers = {
    "Authorization": "Bearer WQ59GRH-1JC4M3R-MS2NN3X-VBQCY7H",  # 替换为你的API Key
    "accept": "application/json"
}

response = requests.get("http://ip:3001/api/v1/workspaces", headers=headers)
if response.status_code == 200:
    print("现有工作空间列表：", response.json())
else:
    print(f"获取失败，状态码：{response.status_code}，错误信息：{response.text}")