Skip to content

Commit 07f3ca6

Browse files
authored
adding instrumentation to create_thread_and_run (#41225)
* adding instrumentation to create_thread_and_run * removing empty sections from changelog since those prevent green build
1 parent c5db108 commit 07f3ca6

File tree

2 files changed

+22
-4
lines changed

2 files changed

+22
-4
lines changed

sdk/ai/azure-ai-agents/CHANGELOG.md

Lines changed: 2 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -2,12 +2,10 @@
22

33
## 1.1.0b1 (2025-05-20)
44

5-
### Breaking Changes
6-
7-
### Features Added
8-
95
### Bugs Fixed
106

7+
* Adding instrumentation for create_thread_and_run
8+
119
## 1.0.0 (2025-05-15)
1210

1311
### Features Added

sdk/ai/azure-ai-agents/azure/ai/agents/telemetry/_ai_agents_instrumentor.py

Lines changed: 20 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -1472,6 +1472,9 @@ def inner(*args, **kwargs): # pylint: disable=R0911
14721472
if class_function_name.startswith("RunsOperations.create_and_process"):
14731473
kwargs.setdefault("merge_span", True)
14741474
return self.trace_create_run(OperationName.PROCESS_THREAD_RUN, function, *args, **kwargs)
1475+
if class_function_name.startswith("AgentsClient.create_thread_and_run"):
1476+
kwargs.setdefault("merge_span", True)
1477+
return self.trace_create_run(OperationName.PROCESS_THREAD_RUN, function, *args, **kwargs)
14751478
if class_function_name.startswith("RunsOperations.submit_tool_outputs"):
14761479
kwargs.setdefault("merge_span", True)
14771480
return self.trace_submit_tool_outputs(False, function, *args, **kwargs)
@@ -1545,6 +1548,9 @@ async def inner(*args, **kwargs): # pylint: disable=R0911
15451548
if class_function_name.startswith("RunsOperations.create_and_process"):
15461549
kwargs.setdefault("merge_span", True)
15471550
return await self.trace_create_run_async(OperationName.PROCESS_THREAD_RUN, function, *args, **kwargs)
1551+
if class_function_name.startswith("AgentsClient.create_thread_and_run"):
1552+
kwargs.setdefault("merge_span", True)
1553+
return await self.trace_create_run_async(OperationName.PROCESS_THREAD_RUN, function, *args, **kwargs)
15481554
if class_function_name.startswith("RunsOperations.submit_tool_outputs"):
15491555
kwargs.setdefault("merge_span", True)
15501556
return await self.trace_submit_tool_outputs_async(False, function, *args, **kwargs)
@@ -1690,6 +1696,13 @@ def _agents_apis(self):
16901696
TraceType.AGENTS,
16911697
"stream",
16921698
),
1699+
(
1700+
"azure.ai.agents",
1701+
"AgentsClient",
1702+
"create_thread_and_run",
1703+
TraceType.AGENTS,
1704+
"create_thread_and_run",
1705+
),
16931706
# Switching off the instrumentation for list method as it requires
16941707
# monkey patching inside pageable class.
16951708
(
@@ -1785,6 +1798,13 @@ def _agents_apis(self):
17851798
TraceType.AGENTS,
17861799
"stream",
17871800
),
1801+
(
1802+
"azure.ai.agents.aio",
1803+
"AgentsClient",
1804+
"create_thread_and_run",
1805+
TraceType.AGENTS,
1806+
"create_thread_and_run",
1807+
),
17881808
# Switching off the instrumentation for list method as it requires
17891809
# monkey patching inside async pageable class.
17901810
(

0 commit comments

Comments
 (0)