Skip to main content
`Ctrl`+`K`

# Sentence Transformers Embeddings

Let’s generate embeddings using the SentenceTransformers integration. SentenceTransformers is a python package that can generate text and image embeddings, originating from Sentence-BERT

    !pip install sentence_transformers > /dev/null

    huggingface/tokenizers: The current process just got forked, after parallelism has already been used. Disabling parallelism to avoid deadlocks...
    To disable this warning, you can either:
    	- Avoid using `tokenizers` before the fork if possible
    	- Explicitly set the environment variable TOKENIZERS_PARALLELISM=(true | false)

    from langchain.embeddings import SentenceTransformerEmbeddings

    embeddings = SentenceTransformerEmbeddings(model="all-MiniLM-L6-v2")

    text = "This is a test document."

    query_result = embeddings.embed_query(text)

    doc_result = embeddings.embed_documents([text, "This is not a test document."])