使用dify的api连接外部知识库,dify连接ragflow的知识库(附java代码)

dify的知识库一般般,但是ragflow的知识库很强大,今天教大家如何使用dify连接ragflow的知识库

 一.ragflow的准备工作

1.在ragflow建立一个知识库,拿到知识库的id,红框圈出来的地方就是这个知识库的id,后面要用到

2.拿到ragflow的api的key 

二.写代码,将ragflow的接口返回的内容按照dify的标准格式进行组装

 这里我使用的是Java代码,本地直接新建一个空的springboot项目,然后将以下代码复制粘贴,修改下参数即可

实体类(无需修改)

public class RetrievalSetting {
    @NotNull
    private int top_k;
    @NotNull
    private float score_threshold;
    // Getters and Setters
    public int getTop_k() {
        return top_k;
    }
    public void setTop_k(int top_k) {
        this.top_k = top_k;
    }
    public float getScore_threshold() {
        return score_threshold;
    }
    public void setScore_threshold(float score_threshold) {
        this.score_threshold = score_threshold;
    }
}

public class RetrievalRequest {
    @NotBlank
    private String knowledge_id;
    @NotBlank
    private String query;
    @NotNull
    private RetrievalSetting retrieval_setting;
    // Getters and Setters
    public String getKnowledge_id() {
        return knowledge_id;
    }
    public void setKnowledge_id(String knowledge_id) {
        this.knowledge_id = knowledge_id;
    }
    public String getQuery() {
        return query;
    }
    public void setQuery(String query) {
        this.query = query;
    }
    public RetrievalSetting getRetrieval_setting() {
        return retrieval_setting;
    }
    public void setRetrieval_setting(RetrievalSetting retrieval_setting) {
        this.retrieval_setting = retrieval_setting;
    }
}

@RestController
@RequestMapping("/")
public class RetrievalController {
    private static final String ORIGINAL_API_URL = "http://你的ragflowapi的主机和端口/api/v1/retrieval";
    @PostMapping("/retrieval")
    public ResponseEntity retrieveChunks(
            @RequestBody RetrievalRequest requestData,
            @RequestHeader("Authorization") String authorization) {
        // 验证 Authorization 头
        if (!authorization.startsWith("Bearer ")) {
            return ResponseEntity.status(HttpStatus.UNAUTHORIZED).body(null);
        }
        String apiKey = authorization.split(" ")[1];
        // 构建请求体
        Map payload = new HashMap();
        payload.put("question", requestData.getQuery());
        payload.put("dataset_ids", Collections.singletonList(requestData.getKnowledge_id()));
        payload.put("top_k", requestData.getRetrieval_setting().getTop_k());
        payload.put("similarity_threshold", requestData.getRetrieval_setting().getScore_threshold());
        // 构建请求头
        HttpHeaders headers = new HttpHeaders();
        headers.setContentType(MediaType.APPLICATION_JSON);
        headers.set("Authorization", "Bearer " + apiKey);
        HttpEntity entity = new HttpEntity(payload, headers);
        // 发送请求
        RestTemplate restTemplate = new RestTemplate();
        try {
            ResponseEntity response = restTemplate.exchange(
                    ORIGINAL_API_URL,
                    HttpMethod.POST,
                    entity,
                    Map.class
            );
            if (response.getStatusCode() != HttpStatus.OK) {
                return ResponseEntity.status(response.getStatusCode()).body(null);
            }
            // 处理响应数据
            Map originalData = response.getBody();
            Map data = (Map) originalData.get("data");
            List chunks = (List) data.get("chunks");
            List records = new ArrayList();
            if (chunks != null) {
                for (Map chunk : chunks) {
                    Map record = new HashMap();
                    record.put("content", chunk.get("content"));
                    record.put("score", chunk.get("similarity"));
                    record.put("title", chunk.getOrDefault("document_keyword", "Unknown Document"));
                    Map metadata = new HashMap();
                    metadata.put("document_id", chunk.get("document_id"));
                    record.put("metadata", metadata);
                    records.add(record);
                }
            }
            Map result = new HashMap();
            result.put("records", records);
            return ResponseEntity.ok(result);
        } catch (HttpClientErrorException e) {
            return ResponseEntity.status(e.getStatusCode()).body(null);
        }
    }
}

在controller中修改主机和端口,能在这里找到:

完事可以用api工具测试一下这个接口通不通

 注意这里请求头要加一个Bearer 然后接上apikey,中间有个空格

三.dify连接ragflow的知识库

1.rag先连接

2.新建知识库 

 3.进行召回测试

这里是我运行的结果,完美

 至此结束,撒花