embedding size

Former-commit-id: 838a095df9
pull/47/head
Kye 1 year ago
parent 09a37ca559
commit 6361653ad9

@ -49,7 +49,7 @@ class VortexWorkerAgent:
self.worker_name = worker_name self.worker_name = worker_name
self.worker_role = worker_role self.worker_role = worker_role
self.embedding_size = 8192 self.embedding_size = embedding_size
self.human_in_the_loop = human_in_the_loop self.human_in_the_loop = human_in_the_loop
self.search_kwargs = search_kwargs self.search_kwargs = search_kwargs
@ -87,7 +87,8 @@ class VortexWorkerAgent:
try: try:
openai_api_key = self.openai_api_key or os.getenv("OPENAI_API_KEY") openai_api_key = self.openai_api_key or os.getenv("OPENAI_API_KEY")
embeddings_model = OpenAIEmbeddings(openai_api_key=openai_api_key) embeddings_model = OpenAIEmbeddings(openai_api_key=openai_api_key)
index = faiss.IndexFlatL2(embedding_size=self.embedding_size) embedding_size = 8192
index = faiss.IndexFlatL2(embedding_size=embedding_size)
return FAISS(embeddings_model, index, InMemoryDocstore({}), {}) return FAISS(embeddings_model, index, InMemoryDocstore({}), {})
except Exception as error: except Exception as error:
logging.error(f"Failed to initialize vector store: {error}") logging.error(f"Failed to initialize vector store: {error}")

Loading…
Cancel
Save