aardio 范例: 知识库检索

指南 | 发送请求。参数训明 | 消耗 token 数，计费规则

import console.int;
import web.rest.volcengine.client;

//调用火山引擎云
//指南: https://www.aardio.com/zh-cn/docs/library-guide/std/web/rest/volcengine.html#knowledge

//创建火山引擎云 API 客户端
var http  = web.rest.volcengine.client(
    accessKeyId = "ak";
    secretAccessKey = "sk"; 
    region =  "cn-north-1"; 
    service = "air";
    accountId = "账号ID"//鼠标放到火山头像上显示的账号ID
);

//声明 API 
var air = http.api("https://api-knowledgebase.ml_platform.cn-beijing.volces.com/api/knowledge/collection/");

//发送请求。参数训明: https://www.volcengine.com/docs/84313/1350012
var resp,err = air.search_knowledge({
    name="知识库名称";
    project="default";//项目名称
    dense_weight=0.5;
    limit=4; //限制最大返回结果数量
    post_processing={
        get_attachment_link=false;
        rerank_only_chunk=false;
        rerank_switch=false     
    };
    query="这里写需要搜索知识库的用户问题"   
})

//获取数据
var data = resp.data;

//提取召回片段
var result = table.map(data.result_list
    ,lambda(v,k) '# ' +v.chunk_title +'\r\n\r\n'+ v.content )

//显示结果
for(k,v in result){
    console.log(v)
    console.more();
}

//消耗 token 数，计费规则: https://www.volcengine.com/docs/82379/1099320
console.dumpJson(data.token_usage)