GCP improvements

- Pin google-auth to 2.27.0 to fix conflicts in Colab notebooks
- Enable passing a user-provided access token to MistralGoogleCloud

Author: hrjn

packages/mistralai_gcp/poetry.lock

@@ -18,7 +18,7 @@ in-project = true
 [tool.poetry.dependencies]
 python = "^3.8"
 eval-type-backport = "^0.2.0"
-google-auth = "^2.31.0"
+google-auth = "2.27.0"
 httpx = "^0.27.0"
 jsonpath-python = "^1.0.6"
 pydantic = "~2.8.2"

packages/mistralai_gcp/pyproject.toml

@@ -32,6 +32,7 @@ def __init__(
         self,
         region: str = "europe-west4",
         project_id: Optional[str] = None,
+        access_token: Optional[str] = None,
         client: Optional[HttpClient] = None,
         async_client: Optional[AsyncHttpClient] = None,
         retry_config: Optional[Nullable[RetryConfig]] = None,
@@ -45,10 +46,11 @@ def __init__(
         :param async_client: The Async HTTP client to use for all asynchronous methods
         :param retry_config: The retry configuration to use for all supported methods
         """
-
+        
         credentials, loaded_project_id = google.auth.default(
             scopes=["https://www.googleapis.com/auth/cloud-platform"],
         )
+        credentials.refresh(google.auth.transport.requests.Request())
 
         if not isinstance(credentials, google.auth.credentials.Credentials):
             raise models.SDKError(
@@ -60,12 +62,14 @@ def __init__(
             raise models.SDKError("project_id must be provided")
 
         def auth_token() -> str:
-            if credentials.expired:
+            if access_token:
+                return access_token
+            else:
                 credentials.refresh(google.auth.transport.requests.Request())
-            token = credentials.token
-            if not token:
-                raise models.SDKError("Failed to get token from credentials")
-            return token
+                token = credentials.token
+                if not token:
+                    raise models.SDKError("Failed to get token from credentials")
+                return token
 
         if client is None:
             client = httpx.Client()

packages/mistralai_gcp/src/mistralai_gcp/sdk.py

@@ -21,7 +21,7 @@ jsonpath-python = "^1.0.6"
 pydantic = "~2.8.2"
 python-dateutil = "^2.9.0.post0"
 typing-inspect = "^0.9.0"
-google-auth = { version = "^2.31.0", optional = true }
+google-auth = { version = "2.27.0", optional = true }
 requests = { version = "^2.32.3", optional = true }
 
 [tool.poetry.group.dev.dependencies]