Spring Boot框架使用chromadb向量库

引入maven依赖



    io.github.amikos-tech
    chromadb-java-client
    0.1.7




    com.alibaba
    dashscope-sdk-java
    2.16.10

默认嵌入函数

下面是人家写好的例子,可以直接用,我复制过来的,文章最后是源码地址和文档地址
package tech.amikos;

import tech.amikos.chromadb.*;
import tech.amikos.chromadb.Collection;
import tech.amikos.chromadb.embeddings.DefaultEmbeddingFunction;

import java.util.*;

public class Main {
    public static void main(String[] args) {
        try {
            Client client = new Client(System.getenv("CHROMA_URL"));
            client.reset();
            EmbeddingFunction ef = new DefaultEmbeddingFunction();
            Collection collection = client.createCollection("test-collection", null, true, ef);
            List> metadata = new ArrayList<>();
            metadata.add(new HashMap() {
  {
                put("type", "scientist");
            }});
            metadata.add(new HashMap() {
  {
                put("type", "spy");
            }});
            collection.add(null, metadata, Arrays.asList("Hello, my name is John. I am a Data Scientist.", "Hello, my name is Bond. I am a Spy."), Arrays.asList("1", "2"));
            Collection.QueryResponse qr = collection.qu

你可能感兴趣的:(ChromaDB,Spring,Boot框架,spring,boot,数据库,人工智能)