Skip to content

Commit a96d3db

Browse files
committed
add agent id to trace context
1 parent 5492d7a commit a96d3db

File tree

1 file changed

+10
-0
lines changed
  • sdk/ai/azure-ai-agentserver-core/azure/ai/agentserver/core/server

1 file changed

+10
-0
lines changed

sdk/ai/azure-ai-agentserver-core/azure/ai/agentserver/core/server/base.py

Lines changed: 10 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -60,10 +60,20 @@ def set_request_id_to_context_var(self, request):
6060
request_context.set(ctx)
6161

6262
def set_run_context_to_context_var(self, run_context):
63+
agent_id = ""
64+
agent_obj = run_context.get_agent_id_object()
65+
if agent_obj:
66+
agent_name = getattr(agent_obj, "name", "")
67+
agent_version = getattr(agent_obj, "version", "")
68+
agent_id = f"{agent_name}:{agent_version}"
69+
6370
res = {
6471
"azure.ai.agentshosting.response_id": run_context.response_id or "",
6572
"azure.ai.agentshosting.conversation_id": run_context.conversation_id or "",
6673
"azure.ai.agentshosting.streaming": str(run_context.stream or False),
74+
"gen_ai.agent.id": agent_id,
75+
"gen_ai.provider.name": "AzureAI Hosted Agents",
76+
"gen_ai.response.id": run_context.response_id or "",
6777
}
6878
ctx = request_context.get() or {}
6979
ctx.update(res)

0 commit comments

Comments
 (0)