embedding size not defined

Former-commit-id: 7f9b8dd58f372dcb520d84b57a045ba0632c3248
pull/160/head
Kye 2 years ago
parent b9fed73fa1
commit 3847fa6905

@ -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