Former-commit-id: c7341d409d3bdc8afe257236ad2b466380d08a21
@ -118,7 +118,7 @@ class WorkerNodeInitializer:
raise e
class WorkerNode:
def __init__(self, openai_api_key):
def __init__(self, openai_api_key, worker_name, worker_role):
if not openai_api_key:
logging.error("OpenAI API key is not provided")
raise ValueError("openai_api_key cannot be None")