diff --git a/tests/unit/test_level_based_platform_query_engine_prepare_context.py b/tests/unit/test_level_based_platform_query_engine_prepare_context.py index e28469c..3fa415e 100644 --- a/tests/unit/test_level_based_platform_query_engine_prepare_context.py +++ b/tests/unit/test_level_based_platform_query_engine_prepare_context.py @@ -2,10 +2,8 @@ import unittest from unittest.mock import patch -from bot.retrievers.forum_summary_retriever import ForumBasedSummaryRetriever from bot.retrievers.retrieve_similar_nodes import RetrieveSimilarNodes from llama_index.schema import NodeWithScore, TextNode -from sqlalchemy.exc import OperationalError from utils.query_engine.level_based_platform_query_engine import ( LevelBasedPlatformQueryEngine, )