fix: Replace the track_event function with track_event_if_configured across various files in the backend (microsoft#58)

Roopan-Microsoft · Prashant-Microsoft · Kiran-Siluveru-Microsoft · web-flow · commit 8f5bc032ecea · 2025-01-28T15:29:34.000+05:30
* fix: ui changes (microsoft#1) * fix: in progress status color after fetch task details (microsoft#5) * cancel notification message updated (microsoft#7) * Update task.js (microsoft#9) * Stages overflow issue fix (microsoft#10) * fix: added space to the agent (microsoft#13) * Approve reject buttons titles disabling buttons and (microsoft#15) * Fix: UX becomes damaged when chat outputs sample code for a task (microsoft#14) * task page UI updates * UI updated code for task * Task page UI updated code * status section UI update in task page * Added custom event (microsoft#24) * feat: added custom event * Logs updated * modify code * added exception logs * added exception logs for cosmos memory --------- Co-authored-by: Roopan P M <v-roopanpm@microsoft.com> * fix: task with zero stages cannot show the page, spins forever and rai test prompt (microsoft#41) * Updated the workflow for build and push docker * updated the repo name * Update requirements.txt * Update requirements.txt * Update requirements.txt * feat: Integrated application insights instrumentation key into the bicep file (microsoft#42) * feat: Integrated application insights instrumentation key into bicep files * added application insights instrumentation key into env and readme file * updated json file * upgraded json file * Update docker-build-and-push.yml * Update docker-build-and-push.yml * Update docker-build-and-push.yml * Update docker-build-and-push.yml * Update docker-build-and-push.yml * Update docker-build-and-push.yml * pyLint issues fixed * lint issues fixed --------- Co-authored-by: Roopan-Microsoft <168007406+Roopan-Microsoft@users.noreply.github.com> Co-authored-by: Roopan P M <v-roopanpm@microsoft.com> * Update test.yml (microsoft#43) * Disabling Text Area functionality (microsoft#47) * fix: Usability and Alignments changes (microsoft#48) * Name update and padding removed * fix home page padding and cards height * remove gap in tasks section * Update docker-build-and-push.yml to debug * Update docker-build-and-push.yml * Update docker-build-and-push.yml to take the correct event name * Update docker-build-and-push.yml * fix: text area background color (microsoft#50) * fix: text area background color * Cursor hover style and Light dark color updated * feat: customize track events (microsoft#54) * feat: customized track event * pylint fix * pylint fix --------- Co-authored-by: Prashant-Microsoft <v-pmalusare@microsoft.com> Co-authored-by: Kiran-Siluveru-Microsoft <v-ksiluveru@microsoft.com> Co-authored-by: Mohan-Microsoft <v-mvenudass@microsoft.com> Co-authored-by: Harmanpreet-Microsoft <v-harmanprka@microsoft.com>
diff --git a/src/backend/agents/base_agent.py b/src/backend/agents/base_agent.py
@@ -21,7 +21,7 @@
     Step,
     StepStatus,
 )
-from azure.monitor.events.extension import track_event
+from event_utils import track_event_if_configured
 
 
 class BaseAgent(RoutedAgent):
@@ -105,7 +105,7 @@ async def handle_action_request(
                 )
             )
 
-            track_event(
+            track_event_if_configured(
                 "Base agent - Added into the cosmos",
                 {
                     "session_id": message.session_id,
@@ -119,7 +119,7 @@ async def handle_action_request(
 
         except Exception as e:
             logging.exception(f"Error during LLM call: {e}")
-            track_event(
+            track_event_if_configured(
                 "Base agent - Error during llm call, captured into the cosmos",
                 {
                     "session_id": message.session_id,
@@ -138,7 +138,7 @@ async def handle_action_request(
         step.agent_reply = result
         await self._model_context.update_step(step)
 
-        track_event(
+        track_event_if_configured(
             "Base agent - Updated step and updated into the cosmos",
             {
                 "status": StepStatus.completed,
diff --git a/src/backend/agents/group_chat_manager.py b/src/backend/agents/group_chat_manager.py
@@ -22,7 +22,7 @@
     StepStatus,
 )
 
-from azure.monitor.events.extension import track_event
+from event_utils import track_event_if_configured
 
 
 @default_subscription
@@ -62,7 +62,7 @@ async def handle_input_task(
             )
         )
 
-        track_event(
+        track_event_if_configured(
             "Group Chat Manager - Received and added input task into the cosmos",
             {
                 "session_id": message.session_id,
@@ -164,7 +164,7 @@ class Step(BaseDataModel):
                     step.status = StepStatus.rejected
                     step.human_approval_status = HumanFeedbackStatus.rejected
                     self._memory.update_step(step)
-                    track_event(
+                    track_event_if_configured(
                         "Group Chat Manager - Steps has been rejected and updated into the cosmos",
                         {
                             "status": StepStatus.rejected,
@@ -188,7 +188,7 @@ class Step(BaseDataModel):
                     step.status = StepStatus.rejected
                     step.human_approval_status = HumanFeedbackStatus.rejected
                     self._memory.update_step(step)
-                    track_event(
+                    track_event_if_configured(
                         "Group Chat Manager - Step has been rejected and updated into the cosmos",
                         {
                             "status": StepStatus.rejected,
@@ -213,7 +213,7 @@ async def _update_step_status(
         step.human_feedback = received_human_feedback
         step.status = StepStatus.completed
         await self._memory.update_step(step)
-        track_event(
+        track_event_if_configured(
             "Group Chat Manager - Received human feedback, Updating step and updated into the cosmos",
             {
                 "status": StepStatus.completed,
@@ -241,7 +241,7 @@ async def _execute_step(self, session_id: str, step: Step):
         # Update step status to 'action_requested'
         step.status = StepStatus.action_requested
         await self._memory.update_step(step)
-        track_event(
+        track_event_if_configured(
             "Group Chat Manager - Update step to action_requested and updated into the cosmos",
             {
                 "status": StepStatus.action_requested,
@@ -304,7 +304,7 @@ async def _execute_step(self, session_id: str, step: Step):
             )
         )
 
-        track_event(
+        track_event_if_configured(
             f"Group Chat Manager - Requesting {step.agent.value.title()} to perform the action and added into the cosmos",
             {
                 "session_id": session_id,
@@ -338,7 +338,7 @@ async def _execute_step(self, session_id: str, step: Step):
             logging.info(
                 "Marking the step as complete - Since we have received the human feedback"
             )
-            track_event(
+            track_event_if_configured(
                 "Group Chat Manager - Steps completed - Received the human feedback and updated into the cosmos",
                 {
                     "session_id": session_id,
diff --git a/src/backend/agents/human.py b/src/backend/agents/human.py
@@ -12,7 +12,7 @@
     AgentMessage,
     Step,
 )
-from azure.monitor.events.extension import track_event
+from event_utils import track_event_if_configured
 
 
 @default_subscription
@@ -57,7 +57,7 @@ async def handle_step_feedback(
             )
         )
         logging.info(f"HumanAgent received feedback for step: {step}")
-        track_event(
+        track_event_if_configured(
             f"Human Agent - Received feedback for step: {step} and added into the cosmos",
             {
                 "session_id": message.session_id,
@@ -81,7 +81,7 @@ async def handle_step_feedback(
         )
         logging.info(f"HumanAgent sent approval request for step: {step}")
 
-        track_event(
+        track_event_if_configured(
             f"Human Agent - Approval request sent for step {step} and added into the cosmos",
             {
                 "session_id": message.session_id,
diff --git a/src/backend/agents/planner.py b/src/backend/agents/planner.py
@@ -26,7 +26,7 @@
     HumanFeedbackStatus,
 )
 
-from azure.monitor.events.extension import track_event
+from event_utils import track_event_if_configured
 
 
 @default_subscription
@@ -74,7 +74,7 @@ async def handle_input_task(self, message: InputTask, ctx: MessageContext) -> Pl
             )
             logging.info(f"Plan generated: {plan.summary}")
 
-            track_event(
+            track_event_if_configured(
                 f"Planner - Generated a plan with {len(steps)} steps and added plan into the cosmos",
                 {
                     "session_id": message.session_id,
@@ -101,7 +101,7 @@ async def handle_input_task(self, message: InputTask, ctx: MessageContext) -> Pl
                     f"Additional information requested: {plan.human_clarification_request}"
                 )
 
-                track_event(
+                track_event_if_configured(
                     "Planner - Additional information requested and added into the cosmos",
                     {
                         "session_id": message.session_id,
@@ -138,7 +138,7 @@ async def handle_plan_clarification(
             )
         )
 
-        track_event(
+        track_event_if_configured(
             "Planner - Store HumanAgent clarification and added into the cosmos",
             {
                 "session_id": message.session_id,
@@ -160,7 +160,7 @@ async def handle_plan_clarification(
         )
         logging.info("Plan updated with HumanClarification.")
 
-        track_event(
+        track_event_if_configured(
             "Planner - Updated with HumanClarification and added into the cosmos",
             {
                 "session_id": message.session_id,
@@ -254,7 +254,7 @@ class StructuredOutputPlan(BaseModel):
             structured_plan = StructuredOutputPlan(**parsed_result)
 
             if not structured_plan.steps:
-                track_event(
+                track_event_if_configured(
                     "Planner agent - No steps found",
                     {
                         "session_id": self._session_id,
@@ -282,7 +282,7 @@ class StructuredOutputPlan(BaseModel):
             # Store the plan in memory
             await self._memory.add_plan(plan)
 
-            track_event(
+            track_event_if_configured(
                 "Planner - Initial plan and added into the cosmos",
                 {
                     "session_id": self._session_id,
@@ -308,7 +308,7 @@ class StructuredOutputPlan(BaseModel):
                     human_approval_status=HumanFeedbackStatus.requested,
                 )
                 await self._memory.add_step(step)
-                track_event(
+                track_event_if_configured(
                     "Planner - Added planned individual step into the cosmos",
                     {
                         "plan_id": plan.id,
@@ -326,7 +326,7 @@ class StructuredOutputPlan(BaseModel):
 
         except Exception as e:
             logging.exception(f"Error in create_structured_plan: {e}")
-            track_event(
+            track_event_if_configured(
                 f"Planner - Error in create_structured_plan: {e} into the cosmos",
                 {
                     "session_id": self._session_id,
diff --git a/src/backend/app.py b/src/backend/app.py
@@ -20,13 +20,20 @@
     PlanWithSteps,
 )
 from utils import initialize_runtime_and_context, retrieve_all_agent_tools, rai_success
+from event_utils import track_event_if_configured
 from fastapi.middleware.cors import CORSMiddleware
 from azure.monitor.opentelemetry import configure_azure_monitor
-from azure.monitor.events.extension import track_event
 
-configure_azure_monitor(
-    connection_string=os.getenv("APPLICATIONINSIGHTS_INSTRUMENTATION_KEY")
-)
+
+# Check if the Application Insights Instrumentation Key is set in the environment variables
+instrumentation_key = os.getenv("APPLICATIONINSIGHTS_INSTRUMENTATION_KEY")
+if instrumentation_key:
+    # Configure Application Insights if the Instrumentation Key is found
+    configure_azure_monitor(connection_string=instrumentation_key)
+    logging.info("Application Insights configured with the provided Instrumentation Key")
+else:
+    # Log a warning if the Instrumentation Key is not found
+    logging.warning("No Application Insights Instrumentation Key found. Skipping configuration")
 
 # Configure logging
 logging.basicConfig(level=logging.INFO)
@@ -113,7 +120,7 @@ async def input_task_endpoint(input_task: InputTask, request: Request):
     if not rai_success(input_task.description):
         print("RAI failed")
 
-        track_event(
+        track_event_if_configured(
             "RAI failed",
             {
                 "status": "Plan not created",
@@ -129,7 +136,7 @@ async def input_task_endpoint(input_task: InputTask, request: Request):
     user_id = authenticated_user["user_principal_id"]
 
     if not user_id:
-        track_event("UserIdNotFound", {"status_code": 400, "detail": "no user"})
+        track_event_if_configured("UserIdNotFound", {"status_code": 400, "detail": "no user"})
 
         raise HTTPException(status_code=400, detail="no user")
     if not input_task.session_id:
@@ -150,7 +157,7 @@ async def input_task_endpoint(input_task: InputTask, request: Request):
     logging.info(f"Plan created: {plan.summary}")
 
     # Log custom event for successful input task processing
-    track_event(
+    track_event_if_configured(
         "InputTaskProcessed",
         {
             "status": f"Plan created:\n {plan.summary}"
@@ -231,7 +238,7 @@ async def human_feedback_endpoint(human_feedback: HumanFeedback, request: Reques
     authenticated_user = get_authenticated_user_details(request_headers=request.headers)
     user_id = authenticated_user["user_principal_id"]
     if not user_id:
-        track_event("UserIdNotFound", {"status_code": 400, "detail": "no user"})
+        track_event_if_configured("UserIdNotFound", {"status_code": 400, "detail": "no user"})
         raise HTTPException(status_code=400, detail="no user")
     # Initialize runtime and context
     runtime, _ = await initialize_runtime_and_context(
@@ -242,7 +249,7 @@ async def human_feedback_endpoint(human_feedback: HumanFeedback, request: Reques
     human_agent_id = AgentId("human_agent", human_feedback.session_id)
     await runtime.send_message(human_feedback, human_agent_id)
 
-    track_event(
+    track_event_if_configured(
         "Completed Feedback received",
         {
             "status": "Feedback received",
@@ -308,7 +315,7 @@ async def human_clarification_endpoint(
     authenticated_user = get_authenticated_user_details(request_headers=request.headers)
     user_id = authenticated_user["user_principal_id"]
     if not user_id:
-        track_event("UserIdNotFound", {"status_code": 400, "detail": "no user"})
+        track_event_if_configured("UserIdNotFound", {"status_code": 400, "detail": "no user"})
         raise HTTPException(status_code=400, detail="no user")
     # Initialize runtime and context
     runtime, _ = await initialize_runtime_and_context(
@@ -319,7 +326,7 @@ async def human_clarification_endpoint(
     planner_agent_id = AgentId("planner_agent", human_clarification.session_id)
     await runtime.send_message(human_clarification, planner_agent_id)
 
-    track_event(
+    track_event_if_configured(
         "Completed Human clarification on the plan",
         {
             "status": "Clarification received",
@@ -390,7 +397,7 @@ async def approve_step_endpoint(
     authenticated_user = get_authenticated_user_details(request_headers=request.headers)
     user_id = authenticated_user["user_principal_id"]
     if not user_id:
-        track_event("UserIdNotFound", {"status_code": 400, "detail": "no user"})
+        track_event_if_configured("UserIdNotFound", {"status_code": 400, "detail": "no user"})
         raise HTTPException(status_code=400, detail="no user")
     # Initialize runtime and context
     runtime, _ = await initialize_runtime_and_context(user_id=user_id)
@@ -405,7 +412,7 @@ async def approve_step_endpoint(
     )
     # Return a status message
     if human_feedback.step_id:
-        track_event(
+        track_event_if_configured(
             "Completed Human clarification with step_id",
             {
                 "status": f"Step {human_feedback.step_id} - Approval:{human_feedback.approved}."
@@ -416,7 +423,7 @@ async def approve_step_endpoint(
             "status": f"Step {human_feedback.step_id} - Approval:{human_feedback.approved}."
         }
     else:
-        track_event(
+        track_event_if_configured(
             "Completed Human clarification without step_id",
             {"status": "All steps approved"},
         )
@@ -488,15 +495,15 @@ async def get_plans(
     authenticated_user = get_authenticated_user_details(request_headers=request.headers)
     user_id = authenticated_user["user_principal_id"]
     if not user_id:
-        track_event("UserIdNotFound", {"status_code": 400, "detail": "no user"})
+        track_event_if_configured("UserIdNotFound", {"status_code": 400, "detail": "no user"})
         raise HTTPException(status_code=400, detail="no user")
 
     cosmos = CosmosBufferedChatCompletionContext(session_id or "", user_id)
 
     if session_id:
         plan = await cosmos.get_plan_by_session(session_id=session_id)
         if not plan:
-            track_event(
+            track_event_if_configured(
                 "GetPlanBySessionNotFound",
                 {"status_code": 400, "detail": "Plan not found"},
             )
@@ -576,7 +583,7 @@ async def get_steps_by_plan(plan_id: str, request: Request) -> List[Step]:
     authenticated_user = get_authenticated_user_details(request_headers=request.headers)
     user_id = authenticated_user["user_principal_id"]
     if not user_id:
-        track_event("UserIdNotFound", {"status_code": 400, "detail": "no user"})
+        track_event_if_configured("UserIdNotFound", {"status_code": 400, "detail": "no user"})
         raise HTTPException(status_code=400, detail="no user")
     cosmos = CosmosBufferedChatCompletionContext("", user_id)
     steps = await cosmos.get_steps_by_plan(plan_id=plan_id)
@@ -634,7 +641,7 @@ async def get_agent_messages(session_id: str, request: Request) -> List[AgentMes
     authenticated_user = get_authenticated_user_details(request_headers=request.headers)
     user_id = authenticated_user["user_principal_id"]
     if not user_id:
-        track_event("UserIdNotFound", {"status_code": 400, "detail": "no user"})
+        track_event_if_configured("UserIdNotFound", {"status_code": 400, "detail": "no user"})
         raise HTTPException(status_code=400, detail="no user")
     cosmos = CosmosBufferedChatCompletionContext(session_id, user_id)
     agent_messages = await cosmos.get_data_by_type("agent_message")
diff --git a/src/backend/event_utils.py b/src/backend/event_utils.py
@@ -0,0 +1,11 @@
+import logging
+import os
+from azure.monitor.events.extension import track_event
+
+
+def track_event_if_configured(event_name: str, event_data: dict):
+    instrumentation_key = os.getenv("APPLICATIONINSIGHTS_INSTRUMENTATION_KEY")
+    if instrumentation_key:
+        track_event(event_name, event_data)
+    else:
+        logging.warning(f"Skipping track_event for {event_name} as Application Insights is not configured")

Original file line number	Diff line number	Diff line change
`@@ -21,7 +21,7 @@`
`21`	`21`	`Step,`
`22`	`22`	`StepStatus,`
`23`	`23`	`)`
`24`		`-from azure.monitor.events.extension import track_event`
	`24`	`+from event_utils import track_event_if_configured`
`25`	`25`
`26`	`26`
`27`	`27`	`class BaseAgent(RoutedAgent):`
`@@ -105,7 +105,7 @@ async def handle_action_request(`
`105`	`105`	`)`
`106`	`106`	`)`
`107`	`107`
`108`		`- track_event(`
	`108`	`+ track_event_if_configured(`
`109`	`109`	`"Base agent - Added into the cosmos",`
`110`	`110`	`{`
`111`	`111`	`"session_id": message.session_id,`
`@@ -119,7 +119,7 @@ async def handle_action_request(`
`119`	`119`
`120`	`120`	`except Exception as e:`
`121`	`121`	`logging.exception(f"Error during LLM call: {e}")`
`122`		`- track_event(`
	`122`	`+ track_event_if_configured(`
`123`	`123`	`"Base agent - Error during llm call, captured into the cosmos",`
`124`	`124`	`{`
`125`	`125`	`"session_id": message.session_id,`
`@@ -138,7 +138,7 @@ async def handle_action_request(`
`138`	`138`	`step.agent_reply = result`
`139`	`139`	`await self._model_context.update_step(step)`
`140`	`140`
`141`		`- track_event(`
	`141`	`+ track_event_if_configured(`
`142`	`142`	`"Base agent - Updated step and updated into the cosmos",`
`143`	`143`	`{`
`144`	`144`	`"status": StepStatus.completed,`
Original file line number	Diff line number	Diff line change
`@@ -22,7 +22,7 @@`
`22`	`22`	`StepStatus,`
`23`	`23`	`)`
`24`	`24`
`25`		`-from azure.monitor.events.extension import track_event`
	`25`	`+from event_utils import track_event_if_configured`
`26`	`26`
`27`	`27`
`28`	`28`	`@default_subscription`
`@@ -62,7 +62,7 @@ async def handle_input_task(`
`62`	`62`	`)`
`63`	`63`	`)`
`64`	`64`
`65`		`- track_event(`
	`65`	`+ track_event_if_configured(`
`66`	`66`	`"Group Chat Manager - Received and added input task into the cosmos",`
`67`	`67`	`{`
`68`	`68`	`"session_id": message.session_id,`
`@@ -164,7 +164,7 @@ class Step(BaseDataModel):`
`164`	`164`	`step.status = StepStatus.rejected`
`165`	`165`	`step.human_approval_status = HumanFeedbackStatus.rejected`
`166`	`166`	`self._memory.update_step(step)`
`167`		`- track_event(`
	`167`	`+ track_event_if_configured(`
`168`	`168`	`"Group Chat Manager - Steps has been rejected and updated into the cosmos",`
`169`	`169`	`{`
`170`	`170`	`"status": StepStatus.rejected,`
`@@ -188,7 +188,7 @@ class Step(BaseDataModel):`
`188`	`188`	`step.status = StepStatus.rejected`
`189`	`189`	`step.human_approval_status = HumanFeedbackStatus.rejected`
`190`	`190`	`self._memory.update_step(step)`
`191`		`- track_event(`
	`191`	`+ track_event_if_configured(`
`192`	`192`	`"Group Chat Manager - Step has been rejected and updated into the cosmos",`
`193`	`193`	`{`
`194`	`194`	`"status": StepStatus.rejected,`
`@@ -213,7 +213,7 @@ async def _update_step_status(`
`213`	`213`	`step.human_feedback = received_human_feedback`
`214`	`214`	`step.status = StepStatus.completed`
`215`	`215`	`await self._memory.update_step(step)`
`216`		`- track_event(`
	`216`	`+ track_event_if_configured(`
`217`	`217`	`"Group Chat Manager - Received human feedback, Updating step and updated into the cosmos",`
`218`	`218`	`{`
`219`	`219`	`"status": StepStatus.completed,`
`@@ -241,7 +241,7 @@ async def _execute_step(self, session_id: str, step: Step):`
`241`	`241`	`# Update step status to 'action_requested'`
`242`	`242`	`step.status = StepStatus.action_requested`
`243`	`243`	`await self._memory.update_step(step)`
`244`		`- track_event(`
	`244`	`+ track_event_if_configured(`
`245`	`245`	`"Group Chat Manager - Update step to action_requested and updated into the cosmos",`
`246`	`246`	`{`
`247`	`247`	`"status": StepStatus.action_requested,`
`@@ -304,7 +304,7 @@ async def _execute_step(self, session_id: str, step: Step):`
`304`	`304`	`)`
`305`	`305`	`)`
`306`	`306`
`307`		`- track_event(`
	`307`	`+ track_event_if_configured(`
`308`	`308`	`f"Group Chat Manager - Requesting {step.agent.value.title()} to perform the action and added into the cosmos",`
`309`	`309`	`{`
`310`	`310`	`"session_id": session_id,`
`@@ -338,7 +338,7 @@ async def _execute_step(self, session_id: str, step: Step):`
`338`	`338`	`logging.info(`
`339`	`339`	`"Marking the step as complete - Since we have received the human feedback"`
`340`	`340`	`)`
`341`		`- track_event(`
	`341`	`+ track_event_if_configured(`
`342`	`342`	`"Group Chat Manager - Steps completed - Received the human feedback and updated into the cosmos",`
`343`	`343`	`{`
`344`	`344`	`"session_id": session_id,`
Original file line number	Diff line number	Diff line change
`@@ -12,7 +12,7 @@`
`12`	`12`	`AgentMessage,`
`13`	`13`	`Step,`
`14`	`14`	`)`
`15`		`-from azure.monitor.events.extension import track_event`
	`15`	`+from event_utils import track_event_if_configured`
`16`	`16`
`17`	`17`
`18`	`18`	`@default_subscription`
`@@ -57,7 +57,7 @@ async def handle_step_feedback(`
`57`	`57`	`)`
`58`	`58`	`)`
`59`	`59`	`logging.info(f"HumanAgent received feedback for step: {step}")`
`60`		`- track_event(`
	`60`	`+ track_event_if_configured(`
`61`	`61`	`f"Human Agent - Received feedback for step: {step} and added into the cosmos",`
`62`	`62`	`{`
`63`	`63`	`"session_id": message.session_id,`
`@@ -81,7 +81,7 @@ async def handle_step_feedback(`
`81`	`81`	`)`
`82`	`82`	`logging.info(f"HumanAgent sent approval request for step: {step}")`
`83`	`83`
`84`		`- track_event(`
	`84`	`+ track_event_if_configured(`
`85`	`85`	`f"Human Agent - Approval request sent for step {step} and added into the cosmos",`
`86`	`86`	`{`
`87`	`87`	`"session_id": message.session_id,`
Original file line number	Diff line number	Diff line change
`@@ -26,7 +26,7 @@`
`26`	`26`	`HumanFeedbackStatus,`
`27`	`27`	`)`
`28`	`28`
`29`		`-from azure.monitor.events.extension import track_event`
	`29`	`+from event_utils import track_event_if_configured`
`30`	`30`
`31`	`31`
`32`	`32`	`@default_subscription`
`@@ -74,7 +74,7 @@ async def handle_input_task(self, message: InputTask, ctx: MessageContext) -> Pl`
`74`	`74`	`)`
`75`	`75`	`logging.info(f"Plan generated: {plan.summary}")`
`76`	`76`
`77`		`- track_event(`
	`77`	`+ track_event_if_configured(`
`78`	`78`	`f"Planner - Generated a plan with {len(steps)} steps and added plan into the cosmos",`
`79`	`79`	`{`
`80`	`80`	`"session_id": message.session_id,`
`@@ -101,7 +101,7 @@ async def handle_input_task(self, message: InputTask, ctx: MessageContext) -> Pl`
`101`	`101`	`f"Additional information requested: {plan.human_clarification_request}"`
`102`	`102`	`)`
`103`	`103`
`104`		`- track_event(`
	`104`	`+ track_event_if_configured(`
`105`	`105`	`"Planner - Additional information requested and added into the cosmos",`
`106`	`106`	`{`
`107`	`107`	`"session_id": message.session_id,`
`@@ -138,7 +138,7 @@ async def handle_plan_clarification(`
`138`	`138`	`)`
`139`	`139`	`)`
`140`	`140`
`141`		`- track_event(`
	`141`	`+ track_event_if_configured(`
`142`	`142`	`"Planner - Store HumanAgent clarification and added into the cosmos",`
`143`	`143`	`{`
`144`	`144`	`"session_id": message.session_id,`
`@@ -160,7 +160,7 @@ async def handle_plan_clarification(`
`160`	`160`	`)`
`161`	`161`	`logging.info("Plan updated with HumanClarification.")`
`162`	`162`
`163`		`- track_event(`
	`163`	`+ track_event_if_configured(`
`164`	`164`	`"Planner - Updated with HumanClarification and added into the cosmos",`
`165`	`165`	`{`
`166`	`166`	`"session_id": message.session_id,`
`@@ -254,7 +254,7 @@ class StructuredOutputPlan(BaseModel):`
`254`	`254`	`structured_plan = StructuredOutputPlan(**parsed_result)`
`255`	`255`
`256`	`256`	`if not structured_plan.steps:`
`257`		`- track_event(`
	`257`	`+ track_event_if_configured(`
`258`	`258`	`"Planner agent - No steps found",`
`259`	`259`	`{`
`260`	`260`	`"session_id": self._session_id,`
`@@ -282,7 +282,7 @@ class StructuredOutputPlan(BaseModel):`
`282`	`282`	`# Store the plan in memory`
`283`	`283`	`await self._memory.add_plan(plan)`
`284`	`284`
`285`		`- track_event(`
	`285`	`+ track_event_if_configured(`
`286`	`286`	`"Planner - Initial plan and added into the cosmos",`
`287`	`287`	`{`
`288`	`288`	`"session_id": self._session_id,`
`@@ -308,7 +308,7 @@ class StructuredOutputPlan(BaseModel):`
`308`	`308`	`human_approval_status=HumanFeedbackStatus.requested,`
`309`	`309`	`)`
`310`	`310`	`await self._memory.add_step(step)`
`311`		`- track_event(`
	`311`	`+ track_event_if_configured(`
`312`	`312`	`"Planner - Added planned individual step into the cosmos",`
`313`	`313`	`{`
`314`	`314`	`"plan_id": plan.id,`
`@@ -326,7 +326,7 @@ class StructuredOutputPlan(BaseModel):`
`326`	`326`
`327`	`327`	`except Exception as e:`
`328`	`328`	`logging.exception(f"Error in create_structured_plan: {e}")`
`329`		`- track_event(`
	`329`	`+ track_event_if_configured(`
`330`	`330`	`f"Planner - Error in create_structured_plan: {e} into the cosmos",`
`331`	`331`	`{`
`332`	`332`	`"session_id": self._session_id,`