diff --git a/firebase_admin/ml.py b/firebase_admin/ml.py index d6c14c7ac..06429b5a1 100644 --- a/firebase_admin/ml.py +++ b/firebase_admin/ml.py @@ -771,8 +771,8 @@ def _validate_page_token(page_token): class _MLService: """Firebase ML service.""" - PROJECT_URL = 'https://mlkit.googleapis.com/v1beta1/projects/{0}/' - OPERATION_URL = 'https://mlkit.googleapis.com/v1beta1/' + PROJECT_URL = 'https://firebaseml.googleapis.com/v1beta2/projects/{0}/' + OPERATION_URL = 'https://firebaseml.googleapis.com/v1beta2/' POLL_EXPONENTIAL_BACKOFF_FACTOR = 1.5 POLL_BASE_WAIT_TIME_SECONDS = 3 diff --git a/tests/test_ml.py b/tests/test_ml.py index 439aeaac3..8813792e6 100644 --- a/tests/test_ml.py +++ b/tests/test_ml.py @@ -24,8 +24,9 @@ from tests import testutils -BASE_URL = 'https://mlkit.googleapis.com/v1beta1/' +BASE_URL = 'https://firebaseml.googleapis.com/v1beta2/' PROJECT_ID = 'my-project-1' + PAGE_TOKEN = 'pageToken' NEXT_PAGE_TOKEN = 'nextPageToken' @@ -306,7 +307,7 @@ def instrument_ml_service(status=200, payload=None, operations=False, app=None): app = firebase_admin.get_app() ml_service = ml._get_ml_service(app) recorder = [] - session_url = 'https://mlkit.googleapis.com/v1beta1/' + session_url = 'https://firebaseml.googleapis.com/v1beta2/' if isinstance(status, list): adapter = testutils.MockMultiRequestAdapter