39 lines
1.5 KiB
Java
39 lines
1.5 KiB
Java
//import com.electromagnetic.industry.software.manage.Application;
|
|
//import jakarta.annotation.Resource;
|
|
//import org.junit.jupiter.api.Test;
|
|
//import org.springframework.ai.document.Document;
|
|
//import org.springframework.ai.document.DocumentReader;
|
|
//import org.springframework.ai.reader.TextReader;
|
|
//import org.springframework.ai.transformer.splitter.TokenTextSplitter;
|
|
//import org.springframework.ai.vectorstore.VectorStore;
|
|
//import org.springframework.boot.test.context.SpringBootTest;
|
|
//
|
|
//import java.io.File;
|
|
//import java.net.MalformedURLException;
|
|
//import java.nio.file.Path;
|
|
//import java.util.List;
|
|
//
|
|
//@SpringBootTest(classes = Application.class)
|
|
//public class Test1 {
|
|
//
|
|
// @Resource
|
|
// private VectorStore vectorStore;
|
|
//
|
|
// @Test
|
|
// public void testTree() throws MalformedURLException {
|
|
// String path = "D:/明朝那些事儿.txt";
|
|
// File file = new File(path);
|
|
// Path tempFile = file.toPath();
|
|
// DocumentReader documentReader = new TextReader(String.valueOf(tempFile.toUri().toURL()));
|
|
// List<Document> documents = new TokenTextSplitter().apply(documentReader.get());
|
|
// documents = documents.subList(0, 200);
|
|
// vectorStore.write(documents);
|
|
//
|
|
/// / Path tempFile = Paths.get(file.getAbsolutePath());
|
|
/// / DocumentReader documentReader = new TextReader(String.valueOf(tempFile.toUri().toURL()));
|
|
/// / List<Document> documents = documentReader.get();
|
|
/// / vectorStore.write(documents);
|
|
// }
|
|
//
|
|
//}
|