lmux-google 0.7.0__tar.gz

lmux_google-0.7.0/PKG-INFO

@@ -0,0 +1,141 @@
+Metadata-Version: 2.4
+Name: lmux-google
+Version: 0.7.0
+Summary: Google (Gemini) provider for lmux
+Keywords: llm,ai,google,gemini,vertex,language-model
+Author: Connor Luebbehusen
+Author-email: Connor Luebbehusen <connor@luebbehusen.dev>
+License-Expression: MIT
+Classifier: Development Status :: 4 - Beta
+Classifier: Programming Language :: Python :: 3
+Classifier: Topic :: Scientific/Engineering :: Artificial Intelligence
+Classifier: Typing :: Typed
+Requires-Dist: lmux~=0.7
+Requires-Dist: google-genai~=1.0
+Requires-Python: >=3.13
+Project-URL: Homepage, https://github.com/cluebbehusen/lmux
+Project-URL: Source, https://github.com/cluebbehusen/lmux/tree/main/packages/lmux-google
+Project-URL: Issues, https://github.com/cluebbehusen/lmux/issues
+Description-Content-Type: text/markdown
+
+# lmux-google
+
+Google (Gemini) provider for [lmux](https://github.com/cluebbehusen/lmux). Wraps the [google-genai](https://pypi.org/project/google-genai/) SDK, which serves Google's models through either backend:
+
+- **Vertex AI** (default) — authenticated with Google Cloud credentials
+- **Gemini Developer API** (AI Studio) — authenticated with an API key (`vertexai=False`)
+
+Supports chat completions, streaming, and embeddings for Google-published models: Gemini and the Gemini/text embedding models.
+
+Part of the [lmux](https://github.com/cluebbehusen/lmux) ecosystem: standardized interface, cost tracking on every response, and registry-based routing across providers.
+
+## Auth
+
+Three authentication methods:
+
+### Application Default Credentials (default)
+
+Uses `google.auth.default()`, which works with `GOOGLE_APPLICATION_CREDENTIALS`, `gcloud` CLI, or instance metadata.
+
+```python
+from lmux_google import GoogleProvider
+
+provider = GoogleProvider(project="my-project", location="us-central1")
+```
+
+### Service Account
+
+```python
+from lmux_google import GoogleServiceAccountAuthProvider
+
+provider = GoogleProvider(
+    project="my-project",
+    location="us-central1",
+    auth=GoogleServiceAccountAuthProvider(service_account_file="/path/to/key.json"),
+)
+```
+
+### API Key
+
+Set `GOOGLE_API_KEY` in your environment:
+
+```python
+from lmux_google import GoogleAPIKeyAuthProvider
+
+provider = GoogleProvider(auth=GoogleAPIKeyAuthProvider(), vertexai=False)
+```
+
+## Usage
+
+### Chat
+
+```python
+from lmux import UserMessage
+
+response = provider.chat("gemini-2.5-pro", [UserMessage(content="Hello")])
+print(response.content)
+print(response.cost)
+```
+
+### Streaming
+
+```python
+for chunk in provider.chat_stream("gemini-2.5-pro", [UserMessage(content="Hello")]):
+    if chunk.delta:
+        print(chunk.delta, end="")
+```
+
+### Embeddings
+
+```python
+response = provider.embed("text-embedding-005", "Hello")
+print(response.embeddings)
+```
+
+### Async
+
+All methods have async variants: `achat`, `achat_stream`, `aembed`.
+
+### Registry
+
+Use with the lmux registry to route across multiple providers:
+
+```python
+from lmux import Registry
+
+registry = Registry()
+registry.register("google", provider)
+response = registry.chat("google/gemini-2.5-pro", messages)
+```
+
+## Provider Params
+
+```python
+from lmux_google import GoogleParams
+
+response = provider.chat(
+    "gemini-2.5-pro",
+    messages,
+    provider_params=GoogleParams(thinking_config={"thinking_budget": 1024}),
+)
+```
+
+| Parameter           | Type                  | Description                      |
+| ------------------- | --------------------- | -------------------------------- |
+| `safety_settings`   | `list[SafetySetting]` | Content safety thresholds        |
+| `presence_penalty`  | `float`               | Presence penalty                 |
+| `frequency_penalty` | `float`               | Frequency penalty                |
+| `seed`              | `int`                 | Deterministic sampling seed      |
+| `labels`            | `dict[str, str]`      | Request labels                   |
+| `thinking_config`   | `dict`                | Thinking/reasoning configuration |
+
+## Constructor Options
+
+```python
+GoogleProvider(
+    auth=...,       # AuthProvider, default: GoogleADCAuthProvider()
+    project=...,    # GCP project ID
+    location=...,   # GCP region
+    vertexai=...,   # Use Vertex AI (default: True) vs. AI Studio
+)
+```

lmux_google-0.7.0/README.md

@@ -0,0 +1,121 @@
+# lmux-google
+
+Google (Gemini) provider for [lmux](https://github.com/cluebbehusen/lmux). Wraps the [google-genai](https://pypi.org/project/google-genai/) SDK, which serves Google's models through either backend:
+
+- **Vertex AI** (default) — authenticated with Google Cloud credentials
+- **Gemini Developer API** (AI Studio) — authenticated with an API key (`vertexai=False`)
+
+Supports chat completions, streaming, and embeddings for Google-published models: Gemini and the Gemini/text embedding models.
+
+Part of the [lmux](https://github.com/cluebbehusen/lmux) ecosystem: standardized interface, cost tracking on every response, and registry-based routing across providers.
+
+## Auth
+
+Three authentication methods:
+
+### Application Default Credentials (default)
+
+Uses `google.auth.default()`, which works with `GOOGLE_APPLICATION_CREDENTIALS`, `gcloud` CLI, or instance metadata.
+
+```python
+from lmux_google import GoogleProvider
+
+provider = GoogleProvider(project="my-project", location="us-central1")
+```
+
+### Service Account
+
+```python
+from lmux_google import GoogleServiceAccountAuthProvider
+
+provider = GoogleProvider(
+    project="my-project",
+    location="us-central1",
+    auth=GoogleServiceAccountAuthProvider(service_account_file="/path/to/key.json"),
+)
+```
+
+### API Key
+
+Set `GOOGLE_API_KEY` in your environment:
+
+```python
+from lmux_google import GoogleAPIKeyAuthProvider
+
+provider = GoogleProvider(auth=GoogleAPIKeyAuthProvider(), vertexai=False)
+```
+
+## Usage
+
+### Chat
+
+```python
+from lmux import UserMessage
+
+response = provider.chat("gemini-2.5-pro", [UserMessage(content="Hello")])
+print(response.content)
+print(response.cost)
+```
+
+### Streaming
+
+```python
+for chunk in provider.chat_stream("gemini-2.5-pro", [UserMessage(content="Hello")]):
+    if chunk.delta:
+        print(chunk.delta, end="")
+```
+
+### Embeddings
+
+```python
+response = provider.embed("text-embedding-005", "Hello")
+print(response.embeddings)
+```
+
+### Async
+
+All methods have async variants: `achat`, `achat_stream`, `aembed`.
+
+### Registry
+
+Use with the lmux registry to route across multiple providers:
+
+```python
+from lmux import Registry
+
+registry = Registry()
+registry.register("google", provider)
+response = registry.chat("google/gemini-2.5-pro", messages)
+```
+
+## Provider Params
+
+```python
+from lmux_google import GoogleParams
+
+response = provider.chat(
+    "gemini-2.5-pro",
+    messages,
+    provider_params=GoogleParams(thinking_config={"thinking_budget": 1024}),
+)
+```
+
+| Parameter           | Type                  | Description                      |
+| ------------------- | --------------------- | -------------------------------- |
+| `safety_settings`   | `list[SafetySetting]` | Content safety thresholds        |
+| `presence_penalty`  | `float`               | Presence penalty                 |
+| `frequency_penalty` | `float`               | Frequency penalty                |
+| `seed`              | `int`                 | Deterministic sampling seed      |
+| `labels`            | `dict[str, str]`      | Request labels                   |
+| `thinking_config`   | `dict`                | Thinking/reasoning configuration |
+
+## Constructor Options
+
+```python
+GoogleProvider(
+    auth=...,       # AuthProvider, default: GoogleADCAuthProvider()
+    project=...,    # GCP project ID
+    location=...,   # GCP region
+    vertexai=...,   # Use Vertex AI (default: True) vs. AI Studio
+)
+```

lmux_google-0.7.0/pyproject.toml

@@ -0,0 +1,31 @@
+[project]
+name = "lmux-google"
+version = "0.7.0"
+description = "Google (Gemini) provider for lmux"
+requires-python = ">=3.13"
+license = "MIT"
+readme = "README.md"
+authors = [{ name = "Connor Luebbehusen", email = "connor@luebbehusen.dev" }]
+keywords = ["llm", "ai", "google", "gemini", "vertex", "language-model"]
+classifiers = [
+  "Development Status :: 4 - Beta",
+  "Programming Language :: Python :: 3",
+  "Topic :: Scientific/Engineering :: Artificial Intelligence",
+  "Typing :: Typed",
+]
+dependencies = [
+  "lmux~=0.7",
+  "google-genai~=1.0",
+]
+
+[project.urls]
+Homepage = "https://github.com/cluebbehusen/lmux"
+Source = "https://github.com/cluebbehusen/lmux/tree/main/packages/lmux-google"
+Issues = "https://github.com/cluebbehusen/lmux/issues"
+
+[tool.uv.sources]
+lmux = { workspace = true }
+
+[build-system]
+requires = ["uv_build>=0.11.0,<0.12.0"]
+build-backend = "uv_build"

lmux_google-0.7.0/src/lmux_google/__init__.py

@@ -0,0 +1,37 @@
+"""lmux-google — Google (Gemini) provider for lmux."""
+
+from lmux_google.auth import (
+    GoogleADCAuthProvider,
+    GoogleAPIKeyAuthProvider,
+    GoogleServiceAccountAuthProvider,
+)
+from lmux_google.cost import calculate_google_cost
+from lmux_google.params import (
+    DynamicRetrievalConfig,
+    GoogleParams,
+    GoogleSearchConfig,
+    GoogleSearchRetrievalConfig,
+    GoogleSearchTypes,
+    SafetySetting,
+)
+from lmux_google.provider import GoogleProvider
+
+__all__ = [
+    "DynamicRetrievalConfig",
+    "GoogleADCAuthProvider",
+    "GoogleAPIKeyAuthProvider",
+    "GoogleParams",
+    "GoogleProvider",
+    "GoogleSearchConfig",
+    "GoogleSearchRetrievalConfig",
+    "GoogleSearchTypes",
+    "GoogleServiceAccountAuthProvider",
+    "SafetySetting",
+    "calculate_google_cost",
+    "preload",
+]
+
+
+def preload() -> None:
+    """Eagerly import the google-genai SDK."""
+    import google.genai  # noqa: PLC0415, F401  # pyright: ignore[reportUnusedImport]

lmux_google-0.7.0/src/lmux_google/_exceptions.py

@@ -0,0 +1,66 @@
+"""Map google-genai SDK exceptions to lmux exception hierarchy."""
+
+from lmux.exceptions import (
+    AuthenticationError,
+    InvalidRequestError,
+    LmuxError,
+    NotFoundError,
+    ProviderError,
+    RateLimitError,
+    TimeoutError,  # noqa: A004
+)
+
+PROVIDER = "google"
+
+_STATUS_CODE_MAP: dict[int, type[LmuxError]] = {
+    400: InvalidRequestError,
+    401: AuthenticationError,
+    403: AuthenticationError,
+    404: NotFoundError,
+    408: TimeoutError,
+    429: RateLimitError,
+}
+
+
+def map_google_error(error: Exception) -> LmuxError:
+    """Convert a google-genai SDK exception to the corresponding lmux exception."""
+    from google.genai import errors as genai_errors  # noqa: PLC0415
+
+    if isinstance(error, genai_errors.ClientError):
+        return _map_client_error(error)
+
+    if isinstance(error, genai_errors.ServerError):
+        return ProviderError(str(error), provider=PROVIDER, status_code=error.code)
+
+    if isinstance(error, genai_errors.APIError):
+        return ProviderError(str(error), provider=PROVIDER, status_code=error.code)
+
+    # google.auth errors
+    auth_error = _check_auth_error(error)
+    if auth_error is not None:
+        return auth_error
+
+    return ProviderError(str(error), provider=PROVIDER)
+
+
+def _map_client_error(error: Exception) -> LmuxError:
+    """Map a google-genai ClientError based on its HTTP status code."""
+    code: int | None = getattr(error, "code", None)
+
+    exc_cls = _STATUS_CODE_MAP.get(code) if code is not None else None
+    if exc_cls is not None:
+        if exc_cls is RateLimitError:
+            return RateLimitError(str(error), provider=PROVIDER, status_code=code)
+        return exc_cls(str(error), provider=PROVIDER, status_code=code)
+
+    return ProviderError(str(error), provider=PROVIDER, status_code=code)
+
+
+def _check_auth_error(error: Exception) -> AuthenticationError | None:
+    """Check if the error is a google.auth exception and map it."""
+    import google.auth.exceptions  # noqa: PLC0415
+
+    if isinstance(error, google.auth.exceptions.DefaultCredentialsError | google.auth.exceptions.RefreshError):
+        return AuthenticationError(str(error), provider=PROVIDER)
+
+    return None

lmux_google-0.7.0/src/lmux_google/_lazy.py

@@ -0,0 +1,34 @@
+"""Lazy google-genai SDK loading internals.
+
+Client creation is isolated here so tests can easily mock it
+without patching sys.modules or using TYPE_CHECKING tricks.
+"""
+
+from typing import TYPE_CHECKING, Any
+
+if TYPE_CHECKING:
+    from google.auth.credentials import Credentials
+    from google.genai import Client
+
+
+def create_client(
+    *,
+    vertexai: bool = True,
+    project: str | None = None,
+    location: str | None = None,
+    credentials: "Credentials | None" = None,
+    api_key: str | None = None,
+) -> "Client":
+    """Create a google-genai Client."""
+    from google import genai  # noqa: PLC0415
+
+    kwargs: dict[str, Any] = {"vertexai": vertexai}
+    if project is not None:
+        kwargs["project"] = project
+    if location is not None:
+        kwargs["location"] = location
+    if credentials is not None:
+        kwargs["credentials"] = credentials
+    if api_key is not None:
+        kwargs["api_key"] = api_key
+    return genai.Client(**kwargs)