.

Author: smoreinis

src/agentex/lib/cli/handlers/deploy_handlers.py

@@ -470,10 +470,19 @@ def deploy_agent(
     helm_repository_name: str | None = None
     oci_registry: str | None = None
 
+    # Track OCI provider for provider-specific features
+    oci_provider: str | None = None
+
     if use_oci:
         oci_registry = agent_env_config.helm_oci_registry  # type: ignore[union-attr]
+        oci_provider = agent_env_config.helm_oci_provider  # type: ignore[union-attr]
         console.print(f"[blue]ℹ[/blue] Using OCI Helm registry: {oci_registry}")
-        login_to_gar_registry(oci_registry)  # type: ignore[arg-type]
+
+        # Only auto-authenticate for GAR provider
+        if oci_provider == "gar":
+            login_to_gar_registry(oci_registry)  # type: ignore[arg-type]
+        else:
+            console.print("[blue]ℹ[/blue] Skipping auto-authentication (no provider specified, assuming already authenticated)")
     else:
         if agent_env_config:
             helm_repository_name = agent_env_config.helm_repository_name
@@ -497,6 +506,9 @@ def deploy_agent(
         if not use_oci:
             console.print("[yellow]⚠[/yellow] --use-latest-chart only works with OCI registries, using default version")
             chart_version = DEFAULT_HELM_CHART_VERSION
+        elif oci_provider != "gar":
+            console.print("[yellow]⚠[/yellow] --use-latest-chart only works with GAR provider (helm_oci_provider: gar), using default version")
+            chart_version = DEFAULT_HELM_CHART_VERSION
         else:
             chart_version = get_latest_gar_chart_version(oci_registry)  # type: ignore[arg-type]
     elif agent_env_config and agent_env_config.helm_chart_version:

src/agentex/lib/sdk/config/environment_config.py

@@ -7,7 +7,7 @@
 
 from __future__ import annotations
 
-from typing import Any, Dict, override
+from typing import Any, Dict, Literal, override
 from pathlib import Path
 
 import yaml
@@ -74,6 +74,12 @@ class AgentEnvironmentConfig(BaseModel):
         description="OCI registry URL for Helm charts (e.g., 'us-west1-docker.pkg.dev/project/repo'). "
                     "When set, OCI mode is used instead of classic helm repo."
     )
+    helm_oci_provider: Literal["gar"] | None = Field(
+        default=None,
+        description="OCI registry provider for provider-specific features. "
+                    "Set to 'gar' for Google Artifact Registry to enable auto-authentication via gcloud "
+                    "and latest version fetching. When not set, assumes user has already authenticated."
+    )
     helm_chart_version: str | None = Field(
         default=None,
         description="Helm chart version to deploy. If not set, uses the default version from the CLI."