手把手教你：基于 AnythingLLM API 训练并调用本地知识库

import requests

# 替换为实际的 API 端点
api_url = "http://localhost:3001/api/v1/workspace/new"

# 身份验证，设置请求头
headers = {
    "accept": "application/json",
    "Authorization": "Bearer your_api_key",  #注意，替换apikey一定要保留Bearer 
    "Content-Type": "application/json"
}

# 准备创建工作区所需的数据
workspace_data = {   
    "name": "New Workspace", #我这里创建的 ddj，替换成你的工作空间
    "similarityThreshold": 0.7,    
    "openAiTemp": 0.7,  
    "openAiHistory": 20, 
    "openAiPrompt": "Custom prompt for responses",    
    "queryRefusalResponse": "Custom refusal message",    
    "chatMode": "chat",   
    "topN": 4
}

try:
    # 发送 POST 请求
    response = requests.post(api_url, headers=headers, json=workspace_data)

    # 检查响应状态码
    if response.status_code == 200:  # 200 表示创建成功
        result = response.json()
        print("工作区创建成功：", result)
    else:
        print(f"工作区创建失败，状态码：{response.status_code}，错误信息：{response.text}")
except requests.RequestException as e:
    print(f"请求发生错误：{e}")

#如下结果则执行成功
{
  "workspace": {
    "id": 4,
    "name": "ddj",
    "slug": "ddj",
    "vectorTag": null,
    "createdAt": "2025-02-10T16:14:58.744Z",
    "openAiTemp": 0.7,
    "openAiHistory": 20,
    "lastUpdatedAt": "2025-02-10T16:14:58.744Z",
    "openAiPrompt": "Custom prompt for responses",
    "similarityThreshold": 0.7,
    "chatProvider": null,
    "chatModel": null,
    "topN": 4,
    "chatMode": "chat",
    "pfpFilename": null,
    "agentProvider": null,
    "agentModel": null,
    "queryRefusalResponse": "Custom refusal message",
    "vectorSearchMode": "default"
  },
  "message": null
}

# Python调用AnythingLLM文档上传API   
import requests   
headers = {       
    "Authorization": "Bearer YOUR_API_KEY",       
    "accept": "application/json"   }   
files = {'file': open('product_guide.pdf', 'rb')}   
response = requests.post(       
    'http://localhost:3001/api/v1/workspace/{workspace_id}/document',       
    headers=headers,       
    files=files   
)   
print(response.json())  # 返回文档ID及处理状态   ``  

import requests
import jsondef 
ask_anythingllm(question, workspace_name, api_key):    
url = f"http://localhost:3001/api/v1/workspace/{workspace_name}/chat"    
headers = {        
    "Authorization": f"Bearer {api_key}",        
    "Content-Type": "application/json",        
    "accept": "application/json"    
}    
data = {        
    "message": question,        
    "mode": "query"  # 可选chat/query模式    
}    
response = requests.post(url, headers=headers, json=data)    
if response.status_code == 200:        
    result = response.json()        
    # 提取有效回答（去除思考过程）        
    answer = result['textResponse'].split('</think>')[-1].strip()        
    sources = result.get('sources', [])        
    return answer, sources    
else:        
    return f"Error: {response.text}", []
    # 示例调用
    api_key = "your_api_key"  #替换成你自己的apikey
    workspace = "product_kb"
    question = "道德经讲的是什么，用50个字概括"
    answer, sources = ask_anythingllm(question, workspace, api_key)
    print("回答:", answer)
    print("来源:", [src['title'] for src in sources])

回答: 道德经讲的是“无为而治、柔弱胜刚强、厚积薄发”，强调以智慧和德行为本，以柔克刚，顺其自然，实现内心与外在的平衡。
来源: ['ddj.txt']