adding agents instrumentation to configure_azure_monitor

sdk/ai/azure-ai-projects/samples/agents/sample_agents_basics_with_azure_monitor_tracing.py

@@ -51,7 +51,7 @@
 
 with tracer.start_as_current_span(scenario):
     with project_client:
-        project_client.telemetry.enable()
+        #project_client.telemetry.enable()
         # [END enable_tracing]
         agent = project_client.agents.create_agent(
             model=os.environ["MODEL_DEPLOYMENT_NAME"], name="my-assistant", instructions="You are helpful assistant"
@@ -62,7 +62,7 @@
         print(f"Created thread, thread ID: {thread.id}")
 
         message = project_client.agents.create_message(
-            thread_id=thread.id, role="user", content="Hello, tell me a joke"
+            thread_id=thread.id, role="user", content="Hello, tell me a hilarious joke"
         )
         print(f"Created message, message ID: {message.id}")
 

sdk/monitor/azure-monitor-opentelemetry/CHANGELOG.md

@@ -4,6 +4,9 @@
 
 ### Features Added
 
+- Enable Azure AI Agents instrumentation
+    ([#40043](https://github.com/Azure/azure-sdk-for-python/pull/40043))
+
 ### Breaking Changes
 
 ### Bugs Fixed

sdk/monitor/azure-monitor-opentelemetry/azure/monitor/opentelemetry/_configure.py

@@ -269,20 +269,42 @@ def _setup_additional_azure_sdk_instrumentations(configurations: Dict[str, Confi
         _logger.debug("Instrumentation skipped for library azure_sdk")
         return
 
+    ai_inference_instrumentor_imported = False
     try:
         from azure.ai.inference.tracing import AIInferenceInstrumentor  # pylint: disable=import-error,no-name-in-module
+        ai_inference_instrumentor_imported = True
     except Exception as ex:  # pylint: disable=broad-except
         _logger.debug(
             "Failed to import AIInferenceInstrumentor from azure-ai-inference",
             exc_info=ex,
         )
-        return
 
+    if ai_inference_instrumentor_imported:
+        try:
+            AIInferenceInstrumentor().instrument()
+        except Exception as ex:  # pylint: disable=broad-except
+            _logger.warning(
+                "Exception occurred when instrumenting: %s.",
+                "azure-ai-inference",
+                exc_info=ex,
+            )
+
+    ai_agents_instrumentor_imported = False
     try:
-        AIInferenceInstrumentor().instrument()
+        from azure.ai.projects.telemetry.agents import AIAgentsInstrumentor  # pylint: disable=import-error,no-name-in-module
+        ai_agents_instrumentor_imported = True
     except Exception as ex:  # pylint: disable=broad-except
-        _logger.warning(
-            "Exception occurred when instrumenting: %s.",
-            "azure-ai-inference",
+        _logger.debug(
+            "Failed to import AIAgentsInstrumentor from azure-ai-projects-telemetry-agents",
             exc_info=ex,
         )
+
+    if ai_agents_instrumentor_imported:
+        try:
+            AIAgentsInstrumentor().instrument()
+        except Exception as ex:  # pylint: disable=broad-except
+            _logger.warning(
+                "Exception occurred when instrumenting: %s.",
+                "azure-ai-projects-telemetry-agents",
+                exc_info=ex,
+            )