feat: Updates the ADK template to direct structured JSON logs to standard output.

PiperOrigin-RevId: 837124358

Author: vertex-sdk-bot

vertexai/agent_engines/templates/adk.py

@@ -27,6 +27,7 @@
 import asyncio
 from collections.abc import Awaitable
 import queue
+import sys
 import threading
 import warnings
 
@@ -437,14 +438,21 @@ def _detect_cloud_resource_id(project_id: str) -> Optional[str]:
                 "opentelemetry-exporter-gcp-logging", needed_for_logging=True
             )
 
+        class _SimpleLogRecordProcessor(opentelemetry.sdk._logs.export.SimpleLogRecordProcessor):
+            def force_flush(self, timeout_millis: int = 30000) -> bool:  # pylint: disable=no-self-use
+                sys.stdout.flush()
+                sys.stderr.flush()
+                return True
+
         logger_provider = opentelemetry.sdk._logs.LoggerProvider(resource=resource)
         logger_provider.add_log_record_processor(
-            opentelemetry.sdk._logs.export.BatchLogRecordProcessor(
+            _SimpleLogRecordProcessor(
                 opentelemetry.exporter.cloud_logging.CloudLoggingExporter(
                     project_id=project_id,
                     default_log_name=os.getenv(
                         "GCP_DEFAULT_LOG_NAME", "adk-on-agent-engine"
                     ),
+                    structured_json_file=sys.stdout,
                 ),
             )
         )

vertexai/preview/reasoning_engines/templates/adk.py

@@ -27,6 +27,7 @@
 import asyncio
 from collections.abc import Awaitable
 import queue
+import sys
 import threading
 
 
@@ -439,14 +440,21 @@ def _detect_cloud_resource_id(project_id: str) -> Optional[str]:
                 "opentelemetry-exporter-gcp-logging", needed_for_logging=True
             )
 
+        class _SimpleLogRecordProcessor(opentelemetry.sdk._logs.export.SimpleLogRecordProcessor):
+            def force_flush(self, timeout_millis: int = 30000) -> bool:  # pylint: disable=no-self-use
+                sys.stdout.flush()
+                sys.stderr.flush()
+                return True
+
         logger_provider = opentelemetry.sdk._logs.LoggerProvider(resource=resource)
         logger_provider.add_log_record_processor(
-            opentelemetry.sdk._logs.export.BatchLogRecordProcessor(
+            _SimpleLogRecordProcessor(
                 opentelemetry.exporter.cloud_logging.CloudLoggingExporter(
                     project_id=project_id,
                     default_log_name=os.getenv(
                         "GCP_DEFAULT_LOG_NAME", "adk-on-agent-engine"
                     ),
+                    structured_json_file=sys.stdout,
                 ),
             )
         )