back to a2a lib

main.py

@@ -1,67 +1,41 @@
+# main.py
 import os
-from fastapi import FastAPI, Request
-from fastapi.responses import JSONResponse
+import uvicorn
 
-app = FastAPI(title="HF A2A Agent")
+# ---- 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
 
-def hello_agent(message: str):
-    return {"text": f"Hello from A2A on HF Spaces! You said: {message}"}
+# Example 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 = {
-    "protocolVersion": "0.3.0",
-    "name": "Hello World Agent",
-    "description": "A simple A2A agent that responds with 'Hello World' to any request",
-    "url": "https://{HOST}/a2a",
-    "preferredTransport": "HTTP+JSON",
-    "version": "1.0.0",
-    "capabilities": {
-        "streaming": False,
-        "pushNotifications": False,
-        "stateTransitionHistory": False
-    },
-    "defaultInputModes": ["text/plain", "application/json"],
-    "defaultOutputModes": ["text/plain", "application/json"],
-    "skills": [
-        {
-            "id": "hello",
-            "name": "Say Hello",
-            "description": "Responds with a friendly 'Hello World' message",
-            "tags": ["greeting", "hello", "simple"],
-            "inputModes": ["text/plain", "application/json"],
-            "outputModes": ["text/plain", "application/json"],
-            "examples": ["Say hello", "Greet me", "Hello"]
-        }
-    ],
-    "provider": {
-        "organization": "A2A Registry Team",
-        "url": "https://github.com/prassanna-ravishankar/a2a-registry"
-    },
-    "documentationUrl": "https://github.com/prassanna-ravishankar/a2a-registry",
-    "author": "A2A Registry Team",
-    "homepage": "https://hello.a2aregistry.org",
-    "license": "MIT"
+    "name": "HF A2A Agent",
+    "version": "1.0",
+    "description": "A2A demo running on Hugging Face Spaces",
+    "endpoints": {
+        "send": "/message/send",
+        "stream": "/message/stream"
+    }
 }
 
-@app.get("/.well-known/agent-card.json")
-async def agent_card(request: Request):
-    host = request.headers.get("x-forwarded-host") or request.headers.get("host") or "localhost:7860"
-    proto = request.headers.get("x-forwarded-proto") or "https"
-    url_base = f"{proto}://{host}"
-    card = dict(AGENT_CARD)
-    card["url"] = AGENT_CARD["url"].replace("{HOST}", host).replace("https://{HOST}", url_base)
-    return JSONResponse(card)
+def build_app():
+    app = A2AStarletteApplication(
+        agent_card=AGENT_CARD,
+        http_handler=DefaultRequestHandler(
+            agent_executor=HelloWorldAgentExecutor(),
+            task_store=InMemoryTaskStore(),
+        ),
+        # extended_agent_card=...  # optional
+    ).build()
+    return app
 
-@app.get("/")
-async def root():
-    return {"status": "ok", "send_endpoint": "/a2a"}
-
-@app.post("/a2a")
-async def a2a_endpoint(request: Request):
-    ctype = request.headers.get("content-type", "")
-    if "application/json" in ctype:
-        body = await request.json()
-        msg = body.get("message") or body.get("text") or str(body)
-        return JSONResponse(hello_agent(msg))
-    text = (await request.body()).decode("utf-8", errors="ignore")
-    return JSONResponse(hello_agent(text))
+if __name__ == "__main__":
+    port = int(os.getenv("PORT", "7860"))  # HF provides PORT
+    uvicorn.run(build_app(), host="0.0.0.0", port=port)
 

simple_main.py

@@ -0,0 +1,67 @@
+import os
+from fastapi import FastAPI, Request
+from fastapi.responses import JSONResponse
+
+app = FastAPI(title="HF A2A Agent")
+
+def hello_agent(message: str):
+    return {"text": f"Hello from A2A on HF Spaces! You said: {message}"}
+
+AGENT_CARD = {
+    "protocolVersion": "0.3.0",
+    "name": "Hello World Agent",
+    "description": "A simple A2A agent that responds with 'Hello World' to any request",
+    "url": "https://{HOST}/a2a",
+    "preferredTransport": "HTTP+JSON",
+    "version": "1.0.0",
+    "capabilities": {
+        "streaming": False,
+        "pushNotifications": False,
+        "stateTransitionHistory": False
+    },
+    "defaultInputModes": ["text/plain", "application/json"],
+    "defaultOutputModes": ["text/plain", "application/json"],
+    "skills": [
+        {
+            "id": "hello",
+            "name": "Say Hello",
+            "description": "Responds with a friendly 'Hello World' message",
+            "tags": ["greeting", "hello", "simple"],
+            "inputModes": ["text/plain", "application/json"],
+            "outputModes": ["text/plain", "application/json"],
+            "examples": ["Say hello", "Greet me", "Hello"]
+        }
+    ],
+    "provider": {
+        "organization": "A2A Registry Team",
+        "url": "https://github.com/prassanna-ravishankar/a2a-registry"
+    },
+    "documentationUrl": "https://github.com/prassanna-ravishankar/a2a-registry",
+    "author": "A2A Registry Team",
+    "homepage": "https://hello.a2aregistry.org",
+    "license": "MIT"
+}
+
+@app.get("/.well-known/agent-card.json")
+async def agent_card(request: Request):
+    host = request.headers.get("x-forwarded-host") or request.headers.get("host") or "localhost:7860"
+    proto = request.headers.get("x-forwarded-proto") or "https"
+    url_base = f"{proto}://{host}"
+    card = dict(AGENT_CARD)
+    card["url"] = AGENT_CARD["url"].replace("{HOST}", host).replace("https://{HOST}", url_base)
+    return JSONResponse(card)
+
+@app.get("/")
+async def root():
+    return {"status": "ok", "send_endpoint": "/a2a"}
+
+@app.post("/a2a")
+async def a2a_endpoint(request: Request):
+    ctype = request.headers.get("content-type", "")
+    if "application/json" in ctype:
+        body = await request.json()
+        msg = body.get("message") or body.get("text") or str(body)
+        return JSONResponse(hello_agent(msg))
+    text = (await request.body()).decode("utf-8", errors="ignore")
+    return JSONResponse(hello_agent(text))
+