@@ -1472,6 +1472,9 @@ def inner(*args, **kwargs): # pylint: disable=R0911
1472
1472
if class_function_name .startswith ("RunsOperations.create_and_process" ):
1473
1473
kwargs .setdefault ("merge_span" , True )
1474
1474
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 )
1475
1478
if class_function_name .startswith ("RunsOperations.submit_tool_outputs" ):
1476
1479
kwargs .setdefault ("merge_span" , True )
1477
1480
return self .trace_submit_tool_outputs (False , function , * args , ** kwargs )
@@ -1545,6 +1548,9 @@ async def inner(*args, **kwargs): # pylint: disable=R0911
1545
1548
if class_function_name .startswith ("RunsOperations.create_and_process" ):
1546
1549
kwargs .setdefault ("merge_span" , True )
1547
1550
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 )
1548
1554
if class_function_name .startswith ("RunsOperations.submit_tool_outputs" ):
1549
1555
kwargs .setdefault ("merge_span" , True )
1550
1556
return await self .trace_submit_tool_outputs_async (False , function , * args , ** kwargs )
@@ -1690,6 +1696,13 @@ def _agents_apis(self):
1690
1696
TraceType .AGENTS ,
1691
1697
"stream" ,
1692
1698
),
1699
+ (
1700
+ "azure.ai.agents" ,
1701
+ "AgentsClient" ,
1702
+ "create_thread_and_run" ,
1703
+ TraceType .AGENTS ,
1704
+ "create_thread_and_run" ,
1705
+ ),
1693
1706
# Switching off the instrumentation for list method as it requires
1694
1707
# monkey patching inside pageable class.
1695
1708
(
@@ -1785,6 +1798,13 @@ def _agents_apis(self):
1785
1798
TraceType .AGENTS ,
1786
1799
"stream" ,
1787
1800
),
1801
+ (
1802
+ "azure.ai.agents.aio" ,
1803
+ "AgentsClient" ,
1804
+ "create_thread_and_run" ,
1805
+ TraceType .AGENTS ,
1806
+ "create_thread_and_run" ,
1807
+ ),
1788
1808
# Switching off the instrumentation for list method as it requires
1789
1809
# monkey patching inside async pageable class.
1790
1810
(
0 commit comments