Former-commit-id: 000ae415e1
000ae415e1
@ -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")