diff --git a/__pycache__/vast_start.cpython-312.pyc b/__pycache__/vast_start.cpython-312.pyc new file mode 100644 index 0000000..6ec5d61 Binary files /dev/null and b/__pycache__/vast_start.cpython-312.pyc differ diff --git a/m2.py b/m2.py index da96e4d..8ab554f 100644 --- a/m2.py +++ b/m2.py @@ -29,11 +29,15 @@ if instance_id > 0 : print("Instance ready:", ip) + + pub_ip=data["public_ipaddr"] + pub_port=data['ports']['22/tcp'][0]['HostPort'] + vs.ssh_and_setup(ip, port, user, "") - - ip=data["public_ipaddr"] - port=data['ports']['22/tcp'][0]['HostPort'] - print(f"ssh root@{ip} -p {port}") + print(f"ssh root@{pub_ip} -p {pub_port}") ollama_port=data['ports']['11434/tcp'][0]['HostPort'] - print(f"ollama up at http://{ip}:{ollama_port}") + print(f"ollama up at http://{pub_ip}:{ollama_port} with mode {model}") + cmd2=["ssh","-t","-p",str(pub_port),f"{user}@{pub_ip}","ollama","run",model] + print(' '.join(cmd2)) + subprocess.run(cmd2) diff --git a/vast_start.py b/vast_start.py index a1f9603..760cac7 100644 --- a/vast_start.py +++ b/vast_start.py @@ -77,7 +77,18 @@ def get_instance(instance_id): data = r.json() return data["instances"] +def list_instances_json(): + r = requests.get( + f"{BASE}/instances/", + headers={"Authorization": f"Bearer {API_KEY}"} + ) + data = r.json() + return data["instances"] +def print_instances(): + instances=list_instances_json() + for instance in instances: + print(f"{instance['id']}") def wait_until_ready(instance_id): while True: data = get_instance(instance_id)