google-adk-extras 0.3.0__tar.gz → 0.3.1__tar.gz

{google_adk_extras-0.3.0/src/google_adk_extras.egg-info → google_adk_extras-0.3.1}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: google-adk-extras
-Version: 0.3.0
+Version: 0.3.1
 Summary: Production-ready services and FastAPI wiring for Google ADK
 Home-page: https://github.com/DeadMeme5441/google-adk-extras
 Author: DeadMeme5441

{google_adk_extras-0.3.0 → google_adk_extras-0.3.1}/mkdocs.yml

@@ -14,6 +14,7 @@ nav:
   - Getting Started: getting-started.md
   - Quickstarts: quickstarts.md
   - FastAPI Integration: fastapi.md
+  - Auth (Optional): auth.md
   - Streaming: streaming.md
   - Services:
       - Durable Services: services.md
@@ -46,10 +47,10 @@ plugins:
         - quickstarts.md: A few end-to-end snippets to copy/paste
         FastAPI & A2A:
         - fastapi.md: Build a FastAPI app with AdkBuilder and enable A2A
+        Auth:
+        - auth.md: Optional inbound API authentication
         Durable Services:
         - services.md: Sessions, artifacts, memory backends and trade-offs
-        Credentials:
-        # credentials docs removed; use ADK documentation for outbound creds
         Programmatic Agents:
         - agent-loading.md: Custom loaders and in-memory registrations
         Configuration:

{google_adk_extras-0.3.0 → google_adk_extras-0.3.1}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "google-adk-extras"
-version = "0.3.0"
+version = "0.3.1"
 description = "Production-ready services and FastAPI wiring for Google ADK"
 readme = "README.md"
 requires-python = ">=3.10,<3.13"

{google_adk_extras-0.3.0 → google_adk_extras-0.3.1}/src/google_adk_extras/__init__.py

@@ -28,4 +28,4 @@ __all__ = [
     "CustomAgentLoader",
 ]
 
-__version__ = "0.3.0"
+__version__ = "0.3.1"

{google_adk_extras-0.3.0 → google_adk_extras-0.3.1}/src/google_adk_extras/adk_builder.py

@@ -683,6 +683,7 @@ class AdkBuilder:
             eval_storage_uri=self._eval_storage_uri,
             allow_origins=self._allow_origins,
             web=self._web_ui,
+            # Expose future override via builder when needed
             a2a=self._a2a,
             programmatic_a2a=self._a2a_expose_programmatic,
             programmatic_a2a_mount_base=self._a2a_programmatic_mount_base,

{google_adk_extras-0.3.0 → google_adk_extras-0.3.1}/src/google_adk_extras/auth/attach.py

@@ -9,7 +9,7 @@ from starlette.middleware.base import BaseHTTPMiddleware
 
 from .config import AuthConfig, JwtIssuerConfig, JwtValidatorConfig
 from .jwt_utils import decode_jwt, encode_jwt, now_ts
-from .sql_store import AuthStore
+from typing import Any
 
 
 def attach_auth(app: FastAPI, cfg: Optional[AuthConfig]) -> None:
@@ -25,9 +25,14 @@ def attach_auth(app: FastAPI, cfg: Optional[AuthConfig]) -> None:
     issuer_cfg = cfg.jwt_issuer
     api_keys = set(cfg.api_keys or [])
     basic_users = cfg.basic_users or {}
-    auth_store: Optional[AuthStore] = None
+    auth_store: Optional[Any] = None
     if issuer_cfg and issuer_cfg.database_url:
-        auth_store = AuthStore(issuer_cfg.database_url)
+        try:
+            from .sql_store import AuthStore  # type: ignore
+            auth_store = AuthStore(issuer_cfg.database_url)
+        except Exception:
+            # SQL store not available; API key issuance and password grants will be unavailable.
+            auth_store = None
 
     # Security helpers
     api_key_header = APIKeyHeader(name="X-API-Key", auto_error=False)

{google_adk_extras-0.3.0 → google_adk_extras-0.3.1}/src/google_adk_extras/auth/jwt_utils.py

@@ -5,8 +5,16 @@ import json
 import time
 from typing import Any, Dict, Optional
 
-import jwt
-from jwt import PyJWKClient
+# Lazy import PyJWT to keep this optional when auth is not used.
+def _import_pyjwt():
+    try:
+        import jwt  # type: ignore
+        from jwt import PyJWKClient  # type: ignore
+        return jwt, PyJWKClient
+    except Exception as e:
+        raise ImportError(
+            "PyJWT is required for JWT encode/decode. Install with: pip install PyJWT"
+        ) from e
 
 
 def _b64url(data: bytes) -> str:
@@ -14,11 +22,13 @@ def _b64url(data: bytes) -> str:
 
 
 def encode_jwt(payload: Dict[str, Any], *, algorithm: str, key: str, headers: Optional[Dict[str, Any]] = None) -> str:
+    jwt, _PyJWKClient = _import_pyjwt()
     return jwt.encode(payload, key, algorithm=algorithm, headers=headers)
 
 
 def decode_jwt(token: str, *, issuer: Optional[str] = None, audience: Optional[str] = None,
                jwks_url: Optional[str] = None, hs256_secret: Optional[str] = None) -> Dict[str, Any]:
+    jwt, PyJWKClient = _import_pyjwt()
     options = {"verify_signature": True, "verify_exp": True, "verify_nbf": True}
     if jwks_url:
         jwk_client = PyJWKClient(jwks_url)
@@ -33,4 +43,3 @@ def decode_jwt(token: str, *, issuer: Optional[str] = None, audience: Optional[s
 def now_ts() -> int:
     return int(time.time())
 
-

{google_adk_extras-0.3.0 → google_adk_extras-0.3.1}/src/google_adk_extras/enhanced_fastapi.py

@@ -10,7 +10,7 @@ import logging
 import os
 from pathlib import Path
 import shutil
-from typing import Any, Mapping, Optional, List, Callable, Dict
+from typing import Any, Mapping, Optional, List, Callable, Dict, Union
 
 import click
 from fastapi import FastAPI
@@ -57,6 +57,7 @@ def get_enhanced_fast_api_app(
     eval_storage_uri: Optional[str] = None,
     allow_origins: Optional[List[str]] = None,
     web: bool = True,
+    web_assets_dir: Optional[Union[str, Path]] = None,
     a2a: bool = False,
     programmatic_a2a: bool = False,
     programmatic_a2a_mount_base: str = "/a2a",
@@ -305,20 +306,53 @@ def get_enhanced_fast_api_app(
             tear_down_observer=tear_down_observer,
         )
 
-    if web:
+    def _auto_find_web_assets() -> Optional[Path]:
         try:
-            # Try to find ADK's web assets
-            from google.adk.cli.fast_api import BASE_DIR
-            ANGULAR_DIST_PATH = BASE_DIR / "browser"
-        except (ImportError, AttributeError):
-            # Fallback if ADK structure changes
-            BASE_DIR = Path(__file__).parent.resolve()
-            ANGULAR_DIST_PATH = BASE_DIR / "browser"
-        
-        if ANGULAR_DIST_PATH.exists():
-            extra_fast_api_args.update(web_assets_dir=ANGULAR_DIST_PATH)
+            # Prefer importlib.resources so this works across ADK versions
+            import importlib.resources as r
+            try:
+                import google.adk.cli.fast_api as fast_api_pkg  # type: ignore
+                base = r.files(fast_api_pkg)
+                candidates = [
+                    base / "browser",
+                    base / "static" / "browser",
+                ]
+            except Exception:
+                import google.adk.cli as cli_pkg  # type: ignore
+                base = r.files(cli_pkg) / "fast_api"
+                candidates = [
+                    base / "browser",
+                    base / "static" / "browser",
+                ]
+            for p in candidates:
+                if p.exists() and (p / "index.html").exists():
+                    # Convert to real filesystem Path if possible
+                    try:
+                        return Path(str(p))
+                    except Exception:
+                        continue
+        except Exception:
+            pass
+        # Fallback to local relative path (for dev builds of this package)
+        local = Path(__file__).parent / "browser"
+        if local.exists() and (local / "index.html").exists():
+            return local
+        return None
+
+    if web:
+        chosen: Optional[Path] = None
+        if web_assets_dir is not None:
+            p = Path(web_assets_dir)
+            if p.exists():
+                chosen = p
+        if chosen is None:
+            chosen = _auto_find_web_assets()
+        if chosen is not None:
+            extra_fast_api_args.update(web_assets_dir=chosen)
         else:
-            logger.warning("Web UI assets not found, web interface will not be available")
+            logger.warning(
+                "Web UI assets not found; set web_assets_dir or install an ADK build that ships fast_api/browser"
+            )
 
     # Create FastAPI app
     app = adk_web_server.get_fast_api_app(

{google_adk_extras-0.3.0 → google_adk_extras-0.3.1/src/google_adk_extras.egg-info}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: google-adk-extras
-Version: 0.3.0
+Version: 0.3.1
 Summary: Production-ready services and FastAPI wiring for Google ADK
 Home-page: https://github.com/DeadMeme5441/google-adk-extras
 Author: DeadMeme5441