fix docs

Author: twishabansal

packages/toolbox-core/README.md

@@ -335,10 +335,9 @@ For Toolbox servers hosted on Google Cloud (e.g., Cloud Run) and requiring
 3. **Connect to the Toolbox Server**
 
     ```python
-    from functools import partial
     from toolbox_core import auth_methods
 
-    auth_token_provider = partial(auth_methods.aget_google_id_token, URL) # can also use sync method
+    auth_token_provider = auth_methods.aget_google_id_token(URL) # can also use sync method
     async with ToolboxClient(
         URL,
         client_headers={"Authorization": auth_token_provider},

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

@@ -20,7 +20,6 @@
 
 Example Usage:
 from toolbox_core import auth_methods
-from functools import partial
 
 URL = "https://toolbox-service-url"
 async with ToolboxClient(

packages/toolbox-langchain/README.md

@@ -269,11 +269,10 @@ For Toolbox servers hosted on Google Cloud (e.g., Cloud Run) and requiring
 3. **Connect to the Toolbox Server**
 
     ```python
-    from functools import partial
     from toolbox_langchain import ToolboxClient
     from toolbox_core import auth_methods
 
-    auth_token_provider = partial(auth_methods.aget_google_id_token, URL) # can also use sync method
+    auth_token_provider = auth_methods.aget_google_id_token(URL) # can also use sync method
     async with ToolboxClient(
         URL,
         client_headers={"Authorization": auth_token_provider},

packages/toolbox-llamaindex/README.md

@@ -248,11 +248,10 @@ For Toolbox servers hosted on Google Cloud (e.g., Cloud Run) and requiring
 3. **Connect to the Toolbox Server**
 
     ```python
-    from functools import partial
     from toolbox_llamaindex import ToolboxClient
     from toolbox_core import auth_methods
 
-    auth_token_provider = partial(auth_methods.aget_google_id_token, URL)
+    auth_token_provider = auth_methods.aget_google_id_token(URL)
     async with ToolboxClient(
         URL,
         client_headers={"Authorization": auth_token_provider},