langchain/docs/modules/models/text_embedding/examples/openai.ipynb

OpenAI

Let's load the OpenAI Embedding class.

In [1]:

from langchain.embeddings import OpenAIEmbeddings

In [2]:

embeddings = OpenAIEmbeddings()

In [3]:

text = "This is a test document."

In [4]:

query_result = embeddings.embed_query(text)

In [5]:

doc_result = embeddings.embed_documents([text])

Let's load the OpenAI Embedding class with first generation models (e.g. text-search-ada-doc-001/text-search-ada-query-001). Note: These are not recommended models - see here

In [ ]:

from langchain.embeddings.openai import OpenAIEmbeddings

In [ ]:

embeddings = OpenAIEmbeddings(model_name="ada")

In [ ]:

text = "This is a test document."

In [ ]:

query_result = embeddings.embed_query(text)

In [ ]:

doc_result = embeddings.embed_documents([text])

In [ ]: