fu-hsi-service/src/test/java/com/supervision/demo/DifyTest.java

package com.supervision.demo;

import com.supervision.police.vo.dify.ChatReqVO;
import com.supervision.utils.DifyApiUtil;
import com.supervision.utils.Document;
import org.junit.jupiter.api.Test;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.boot.test.context.SpringBootTest;

import java.util.List;
import java.util.Map;

@SpringBootTest
public class DifyTest {
    @Autowired
    private DifyApiUtil difyApiUtil;

    @Test
    public void testCreateDataset() {
        System.out.println(difyApiUtil.createDataset("张三1", "张三1"));
    }

    @Test
    public void testDeleteDataset() {
        difyApiUtil.deleteDataset("65c3a191-2433-4f79-9fd2-5c4dfaf4264d");
    }

    @Test
    public void testCreateDocumentByFile() {
        System.out.println(difyApiUtil.createDocumentByFile("92e613e7-7369-46b1-ab12-b3d42dd5b646", "1823953980884635650"));
    }

    @Test
    public void testDeleteDocument() {
        difyApiUtil.deleteDocument("92e613e7-7369-46b1-ab12-b3d42dd5b646", "3f2a3719-1ee8-4cfd-b50c-d469954ee72e");
    }

    @Test
    public void testChat() {
        System.out.println("Test");
        ChatReqVO chatReqVO = new ChatReqVO();
        chatReqVO.setUser("admin");
        chatReqVO.setQuery("Who are you?");
        chatReqVO.setInputs(Map.of("dataset_id", "13c60b8c-341f-43ea-b3cc-5289a518abd9"));
        difyApiUtil.chat(chatReqVO);
    }

    @Test
    public void testQueryDocuments() {
        List<Document> documents = difyApiUtil.queryDocuments("d6c3e9fa-05a3-4d10-b482-d2797d7eee25");

        System.out.println(documents.size());
    }
}