fix on object

main.py

@@ -2,40 +2,60 @@
 import os
 import uvicorn
 
-# ---- Adjust imports to match a2a-python ----
-# These paths are common in the tutorial; if yours differ, change them accordingly.
 from a2a.server.apps import A2AStarletteApplication
 from a2a.server.request_handlers import DefaultRequestHandler
 from a2a.server.tasks import InMemoryTaskStore
 
-# Example minimal executor
+# use the typed models from the SDK
+from a2a.types import AgentCard, AgentCapabilities, AgentSkill
+
+# Minimal executor
 class HelloWorldAgentExecutor:
     async def call(self, message: str, **kwargs):
         return {"text": f"Hello from A2A on HF Spaces! You said: {message}"}
 
-# Example minimal public AgentCard (adjust fields to your spec)
-AGENT_CARD = {
-    "name": "HF A2A Agent",
-    "version": "1.0",
-    "description": "A2A demo running on Hugging Face Spaces",
-    "endpoints": {
-        "send": "/message/send",
-        "stream": "/message/stream"
-    }
-}
-
 def build_app():
-    app = A2AStarletteApplication(
-        agent_card=AGENT_CARD,
+    # Define at least one skill (IDs are free-form but should be stable)
+    hello_skill = AgentSkill(
+        id="hello",
+        name="Say Hello",
+        description="Return a friendly greeting",
+        tags=["hello", "demo"],
+        examples=["hi", "hello there"],
+    )
+
+    # HF Spaces gives you a PORT; URL here is informational for clients
+    public_url = os.getenv("SPACE_HOST")  # e.g. "username-space.hf.space"
+    if public_url:
+        public_url = f"https://{public_url}/"
+    else:
+        # local/dev fallback
+        public_url = f"http://0.0.0.0:{os.getenv('PORT', '7860')}/"
+
+    agent_card = AgentCard(
+        name="HF A2A Agent",
+        description="A2A demo running on Hugging Face Spaces",
+        version="1.0.0",
+        url=public_url,
+        protocol_version="0.3.0",
+        default_input_modes=["text"],
+        default_output_modes=["text"],
+        capabilities=AgentCapabilities(streaming=False),
+        skills=[hello_skill],
+        # keep it simple: no authenticated extended card for now
+        supports_authenticated_extended_card=False,
+    )
+
+    app_builder = A2AStarletteApplication(
+        agent_card=agent_card,
         http_handler=DefaultRequestHandler(
             agent_executor=HelloWorldAgentExecutor(),
             task_store=InMemoryTaskStore(),
         ),
-        # extended_agent_card=...  # optional
-    ).build()
-    return app
+        # extended_agent_card=...  # only if you actually support it
+    )
+    return app_builder.build()
 
 if __name__ == "__main__":
-    port = int(os.getenv("PORT", "7860"))  # HF provides PORT
+    port = int(os.getenv("PORT", "7860"))
     uvicorn.run(build_app(), host="0.0.0.0", port=port)
-