handle passed-in key

Author: njbrake

src/any_llm/providers/google/google.py

@@ -203,10 +203,11 @@ def __init__(self, config: ApiConfig) -> None:
             self.client = genai.Client(vertexai=True, project=self.project_id, location=self.location)
         else:
             # Gemini Developer API configuration
-            api_key = os.getenv("GEMINI_API_KEY") or os.getenv("GOOGLE_API_KEY")
+            # Use api_key from config if provided, otherwise fall back to environment variables
+            api_key = getattr(config, "api_key", None) or os.getenv("GEMINI_API_KEY") or os.getenv("GOOGLE_API_KEY")
 
             if not api_key:
-                msg = "GEMINI_API_KEY or GOOGLE_API_KEY environment variable is required for Gemini Developer API"
+                msg = "API key is required for Gemini Developer API. Provide it via ApiConfig or set GEMINI_API_KEY/GOOGLE_API_KEY environment variable"
                 raise ValueError(msg)
 
             # Initialize client for Gemini Developer API

tests/integration/test_completion.py

@@ -29,7 +29,8 @@ def test_providers(provider: str) -> None:
         if "API key provided" in str(e):
             pytest.skip(f"{provider} API key not provided, skipping")
         raise e
-    except httpx.HTTPStatusError as e:
-        if e.response.status_code == 404 and provider == "ollama":
+    except (httpx.HTTPStatusError, httpx.ConnectError):
+        if provider == "ollama":
             pytest.skip("Ollama is not set up, skipping")
+        raise
     assert result.choices[0].message.content is not None