download vast.ai's root certificate in order to make pyworker requests (#25)

This commit is contained in:
Nader Arbabian
2025-08-08 17:04:16 -07:00
committed by GitHub
parent 4016cf9a53
commit 8804e17201
5 changed files with 34 additions and 3 deletions
+3
View File
@@ -10,6 +10,7 @@ from collections import Counter
from dataclasses import dataclass, field, asdict
from urllib.parse import urljoin
from utils.endpoint_util import Endpoint
from utils.ssl import get_cert_file_path
import requests
from lib.data_types import AuthData, ApiPayload
@@ -120,9 +121,11 @@ class ClientState:
self.url = worker_address
url = urljoin(worker_address, self.worker_endpoint)
self.status = ClientStatus.Generating
response = requests.post(
url,
json=req_data,
verify=get_cert_file_path(),
)
if response.status_code != 200:
self.infer_error.append(