close the client to avoid error

Author: Fr4nc3

src/backend/app_kernel.py

@@ -124,10 +124,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]
@@ -161,7 +168,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,
@@ -249,8 +260,18 @@ async def human_feedback_endpoint(human_feedback: HumanFeedback, request: Reques
     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
+        session_id=human_feedback.session_id,
+        user_id=user_id,
+        memory_store=memory_store,
+        client=client,
     )
 
     # Send the feedback to the human agent
@@ -267,7 +288,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,
@@ -333,10 +358,16 @@ async def human_clarification_endpoint(
     kernel, memory_store = await initialize_runtime_and_context(
         human_clarification.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_clarification.session_id,
         user_id=user_id,
         memory_store=memory_store,
+        client=client,
     )
 
     # Send the feedback to the human agent
@@ -354,7 +385,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,
@@ -427,17 +462,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

@@ -239,6 +239,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 +266,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 +288,7 @@ async def create_all_agents(
                 session_id=session_id,
                 user_id=user_id,
                 temperature=temperature,
+                client=client,
                 memory_store=memory_store,
             )
 
@@ -305,6 +314,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 +334,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