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