comment out azure monitor

Author: Fr4nc3

src/backend/app_kernel.py

@@ -43,18 +43,18 @@
 from utils_kernel import get_agents, initialize_runtime_and_context, rai_success
 
 # # Check if the Application Insights Instrumentation Key is set in the environment variables
-connection_string = os.getenv("APPLICATIONINSIGHTS_CONNECTION_STRING")
-if connection_string:
-    # Configure Application Insights if the Instrumentation Key is found
-    configure_azure_monitor(connection_string=connection_string)
-    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"
-    )
+# connection_string = os.getenv("APPLICATIONINSIGHTS_CONNECTION_STRING")
+# if connection_string:
+#     # Configure Application Insights if the Instrumentation Key is found
+#     configure_azure_monitor(connection_string=connection_string)
+#     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)
@@ -129,10 +129,17 @@ async def input_task_endpoint(input_task: InputTask, request: Request):
         kernel, memory_store = await initialize_runtime_and_context(
             input_task.session_id, user_id
         )
+        client = None
+        try:
+            client = config.get_ai_project_client()
+        except Exception as client_exc:
+            logging.error(f"Error creating AIProjectClient: {client_exc}")
+
         agents = await AgentFactory.create_all_agents(
             session_id=input_task.session_id,
             user_id=user_id,
             memory_store=memory_store,
+            client=client,
         )
 
         group_chat_manager = agents[AgentType.GROUP_CHAT_MANAGER.value]
@@ -166,7 +173,11 @@ async def input_task_endpoint(input_task: InputTask, request: Request):
                 "description": input_task.description,
             },
         )
-
+        if client:
+            try:
+                client.close()
+            except Exception as e:
+                logging.error(f"Error sending to AIProjectClient: {e}")
         return {
             "status": f"Plan created with ID: {plan.id}",
             "session_id": input_task.session_id,
@@ -254,12 +265,31 @@ async def human_feedback_endpoint(human_feedback: HumanFeedback, request: Reques
     kernel, memory_store = await initialize_runtime_and_context(
         human_feedback.session_id, user_id
     )
-    agents = await AgentFactory.create_all_agents(
-        session_id=human_feedback.session_id, user_id=user_id, memory_store=memory_store
+
+    client = None
+    try:
+        client = config.get_ai_project_client()
+    except Exception as client_exc:
+        logging.error(f"Error creating AIProjectClient: {client_exc}")
+
+    human_agent = await AgentFactory.create_agent(
+        agent_type=AgentType.HUMAN,
+        session_id=human_feedback.session_id,
+        user_id=user_id,
+        memory_store=memory_store,
+        client=client,
     )
 
-    # Send the feedback to the human agent
-    human_agent = agents[AgentType.HUMAN.value]
+    if human_agent is None:
+        track_event_if_configured(
+            "AgentNotFound",
+            {
+                "status": "Agent not found",
+                "session_id": human_feedback.session_id,
+                "step_id": human_feedback.step_id,
+            },
+        )
+        raise HTTPException(status_code=404, detail="Agent not found")
 
     # Use the human agent to handle the feedback
     await human_agent.handle_human_feedback(human_feedback=human_feedback)
@@ -272,7 +302,11 @@ async def human_feedback_endpoint(human_feedback: HumanFeedback, request: Reques
             "step_id": human_feedback.step_id,
         },
     )
-
+    if client:
+        try:
+            client.close()
+        except Exception as e:
+            logging.error(f"Error sending to AIProjectClient: {e}")
     return {
         "status": "Feedback received",
         "session_id": human_feedback.session_id,
@@ -338,14 +372,30 @@ async def human_clarification_endpoint(
     kernel, memory_store = await initialize_runtime_and_context(
         human_clarification.session_id, user_id
     )
-    agents = await AgentFactory.create_all_agents(
+    client = None
+    try:
+        client = config.get_ai_project_client()
+    except Exception as client_exc:
+        logging.error(f"Error creating AIProjectClient: {client_exc}")
+
+    human_agent = await AgentFactory.create_agent(
+        agent_type=AgentType.HUMAN,
         session_id=human_clarification.session_id,
         user_id=user_id,
         memory_store=memory_store,
+        client=client,
     )
 
-    # Send the feedback to the human agent
-    human_agent = agents[AgentType.HUMAN.value]
+    if human_agent is None:
+        track_event_if_configured(
+            "AgentNotFound",
+            {
+                "status": "Agent not found",
+                "session_id": human_clarification.session_id,
+                "step_id": human_clarification.step_id,
+            },
+        )
+        raise HTTPException(status_code=404, detail="Agent not found")
 
     # Use the human agent to handle the feedback
     await human_agent.handle_human_clarification(
@@ -359,7 +409,11 @@ async def human_clarification_endpoint(
             "session_id": human_clarification.session_id,
         },
     )
-
+    if client:
+        try:
+            client.close()
+        except Exception as e:
+            logging.error(f"Error sending to AIProjectClient: {e}")
     return {
         "status": "Clarification received",
         "session_id": human_clarification.session_id,
@@ -432,17 +486,28 @@ async def approve_step_endpoint(
     kernel, memory_store = await initialize_runtime_and_context(
         human_feedback.session_id, user_id
     )
+    client = None
+    try:
+        client = config.get_ai_project_client()
+    except Exception as client_exc:
+        logging.error(f"Error creating AIProjectClient: {client_exc}")
     agents = await AgentFactory.create_all_agents(
         session_id=human_feedback.session_id,
         user_id=user_id,
         memory_store=memory_store,
+        client=client,
     )
 
     # Send the approval to the group chat manager
     group_chat_manager = agents[AgentType.GROUP_CHAT_MANAGER.value]
 
     await group_chat_manager.handle_human_feedback(human_feedback)
 
+    if client:
+        try:
+            client.close()
+        except Exception as e:
+            logging.error(f"Error sending to AIProjectClient: {e}")
     # Return a status message
     if human_feedback.step_id:
         track_event_if_configured(

src/backend/kernel_agents/agent_factory.py

@@ -158,7 +158,6 @@ async def create_agent(
 
         # Build the agent definition (functions schema)
         definition = None
-        client = None
 
         try:
             if client is None:
@@ -239,6 +238,7 @@ async def create_all_agents(
         user_id: str,
         temperature: float = 0.0,
         memory_store: Optional[CosmosMemoryContext] = None,
+        client: Optional[Any] = None,
     ) -> Dict[AgentType, BaseAgent]:
         """Create all agent types for a session in a specific order.
 
@@ -265,6 +265,13 @@ async def create_all_agents(
         planner_agent_type = AgentType.PLANNER
         group_chat_manager_type = AgentType.GROUP_CHAT_MANAGER
 
+        try:
+            if client is None:
+                # Create the AIProjectClient instance using the config
+                # This is a placeholder; replace with actual client creation logic
+                client = config.get_ai_project_client()
+        except Exception as client_exc:
+            logger.error(f"Error creating AIProjectClient: {client_exc}")
         # Initialize cache for this session if it doesn't exist
         if session_id not in cls._agent_cache:
             cls._agent_cache[session_id] = {}
@@ -280,6 +287,7 @@ async def create_all_agents(
                 session_id=session_id,
                 user_id=user_id,
                 temperature=temperature,
+                client=client,
                 memory_store=memory_store,
             )
 
@@ -305,6 +313,7 @@ async def create_all_agents(
             user_id=user_id,
             temperature=temperature,
             agent_instances=agent_instances,  # Pass agent instances to the planner
+            client=client,
             response_format=ResponseFormatJsonSchemaType(
                 json_schema=ResponseFormatJsonSchema(
                     name=PlannerResponsePlan.__name__,
@@ -324,6 +333,7 @@ async def create_all_agents(
             session_id=session_id,
             user_id=user_id,
             temperature=temperature,
+            client=client,
             agent_instances=agent_instances,  # Pass agent instances to the planner
         )
         agents[group_chat_manager_type] = group_chat_manager