feat(letsgo): add Gemini and Azure support (#5706)

Add remote::gemini and remote::azure to the provider candidates probed
by `ogx letsgo`. Both require API credentials that are checked before
any network probe is attempted.

Extend _probe_endpoint with an optional api_key_header parameter to
support providers that use a custom auth header instead of
Authorization: Bearer. Use it for Anthropic (x-api-key) and Azure
(api-key), and drop the redundant x-api-key header from the default
Bearer path.

Signed-off-by: Sébastien Han <seb@redhat.com>
Co-authored-by: Sébastien Han <seb@redhat.com>
Co-authored-by: Claude Opus 4.6 (1M context) <noreply@anthropic.com>

Author: 3 people

src/ogx/cli/stack/lets_go.py

@@ -206,11 +206,20 @@ def _autodetect_providers() -> str:
     when the key environment variable is not set.
     """
     candidates = [
-        # provider_type, env_for_base_url, default_base_url, probe_path, requires_api_key, api_key_env, extra_headers
-        ("remote::ollama", "OLLAMA_URL", "http://localhost:11434/v1", "models", False, None, {}),
-        ("remote::vllm", "VLLM_URL", "http://localhost:8000/v1", "health", False, None, {}),
-        ("remote::llama-cpp-server", "LLAMA_CPP_SERVER_URL", "http://localhost:8080/v1", "models", False, None, {}),
-        ("remote::openai", "OPENAI_BASE_URL", "https://api.openai.com/v1", "models", True, "OPENAI_API_KEY", {}),
+        # provider_type, env_for_base_url, default_base_url, probe_path, requires_api_key, api_key_env, extra_headers, api_key_header
+        ("remote::ollama", "OLLAMA_URL", "http://localhost:11434/v1", "models", False, None, {}, None),
+        ("remote::vllm", "VLLM_URL", "http://localhost:8000/v1", "health", False, None, {}, None),
+        (
+            "remote::llama-cpp-server",
+            "LLAMA_CPP_SERVER_URL",
+            "http://localhost:8080/v1",
+            "models",
+            False,
+            None,
+            {},
+            None,
+        ),
+        ("remote::openai", "OPENAI_BASE_URL", "https://api.openai.com/v1", "models", True, "OPENAI_API_KEY", {}, None),
         (
             "remote::llama-openai-compat",
             "LLAMA_API_BASE_URL",
@@ -219,6 +228,7 @@ def _autodetect_providers() -> str:
             True,
             "LLAMA_API_KEY",
             {},
+            None,
         ),
         (
             "remote::anthropic",
@@ -228,12 +238,42 @@ def _autodetect_providers() -> str:
             True,
             "ANTHROPIC_API_KEY",
             {"anthropic-version": "2023-06-01"},
+            "x-api-key",
+        ),
+        (
+            "remote::gemini",
+            None,
+            "https://generativelanguage.googleapis.com/v1beta/openai",
+            "models",
+            True,
+            "GEMINI_API_KEY",
+            {},
+            None,
+        ),
+        (
+            "remote::azure",
+            "AZURE_API_BASE",
+            "",
+            "openai/models?api-version=2024-12-01-preview",
+            True,
+            "AZURE_API_KEY",
+            {},
+            "api-key",
         ),
     ]
 
     passed: list[str] = []
     cprint("Scanning for available providers...", color="cyan")
-    for provider_type, base_env, default_base, probe_path, requires_key, key_env, extra_headers in candidates:
+    for (
+        provider_type,
+        base_env,
+        default_base,
+        probe_path,
+        requires_key,
+        key_env,
+        extra_headers,
+        api_key_header,
+    ) in candidates:
         env_val: str | None = os.getenv(base_env) if base_env else None
         if env_val:
             base = env_val
@@ -242,7 +282,7 @@ def _autodetect_providers() -> str:
             base = default_base
             base_source = "default"
 
-        status = _probe_endpoint(base, probe_path, requires_key, key_env, extra_headers)
+        status = _probe_endpoint(base, probe_path, requires_key, key_env, extra_headers, api_key_header)
 
         # Build annotation parts
         parts = [f"{base}, {base_source}"]
@@ -290,6 +330,7 @@ def _probe_endpoint(
     requires_key: bool,
     key_env: str | None,
     extra_headers: dict[str, str] | None = None,
+    api_key_header: str | None = None,
 ) -> _ProbeStatus:
     """Perform a lightweight HTTP probe for a provider."""
     if not base_url:
@@ -302,8 +343,10 @@ def _probe_endpoint(
         if not key_env or not os.getenv(key_env):
             return _ProbeStatus.NO_KEY
         key: str = os.getenv(key_env, "")
-        headers["Authorization"] = f"Bearer {key}"
-        headers["x-api-key"] = key
+        if api_key_header:
+            headers[api_key_header] = key
+        else:
+            headers["Authorization"] = f"Bearer {key}"
 
     try:
         resp = cast(httpx.Response, httpx.get(url, headers=headers, timeout=2.0))

tests/unit/cli/test_stack_lets_go.py

@@ -196,7 +196,7 @@ def test_auth_headers_set_when_key_present(self, monkeypatch: pytest.MonkeyPatch
             _probe_endpoint("https://api.openai.com/v1", "models", True, "OPENAI_API_KEY")
         headers = mock_get.call_args.kwargs["headers"]
         assert headers["Authorization"] == "Bearer sk-secret"
-        assert headers["x-api-key"] == "sk-secret"
+        assert "x-api-key" not in headers
 
 
 class TestAutodetect:
@@ -230,14 +230,19 @@ def test_autodetect_all_ok(self, mock_probe: MagicMock):
         assert "inference=remote::anthropic" in parts
         assert "files=inline::localfs" in parts
         assert "responses=inline::builtin" in parts
-        assert len(parts) == 12  # 6 probed + 6 inline
+        assert len(parts) == 14  # 8 probed + 6 inline
 
     @patch("ogx.cli.stack.lets_go._probe_endpoint")
     def test_autodetect_only_ollama(self, mock_probe: MagicMock):
         from ogx.cli.stack.lets_go import _autodetect_providers
 
         def side_effect(
-            base_url: str, probe_path: str, requires_key: bool, key_env: object, extra_headers: object = None
+            base_url: str,
+            probe_path: str,
+            requires_key: bool,
+            key_env: object,
+            extra_headers: object = None,
+            api_key_header: object = None,
         ) -> _ProbeStatus:
             if "11434" in base_url:
                 return _ProbeStatus.OK
@@ -258,7 +263,12 @@ def test_autodetect_uses_env_var_base_url(self, mock_probe: MagicMock, monkeypat
         captured: list[str] = []
 
         def side_effect(
-            base_url: str, probe_path: str, requires_key: bool, key_env: object, extra_headers: object = None
+            base_url: str,
+            probe_path: str,
+            requires_key: bool,
+            key_env: object,
+            extra_headers: object = None,
+            api_key_header: object = None,
         ) -> _ProbeStatus:
             captured.append(base_url)
             return _ProbeStatus.UNREACHABLE
@@ -276,7 +286,12 @@ def test_autodetect_result_order_matches_candidate_order(
         monkeypatch.setenv("OPENAI_API_KEY", "sk-test")
 
         def side_effect(
-            base_url: str, probe_path: str, requires_key: bool, key_env: object, extra_headers: object = None
+            base_url: str,
+            probe_path: str,
+            requires_key: bool,
+            key_env: object,
+            extra_headers: object = None,
+            api_key_header: object = None,
         ) -> _ProbeStatus:
             if "11434" in base_url or "openai.com" in base_url:
                 return _ProbeStatus.OK