Former-commit-id: e329e109bf6fbc0a0fa858707dbeac14a7e86baf
pull/160/head
Kye 2 years ago
parent c352b9f525
commit 822e863c72

@ -180,7 +180,7 @@ class Worker:
""" """
try: try:
embeddings_model = OpenAIEmbeddings(openai_api_key=self.openai_api_key) embeddings_model = OpenAIEmbeddings(openai_api_key=self.openai_api_key)
embedding_size = 4096 embedding_size = 1892
index = faiss.IndexFlatL2(embedding_size) index = faiss.IndexFlatL2(embedding_size)
self.vectorstore = FAISS(embeddings_model.embed_query, index, InMemoryDocstore({}), {}) self.vectorstore = FAISS(embeddings_model.embed_query, index, InMemoryDocstore({}), {})

Loading…
Cancel
Save