change docs

Author: twishabansal

packages/toolbox-core/README.md

@@ -335,9 +335,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_core import auth_methods
 
-    auth_token_provider = auth_methods.aget_google_id_token # can also use sync method
+    auth_token_provider = partial(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

@@ -22,9 +22,10 @@
 from toolbox_core import auth_methods
 from functools import partial
 
+URL = "https://toolbox-service-url"
 auth_token_provider = functools.partial(
     auth_methods.aget_google_id_token,
-    "https://toolbox-service-url"
+    URL
 )
 async with ToolboxClient(
     URL,

packages/toolbox-langchain/README.md

@@ -269,10 +269,11 @@ 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 = auth_methods.aget_google_id_token # can also use sync method
+    auth_token_provider = partial(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,10 +248,11 @@ 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 = auth_methods.aget_google_id_token # can also use sync method
+    auth_token_provider = partial(auth_methods.aget_google_id_token, URL)
     async with ToolboxClient(
         URL,
         client_headers={"Authorization": auth_token_provider},