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,