|
"""
|
|
Manually register workers.
|
|
|
|
Usage:
|
|
python3 -m fastchat.serve.register_worker --controller http://localhost:21001 --worker-name http://localhost:21002
|
|
"""
|
|
|
|
import argparse
|
|
|
|
import requests
|
|
|
|
if __name__ == "__main__":
|
|
parser = argparse.ArgumentParser()
|
|
parser.add_argument("--controller-address", type=str)
|
|
parser.add_argument("--worker-name", type=str)
|
|
parser.add_argument("--check-heart-beat", action="store_true")
|
|
args = parser.parse_args()
|
|
|
|
url = args.controller_address + "/register_worker"
|
|
data = {
|
|
"worker_name": args.worker_name,
|
|
"check_heart_beat": args.check_heart_beat,
|
|
"worker_status": None,
|
|
}
|
|
r = requests.post(url, json=data)
|
|
assert r.status_code == 200
|
|
|