embedding size not defined

pull/45/head
Kye 2 years ago
parent de98c6a03b
commit 53ea599a68

@ -31,7 +31,7 @@ class WorkerNodeInitializer:
openai_api_key: str, openai_api_key: str,
llm: Optional[Union[InMemoryDocstore, ChatOpenAI]] = None, llm: Optional[Union[InMemoryDocstore, ChatOpenAI]] = None,
tools: Optional[List[Tool]] = None, tools: Optional[List[Tool]] = None,
embedding_size: Optional[int] = 1926, embedding_size: Optional[int] = 8192,
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,
@ -45,7 +45,7 @@ class WorkerNodeInitializer:
self.worker_name = worker_name self.worker_name = worker_name
self.worker_role = worker_role self.worker_role = worker_role
self.embedding_size = embedding_size self.embedding_size = 8192
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
self.verbose = verbose self.verbose = verbose

Loading…
Cancel
Save