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

JAY.LIN 收录于未分类

2025-03-11 约 804 字预计阅读 2 分钟

https://bing.ee123.net/img/rand?artid=146172342

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

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

一.ragflow的准备工作

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

拿到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<Map<String, List<Map<String, Object>>>> 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<String, Object> 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<Map<String, Object>> entity = new HttpEntity<>(payload, headers);

        // 发送请求
        RestTemplate restTemplate = new RestTemplate();
        try {
            ResponseEntity<Map> response = restTemplate.exchange(
                    ORIGINAL_API_URL,
                    HttpMethod.POST,
                    entity,
                    Map.class
            );

            if (response.getStatusCode() != HttpStatus.OK) {
                return ResponseEntity.status(response.getStatusCode()).body(null);
            }

            // 处理响应数据
            Map<String, Object> originalData = response.getBody();
            Map<String, Object> data = (Map<String, Object>) originalData.get("data");
            List<Map<String, Object>> chunks = (List<Map<String, Object>>) data.get("chunks");

            List<Map<String, Object>> records = new ArrayList<>();
            if (chunks != null) {
                for (Map<String, Object> chunk : chunks) {
                    Map<String, Object> 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<String, Object> metadata = new HashMap<>();
                    metadata.put("document_id", chunk.get("document_id"));
                    record.put("metadata", metadata);
                    records.add(record);
                }
            }

            Map<String, List<Map<String, Object>>> result = new HashMap<>();
            result.put("records", records);
            return ResponseEntity.ok(result);
        } catch (HttpClientErrorException e) {
            return ResponseEntity.status(e.getStatusCode()).body(null);
        }
    }
}