isage-middleware 0.1.4__py3-none-any.whl → 0.1.5.1__py3-none-any.whl

sage/middleware/components/neuromem/tests/core_test/collection_test/__init__.py

@@ -1,37 +0,0 @@
-"""
-SAGE - Streaming-Augmented Generative Execution
-"""
-
-# 动态版本加载
-def _load_version():
-    """从项目根目录动态加载版本信息"""
-    from pathlib import Path
-    
-    # 获取项目根目录
-    current_file = Path(__file__).resolve()
-    root_dir = current_file.parent.parent.parent.parent.parent.parent.parent.parent.parent.parent
-    version_file = root_dir / "_version.py"
-    
-    # 加载版本信息
-    if version_file.exists():
-        version_globals = {}
-        with open(version_file, 'r', encoding='utf-8') as f:
-            exec(f.read(), version_globals)
-        return {
-            'version': version_globals.get('__version__', '0.1.4'),
-            'author': version_globals.get('__author__', 'SAGE Team'),
-            'email': version_globals.get('__email__', 'shuhao_zhang@hust.edu.cn')
-        }
-    
-    # 默认值
-    return {
-        'version': '0.1.4',
-        'author': 'SAGE Team', 
-        'email': 'shuhao_zhang@hust.edu.cn'
-    }
-
-# 加载信息
-_info = _load_version()
-__version__ = _info['version']
-__author__ = _info['author']
-__email__ = _info['email']

sage/middleware/components/neuromem/tests/core_test/collection_test/kv_collection_test.py

@@ -1,60 +0,0 @@
-import pytest
-from sage.middleware.components.neuromem.memory_collection.kv_collection import KVMemoryCollection
-from sage.middleware.components.neuromem.memory_collection.base_collection import get_default_data_dir
-
-@pytest.fixture
-def setup_kv_collection():
-    col = KVMemoryCollection(name="demo")
-
-    col.insert("Hello Jack.", {})
-    col.insert("Hello Tom.", {})
-    col.insert("Hello Alice.", {})
-    col.insert("Jack and Tom say hi.", {})
-    col.insert("Alice in Wonderland.", {})
-    col.insert("Jacky is not Jack.", {})
-
-    col.create_index("global_index")
-    return col
-
-
-def test_kv_retrieve(setup_kv_collection):
-    col = setup_kv_collection
-
-    res1 = col.retrieve("Jack", topk=3, index_name="global_index")
-    assert set(res1) == {"Hello Jack.", "Jack and Tom say hi.", "Jacky is not Jack."}
-
-    res2 = col.retrieve("Alice", topk=2, index_name="global_index")
-    assert set(res2) == {"Hello Alice.", "Alice in Wonderland."}
-
-    res3 = col.retrieve("Tom", topk=2, index_name="global_index")
-    assert set(res3) == {"Hello Tom.", "Jack and Tom say hi."}
-
-
-def test_kv_update_and_delete(setup_kv_collection):
-    col = setup_kv_collection
-
-    col.delete("Hello Tom.")
-    res4 = col.retrieve("Tom", index_name="global_index")
-    assert "Hello Tom." not in res4
-
-    col.update("Hello Jack.", "Hello Jacky.", {}, "global_index")
-    res5 = col.retrieve("Jacky", index_name="global_index")
-
-    assert {"Jacky is not Jack.", "Hello Jacky."}.issubset(set(res5))
-
-
-def test_kv_persistence(setup_kv_collection):
-    col = setup_kv_collection
-    store_path = get_default_data_dir()
-    col_name = "demo"
-    col.store(store_path)
-
-    del col
-
-    col2 = KVMemoryCollection.load(col_name)
-    res = col2.retrieve("Jacky", index_name="global_index")
-
-
-    assert "Jacky is not Jack." in res
-
-    KVMemoryCollection.clear(col_name)

sage/middleware/components/neuromem/tests/core_test/collection_test/vdb_collection_test.py

@@ -1,88 +0,0 @@
-# file sage/sage_tests/neuromem_test/core_test/collection_test/vdb_collection_test.py
-
-import os
-import time
-import pytest
-from sage.middleware.components.neuromem.memory_collection.base_collection import get_default_data_dir
-from sage.utils.embedding_methods.mockembedder import MockTextEmbedder
-from sage.middleware.components.neuromem.memory_collection.vdb_collection import VDBMemoryCollection
-
-
-def almost_equal_dict(d1, d2, float_tol=1e-3):
-    # 只对所有值都是float的dict做容忍，否则严格等价
-    if d1.keys() != d2.keys():
-        return False
-    for k in d1:
-        v1, v2 = d1[k], d2[k]
-        if isinstance(v1, float) and isinstance(v2, float):
-            if abs(v1 - v2) > float_tol:
-                return False
-        else:
-            if v1 != v2:
-                return False
-    return True
-
-@pytest.fixture
-def setup_vdb_collection():
-    default_model = MockTextEmbedder(fixed_dim=16)
-    col = VDBMemoryCollection("vdb_demo", default_model, 16)
-    col.add_metadata_field("source")
-    col.add_metadata_field("lang")
-    col.add_metadata_field("timestamp")
-
-    current_time = time.time()
-    texts = [
-        ("hello world", {"source": "user", "lang": "en", "timestamp": current_time - 3600}),
-        ("你好，世界", {"source": "user", "lang": "zh", "timestamp": current_time - 1800}),
-        ("bonjour le monde", {"source": "web", "lang": "fr", "timestamp": current_time}),
-    ]
-    for t, meta in texts:
-        col.insert(t, meta)
-
-    col.create_index("global_index")
-    col.create_index("en_index", metadata_filter_func=lambda m: m.get("lang") == "en", description="English only")
-    col.create_index("user_index", metadata_filter_func=lambda m: m.get("source") == "user")
-    return col, current_time
-
-def test_vdb_persist(setup_vdb_collection):
-    col, current_time = setup_vdb_collection
-
-    # 检索校验
-    res = col.retrieve("hello", topk=3, index_name="global_index")
-    assert set(res) & {"hello world"} == {"hello world"}
-
-    res = col.retrieve("hello", index_name="en_index")
-    assert set(res) & {"hello world"} == {"hello world"}
-
-    res = col.retrieve("你好", index_name="user_index")
-    assert set(res) & {"你好，世界"} == {"你好，世界"}
-
-    # 持久化保存
-    store_path = get_default_data_dir()
-    col_name = "vdb_demo"
-    col.store(store_path)
-
-    # 清除内存对象
-    del col
-
-    # 恢复对象并回归测试
-    default_model2 = MockTextEmbedder(fixed_dim=16)
-    col2 = VDBMemoryCollection.load(col_name, embedding_model=default_model2)
-
-    # 再检索
-    res = col2.retrieve("hello", index_name="global_index")
-    assert set(res) & {"hello world"} == {"hello world"}
-
-    res = col2.retrieve("你好", index_name="user_index")
-    assert set(res) & {"你好，世界"} == {"你好，世界"}
-
-    # 校验metadata一致性
-    meta = col2.metadata_storage.get(col2._get_stable_id("hello world"))
-    assert almost_equal_dict(meta, {"source": "user", "lang": "en", "timestamp": current_time - 3600})
-
-    # 校验索引条件
-    idx_meta = col2.indexes["en_index"]
-    assert idx_meta.get("description", "") == "English only"
-
-    # 删除磁盘数据
-    VDBMemoryCollection.clear(col_name)

sage/middleware/components/neuromem/tests/core_test/manager_test.py

@@ -1,154 +0,0 @@
-# file sage/sage_tests/neuromem_test/core_test/manager_test.py
-# python -m sage.sage_tests.neuromem_test.core_test.manager_test
-
-from time import sleep
-
-
-def test_vdb(do_reload=True, do_delete=True):
-    from sage.middleware.components.neuromem.memory_manager import MemoryManager
-    from sage.utils.embedding_methods.mockembedder import MockTextEmbedder
-    import os
-
-    def colored(text, color):
-        colors = {"green": "\033[92m", "red": "\033[91m", "yellow": "\033[93m", "reset": "\033[0m"}
-        return colors.get(color, "") + text + colors["reset"]
-        # return text  # 自动化测试不需要彩色输出
-
-    def print_test_case(desc, expected, actual):
-        status = "通过" if expected == actual or (isinstance(expected, set) and set(expected) == set(actual)) else "不通过"
-        print(f"【{desc}】")
-        print(f"预期结果：{expected}")
-        print(f"实际结果：{actual}")
-        print(f"测试情况：{status}\n")
-
-    # 1. 创建 MemoryManager 和 VDBCollection
-    mgr = MemoryManager()
-    embedder = MockTextEmbedder(fixed_dim=16)
-    col = mgr.create_collection(name="test_vdb", backend_type="VDB", description="测试VDB集合", embedding_model=embedder, dim=16)
-    col.add_metadata_field("tag")  # 注册元数据字段tag
-
-    # 2. 插入数据
-    col.insert("Alpha", {"tag": "A"})
-    col.insert("Beta", {"tag": "B"})
-    col.insert("Gamma", {"tag": "A"})
-
-    # 3. 创建索引
-    col.create_index("global_index")
-    col.create_index("tag_A_index", metadata_filter_func=lambda m: m.get("tag") == "A")
-
-    # 4. 检索测试
-    res1 = col.retrieve("Alpha", topk=1, index_name="global_index")
-    print_test_case("检索Alpha", ["Alpha"], res1)
-
-    res2 = set(col.retrieve("Alpha", topk=5, index_name="tag_A_index"))  # Alpha 的向量检索，tag为A
-    print_test_case("tag_A_index 检索", {'Gamma', 'Alpha', 'Beta'}, res2)
-
-    # 5. 持久化
-    mgr.store_collection()
-    print("数据已保存到磁盘！")
-    data_dir = mgr.data_dir
-    print("目录为：", os.path.join(data_dir, "vdb_collection", "test_vdb"))
-
-    # 6. 清空对象
-    del mgr, col
-    print("内存对象已清除。")
-
-    # 7. 读取持久化数据，自动测试时直接执行
-    if do_reload:
-    from sage.middleware.components.neuromem.memory_manager import MemoryManager
-        embedder2 = MockTextEmbedder(fixed_dim=16)
-        mgr2 = MemoryManager()
-        col2 = mgr2.connect_collection("test_vdb", embedding_model=embedder2)
-        res3 = set(col2.retrieve("Alpha", topk=5, index_name="tag_A_index"))
-        print_test_case("恢复后tag_A_index 检索", {'Gamma', 'Alpha', 'Beta'}, res3)
-    else:
-        print("跳过加载测试。")
-
-    sleep(10)  # 等待一会儿，确保所有操作完成
-    # 8. 删除所有数据
-    if do_delete:
-    from sage.middleware.components.neuromem.memory_collection.vdb_collection import VDBMemoryCollection
-        VDBMemoryCollection.clear("test_vdb", data_dir)
-        manager_json = os.path.join(data_dir, "manager.json")
-        if os.path.exists(manager_json):
-            os.remove(manager_json)
-        print("所有数据已删除！")
-    else:
-        print("未执行删除。")
-
-# 作为测试脚本被自动调用
-if __name__ == "__main__":
-    test_vdb()
-
-
-# def kvtest():
-#     from sage.core.sage.middleware.services.neuromem..memory_manager import MemoryManager
-#     import os
-
-#     def colored(text, color):
-#         colors = {"green": "\033[92m", "red": "\033[91m", "yellow": "\033[93m", "reset": "\033[0m"}
-#         return colors.get(color, "") + text + colors["reset"]
-
-#     def print_test_case(desc, expected, actual):
-#         status = "通过" if expected == actual or (isinstance(expected, set) and set(expected) == set(actual)) else "不通过"
-#         color = "green" if status == "通过" else "red"
-#         print(f"【{desc}】")
-#         print(f"预期结果：{expected}")
-#         print(f"实际结果：{actual}")
-#         print(f"测试情况：{colored(status, color)}\n")
-
-#     # 1. 创建 MemoryManager 和 KVCollection
-#     mgr = MemoryManager()
-#     col = mgr.create_collection(name="test_kv", backend_type="KV", description="测试KV集合")
-#     col.add_metadata_field("tag")  # 注册元数据字段tag
-
-#     # 2. 插入数据
-#     col.insert("Alpha", {"tag": "A"})
-#     col.insert("Beta", {"tag": "B"})
-#     col.insert("Gamma", {"tag": "A"})
-
-#     # 3. 创建索引
-#     col.create_index("global_index")
-#     col.create_index("tag_A_index", metadata_filter_func=lambda m: m.get("tag") == "A")
-
-#     # 4. 检索测试
-#     res1 = col.retrieve("Alpha", topk=1, index_name="global_index")
-#     print_test_case("检索Alpha", ["Alpha"], res1)
-
-#     res2 = set(col.retrieve("a", topk=5, index_name="tag_A_index"))  # 检索包含'a'的文本，tag为A
-#     print_test_case("tag_A_index 检索", {"Alpha", "Gamma"}, res2)
-
-#     # 5. 持久化
-#     mgr.store_collection()
-#     print(colored("数据已保存到磁盘！", "yellow"))
-#     data_dir = mgr.data_dir
-#     print("目录为：", os.path.join(data_dir, "kv_collection", "test_kv"))
-
-#     # 6. 清空对象
-#     del mgr, col
-#     print(colored("内存对象已清除。", "yellow"))
-
-#     # 7. 读取持久化数据
-#     user_input = input(colored("输入 yes 加载刚才保存的数据: ", "yellow"))
-#     if user_input.strip().lower() == "yes":
-#         mgr2 = MemoryManager()  # 会自动加载 manager.json 和 collection
-#         col2 = mgr2.get_collection("test_kv")
-#         # 检查是否正常恢复
-#         res3 = set(col2.retrieve("a", topk=5, index_name="tag_A_index"))
-#         print_test_case("恢复后tag_A_index 检索", {"Alpha", "Gamma"}, res3)
-#     else:
-#         print(colored("跳过加载测试。", "yellow"))
-
-#     # 8. 删除所有数据
-#     user_input = input(colored("输入 yes 删除磁盘所有数据: ", "yellow"))
-#     if user_input.strip().lower() == "yes":
-#         from sage.core.sage.middleware.services.neuromem.memory_collection.kv_collection import KVMemoryCollection
-#         KVMemoryCollection.clear("test_kv", data_dir)
-#         manager_json = os.path.join(data_dir, "manager.json")
-#         if os.path.exists(manager_json):
-#             os.remove(manager_json)
-#         print(colored("所有数据已删除！", "green"))
-#     else:
-#         print(colored("未执行删除。", "yellow"))
-
-# kvtest()

sage/middleware/components/neuromem/tests/test_memory_service.py

@@ -1,293 +0,0 @@
-"""
-测试Memory Service的集成测试用例
-"""
-import os
-import shutil
-import traceback
-from sage.middleware.components.neuromem.memory_service import MemoryService
-from sage.middleware.utils.embedding.embedding_api import apply_embedding_model
-
-def test_memory_service():
-    """测试Memory Service的主要功能（直接测试，不使用服务框架）"""
-    print("🚀 Starting Memory Service test...")
-    
-    try:
-        # 1. 直接创建MemoryService实例
-        # 使用默认的embedding model
-        embedding_model = apply_embedding_model("default")
-        dim = embedding_model.get_dim()
-        # 指定临时测试目录
-        test_data_dir = os.path.join(os.path.dirname(__file__), "test_data")
-        os.makedirs(test_data_dir, exist_ok=True)
-        memory_service = MemoryService(data_dir=test_data_dir)
-        
-        print("✅ Memory service created, testing operations...")
-        
-        # 2. 测试创建collection
-        result1 = memory_service.create_collection(
-            name="test_collection",
-            backend_type="VDB",
-            description="Test collection",
-            embedding_model=embedding_model,
-            dim=dim
-        )
-        print(f"Create collection result: {result1}")
-        assert result1["status"] == "success", f"Create collection failed: {result1}"
-        
-        # 3. 测试插入数据
-        result2 = memory_service.insert_data(
-            collection_name="test_collection",
-            text="This is a test document",
-            metadata={"type": "test", "date": "2025-07-26"}
-        )
-        print(f"Insert data result: {result2}")
-        assert result2["status"] == "success", f"Insert data failed: {result2}"
-        
-        # 4. 测试创建索引
-        result3 = memory_service.create_index(
-            collection_name="test_collection",
-            index_name="test_index",
-            description="Test index"
-        )
-        print(f"Create index result: {result3}")
-        assert result3["status"] == "success", f"Create index failed: {result3}"
-        
-        # 5. 测试检索数据
-        result4 = memory_service.retrieve_data(
-            collection_name="test_collection",
-            query_text="test document",
-            topk=5,
-            index_name="test_index",
-            with_metadata=True
-        )
-        print(f"Retrieve data result: {result4}")
-        assert result4["status"] == "success", f"Retrieve data failed: {result4}"
-        
-        # 6. 测试插入更多数据
-        for i in range(3):
-            result = memory_service.insert_data(
-                collection_name="test_collection",
-                text=f"Test document {i}",
-                metadata={"type": "test", "index": i}
-            )
-            print(f"Insert data {i} result: {result}")
-            assert result["status"] == "success", f"Insert data {i} failed: {result}"
-        
-        # 7. 测试列出collections
-        final_result = memory_service.list_collections()
-        print(f"Collections list: {final_result}")
-        assert final_result["status"] == "success", f"List collections failed: {final_result}"
-        assert len(final_result["collections"]) == 1, "Should have exactly 1 collection"
-        
-        # 8. 测试获取collection信息
-        info_result = memory_service.get_collection_info("test_collection")
-        print(f"Collection info: {info_result}")
-        assert info_result["status"] == "success", f"Get collection info failed: {info_result}"
-        
-        # 9. 测试列出索引
-        index_result = memory_service.list_indexes("test_collection")
-        print(f"Indexes list: {index_result}")
-        assert index_result["status"] == "success", f"List indexes failed: {index_result}"
-        
-        # 10. 测试存储单个collection
-        store_collection_result = memory_service.store_collection("test_collection")
-        print(f"Store collection result: {store_collection_result}")
-        assert store_collection_result["status"] == "success", f"Store collection failed: {store_collection_result}"
-        
-        # 11. 创建第二个collection来测试选择性存储
-        result_col2 = memory_service.create_collection(
-            name="temp_collection",
-            backend_type="VDB",
-            description="Temporary test collection",
-            embedding_model=embedding_model,
-            dim=dim
-        )
-        print(f"Create temp collection result: {result_col2}")
-        assert result_col2["status"] == "success", f"Create temp collection failed: {result_col2}"
-        
-        # 12. 向第二个collection插入数据
-        temp_insert_result = memory_service.insert_data(
-            collection_name="temp_collection",
-            text="This is temporary data",
-            metadata={"type": "temp", "temporary": True}
-        )
-        print(f"Insert temp data result: {temp_insert_result}")
-        assert temp_insert_result["status"] == "success", f"Insert temp data failed: {temp_insert_result}"
-        
-        # 13. 测试存储所有manager数据
-        store_all_result = memory_service.store()
-        print(f"Store all result: {store_all_result}")
-        assert store_all_result["status"] == "success", f"Store all failed: {store_all_result}"
-        
-        print("✅ All operations including storage completed successfully!")
-        
-        return True
-        
-    except Exception as e:
-        print(f"❌ Test failed: {e}")
-        traceback.print_exc()
-        return False
-        
-    finally:
-        # 清理资源
-        try:
-            # 清理测试数据目录
-            test_data_dir = os.path.join(os.path.dirname(__file__), "test_data")
-            if os.path.exists(test_data_dir):
-                shutil.rmtree(test_data_dir)
-            print("🧹 Cleanup completed")
-        except Exception as e:
-            print(f"⚠️ Cleanup error: {e}")
-
-
-def test_data_loss_without_storage():
-    """测试未保存collection导致的数据丢失"""
-    print("\n🔍 Starting data loss test without storage...")
-    
-    try:
-        # 使用默认的embedding model
-        embedding_model = apply_embedding_model("default")
-        dim = embedding_model.get_dim()
-        
-        # 创建独立的测试目录
-        test_data_dir = os.path.join(os.path.dirname(__file__), "test_data_loss")
-        os.makedirs(test_data_dir, exist_ok=True)
-        
-        print("✅ Starting data loss test...")
-        
-        # 阶段1: 创建两个collection并插入数据
-        print("\n📝 Phase 1: Creating collections and inserting data...")
-        
-        memory_service = MemoryService(data_dir=test_data_dir)
-        
-        # 创建第一个collection
-        result1 = memory_service.create_collection(
-            name="saved_collection",
-            backend_type="VDB", 
-            description="This will be saved",
-            embedding_model=embedding_model,
-            dim=dim
-        )
-        assert result1["status"] == "success"
-        
-        # 创建第二个collection
-        result2 = memory_service.create_collection(
-            name="unsaved_collection",
-            backend_type="VDB",
-            description="This will NOT be saved",
-            embedding_model=embedding_model,
-            dim=dim
-        )
-        assert result2["status"] == "success"
-        
-        # 向两个collection插入数据
-        for i in range(3):
-            # 保存的collection
-            memory_service.insert_data(
-                collection_name="saved_collection",
-                text=f"Saved document {i}",
-                metadata={"type": "saved", "index": i}
-            )
-            # 未保存的collection  
-            memory_service.insert_data(
-                collection_name="unsaved_collection", 
-                text=f"Unsaved document {i}",
-                metadata={"type": "unsaved", "index": i}
-            )
-        
-        # 只保存第一个collection
-        print("\n💾 Phase 2: Storing only the first collection...")
-        store_result = memory_service.store_collection("saved_collection")
-        assert store_result["status"] == "success"
-        print(f"✅ Stored 'saved_collection': {store_result}")
-        
-        # 验证两个collection都还在内存中
-        list_result = memory_service.list_collections()
-        assert list_result["status"] == "success"
-        assert len(list_result["collections"]) == 2
-        print(f"✅ Both collections still in memory: {[c['name'] for c in list_result['collections']]}")
-        
-        # 释放当前service实例
-        print("\n🔄 Phase 3: Creating new service instance to simulate restart...")
-        del memory_service
-        
-        # 重新创建服务实例 (模拟重启后从磁盘加载)
-        memory_service2 = MemoryService(data_dir=test_data_dir)
-        
-        # 验证数据丢失情况
-        print("\n🔍 Phase 4: Checking data after restart...")
-        
-        # 检查collections列表
-        list_result_after = memory_service2.list_collections()
-        assert list_result_after["status"] == "success"
-        collection_names = [c['name'] for c in list_result_after['collections']]
-        print(f"Collections after restart: {collection_names}")
-        
-        # 验证保存的collection仍然存在
-        if "saved_collection" in collection_names:
-            print("✅ 'saved_collection' found - data persisted correctly")
-            # 尝试检索数据验证完整性
-            retrieve_saved = memory_service2.retrieve_data(
-                collection_name="saved_collection",
-                query_text="Saved document",
-                topk=5,
-                with_metadata=True
-            )
-            if retrieve_saved["status"] == "success" and len(retrieve_saved["results"]) > 0:
-                print(f"✅ Saved collection data intact: {len(retrieve_saved['results'])} documents found")
-            else:
-                print("⚠️ Saved collection exists but data may be incomplete")
-        else:
-            print("❌ ERROR: 'saved_collection' not found after restart!")
-            
-        # 验证未保存的collection丢失
-        if "unsaved_collection" not in collection_names:
-            print("✅ 'unsaved_collection' correctly lost - demonstrates need for storage")
-        else:
-            print("⚠️ WARNING: 'unsaved_collection' unexpectedly persisted")
-            
-        print("\n🎯 Data loss test demonstrates:")
-        print("  - Collections must be explicitly stored to persist")
-        print("  - Only stored collections survive service restart") 
-        print("  - store_collection() provides selective persistence")
-        
-        return True
-        
-    except Exception as e:
-        print(f"❌ Data loss test failed: {e}")
-        traceback.print_exc()
-        return False
-        
-    finally:
-        # 清理测试数据
-        try:
-            test_data_dir = os.path.join(os.path.dirname(__file__), "test_data_loss")
-            if os.path.exists(test_data_dir):
-                shutil.rmtree(test_data_dir)
-            print("🧹 Data loss test cleanup completed")
-        except Exception as e:
-            print(f"⚠️ Data loss test cleanup error: {e}")
-
-
-if __name__ == "__main__":
-    print("=" * 60)
-    print("SAGE Memory Service Integration Test")
-    print("=" * 60)
-    
-    # 运行基本功能测试
-    success = test_memory_service()
-    
-    if success:
-        print("\n🎉 Basic tests passed! Running data loss test...")
-        # 运行数据丢失测试
-        loss_test_success = test_data_loss_without_storage()
-        
-        if loss_test_success:
-            print("\n🎉 All tests passed! Memory service system is working correctly.")
-            print("✨ Storage functionality verified - data persistence works as expected.")
-        else:
-            print("\n💥 Data loss tests failed! Please check the logs above.")
-            exit(1)
-    else:
-        print("\n💥 Basic tests failed! Please check the logs above.")
-        exit(1)