Skip to content

Commit 2851a87

Browse files
Update src/a2a/client/transports/jsonrpc.py
Co-authored-by: gemini-code-assist[bot] <176961590+gemini-code-assist[bot]@users.noreply.github.com>
1 parent d78503b commit 2851a87

File tree

1 file changed

+4
-4
lines changed

1 file changed

+4
-4
lines changed

src/a2a/client/transports/jsonrpc.py

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -147,10 +147,10 @@ async def send_message_streaming(
147147
context,
148148
)
149149

150-
modified_kwargs['timeout'] = self.httpx_client.timeout.as_dict().get(
151-
'read', None
152-
)
153-
modified_kwargs['headers'] = dict(self.httpx_client.headers.items())
150+
modified_kwargs.setdefault('timeout', self.httpx_client.timeout.as_dict().get('read', None))
151+
headers = dict(self.httpx_client.headers.items())
152+
headers.update(modified_kwargs.get('headers', {}))
153+
modified_kwargs['headers'] = headers
154154

155155
async with aconnect_sse(
156156
self.httpx_client,

0 commit comments

Comments
 (0)