Add debug trace endpoint in api server

Details:
- Add a in-memory SpanExporter to capture all trace information.
- Add /debug/trace/session/{session_id} endpoint to retrieve traces from the in-memory exporter.
- Add Session ID in Telemetry spans.

PiperOrigin-RevId: 757984565
This commit is contained in:
Yifan Wang
2025-05-12 17:56:20 -07:00
committed by Copybara-Service
parent d35b99e6dd
commit 80813a75cf
2 changed files with 61 additions and 2 deletions

View File

@@ -111,6 +111,9 @@ def trace_call_llm(
span.set_attribute(
'gcp.vertex.agent.invocation_id', invocation_context.invocation_id
)
span.set_attribute(
'gcp.vertex.agent.session_id', invocation_context.session.id
)
span.set_attribute('gcp.vertex.agent.event_id', event_id)
# Consider removing once GenAI SDK provides a way to record this info.
span.set_attribute(