Skip to content

Commit 07427dd

Browse files
committed
feat(vefaas): update remoteveagent
1 parent 3740e07 commit 07427dd

1 file changed

Lines changed: 5 additions & 2 deletions

File tree

veadk/a2a/remote_ve_agent.py

Lines changed: 5 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -314,10 +314,13 @@ async def _pre_run(self, ctx: InvocationContext) -> None:
314314
else str(uuid.uuid4())
315315
)
316316
x_session_id = f"{parent_name}_{user_id}_{session_id}"
317+
x_session_id_key = "x-session-id-veadk"
317318
self._a2a_client._transport.httpx_client.headers.update(
318-
{"x-session-id-veadk": x_session_id}
319+
{x_session_id_key: x_session_id}
320+
)
321+
logger.debug(
322+
f"a2a client inject {x_session_id_key} to header: {x_session_id}"
319323
)
320-
logger.debug(f"Auto-injected session_id header: {x_session_id}")
321324

322325
async def _inject_auth_token(self, ctx: InvocationContext) -> None:
323326
"""Inject authentication token from credential service into the HTTP client.

0 commit comments

Comments
 (0)