aardio 范例: ROA:知识库检索

阿里云知识库 API 指南 | 阿里云知识库 API 参考

import console;
import web.rest.aliyun;

//阿里云知识库 API 指南: https://help.aliyun.com/zh/model-studio/rag-knowledge-base-api-guide
//阿里云知识库 API 参考: https://help.aliyun.com/zh/model-studio/api-bailian-2023-12-29-retrieve

// 配置阿里云凭证
var aliyunClient = web.rest.aliyun(
    accessKeyId = "你的AccessKeyId";     
    accessKeySecret = "你的AccessKeySecret"; 
    version = "2023-12-29"; // 百炼 API 版本
    acsAction = "Retrieve";  // 用于指定 x-acs-action 请求头并切换到 ROA 风格签名
)

// 声明 API 接口
var bailian = aliyunClient.api("https://bailian.cn-beijing.aliyuncs.com")

// 发起检索请求
var result, err = bailian["llm-*******"/*业务空间 ID*/].index.retrieve(
    Query = "1 循环到 10 怎么写？"; // 用户查询词
    IndexId = "l18x*******";   // 知识库 ID;
    DenseSimilarityTopK = 5; // 召回切片数量
    EnableReranking = true;  // 开启重排序以提高准确度
    EnableRewrite = false
)

// 处理结果
if(err) console.error("请求失败：", err); 
else {
    if(result.Success){
        console.success("检索成功！");

        // 遍历命中的文本切片
        for(i, node in result.Data.Nodes){
            console.log("---- 命中切片 [" + i + "] ----");
            console.log("得分:", node.Score);
            console.log("内容:", node.Text);

            // 获取元数据（如原始文件名等）
            //console.dumpJson(node.Metadata);
        }
    }
    else {
        console.error("业务错误码：", result.Code);
        console.error("错误信息：", result.Message);
    }
}

console.pause();