From a8085e50741471fa5c17a5b36b69adaab1da4f25 Mon Sep 17 00:00:00 2001
From: A Vertex SDK engineer <vertex-sdk-bot@google.com>
Date: Tue, 7 Apr 2026 23:30:14 -0700
Subject: [PATCH] feat: Set agent_framework for A2aAgent class.

PiperOrigin-RevId: 896290492
---
 vertexai/_genai/_agent_engines_utils.py             | 1 +
 vertexai/preview/reasoning_engines/templates/a2a.py | 2 ++
 2 files changed, 3 insertions(+)

diff --git a/vertexai/_genai/_agent_engines_utils.py b/vertexai/_genai/_agent_engines_utils.py
index ba388f4d38..fc4fa850bc 100644
--- a/vertexai/_genai/_agent_engines_utils.py
+++ b/vertexai/_genai/_agent_engines_utils.py
@@ -152,6 +152,7 @@
         "ag2",
         "llama-index",
         "custom",
+        "a2a",
     ]
 )
 _DEFAULT_ASYNC_METHOD_NAME = "async_query"
diff --git a/vertexai/preview/reasoning_engines/templates/a2a.py b/vertexai/preview/reasoning_engines/templates/a2a.py
index 724e2af41e..fec888f610 100644
--- a/vertexai/preview/reasoning_engines/templates/a2a.py
+++ b/vertexai/preview/reasoning_engines/templates/a2a.py
@@ -165,6 +165,8 @@ async def cancel(
 class A2aAgent:
     """A class to initialize and set up an Agent-to-Agent application."""
 
+    agent_framework = "a2a"
+
     # TODO: Add instrumentation for the A2A agent.
     def __init__(
         self,