args in create worker node

Former-commit-id: 7d2594e03f
pull/47/head
Kye 2 years ago
parent 5090c8c7d0
commit ea016cb0ca

@ -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.")
@ -175,14 +175,14 @@ class WorkerNode:
ai_name=worker_name,
ai_role=worker_role,
human_in_the_loop=human_in_the_loop,
search_kwargs=search_kwargs,
verbose=verbose
search_kwargs=search_kwargs,
)
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