fix sync method

Author: twishabansal

packages/toolbox-core/pyproject.toml

@@ -13,6 +13,8 @@ dependencies = [
     "pydantic>=2.7.0,<3.0.0",
     "aiohttp>=3.8.6,<4.0.0",
     "deprecated>=1.2.15,<2.0.0",
+    "google-auth>=2.0.0,<3.0.0",
+    "requests>=2.19.0,<3.0.0"
 ]
 
 classifiers = [

packages/toolbox-core/requirements.txt

@@ -1,3 +1,5 @@
 aiohttp==3.12.14
 pydantic==2.11.7
-deprecated==1.2.18
+deprecated==1.2.18
+requests==2.32.4
+google-auth==2.40.3

packages/toolbox-core/src/toolbox_core/auth_methods.py

@@ -23,20 +23,16 @@
 from functools import partial
 
 URL = "https://toolbox-service-url"
-auth_token_provider = functools.partial(
-    auth_methods.aget_google_id_token,
-    URL
-)
 async with ToolboxClient(
     URL,
-    client_headers={"Authorization": auth_token_provider},
-) as toolbox:
+    client_headers={"Authorization": auth_methods.aget_google_id_token}) 
+as toolbox:
     tools = await toolbox.load_toolset()
 """
 
 import asyncio
 from datetime import datetime, timedelta, timezone
-from typing import Any, Dict, Optional
+from typing import Any, Dict, Optional, Callable, Coroutine
 
 import google.auth
 from google.auth.exceptions import GoogleAuthError
@@ -92,25 +88,8 @@ def _update_cache(new_token: str) -> None:
         _token_cache["token"] = None
         _token_cache["expires_at"] = datetime.min.replace(tzinfo=timezone.utc)
         raise ValueError(f"Failed to validate and cache the new token: {e}") from e
-
-
-def get_google_id_token(audience: Optional[str] = None) -> str:
-    """
-    Synchronously fetches a Google ID token for a specific audience.
-    This function uses Application Default Credentials for local systems
-    and standard google auth libraries for Google Cloud environments.
-    It caches the token in memory.
-
-    Args:
-        audience: The audience for the ID token (e.g., a service URL or client ID).
-
-    Returns:
-        A string in the format "Bearer <google_id_token>".
-
-    Raises:
-        GoogleAuthError: If fetching credentials or the token fails.
-        ValueError: If the fetched token is invalid.
-    """
+    
+def get_google_token_from_aud(audience: Optional[str] = None) -> str:
     if _is_token_valid():
         return BEARER_TOKEN_PREFIX + _token_cache["token"]
 
@@ -145,7 +124,30 @@ def get_google_id_token(audience: Optional[str] = None) -> str:
         ) from e
 
 
-async def aget_google_id_token(audience: Optional[str] = None) -> str:
+def get_google_id_token(audience: Optional[str] = None) -> Callable[[], str]:
+    """
+    Synchronously fetches a Google ID token for a specific audience.
+    This function uses Application Default Credentials for local systems
+    and standard google auth libraries for Google Cloud environments.
+    It caches the token in memory.
+
+    Args:
+        audience: The audience for the ID token (e.g., a service URL or client ID).
+
+    Returns:
+        A string in the format "Bearer <google_id_token>".
+
+    Raises:
+        GoogleAuthError: If fetching credentials or the token fails.
+        ValueError: If the fetched token is invalid.
+    """
+    def _token_getter() -> str:
+        return get_google_token_from_aud(audience)
+
+    return _token_getter
+
+
+async def aget_google_id_token(audience: Optional[str] = None) -> Callable[[], Coroutine[Any, Any, str]]:
     """
     Asynchronously fetches a Google ID token for a specific audience.
     This function uses Application Default Credentials for local systems
@@ -162,5 +164,7 @@ async def aget_google_id_token(audience: Optional[str] = None) -> str:
         GoogleAuthError: If fetching credentials or the token fails.
         ValueError: If the fetched token is invalid.
     """
-    token = await asyncio.to_thread(get_google_id_token, audience)
-    return token
+    async def _token_getter() -> str:
+        return await asyncio.to_thread(get_google_token_from_aud, audience)
+    
+    return _token_getter