Update the agent/contact_lookup sample to use the a2ui-agent python SDK

Tested:
- [x] The contact_lookup client successfully connected to the contact_lookup
agent and rendered the response correctly.

Author: nan-yu

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

@@ -36,7 +36,8 @@
     find_repo_root,
 )
 from .catalog import CustomCatalogConfig, A2uiCatalog
-from ...extension.a2ui_extension import INLINE_CATALOGS_KEY, SUPPORTED_CATALOG_IDS_KEY
+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:
@@ -326,3 +327,7 @@ def generate_system_prompt(
         parts.append(f"### Examples:\n{examples_str}")
 
     return "\n\n".join(parts)
+
+  def get_agent_extension(self) -> AgentExtension:
+    catalog_ids = self.catalog_schemas.keys()
+    return get_a2ui_agent_extension(supported_catalog_ids=list(catalog_ids))

samples/agent/adk/contact_lookup/__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,19 @@ 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}"
+        ui_agent = ContactAgent(base_url=base_url, use_ui=True)
+        text_agent = ContactAgent(base_url=base_url, use_ui=False)
 
-        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)
+        agent_executor = ContactAgentExecutor(ui_agent=ui_agent, text_agent=text_agent)
 
         request_handler = DefaultRequestHandler(
             agent_executor=agent_executor,
             task_store=InMemoryTaskStore(),
         )
         server = A2AStarletteApplication(
-            agent_card=agent_card, http_handler=request_handler
+            agent_card=ui_agent.get_agent_card(), http_handler=request_handler
         )
         import uvicorn