mawei/StarPilot
1
0
Fork 0
mirror of https://github.com/firestar5683/StarPilot.git synced 2026-09-01 05:33:49 +08:00
Code Issues Packages Projects Releases Wiki Activity

tools: add retry mechanism for API requests (#1480)

Browse Source
This commit is contained in:
Jason Wen
2025-11-16 01:55:19 -05:00
committed by GitHub
parent 90cbb09482
commit cb03d08397
1 changed files with 4 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files
tools/lib/api.py
+4
View File
@@ -1,5 +1,6 @@
import os import os
import requests import requests
from requests.adapters import HTTPAdapter, Retry
API_HOST = os.getenv('API_HOST', 'https://api.commadotai.com') API_HOST = os.getenv('API_HOST', 'https://api.commadotai.com')
# TODO: this should be merged into common.api # TODO: this should be merged into common.api
@@ -11,6 +12,9 @@ class CommaApi:
if token: if token:
self.session.headers['Authorization'] = 'JWT ' + token self.session.headers['Authorization'] = 'JWT ' + token
retries = Retry(total=5, backoff_factor=1, status_forcelist=[500, 502, 503, 504])
self.session.mount('https://', HTTPAdapter(max_retries=retries))
def request(self, method, endpoint, **kwargs): def request(self, method, endpoint, **kwargs):
with self.session.request(method, API_HOST + '/' + endpoint, **kwargs) as resp: with self.session.request(method, API_HOST + '/' + endpoint, **kwargs) as resp:
resp_json = resp.json() resp_json = resp.json()