package com.supervision.demo;

import com.supervision.police.vo.dify.DifyChatReqVO;
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;

@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("d6c3e9fa-05a3-4d10-b482-d2797d7eee25", "1823953980884635650"));
//        System.out.println(difyApiUtil.createDocumentByFile("d6c3e9fa-05a3-4d10-b482-d2797d7eee25", "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");
        DifyChatReqVO difyChatReqVO = new DifyChatReqVO();
        difyChatReqVO.setUser("admin");
        difyChatReqVO.setQuery("裴金禄的户籍地址");
        System.out.println(difyApiUtil.chat(difyChatReqVO));
    }

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

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

    @Test
    public void createDocumentByTextTest() {
        String documentId = difyApiUtil.createDocumentByText("d5da0a24-a8e9-4ceb-8004-6da4847fbb04", "test-text", "坚定不移中国特色社会主义道路");
        System.out.println(documentId);
    }
}