worker name init

Former-commit-id: 000ae415e1
pull/47/head
Kye 2 years ago
parent e64763ee67
commit 769c6baa16

@ -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