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