Skip to main content

AwaEmbedding

This notebook explains how to use AwaEmbedding, which is included in awadb, to embedding texts in langchain.

# pip install awadb

import the library​

from langchain.embeddings import AwaEmbeddings

API Reference:

Embedding = AwaEmbeddings()

Set embedding model

Users can use Embedding.set_model() to specify the embedding model. \ The input of this function is a string which represents the model's name. \ The list of currently supported models can be obtained here \ \

The default model is all-mpnet-base-v2, it can be used without setting.

text = "our embedding test"

Embedding.set_model("all-mpnet-base-v2")
res_query = Embedding.embed_query("The test information")
res_document = Embedding.embed_documents(["test1", "another test"])