Skip to content

FEAT: Add A2ATarget for Agent-to-Agent protocol communication #2771

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
Radoslaw Brus (rbrus) wants to merge 6 commits into
base: main
Choose a base branch
from
Open

FEAT: Add A2ATarget for Agent-to-Agent protocol communication #2771

Show file tree
Hide file tree
Changes from all commits
Commits
Show all changes
6 commits
Select commit Hold shift + click to select a range
2e18344
feat: Add A2ATarget for Agent-to-Agent protocol communication
rbrus Sep 22, 2026
6dac262
docs(targets): document A2ATarget in prompt targets overview table
rbrus Sep 23, 2026
439d91f
FIX: A2ATarget context continuity, task polling, retries and error re…
rbrus Sep 24, 2026
f8a19ee
Merge branch 'main' into feat/a2a-prompt-target
rbrus Sep 24, 2026
0f76903
FEAT: Refactor A2ATarget to use a2a-sdk with safe polling and lifecyc…
rbrus Sep 26, 2026
dd9aceb
Merge branch 'main' into feat/a2a-prompt-target
rbrus Sep 26, 2026
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
1 change: 1 addition & 0 deletions doc/code/targets/0_prompt_targets.md
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -73,6 +73,7 @@ Here are some examples:
| **OpenAIChatTarget** (e.g., GPT-4) | **Yes** (multi-turn + editable history) | Designed for conversational prompts (system messages, conversation history, etc.). |
| **OpenAIImageTarget** | **No** | Used for image generation; does not manage conversation history. |
| **HTTPTarget** | **No** | Generic HTTP target. Some apps might allow conversation history, but this target doesn't handle it. |
| **A2ATarget** | **Yes** (per-conversation A2A context) | Target for Agent-to-Agent (A2A) protocol endpoints using the official a2a-sdk (v0.3 / v1.0). |
| **AzureBlobStorageTarget** | **No** | Used primarily for storage; not for conversation-based AI. |

## Target Capabilities
Expand Down
5 changes: 5 additions & 0 deletions pyproject.toml
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -136,8 +136,13 @@ litellm = [
"litellm>=1.84.0",
]

a2a = [
"a2a-sdk>=1.1.0",
]

# all includes all functional dependencies excluding the ones from the "dev" dependency group
all = [
"a2a-sdk>=1.1.0",
"accelerate>=1.7.0",
"azure-ai-ml>=1.32.0",
"azure-cognitiveservices-speech>=1.44.0",
Expand Down
2 changes: 2 additions & 0 deletions pyrit/prompt_target/__init__.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -14,6 +14,7 @@
from pyrit.common.lazy_imports import get_lazy_dir, resolve_lazy_export

if TYPE_CHECKING:
from pyrit.prompt_target.a2a_target import A2ATarget
from pyrit.prompt_target.azure_blob_storage_target import AzureBlobStorageTarget
from pyrit.prompt_target.azure_ml_chat_target import AzureMLChatTarget
from pyrit.prompt_target.common.conversation_normalization_pipeline import ConversationNormalizationPipeline
Expand Down Expand Up @@ -73,6 +74,7 @@
from pyrit.prompt_target.websocket_target import WebsocketTarget

_LAZY_EXPORTS: dict[str, str | tuple[str, str | None]] = {
"A2ATarget": "pyrit.prompt_target.a2a_target",
"TargetTraceConfig": "pyrit.prompt_target.common.target_trace_config",
"AzureBlobStorageTarget": "pyrit.prompt_target.azure_blob_storage_target",
"AzureMLChatTarget": "pyrit.prompt_target.azure_ml_chat_target",
Expand Down
Loading