Firebase ML Changing service endpoint (#421)

Author: ifielker

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
 

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