Skip to content

feat(a2a): updates advanced server customization example code to demonstrate passing additional keyword arguments to FastAPI and Starlette constructors #342

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Open
wants to merge 1 commit into
base: main
Choose a base branch
from
Open

feat(a2a): updates advanced server customization example code to demonstrate passing additional keyword arguments to FastAPI and Starlette constructors #342

Changes from all commits
Commits
Show all changes
1 commit
Select commit
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
14 changes: 12 additions & 2 deletions docs/user-guide/concepts/multi-agent/agent-to-agent.md
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -86,6 +86,7 @@ The `A2AServer` constructor accepts several configuration options:
The `A2AServer` provides access to the underlying FastAPI or Starlette application objects allowing you to further customize server behavior.

```python
from contextlib import asynccontextmanager
from strands import Agent
from strands.multiagent.a2a import A2AServer
import uvicorn
Expand All @@ -94,13 +95,22 @@ import uvicorn
agent = Agent(name="My Agent", description="A customizable agent", callback_handler=None)
a2a_server = A2AServer(agent=agent)

@asynccontextmanager
async def lifespan(app: FastAPI):
"""Manage application lifespan with proper error handling."""
# Startup tasks
yield # Application runs here
# Shutdown tasks

# Access the underlying FastAPI app
fastapi_app = a2a_server.to_fastapi_app()
# Allows passing keyword arguments to FastAPI constructor for further customization
fastapi_app = a2a_server.to_fastapi_app(lifespan=lifespan)
# Add custom middleware, routes, or configuration
fastapi_app.add_middleware(...)

# Or access the Starlette app
starlette_app = a2a_server.to_starlette_app()
# Allows passing keyword arguments to FastAPI constructor for further customization
starlette_app = a2a_server.to_starlette_app(lifespan=lifespan)
# Customize as needed

# You can then serve the customized app directly
Expand Down