Skip to content

Commit 78a6df8

Browse files
authored
Merge pull request #110 from pdettori/change-trunc
🌱 add logging for auth header in weather agent
2 parents b0d2ff0 + 2c751e8 commit 78a6df8

File tree

1 file changed

+16
-2
lines changed
  • a2a/weather_service/src/weather_service

1 file changed

+16
-2
lines changed

β€Ža2a/weather_service/src/weather_service/agent.pyβ€Ž

Lines changed: 16 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -94,6 +94,10 @@ async def execute(self, context: RequestContext, event_queue: EventQueue):
9494
"""
9595
The agent allows to retrieve weather info through a natural language conversational interface
9696
"""
97+
98+
# Log authorization header for debugging
99+
auth_header = context.call_context.get("authorization") if hasattr(context, 'call_context') else None
100+
logger.info(f"πŸ” Authorization header received: {auth_header[:50] + '...' if auth_header and len(auth_header) > 50 else auth_header}")
97101

98102
# Setup Event Emitter
99103
task = context.current_task
@@ -165,5 +169,15 @@ def run():
165169
agent_card=agent_card,
166170
http_handler=request_handler,
167171
)
168-
169-
uvicorn.run(server.build(), host="0.0.0.0", port=8000)
172+
173+
# Add middleware to log all incoming requests with headers
174+
app = server.build()
175+
176+
@app.middleware("http")
177+
async def log_authorization_header(request, call_next):
178+
auth_header = request.headers.get("authorization", "No Authorization header")
179+
logger.info(f"πŸ” Incoming request to {request.url.path} with Authorization: {auth_header[:80] + '...' if len(auth_header) > 80 else auth_header}")
180+
response = await call_next(request)
181+
return response
182+
183+
uvicorn.run(app, host="0.0.0.0", port=8000)

0 commit comments

Comments
Β (0)