Former-commit-id: f1b9053c4e486babeae93ccbd5a243bb7f069e51
pull/160/head
Kye 2 years ago
parent fd47011dc7
commit 76ac2903bf

@ -53,7 +53,7 @@ class BossNodeInitializer:
""" """
try: try:
embeddings_model = OpenAIEmbeddings(openai_api_key=self.openai_api_key) embeddings_model = OpenAIEmbeddings(openai_api_key=self.openai_api_key)
embedding_size = self.embedding_size embedding_size = 8192
index = faiss.IndexFlatL2(embedding_size) index = faiss.IndexFlatL2(embedding_size)
return FAISS(embeddings_model.embed_query, index, InMemoryDocstore({}), {}) return FAISS(embeddings_model.embed_query, index, InMemoryDocstore({}), {})
except Exception as e: except Exception as e:

Loading…
Cancel
Save