From c03a65c6dc7c3e606e7367890975560efdaca4c3 Mon Sep 17 00:00:00 2001 From: Adilzhan Ismailov <13088690+aismlv@users.noreply.github.com> Date: Thu, 20 Apr 2023 04:49:33 +0100 Subject: [PATCH] Fix from_embeddings method examples (#3174) Fix examples for `from_embeddings` method for annoy and faiss vectorstores --- langchain/vectorstores/annoy.py | 4 +++- langchain/vectorstores/faiss.py | 4 +++- 2 files changed, 6 insertions(+), 2 deletions(-) diff --git a/langchain/vectorstores/annoy.py b/langchain/vectorstores/annoy.py index 05d5a5d0..3a3ea156 100644 --- a/langchain/vectorstores/annoy.py +++ b/langchain/vectorstores/annoy.py @@ -367,7 +367,9 @@ class Annoy(VectorStore): from langchain import Annoy from langchain.embeddings import OpenAIEmbeddings embeddings = OpenAIEmbeddings() - db = Annoy.from_texts(texts, embeddings) + text_embeddings = embeddings.embed_documents(texts) + text_embedding_pairs = list(zip(texts, text_embeddings)) + db = Annoy.from_embeddings(text_embedding_pairs, embeddings) """ texts = [t[0] for t in text_embeddings] embeddings = [t[1] for t in text_embeddings] diff --git a/langchain/vectorstores/faiss.py b/langchain/vectorstores/faiss.py index 27f7a2ba..f391caed 100644 --- a/langchain/vectorstores/faiss.py +++ b/langchain/vectorstores/faiss.py @@ -396,7 +396,9 @@ class FAISS(VectorStore): from langchain import FAISS from langchain.embeddings import OpenAIEmbeddings embeddings = OpenAIEmbeddings() - faiss = FAISS.from_texts(texts, embeddings) + text_embeddings = embeddings.embed_documents(texts) + text_embedding_pairs = list(zip(texts, text_embeddings)) + faiss = FAISS.from_embeddings(text_embedding_pairs, embeddings) """ texts = [t[0] for t in text_embeddings] embeddings = [t[1] for t in text_embeddings]