worker name init

Former-commit-id: c7341d409d3bdc8afe257236ad2b466380d08a21
pull/160/head
Kye 2 years ago
parent 764e6bf6d2
commit f5a6264dcb

@ -118,7 +118,7 @@ class WorkerNodeInitializer:
raise e raise e
class WorkerNode: class WorkerNode:
def __init__(self, openai_api_key): def __init__(self, openai_api_key, worker_name, worker_role):
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")
raise ValueError("openai_api_key cannot be None") raise ValueError("openai_api_key cannot be None")

Loading…
Cancel
Save