parent
f5bed1dc2c
commit
648341b684
@ -0,0 +1,30 @@
|
|||||||
|
package xyz.wbsite.ai;
|
||||||
|
|
||||||
|
import dev.langchain4j.data.embedding.Embedding;
|
||||||
|
import dev.langchain4j.data.segment.TextSegment;
|
||||||
|
import dev.langchain4j.model.embedding.EmbeddingModel;
|
||||||
|
import dev.langchain4j.model.embedding.onnx.allminilml6v2.AllMiniLmL6V2EmbeddingModel;
|
||||||
|
import dev.langchain4j.store.embedding.EmbeddingStore;
|
||||||
|
import dev.langchain4j.store.embedding.qdrant.QdrantEmbeddingStore;
|
||||||
|
|
||||||
|
import static dev.langchain4j.internal.Utils.randomUUID;
|
||||||
|
|
||||||
|
/**
|
||||||
|
* 语义相似度比较
|
||||||
|
*/
|
||||||
|
public class Qdrant_Embedding_Example {
|
||||||
|
|
||||||
|
public static void main(String[] args) {
|
||||||
|
EmbeddingStore<TextSegment> embeddingStore =
|
||||||
|
QdrantEmbeddingStore.builder()
|
||||||
|
.host("127.0.0.1")
|
||||||
|
.port(6334)
|
||||||
|
.collectionName("langchain4j-" + randomUUID())
|
||||||
|
.build();
|
||||||
|
|
||||||
|
EmbeddingModel embeddingModel = new AllMiniLmL6V2EmbeddingModel();
|
||||||
|
TextSegment segment1 = TextSegment.from("I've been to France twice.");
|
||||||
|
Embedding embedding1 = embeddingModel.embed(segment1).content();
|
||||||
|
embeddingStore.add(embedding1, segment1);
|
||||||
|
}
|
||||||
|
}
|
Loading…
Reference in new issue