Skip to content

Commit 8065361

Browse files
committed
feat: Add custom User-Agent header to RestApiTool requests
1 parent 54ed079 commit 8065361

File tree

1 file changed

+6
-0
lines changed

1 file changed

+6
-0
lines changed

src/google/adk/tools/openapi_tool/openapi_spec_parser/rest_api_tool.py

Lines changed: 6 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -245,6 +245,8 @@ def _prepare_request_params(
245245
Example:
246246
self._prepare_request_params({"input_id": "test-id"})
247247
"""
248+
from .... import __version__ as adk_version
249+
248250
method = self.endpoint.method.lower()
249251
if not method:
250252
raise ValueError("Operation method not found.")
@@ -254,6 +256,10 @@ def _prepare_request_params(
254256
header_params: Dict[str, Any] = {}
255257
cookie_params: Dict[str, Any] = {}
256258

259+
# Set the custom User-Agent header
260+
user_agent = f"google-adk/{adk_version}"
261+
header_params["User-Agent"] = user_agent
262+
257263
params_map: Dict[str, ApiParameter] = {p.py_name: p for p in parameters}
258264

259265
# Fill in path, query, header and cookie parameters to the request

0 commit comments

Comments
 (0)