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_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
@@ -49,39 +47,18 @@ def main(host, port):
                     "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 = ContactAgent(base_url=base_url, use_ui=True)
 
-        agent_executor = ContactAgentExecutor(base_url=base_url)
+        agent_executor = ContactAgentExecutor(agent=agent)
 
         request_handler = DefaultRequestHandler(
             agent_executor=agent_executor,
             task_store=InMemoryTaskStore(),
         )
         server = A2AStarletteApplication(
-            agent_card=agent_card, http_handler=request_handler
+            agent_card=agent.get_agent_card(), http_handler=request_handler
         )
         import uvicorn
 

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__)
 
@@ -34,71 +33,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"