From ff31f57dc95149f8f309f83f2ec983ef40f1122c Mon Sep 17 00:00:00 2001 From: Ankur Sharma Date: Tue, 22 Jul 2025 13:20:10 -0700 Subject: [PATCH] fix: Move some API request and responses to DEBUG logging level Please set --log_level to DEBUG, if you are interested in having those API request and responses in logs. NOTE: Generally it is not recommended to have DEBUG log level for services that run in a production setting. It is our recommendation to only use DEBUG log level in a debug or development setting. PiperOrigin-RevId: 785972338 --- src/google/adk/agents/remote_a2a_agent.py | 4 ++-- src/google/adk/cli/fast_api.py | 5 +++-- src/google/adk/memory/vertex_ai_memory_bank_service.py | 6 ++++-- src/google/adk/models/anthropic_llm.py | 3 ++- src/google/adk/models/google_llm.py | 7 ++++--- src/google/adk/sessions/vertex_ai_session_service.py | 3 ++- 6 files changed, 17 insertions(+), 11 deletions(-) diff --git a/src/google/adk/agents/remote_a2a_agent.py b/src/google/adk/agents/remote_a2a_agent.py index 117e2003..58d0057e 100644 --- a/src/google/adk/agents/remote_a2a_agent.py +++ b/src/google/adk/agents/remote_a2a_agent.py @@ -481,11 +481,11 @@ class RemoteA2aAgent(BaseAgent): ), ) - logger.info(build_a2a_request_log(a2a_request)) + logger.debug(build_a2a_request_log(a2a_request)) try: a2a_response = await self._a2a_client.send_message(request=a2a_request) - logger.info(build_a2a_response_log(a2a_response)) + logger.debug(build_a2a_response_log(a2a_response)) event = await self._handle_a2a_response(a2a_response, ctx) diff --git a/src/google/adk/cli/fast_api.py b/src/google/adk/cli/fast_api.py index a9d7b691..2f97d547 100644 --- a/src/google/adk/cli/fast_api.py +++ b/src/google/adk/cli/fast_api.py @@ -894,7 +894,8 @@ def get_fast_api_app( new_message=req.new_message, ) ] - logger.info("Generated %s events in agent run: %s", len(events), events) + logger.info("Generated %s events in agent run", len(events)) + logger.debug("Events generated: %s", events) return events @app.post("/run_sse") @@ -920,7 +921,7 @@ def get_fast_api_app( ): # Format as SSE data sse_event = event.model_dump_json(exclude_none=True, by_alias=True) - logger.info("Generated event in agent run streaming: %s", sse_event) + logger.debug("Generated event in agent run streaming: %s", sse_event) yield f"data: {sse_event}\n\n" except Exception as e: logger.exception("Error in event_generator: %s", e) diff --git a/src/google/adk/memory/vertex_ai_memory_bank_service.py b/src/google/adk/memory/vertex_ai_memory_bank_service.py index f8da6acd..c4d7eb22 100644 --- a/src/google/adk/memory/vertex_ai_memory_bank_service.py +++ b/src/google/adk/memory/vertex_ai_memory_bank_service.py @@ -86,7 +86,8 @@ class VertexAiMemoryBankService(BaseMemoryService): path=f'reasoningEngines/{self._agent_engine_id}/memories:generate', request_dict=request_dict, ) - logger.info('Generate memory response: %s', api_response) + logger.info('Generate memory response received.') + logger.debug('Generate memory response: %s', api_response) else: logger.info('No events to add to memory.') @@ -108,7 +109,8 @@ class VertexAiMemoryBankService(BaseMemoryService): }, ) api_response = _convert_api_response(api_response) - logger.info('Search memory response: %s', api_response) + logger.info('Search memory response received.') + logger.debug('Search memory response: %s', api_response) if not api_response or not api_response.get('retrievedMemories', None): return SearchMemoryResponse() diff --git a/src/google/adk/models/anthropic_llm.py b/src/google/adk/models/anthropic_llm.py index 15d78d72..7556a5e6 100644 --- a/src/google/adk/models/anthropic_llm.py +++ b/src/google/adk/models/anthropic_llm.py @@ -174,7 +174,8 @@ def content_block_to_part( def message_to_generate_content_response( message: anthropic_types.Message, ) -> LlmResponse: - logger.info( + logger.info("Received response from Claude.") + logger.debug( "Claude response: %s", message.model_dump_json(indent=2, exclude_none=True), ) diff --git a/src/google/adk/models/google_llm.py b/src/google/adk/models/google_llm.py index 8934b131..983df546 100644 --- a/src/google/adk/models/google_llm.py +++ b/src/google/adk/models/google_llm.py @@ -96,7 +96,7 @@ class Gemini(BaseLlm): self._api_backend, stream, ) - logger.info(_build_request_log(llm_request)) + logger.debug(_build_request_log(llm_request)) # add tracking headers to custom headers given it will override the headers # set in the api client constructor @@ -121,7 +121,7 @@ class Gemini(BaseLlm): # previous partial content. The only difference is bidi rely on # complete_turn flag to detect end while sse depends on finish_reason. async for response in responses: - logger.info(_build_response_log(response)) + logger.debug(_build_response_log(response)) llm_response = LlmResponse.create(response) usage_metadata = llm_response.usage_metadata if ( @@ -179,7 +179,8 @@ class Gemini(BaseLlm): contents=llm_request.contents, config=llm_request.config, ) - logger.info(_build_response_log(response)) + logger.info('Response received from the model.') + logger.debug(_build_response_log(response)) yield LlmResponse.create(response) @cached_property diff --git a/src/google/adk/sessions/vertex_ai_session_service.py b/src/google/adk/sessions/vertex_ai_session_service.py index 12f5312e..9778352d 100644 --- a/src/google/adk/sessions/vertex_ai_session_service.py +++ b/src/google/adk/sessions/vertex_ai_session_service.py @@ -109,7 +109,8 @@ class VertexAiSessionService(BaseSessionService): request_dict=session_json_dict, ) api_response = _convert_api_response(api_response) - logger.info(f'Create Session response {api_response}') + logger.info('Create session response received.') + logger.debug('Create session response: %s', api_response) session_id = api_response['name'].split('/')[-3] operation_id = api_response['name'].split('/')[-1]