agent/agent-framework
2
0
Fork 0
mirror of https://github.com/microsoft/agent-framework.git synced 2026-06-16 21:04:09 +08:00
Code Issues Packages Projects Releases Wiki Activity

Python: A2A AuthInterceptor Support (#1317)

Browse Source 
* a2a authinterceptor support

* Update python/packages/a2a/agent_framework_a2a/_agent.py

Co-authored-by: Copilot <175728472+Copilot@users.noreply.github.com>

* small fix

---------

Co-authored-by: Copilot <175728472+Copilot@users.noreply.github.com>
This commit is contained in:
Giles Odigwe
2025-10-09 11:56:54 -07:00
committed by GitHub
Unverified
parent 9fe4a61dd0
commit 2f53ce4abd
2 changed files with 23 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files
python/packages/a2a/agent_framework_a2a/_agent.py
+5 -1
View File
@@ -9,6 +9,7 @@ from typing import Any, cast
import httpx
from a2a.client import Client, ClientConfig, ClientFactory, minimal_agent_card
from a2a.client.auth.interceptor import AuthInterceptor
from a2a.types import (
AgentCard,
Artifact,
@@ -78,6 +79,7 @@ class A2AAgent(BaseAgent):
url: str | None = None,
client: Client | None = None,
http_client: httpx.AsyncClient | None = None,
auth_interceptor: AuthInterceptor | None = None,
**kwargs: Any,
) -> None:
"""Initialize the A2AAgent.
@@ -90,6 +92,7 @@ class A2AAgent(BaseAgent):
url: The URL for the A2A server.
client: The A2A client for the agent.
http_client: Optional httpx.AsyncClient to use.
auth_interceptor: Optional authentication interceptor for secured endpoints.
kwargs: any additional properties, passed to BaseAgent.
"""
super().__init__(id=id, name=name, description=description, **kwargs)
@@ -123,7 +126,8 @@ class A2AAgent(BaseAgent):
supported_transports=[TransportProtocol.jsonrpc],
)
factory = ClientFactory(config)
self.client = factory.create(agent_card)
interceptors = [auth_interceptor] if auth_interceptor is not None else None
self.client = factory.create(agent_card, interceptors=interceptors) # type: ignore
async def __aenter__(self) -> "A2AAgent":
"""Async context manager entry."""