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

Loading…
Cancel
Save