Update contact_multiple_surfaces sample

It updates the sample to use the A2uiSchemaManager from the a2ui-agent
python SDK.

Tested:
- [x] The `contact` lit client successfully connected to the
  `contact_multiple_surfaces` agent and rendered the response correctly.

Author: nan-yu

a2a_agents/python/a2ui_agent/src/a2ui/inference/schema/manager.py

@@ -37,6 +37,7 @@
 )
 from .catalog import CustomCatalogConfig, A2uiCatalog
 from ...extension.a2ui_extension import INLINE_CATALOGS_KEY, SUPPORTED_CATALOG_IDS_KEY, get_a2ui_agent_extension
+from a2a.types import AgentExtension
 
 
 def _load_basic_component(version: str, spec_name: str) -> Dict:

samples/agent/adk/contact_lookup/agent.py

@@ -42,6 +42,21 @@ class ContactAgent:
 
   SUPPORTED_CONTENT_TYPES = ["text", "text/plain"]
 
+  def __init__(self, base_url: str, use_ui: bool = False):
+    self.base_url = base_url
+    self.use_ui = use_ui
+    self._schema_manager = (
+        A2uiSchemaManager("0.8", basic_examples_path="examples") if use_ui else None
+    )
+    self._agent = self._build_agent(use_ui)
+    self._user_id = "remote_agent"
+    self._runner = Runner(
+        app_name=self._agent.name,
+        agent=self._agent,
+        artifact_service=InMemoryArtifactService(),
+        session_service=InMemorySessionService(),
+        memory_service=InMemoryMemoryService(),
+    )
   def __init__(self, base_url: str, use_ui: bool = False):
     self.base_url = base_url
     self.use_ui = use_ui
@@ -71,10 +86,7 @@ def get_agent_card(self) -> AgentCard:
             " team)."
         ),
         tags=["contact", "directory", "people", "finder"],
-        examples=[
-            "Who is David Chen in marketing?",
-            "Find Sarah Lee from engineering",
-        ],
+        examples=["Who is David Chen in marketing?", "Find Sarah Lee from engineering"],
     )
 
     return AgentCard(
@@ -89,6 +101,18 @@ def get_agent_card(self) -> AgentCard:
         capabilities=capabilities,
         skills=[skill],
     )
+    return AgentCard(
+        name="Contact Lookup Agent",
+        description=(
+            "This agent helps find contact info for people in your organization."
+        ),
+        url=self.base_url,
+        version="1.0.0",
+        default_input_modes=ContactAgent.SUPPORTED_CONTENT_TYPES,
+        default_output_modes=ContactAgent.SUPPORTED_CONTENT_TYPES,
+        capabilities=capabilities,
+        skills=[skill],
+    )
 
   def get_processing_message(self) -> str:
     return "Looking up contact information..."

samples/agent/adk/contact_multiple_surfaces/__main__.py

@@ -19,8 +19,6 @@
 from a2a.server.apps import A2AStarletteApplication
 from a2a.server.request_handlers import DefaultRequestHandler
 from a2a.server.tasks import InMemoryTaskStore
-from a2a.types import AgentCapabilities, AgentCard, AgentSkill
-from a2ui.extension.a2ui_extension import get_a2ui_agent_extension
 from agent import ContactAgent
 from agent_executor import ContactAgentExecutor
 from dotenv import load_dotenv
@@ -36,80 +34,53 @@
 class MissingAPIKeyError(Exception):
   """Exception for missing API key."""
 
-
-@click.command()
-@click.option("--host", default="localhost")
-@click.option("--port", default=10004)
-def main(host, port):
-  try:
-    # Check for API key only if Vertex AI is not configured
-    if not os.getenv("GOOGLE_GENAI_USE_VERTEXAI") == "TRUE":
-      if not os.getenv("GEMINI_API_KEY"):
-        raise MissingAPIKeyError(
-            "GEMINI_API_KEY environment variable not set and GOOGLE_GENAI_USE_VERTEXAI"
-            " is not TRUE."
-        )
-
-    capabilities = AgentCapabilities(
-        streaming=True,
-        extensions=[get_a2ui_agent_extension()],
-    )
-    skill = AgentSkill(
-        id="find_contact",
-        name="Find Contact Tool",
-        description=(
-            "Helps find contact information for colleagues (e.g., email, location,"
-            " team)."
-        ),
-        tags=["contact", "directory", "people", "finder"],
-        examples=["Who is David Chen in marketing?", "Find Sarah Lee from engineering"],
-    )
-
-    base_url = f"http://{host}:{port}"
-
-    agent_card = AgentCard(
-        name="Contact Lookup Agent",
-        description=(
-            "This agent helps find contact info for people in your organization."
-        ),
-        url=base_url,  # <-- Use base_url here
-        version="1.0.0",
-        default_input_modes=ContactAgent.SUPPORTED_CONTENT_TYPES,
-        default_output_modes=ContactAgent.SUPPORTED_CONTENT_TYPES,
-        capabilities=capabilities,
-        skills=[skill],
-    )
-
-    agent_executor = ContactAgentExecutor(base_url=base_url)
-
-    request_handler = DefaultRequestHandler(
-        agent_executor=agent_executor,
-        task_store=InMemoryTaskStore(),
-    )
-    server = A2AStarletteApplication(
-        agent_card=agent_card, http_handler=request_handler
-    )
-    import uvicorn
-
-    app = server.build()
-
-    app.add_middleware(
-        CORSMiddleware,
-        allow_origins=["http://localhost:5173"],
-        allow_credentials=True,
-        allow_methods=["*"],
-        allow_headers=["*"],
-    )
-
-    app.mount("/static", StaticFiles(directory="images"), name="static")
-
-    uvicorn.run(app, host=host, port=port)
-  except MissingAPIKeyError as e:
-    logger.error(f"Error: {e}")
-    exit(1)
-  except Exception as e:
-    logger.error(f"An error occurred during server startup: {e}")
-    exit(1)
+  @click.command()
+  @click.option("--host", default="localhost")
+  @click.option("--port", default=10004)
+  def main(host, port):
+    try:
+      # Check for API key only if Vertex AI is not configured
+      if not os.getenv("GOOGLE_GENAI_USE_VERTEXAI") == "TRUE":
+        if not os.getenv("GEMINI_API_KEY"):
+          raise MissingAPIKeyError(
+              "GEMINI_API_KEY environment variable not set and"
+              " GOOGLE_GENAI_USE_VERTEXAI is not TRUE."
+          )
+
+      base_url = f"http://{host}:{port}"
+
+      agent = ContactAgent(base_url=base_url, use_ui=True)
+
+      agent_executor = ContactAgentExecutor(agent=agent)
+
+      request_handler = DefaultRequestHandler(
+          agent_executor=agent_executor,
+          task_store=InMemoryTaskStore(),
+      )
+      server = A2AStarletteApplication(
+          agent_card=agent.get_agent_card(), http_handler=request_handler
+      )
+      import uvicorn
+
+      app = server.build()
+
+      app.add_middleware(
+          CORSMiddleware,
+          allow_origins=["http://localhost:5173"],
+          allow_credentials=True,
+          allow_methods=["*"],
+          allow_headers=["*"],
+      )
+
+      app.mount("/static", StaticFiles(directory="images"), name="static")
+
+      uvicorn.run(app, host=host, port=port)
+    except MissingAPIKeyError as e:
+      logger.error(f"Error: {e}")
+      exit(1)
+    except Exception as e:
+      logger.error(f"An error occurred during server startup: {e}")
+      exit(1)
 
 
 if __name__ == "__main__":

samples/agent/adk/contact_multiple_surfaces/a2ui_examples.py

@@ -18,7 +18,6 @@
 from pathlib import Path
 
 import jsonschema
-from a2ui_schema import A2UI_SCHEMA
 
 logger = logging.getLogger(__name__)
 
@@ -35,77 +34,6 @@
 FLOOR_PLAN_FILE = "floor_plan.json"
 
 
-def load_examples(base_url: str = "http://localhost:10004") -> str:
-  """
-  Loads, validates, and formats the UI examples from JSON files.
-
-  Args:
-      base_url: The base URL to replace placeholder URLs with.
-                (Currently examples have http://localhost:10004 hardcoded,
-                 but we can make this dynamic if needed).
-
-  Returns:
-      A string containing all formatted examples for the prompt.
-  """
-
-  # Pre-parse validator
-  try:
-    single_msg_schema = json.loads(A2UI_SCHEMA)
-    # Examples are typically lists of messages
-    list_schema = {"type": "array", "items": single_msg_schema}
-  except json.JSONDecodeError:
-    logger.error("Failed to parse A2UI_SCHEMA for validation")
-    list_schema = None
-
-  examples_dir = Path(os.path.dirname(__file__)) / "examples"
-  formatted_output = []
-
-  for curr_name, filename in EXAMPLE_FILES.items():
-    file_path = examples_dir / filename
-    try:
-      content = file_path.read_text(encoding="utf-8")
-
-      # basic replacement if we decide to template the URL in JSON files
-      # content = content.replace("{{BASE_URL}}", base_url)
-
-      # Validation
-      if list_schema:
-        try:
-          data = json.loads(content)
-          jsonschema.validate(instance=data, schema=list_schema)
-        except (json.JSONDecodeError, jsonschema.ValidationError) as e:
-          logger.warning(f"Example {filename} validation failed: {e}")
-
-      formatted_output.append(f"---BEGIN {curr_name}---")
-      # Handle examples that include user/model text
-      if curr_name == "ORG_CHART_EXAMPLE":
-        formatted_output.append("User: Show me the org chart for Casey Smith")
-        formatted_output.append("Model: Here is the organizational chart.")
-        formatted_output.append("---a2ui_JSON---")
-      elif curr_name == "MULTI_SURFACE_EXAMPLE":
-        formatted_output.append("User: Full profile for Casey Smith")
-        formatted_output.append(
-            "Model: Here is the full profile including contact details and org chart."
-        )
-        formatted_output.append("---a2ui_JSON---")
-      elif curr_name == "CHART_NODE_CLICK_EXAMPLE":
-        formatted_output.append(
-            'User: ACTION: chart_node_click (context: clickedNodeName="John Smith")'
-            " (from modal)"
-        )
-        formatted_output.append("Model: Here is the profile for John Smith.")
-        formatted_output.append("---a2ui_JSON---")
-
-      formatted_output.append(content.strip())
-      formatted_output.append(f"---END {curr_name}---")
-      formatted_output.append("")  # Newline
-
-    except FileNotFoundError:
-      logger.error(f"Example file not found: {file_path}")
-
-  return "\n".join(formatted_output)
-
-
 def load_floor_plan_example() -> str:
   """Loads the floor plan example specifically."""
   examples_dir = Path(os.path.dirname(__file__)) / "examples"