Former-commit-id: e20903522e
pull/47/head
Kye 2 years ago
parent 788f027634
commit eecece615c

@ -34,7 +34,7 @@ class WorkerNodeInitializer:
llm: Optional[Union[InMemoryDocstore, ChatOpenAI]] = None,
tools: Optional[List[Tool]] = None,
# vectorstore: Optional[FAISS] = None,
embedding_size: Optional[int] = 1926,
# embedding_size: Optional[int] = 1926,
worker_name: Optional[str] = "Swarm Worker AI Assistant",
worker_role: Optional[str] = "Assistant",
human_in_the_loop: Optional[bool] = False,
@ -92,7 +92,7 @@ class WorkerNodeInitializer:
def initialize_vectorstore(self):
try:
embeddings_model = OpenAIEmbeddings(openai_api_key=self.openai_api_key)
embedding_size = self.embedding_size
embedding_size = 8192
index = faiss.IndexFlatL2(embedding_size=embedding_size)
return FAISS(embeddings_model.embed_query, index, InMemoryDocstore({}), {})
@ -123,7 +123,7 @@ class WorkerNode:
llm: Optional[Union[InMemoryDocstore, ChatOpenAI]] = None,
tools: Optional[List[Tool]] = None,
# vectorstore: Optional[FAISS] = None,
embedding_size: Optional[int] = 4026,
# embedding_size: Optional[int] = 4026,
worker_name: Optional[str] = "Swarm Worker AI Assistant",
worker_role: Optional[str] = "Assistant",
human_in_the_loop: Optional[bool] = False,

Loading…
Cancel
Save