Skip to content

Commit ed827d3

Browse files
committed
fix
1 parent 044e4bd commit ed827d3

File tree

3 files changed

+5
-3
lines changed

3 files changed

+5
-3
lines changed

veadk/a2a/remote_ve_agent.py

Lines changed: 4 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -312,8 +312,10 @@ async def _inject_auth_token(self, ctx: InvocationContext) -> None:
312312
callback_context=CallbackContext(ctx),
313313
)
314314

315-
if headers := generate_headers(tip_credential):
316-
self._a2a_client._transport.httpx_client.headers.update(headers)
315+
if tip_credential:
316+
self._a2a_client._transport.httpx_client.headers.update(
317+
{VE_TIP_TOKEN_HEADER: tip_credential.api_key}
318+
)
317319
logger.debug(
318320
f"Injected TIP token via header for app={ctx.app_name}, user={ctx.user_id}"
319321
)

veadk/configs/auth_configs.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -21,6 +21,7 @@
2121
# For runtime, use Any to avoid circular import issues
2222
IdentityClient = Any
2323

24+
2425
class VeIdentityConfig(BaseSettings):
2526
"""Configuration for VolcEngine Identity Service.
2627

veadk/utils/auth.py

Lines changed: 0 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -31,7 +31,6 @@
3131
HttpAuth,
3232
HttpCredentials,
3333
)
34-
from google.adk.auth.auth_schemes import AuthSchemeType
3534
from fastapi.openapi.models import APIKey, HTTPBearer, APIKeyIn
3635
from google.adk.auth.auth_tool import AuthConfig
3736

0 commit comments

Comments
 (0)