remove files

Author: twishabansal

packages/toolbox-llamaindex/README.md

@@ -204,16 +204,27 @@ that fresh credentials or header values can be used.
 
 ### Configuration
 
-You can configure these dynamic headers as follows:
+You can configure these dynamic headers in two ways:
 
-```python
-from toolbox_llamaindex import ToolboxClient
+1. **During Client Initialization**
 
-client = ToolboxClient(
-    "toolbox-url", 
-    client_headers={"header1": header1_getter, "header2": header2_getter, ...}
-)
-```
+    ```python
+    from toolbox_llamaindex import ToolboxClient
+
+    client = ToolboxClient(
+        "toolbox-url", 
+        client_headers={"header1": header1_getter, "header2": header2_getter, ...}
+    )
+    ```
+
+1. **After Client Initialization**
+
+    ```python
+    from toolbox_llamaindex import ToolboxClient
+
+    client = ToolboxClient("toolbox-url")
+    client.add_headers({"header1": header1_getter, "header2": header2_getter, ...})
+    ```
 
 ### Authenticating with Google Cloud Servers
 

packages/toolbox-llamaindex/src/toolbox_llamaindex/async_client.py

@@ -190,3 +190,16 @@ def load_toolset(
         strict: bool = False,
     ) -> list[AsyncToolboxTool]:
         raise NotImplementedError("Synchronous methods not supported by async client.")
+
+    def add_headers(
+        self,
+        headers: Mapping[str, Union[Callable[[], str], Callable[[], Awaitable[str]]]],
+    ) -> None:
+        """
+        Add headers to be included in each request sent through this client.
+        Args:
+            headers: Headers to include in each request sent through this client.
+        Raises:
+            ValueError: If any of the headers are already registered in the client.
+        """
+        self.__core_client.add_headers(headers)

packages/toolbox-llamaindex/src/toolbox_llamaindex/client.py

@@ -292,3 +292,16 @@ def load_toolset(
         for core_sync_tool in core_sync_tools:
             tools.append(ToolboxTool(core_tool=core_sync_tool))
         return tools
+
+    def add_headers(
+        self,
+        headers: Mapping[str, Union[Callable[[], str], Callable[[], Awaitable[str]]]],
+    ) -> None:
+        """
+        Add headers to be included in each request sent through this client.
+        Args:
+            headers: Headers to include in each request sent through this client.
+        Raises:
+            ValueError: If any of the headers are already registered in the client.
+        """
+        self.__core_client.add_headers(headers)

packages/toolbox-llamaindex/tests/test_async_client.py

@@ -350,3 +350,11 @@ async def test_init_with_client_headers(
         mock_core_client_constructor.assert_called_once_with(
             url=URL, session=mock_session, client_headers=headers
         )
+
+    async def test_add_headers(self, mock_client):
+        """Tests that add_headers calls the core client's add_headers."""
+        headers = {"X-Another-Header": lambda: "dynamic_value"}
+        mock_client.add_headers(headers)
+        mock_client._AsyncToolboxClient__core_client.add_headers.assert_called_once_with(
+            headers
+        )

packages/toolbox-llamaindex/tests/test_client.py

@@ -431,3 +431,12 @@ def test_init_with_client_headers(self, mock_core_client_constructor):
         mock_core_client_constructor.assert_called_once_with(
             url=URL, client_headers=headers
         )
+
+    @patch("toolbox_llamaindex.client.ToolboxCoreSyncClient")
+    def test_add_headers(self, mock_core_client_constructor):
+        """Tests that add_headers calls the core client's add_headers."""
+        mock_core_instance = mock_core_client_constructor.return_value
+        client = ToolboxClient(URL)
+        headers = {"X-Another-Header": "dynamic_value"}
+        client.add_headers(headers)
+        mock_core_instance.add_headers.assert_called_once_with(headers)