worker name init

pull/30/head
Kye 1 year ago
parent 0d047ac61b
commit 000ae415e1

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