Skip to content

Commit c452192

Browse files
committed
refactor(client)!: reorganize ClientFactory API
1 parent a61f6d4 commit c452192

6 files changed

Lines changed: 291 additions & 195 deletions

File tree

itk/main.py

Lines changed: 2 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -12,7 +12,7 @@
1212

1313
from pyproto import instruction_pb2
1414

15-
from a2a.client import ClientConfig, ClientFactory
15+
from a2a.client import ClientConfig, create_client
1616
from a2a.compat.v0_3 import a2a_v0_3_pb2_grpc
1717
from a2a.compat.v0_3.grpc_handler import CompatGrpcHandler
1818
from a2a.server.agent_execution import AgentExecutor, RequestContext
@@ -128,10 +128,7 @@ async def handle_call_agent(call: instruction_pb2.CallAgent) -> list[str]:
128128
)
129129

130130
try:
131-
client = await ClientFactory.connect(
132-
call.agent_card_uri,
133-
client_config=config,
134-
)
131+
client = await create_client(call.agent_card_uri, client_config=config)
135132

136133
# Wrap nested instruction
137134
async with client:

samples/cli.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -9,7 +9,7 @@
99
import grpc
1010
import httpx
1111

12-
from a2a.client import A2ACardResolver, ClientConfig, ClientFactory
12+
from a2a.client import A2ACardResolver, ClientConfig, create_client
1313
from a2a.types import Message, Part, Role, SendMessageRequest, TaskState
1414

1515

@@ -79,7 +79,7 @@ async def main() -> None:
7979
print('\n✓ Agent Card Found:')
8080
print(f' Name: {card.name}')
8181

82-
client = await ClientFactory.connect(card, client_config=config)
82+
client = await create_client(card, client_config=config)
8383

8484
actual_transport = getattr(client, '_transport', client)
8585
print(f' Picked Transport: {actual_transport.__class__.__name__}')

src/a2a/client/__init__.py

Lines changed: 6 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -12,7 +12,11 @@
1212
ClientCallContext,
1313
ClientConfig,
1414
)
15-
from a2a.client.client_factory import ClientFactory, minimal_agent_card
15+
from a2a.client.client_factory import (
16+
ClientFactory,
17+
create_client,
18+
minimal_agent_card,
19+
)
1620
from a2a.client.errors import (
1721
A2AClientError,
1822
A2AClientTimeoutError,
@@ -36,6 +40,7 @@
3640
'ClientFactory',
3741
'CredentialService',
3842
'InMemoryContextCredentialStore',
43+
'create_client',
3944
'create_text_message_object',
4045
'minimal_agent_card',
4146
]

0 commit comments

Comments
 (0)