add turn id to agent span attrs

This commit is contained in:
Dinesh Yeduguru 2024-12-03 16:27:16 -08:00
parent 5d0c502cdb
commit 32af1f9dd4

View file

@ -169,6 +169,7 @@ class ChatAgent(ShieldRunnerMixin):
messages.extend(request.messages) messages.extend(request.messages)
turn_id = str(uuid.uuid4()) turn_id = str(uuid.uuid4())
span.set_attribute("turn_id", turn_id)
start_time = datetime.now() start_time = datetime.now()
yield AgentTurnResponseStreamChunk( yield AgentTurnResponseStreamChunk(
event=AgentTurnResponseEvent( event=AgentTurnResponseEvent(