From a165f21f30bc0bd4bcabdf0788e8185f67c32283 Mon Sep 17 00:00:00 2001
From: Ayaz-Microsoft <v-ayazkhan@microsoft.com>
Date: Tue, 21 Jul 2026 14:03:00 +0530
Subject: [PATCH] refactor: remove unused imports in test files for improved
 code quality

---
 src/tests/backend/callbacks/test_response_handlers.py        | 5 +----
 src/tests/backend/models/test_plan_models.py                 | 2 --
 .../orchestration/helper/test_plan_to_mplan_converter.py     | 1 -
 src/tests/backend/services/test_team_service.py              | 2 +-
 4 files changed, 2 insertions(+), 8 deletions(-)

diff --git a/src/tests/backend/callbacks/test_response_handlers.py b/src/tests/backend/callbacks/test_response_handlers.py
index d44a712f5..a150c973a 100644
--- a/src/tests/backend/callbacks/test_response_handlers.py
+++ b/src/tests/backend/callbacks/test_response_handlers.py
@@ -1,11 +1,8 @@
 """Unit tests for response_handlers module."""
 
-import asyncio
-import logging
 import os
 import sys
-import time
-from unittest.mock import AsyncMock, MagicMock, Mock, patch
+from unittest.mock import AsyncMock, Mock, patch
 
 import pytest
 
diff --git a/src/tests/backend/models/test_plan_models.py b/src/tests/backend/models/test_plan_models.py
index ee3bea16d..84920bfda 100644
--- a/src/tests/backend/models/test_plan_models.py
+++ b/src/tests/backend/models/test_plan_models.py
@@ -4,8 +4,6 @@
 
 import uuid
 
-import pytest
-
 from backend.models.plan_models import (AgentDefinition, MPlan, MStep,
                                         PlannerResponsePlan,
                                         PlannerResponseStep, PlanStatus)
diff --git a/src/tests/backend/orchestration/helper/test_plan_to_mplan_converter.py b/src/tests/backend/orchestration/helper/test_plan_to_mplan_converter.py
index cf73da440..cdcded513 100644
--- a/src/tests/backend/orchestration/helper/test_plan_to_mplan_converter.py
+++ b/src/tests/backend/orchestration/helper/test_plan_to_mplan_converter.py
@@ -10,7 +10,6 @@
 """
 
 import os
-import re
 import sys
 import unittest
 
diff --git a/src/tests/backend/services/test_team_service.py b/src/tests/backend/services/test_team_service.py
index 7d452c314..680320b78 100644
--- a/src/tests/backend/services/test_team_service.py
+++ b/src/tests/backend/services/test_team_service.py
@@ -66,7 +66,7 @@ class MockDatabaseBase:
 
 # Mock common.models.messages with real dataclasses
 from dataclasses import dataclass, field
-from typing import Any, List, Optional
+from typing import Any, List
 
 
 @dataclass