feat: durable runtime support

src/google/adk/a2a/converters/event_converter.py

@@ -22,7 +22,6 @@
 from typing import Dict
 from typing import List
 from typing import Optional
-import uuid
 
 from a2a.server.events import Event as A2AEvent
 from a2a.types import DataPart
@@ -34,6 +33,8 @@
 from a2a.types import TaskStatus
 from a2a.types import TaskStatusUpdateEvent
 from a2a.types import TextPart
+from google.adk.platform import time as platform_time
+from google.adk.platform import uuid as platform_uuid
 from google.genai import types as genai_types
 
 from ...agents.invocation_context import InvocationContext
@@ -254,7 +255,7 @@ def convert_a2a_task_to_event(
         invocation_id=(
             invocation_context.invocation_id
             if invocation_context
-            else str(uuid.uuid4())
+            else platform_uuid.new_uuid()
         ),
         author=author or "a2a agent",
         branch=invocation_context.branch if invocation_context else None,
@@ -299,7 +300,7 @@ def convert_a2a_message_to_event(
         invocation_id=(
             invocation_context.invocation_id
             if invocation_context
-            else str(uuid.uuid4())
+            else platform_uuid.new_uuid()
         ),
         author=author or "a2a agent",
         branch=invocation_context.branch if invocation_context else None,
@@ -349,7 +350,7 @@ def convert_a2a_message_to_event(
         invocation_id=(
             invocation_context.invocation_id
             if invocation_context
-            else str(uuid.uuid4())
+            else platform_uuid.new_uuid()
         ),
         author=author or "a2a agent",
         branch=invocation_context.branch if invocation_context else None,
@@ -406,7 +407,7 @@ def convert_event_to_a2a_message(
 
     if output_parts:
       return Message(
-          message_id=str(uuid.uuid4()), role=role, parts=output_parts
+          message_id=platform_uuid.new_uuid(), role=role, parts=output_parts
       )
 
   except Exception as e:
@@ -447,7 +448,7 @@ def _create_error_status_event(
       status=TaskStatus(
           state=TaskState.failed,
           message=Message(
-              message_id=str(uuid.uuid4()),
+              message_id=platform_uuid.new_uuid(),
               role=Role.agent,
               parts=[TextPart(text=error_message)],
               metadata={
@@ -456,7 +457,9 @@ def _create_error_status_event(
               if event.error_code
               else {},
           ),
-          timestamp=datetime.now(timezone.utc).isoformat(),
+          timestamp=datetime.fromtimestamp(
+              platform_time.get_time(), tz=timezone.utc
+          ).isoformat(),
       ),
       final=False,
   )
@@ -484,7 +487,9 @@ def _create_status_update_event(
   status = TaskStatus(
       state=TaskState.working,
       message=message,
-      timestamp=datetime.now(timezone.utc).isoformat(),
+      timestamp=datetime.fromtimestamp(
+          platform_time.get_time(), tz=timezone.utc
+      ).isoformat(),
   )
 
   if any(

src/google/adk/a2a/executor/a2a_agent_executor.py

@@ -21,7 +21,6 @@
 from typing import Awaitable
 from typing import Callable
 from typing import Optional
-import uuid
 
 from a2a.server.agent_execution import AgentExecutor
 from a2a.server.agent_execution.context import RequestContext
@@ -34,6 +33,8 @@
 from a2a.types import TaskStatus
 from a2a.types import TaskStatusUpdateEvent
 from a2a.types import TextPart
+from google.adk.platform import time as platform_time
+from google.adk.platform import uuid as platform_uuid
 from google.adk.runners import Runner
 from typing_extensions import override
 
@@ -147,7 +148,9 @@ async def execute(
               status=TaskStatus(
                   state=TaskState.submitted,
                   message=context.message,
-                  timestamp=datetime.now(timezone.utc).isoformat(),
+                  timestamp=datetime.fromtimestamp(
+                      platform_time.get_time(), tz=timezone.utc
+                  ).isoformat(),
               ),
               context_id=context.context_id,
               final=False,
@@ -166,9 +169,11 @@ async def execute(
                 task_id=context.task_id,
                 status=TaskStatus(
                     state=TaskState.failed,
-                    timestamp=datetime.now(timezone.utc).isoformat(),
+                    timestamp=datetime.fromtimestamp(
+                        platform_time.get_time(), tz=timezone.utc
+                    ).isoformat(),
                     message=Message(
-                        message_id=str(uuid.uuid4()),
+                        message_id=platform_uuid.new_uuid(),
                         role=Role.agent,
                         parts=[TextPart(text=str(e))],
                     ),
@@ -219,7 +224,9 @@ async def _handle_request(
             task_id=context.task_id,
             status=TaskStatus(
                 state=TaskState.working,
-                timestamp=datetime.now(timezone.utc).isoformat(),
+                timestamp=datetime.fromtimestamp(
+                    platform_time.get_time(), tz=timezone.utc
+                ).isoformat(),
             ),
             context_id=context.context_id,
             final=False,
@@ -267,7 +274,7 @@ async def _handle_request(
               last_chunk=True,
               context_id=context.context_id,
               artifact=Artifact(
-                  artifact_id=str(uuid.uuid4()),
+                  artifact_id=platform_uuid.new_uuid(),
                   parts=task_result_aggregator.task_status_message.parts,
               ),
           )
@@ -277,7 +284,9 @@ async def _handle_request(
           task_id=context.task_id,
           status=TaskStatus(
               state=TaskState.completed,
-              timestamp=datetime.now(timezone.utc).isoformat(),
+              timestamp=datetime.fromtimestamp(
+                  platform_time.get_time(), tz=timezone.utc
+              ).isoformat(),
           ),
           context_id=context.context_id,
           final=True,
@@ -287,7 +296,9 @@ async def _handle_request(
           task_id=context.task_id,
           status=TaskStatus(
               state=task_result_aggregator.task_state,
-              timestamp=datetime.now(timezone.utc).isoformat(),
+              timestamp=datetime.fromtimestamp(
+                  platform_time.get_time(), tz=timezone.utc
+              ).isoformat(),
               message=task_result_aggregator.task_status_message,
           ),
           context_id=context.context_id,

src/google/adk/agents/invocation_context.py

@@ -15,9 +15,10 @@
 from __future__ import annotations
 
 from typing import Any
+from typing import cast
 from typing import Optional
-import uuid
 
+from google.adk.platform import uuid as platform_uuid
 from google.genai import types
 from pydantic import BaseModel
 from pydantic import ConfigDict
@@ -413,4 +414,4 @@ def _find_matching_function_call(
 
 
 def new_invocation_context_id() -> str:
-  return "e-" + str(uuid.uuid4())
+  return "e-" + str(cast(str, platform_uuid.new_uuid()))

src/google/adk/agents/remote_a2a_agent.py

@@ -24,7 +24,6 @@
 from typing import Optional
 from typing import Union
 from urllib.parse import urlparse
-import uuid
 
 from a2a.client import Client as A2AClient
 from a2a.client import ClientEvent as A2AClientEvent
@@ -43,6 +42,7 @@
 from a2a.types import TaskState
 from a2a.types import TaskStatusUpdateEvent as A2ATaskStatusUpdateEvent
 from a2a.types import TransportProtocol as A2ATransport
+from google.adk.platform import uuid as platform_uuid
 from google.genai import types as genai_types
 import httpx
 from pydantic import BaseModel
@@ -629,7 +629,7 @@ async def _run_async_impl(
         return
 
       a2a_request = A2AMessage(
-          message_id=str(uuid.uuid4()),
+          message_id=platform_uuid.new_uuid(),
           parts=message_parts,
           role="user",
           context_id=context_id,

src/google/adk/artifacts/base_artifact_service.py

@@ -21,6 +21,7 @@
 from typing import Optional
 from typing import Union
 
+from google.adk.platform import time as platform_time
 from google.genai import types
 from pydantic import alias_generators
 from pydantic import BaseModel
@@ -51,7 +52,7 @@ class ArtifactVersion(BaseModel):
       description="Optional user-supplied metadata stored with the artifact.",
   )
   create_time: float = Field(
-      default_factory=lambda: datetime.now().timestamp(),
+      default_factory=lambda: platform_time.get_time(),
       description=(
           "Unix timestamp (seconds) when the version record was created."
       ),

src/google/adk/events/event.py

@@ -14,10 +14,11 @@
 
 from __future__ import annotations
 
-from datetime import datetime
 from typing import Optional
-import uuid
+from typing import cast
 
+from google.adk.platform import time as platform_time
+from google.adk.platform import uuid as platform_uuid
 from google.genai import types
 from pydantic import alias_generators
 from pydantic import ConfigDict
@@ -70,7 +71,7 @@ class Event(LlmResponse):
   # Do not assign the ID. It will be assigned by the session.
   id: str = ''
   """The unique identifier of the event."""
-  timestamp: float = Field(default_factory=lambda: datetime.now().timestamp())
+  timestamp: float = Field(default_factory=lambda: platform_time.get_time())
   """The timestamp of the event."""
 
   def model_post_init(self, __context):
@@ -124,5 +125,5 @@ def has_trailing_code_execution_result(
     return False
 
   @staticmethod
-  def new_id():
-    return str(uuid.uuid4())
+  def new_id() -> str:
+    return cast(str, platform_uuid.new_uuid())

src/google/adk/flows/llm_flows/_code_execution.py

@@ -27,6 +27,7 @@
 from typing import Optional
 from typing import TYPE_CHECKING
 
+from google.adk.platform import time as platform_time
 from google.genai import types
 from typing_extensions import override
 
@@ -288,7 +289,9 @@ async def _run_post_processor(
         if part.inline_data.display_name:
           file_name = part.inline_data.display_name
         else:
-          now = datetime.datetime.now().astimezone()
+          now = datetime.datetime.fromtimestamp(
+              platform_time.get_time()
+          ).astimezone()
           timestamp = now.strftime('%Y%m%d_%H%M%S')
           file_extension = part.inline_data.mime_type.split('/')[-1]
           file_name = f'{timestamp}.{file_extension}'

src/google/adk/flows/llm_flows/audio_cache_manager.py

@@ -15,9 +15,9 @@
 from __future__ import annotations
 
 import logging
-import time
 from typing import TYPE_CHECKING
 
+from google.adk.platform import time as platform_time
 from google.genai import types
 
 from ...agents.invocation_context import RealtimeCacheEntry
@@ -70,7 +70,7 @@ def cache_audio(
       raise ValueError("cache_type must be either 'input' or 'output'")
 
     audio_entry = RealtimeCacheEntry(
-        role=role, data=audio_blob, timestamp=time.time()
+        role=role, data=audio_blob, timestamp=platform_time.get_time()
     )
     cache.append(audio_entry)
 

src/google/adk/flows/llm_flows/base_llm_flow.py

@@ -16,14 +16,14 @@
 
 from abc import ABC
 import asyncio
-import datetime
 import inspect
 import logging
 from typing import AsyncGenerator
 from typing import cast
 from typing import Optional
 from typing import TYPE_CHECKING
 
+from google.adk.platform import time as platform_time
 from google.genai import types
 from websockets.exceptions import ConnectionClosed
 from websockets.exceptions import ConnectionClosedOK
@@ -845,7 +845,7 @@ async def _run_one_step_async(
           async for event in agen:
             # Update the mutable event id to avoid conflict
             model_response_event.id = Event.new_id()
-            model_response_event.timestamp = datetime.datetime.now().timestamp()
+            model_response_event.timestamp = platform_time.get_time()
             yield event
 
   async def _preprocess_async(

src/google/adk/flows/llm_flows/functions.py

@@ -31,8 +31,8 @@
 from typing import Dict
 from typing import Optional
 from typing import TYPE_CHECKING
-import uuid
 
+from google.adk.platform import uuid as platform_uuid
 from google.adk.tools.computer_use.computer_use_tool import ComputerUseTool
 from google.genai import types
 
@@ -178,7 +178,7 @@ def run_async_tool_in_new_loop():
 
 
 def generate_client_function_call_id() -> str:
-  return f'{AF_FUNCTION_CALL_ID_PREFIX}{uuid.uuid4()}'
+  return f'{AF_FUNCTION_CALL_ID_PREFIX}{platform_uuid.new_uuid()}'
 
 
 def populate_client_function_call_id(model_response_event: Event) -> None:

src/google/adk/flows/llm_flows/transcription_manager.py

@@ -15,9 +15,9 @@
 from __future__ import annotations
 
 import logging
-import time
 from typing import TYPE_CHECKING
 
+from google.adk.platform import time as platform_time
 from google.genai import types
 
 from ...events.event import Event
@@ -89,7 +89,7 @@ async def _create_and_save_transcription_event(
           author=author,
           input_transcription=transcription if is_input else None,
           output_transcription=transcription if not is_input else None,
-          timestamp=time.time(),
+          timestamp=platform_time.get_time(),
       )
 
       # Save transcription event to session