Skip to content
Open
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
16 changes: 11 additions & 5 deletions vast.py
Original file line number Diff line number Diff line change
Expand Up @@ -67,8 +67,8 @@


#server_url_default = "https://vast.ai"
server_url_default = os.getenv("VAST_URL") or "https://console.vast.ai"
#server_url_default = "http://localhost:5002"
server_url_default = os.getenv("VAST_URL") or "https://console.vast.ai"
#server_url_default = "host.docker.internal"
#server_url_default = "http://localhost:5002"
#server_url_default = "https://vast.ai/api/v0"
Expand Down Expand Up @@ -727,10 +727,11 @@ def deindent(message: str) -> str:
cluster_fields = (
("id", "ID", "{}", None, True),
("subnet", "Subnet", "{}", None, True),
("status", "Status", "{}", None, True),
("node_count", "Nodes", "{}", None, True),
("manager_id", "Manager ID", "{}", None, True),
("manager_ip", "Manager IP", "{}", None, True),
("machine_ids", "Machine ID's", "{}", None, True)
("machines", "Machines", "{}", None, True),
)

overlay_fields = (
Expand Down Expand Up @@ -5161,17 +5162,22 @@ def show__clusters(args: argparse.Namespace):

rows = []
for cluster_id, cluster_data in response_data['clusters'].items():
machine_ids = [ node["machine_id"] for node in cluster_data["nodes"]]
machines = {member['machine_id']: member['actual_status'] for member in cluster_data["nodes"]}

for node in cluster_data['nodes']:
if node["is_cluster_manager"]:
manager_node = node
break

manager_node = next(node for node in cluster_data['nodes'] if node['is_cluster_manager'])

row_data = {
'id': cluster_id,
'subnet': cluster_data['subnet'],
'node_count': len(cluster_data['nodes']),
'machine_ids': str(machine_ids),
'machines': machines,
'manager_id': str(manager_node['machine_id']),
'manager_ip': manager_node['local_ip'],
'status': cluster_data['actual_status'],
}

rows.append(row_data)
Expand Down