From 6958b08ba69211d3ce63126ceae73e34a19d40bb Mon Sep 17 00:00:00 2001 From: Kye Date: Sat, 29 Jul 2023 12:53:45 -0400 Subject: [PATCH] params Former-commit-id: 5f905bd2c94590e5ee4e5b6119fe3d0e413564cd --- swarms/workers/worker_node.py | 15 +++++++++++++-- 1 file changed, 13 insertions(+), 2 deletions(-) diff --git a/swarms/workers/worker_node.py b/swarms/workers/worker_node.py index 225197fc..dd037f24 100644 --- a/swarms/workers/worker_node.py +++ b/swarms/workers/worker_node.py @@ -118,8 +118,19 @@ class WorkerNodeInitializer: raise e class WorkerNode: - def __init__(self, openai_api_key, worker_name, worker_role, worker_prompt): - if not openai_api_key: + def __init__(self, + openai_api_key: str, + llm: Optional[Union[InMemoryDocstore, ChatOpenAI]] = None, + tools: Optional[List[Tool]] = None, + vectorstore: Optional[FAISS] = None, + 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, + search_kwargs: dict = {}, + verbose: Optional[bool] = False, + chat_history_file: str = "chat_history.txt"): + if not openai_api_key: logging.error("OpenAI API key is not provided") raise ValueError("openai_api_key cannot be None")