args in create worker node

pull/30/head
Kye 2 years ago
parent c974f18ffe
commit 7d2594e03f

@ -160,7 +160,7 @@ class WorkerNode:
except Exception as e:
logging.error(f"Failed to initialize tools: {e}")
def create_worker_node(self, llm_class=ChatOpenAI, worker_name, worker_role, human_in_the_loop, search_kwargs={}, verbose):
def create_worker_node(self, worker_name, worker_role, human_in_the_loop, llm_class=ChatOpenAI, search_kwargs={}, **kwargs):
if not llm_class:
logging.error("llm_class cannot be None.")
raise ValueError("llm_class cannot be None.")
@ -176,13 +176,13 @@ class WorkerNode:
ai_role=worker_role,
human_in_the_loop=human_in_the_loop,
search_kwargs=search_kwargs,
verbose=verbose
)
return worker_node
except Exception as e:
logging.error(f"Failed to create worker node: {e}")
raise
def worker_node(openai_api_key):
if not openai_api_key:
logging.error("OpenAI API key is not provided")

Loading…
Cancel
Save