Spring AI 使用本地 Ollama Embeddings

import com.salesforce.ai.lib.embeddings.Embedding;
import com.salesforce.ai.lib.embeddings.factory.EmbeddingFactory;
import com.salesforce.ai.lib.embeddings.factory.OllamaEmbeddingFactory;
import org.apache.commons.lang3.StringUtils;
 
// 假设已经有了EmbeddingFactory实例ollamaEmbeddingFactory
 
// 获取Ollama Embedding
public Embedding getOllamaEmbedding(String text) {
    if (StringUtils.isBlank(text)) {
        throw new IllegalArgumentException("输入文本不能为空");
    }
    return ollamaEmbeddingFactory.getEmbedding(text);
}
 
// 使用Ollama Embedding
public float[] getEmbeddingVector(String text) {
    Embedding embedding = getOllamaEmbedding(text);
    return embedding.getVector();
}

这个代码示例展示了如何在Java中使用OllamaEmbeddingFactory来获取文本的Ollama Embedding，并获取其向量表示。在实际应用中，你需要确保ollamaEmbeddingFactory已经被正确初始化。