lollms-client 0.33.0__py3-none-any.whl → 1.0.0__py3-none-any.whl

lollms_client-0.33.0.dist-info/RECORD

@@ -1,101 +0,0 @@
-examples/console_discussion.py,sha256=JxjVaAxtt1WEXLN8vCJosu-cYNgfIX2JGO25kg1FFNY,20490
-examples/function_calling_with_local_custom_mcp.py,sha256=g6wOFRB8-p9Cv7hKmQaGzPvtMX3H77gas01QVNEOduM,12407
-examples/generate_a_benchmark_for_safe_store.py,sha256=bkSt0mrpNsN0krZAUShm0jgVM1ukrPpjI7VwSgcNdSA,3974
-examples/generate_text_with_multihop_rag_example.py,sha256=riEyVYo97r6ZYdySL-NJkRhE4MnpwbZku1sN8RNvbvs,11519
-examples/gradio_chat_app.py,sha256=ZZ_D1U0wvvwE9THmAPXUvNKkFG2gi7tQq1f2pQx_2ug,15315
-examples/gradio_lollms_chat.py,sha256=z5FDE62dmPU3nb16zbZX6jkVitML1PMfPxYyWr8VLz8,10135
-examples/internet_search_with_rag.py,sha256=ioTb_WI2M6kFeh1Dg-EGcKjccphnCsIGD_e9PZgZshw,12314
-examples/lollms_discussions_test.py,sha256=Jk1cCUDBBhTcK5glI50jAgzfB3IOiiUlnK3q7RYfMkA,6796
-examples/simple_text_gen_test.py,sha256=RoX9ZKJjGMujeep60wh5WT_GoBn0O9YKJY6WOy-ZmOc,8710
-examples/simple_text_gen_with_image_test.py,sha256=rR1O5Prcb52UHtJ3c6bv7VuTd1cvbkr5aNZU-v-Rs3Y,9263
-examples/text_2_audio.py,sha256=MfL4AH_NNwl6m0I0ywl4BXRZJ0b9Y_9fRqDIe6O-Sbw,3523
-examples/text_2_image.py,sha256=CwRdB9K-38Ghsezg3B7_daPtFtsvDcV2hovaMRUjueg,6495
-examples/text_2_image_diffusers.py,sha256=oIVS--ovy3FQ8qwynY3bnoDRkQPDzVXwyozbZnrQ4fc,14398
-examples/text_and_image_2_audio.py,sha256=QLvSsLff8VZZa7k7K1EFGlPpQWZy07zM4Fnli5btAl0,2074
-examples/text_gen.py,sha256=_feyK65OQKG5j3htEXXFvC1bYTamZSOhkrz4yOn9PK4,1053
-examples/text_gen_system_prompt.py,sha256=jRQeGe1IVu_zRHX09CFiDYi7WrK9Zd5FlMqC_gnVH-g,1018
-examples/article_summary/article_summary.py,sha256=CR8mCBNcZEVCR-q34uOmrJyMlG-xk4HkMbsV-TOZEnk,1978
-examples/console_discussion/console_app.py,sha256=56ar4EKczbVTXMhTLUet5mj5WCY77pm_egZnKGQeXjs,11509
-examples/deep_analyze/deep_analyse.py,sha256=fZNmDrfEAuxEAfdbjAgJYIh1k6wbiuZ4RvwHRvtyUs8,971
-examples/deep_analyze/deep_analyze_multiple_files.py,sha256=fOryShA33P4IFxcxUDe-nJ2kW0v9w9yW8KsToS3ETl8,1032
-examples/generate_and_speak/generate_and_speak.py,sha256=RAlvRwtEKXCh894l9M3iQbADe8CvF5N442jtRurK02I,13908
-examples/generate_game_sfx/generate_game_fx.py,sha256=MgLNGi4hGBRoyr4bqYuCUdCSqd-ldDVfF0VSDUjgzsg,10467
-examples/lollms_chat/calculator.py,sha256=0mIakM3XehyUaoCsBS0l3eDTrqc81E88itfdHmdMYYc,2484
-examples/lollms_chat/derivative.py,sha256=1a4LKx_mD6Mq9oh-iGON0nfINP3Ul_TUSeDjEX7vSzE,1896
-examples/lollms_chat/test_openai_compatible_with_lollms_chat.py,sha256=JsooHpu7Hk4HSZNXvlL6N76-K6eFyq4gOH1lcsv_BcA,419
-examples/mcp_examples/external_mcp.py,sha256=swx1KCOz6jk8jGTAycq-xu7GXPAhRMDe1x--SKocugE,13371
-examples/mcp_examples/local_mcp.py,sha256=w40dgayvHYe01yvekEE0LjcbkpwKjWwJ-9v4_wGYsUk,9113
-examples/mcp_examples/openai_mcp.py,sha256=7IEnPGPXZgYZyiES_VaUbQ6viQjenpcUxGiHE-pGeFY,11060
-examples/mcp_examples/run_remote_mcp_example_v2.py,sha256=bbNn93NO_lKcFzfIsdvJJijGx2ePFTYfknofqZxMuRM,14626
-examples/mcp_examples/run_standard_mcp_example.py,sha256=GSZpaACPf3mDPsjA8esBQVUsIi7owI39ca5avsmvCxA,9419
-examples/test_local_models/local_chat.py,sha256=slakja2zaHOEAUsn2tn_VmI4kLx6luLBrPqAeaNsix8,456
-lollms_client/__init__.py,sha256=47yhAwjjK7irs9nhUU68Bk1NDWYRMIP23TCDociofi8,1147
-lollms_client/lollms_config.py,sha256=goEseDwDxYJf3WkYJ4IrLXwg3Tfw73CXV2Avg45M_hE,21876
-lollms_client/lollms_core.py,sha256=gDhpB62AluEmbVFvPm7vdnZgP2hGBymDLun57K1jrOM,177352
-lollms_client/lollms_discussion.py,sha256=RDX4V5R4sKCxuTjL2ILyGrDlHlyefFPf3x98_mnra3M,85608
-lollms_client/lollms_js_analyzer.py,sha256=01zUvuO2F_lnUe_0NLxe1MF5aHE1hO8RZi48mNPv-aw,8361
-lollms_client/lollms_llm_binding.py,sha256=8jJ8TPtkxh6TSKaYdrUrEI9ulY9gvN1RGHIrbPOWsDs,35963
-lollms_client/lollms_mcp_binding.py,sha256=0rK9HQCBEGryNc8ApBmtOlhKE1Yfn7X7xIQssXxS2Zc,8933
-lollms_client/lollms_mcp_security.py,sha256=FhVTDhSBjksGEZnopVnjFmEF5dv7D8bBTqoaj4BiF0E,3562
-lollms_client/lollms_personality.py,sha256=O-9nqZhazcITOkxjT24ENTxTmIoZLgqIsQ9WtWs0Id0,8719
-lollms_client/lollms_python_analyzer.py,sha256=7gf1fdYgXCOkPUkBAPNmr6S-66hMH4_KonOMsADASxc,10246
-lollms_client/lollms_stt_binding.py,sha256=jAUhLouEhh2hmm1bK76ianfw_6B59EHfY3FmLv6DU-g,5111
-lollms_client/lollms_tti_binding.py,sha256=afO0-d-Kqsmh8UHTijTvy6dZAt-XDB6R-IHmdbf-_fs,5928
-lollms_client/lollms_ttm_binding.py,sha256=FjVVSNXOZXK1qvcKEfxdiX6l2b4XdGOSNnZ0utAsbDg,4167
-lollms_client/lollms_tts_binding.py,sha256=5cJYECj8PYLJAyB6SEH7_fhHYK3Om-Y3arkygCnZ24o,4342
-lollms_client/lollms_ttv_binding.py,sha256=KkTaHLBhEEdt4sSVBlbwr5i_g_TlhcrwrT-7DjOsjWQ,4131
-lollms_client/lollms_types.py,sha256=0iSH1QHRRD-ddBqoL9EEKJ8wWCuwDUlN_FrfbCdg7Lw,3522
-lollms_client/lollms_utilities.py,sha256=3DAsII2X9uhRzRL-D0QlALcEdRg82y7OIL4yHVF32gY,19446
-lollms_client/llm_bindings/__init__.py,sha256=9sWGpmWSSj6KQ8H4lKGCjpLYwhnVdL_2N7gXCphPqh4,14
-lollms_client/llm_bindings/azure_openai/__init__.py,sha256=8C-gXoVa-OI9FmFM3PaMgrTfzqCLbs4f7CHJHxKuAR8,16675
-lollms_client/llm_bindings/claude/__init__.py,sha256=CsWILXAFytXtxp1ZAoNwq8KycW0POQ2MCmpT6Bz0Hd0,24877
-lollms_client/llm_bindings/gemini/__init__.py,sha256=ZflZVwAkAa-GfctuehOWIav977oTCdXUisQy253PFsk,21611
-lollms_client/llm_bindings/grok/__init__.py,sha256=5tIf3348RgAEaSp6FdG-LM9N8R7aR0t7OFspHf3XATs,23141
-lollms_client/llm_bindings/groq/__init__.py,sha256=qDCZXIzldblNHyb3LfcZkM0a8m1hSWilb2m9l-xvkZ8,12180
-lollms_client/llm_bindings/hugging_face_inference_api/__init__.py,sha256=PxgeRqT8dpa9GZoXwtSncy9AUgAN2cDKrvp_nbaWq0E,14027
-lollms_client/llm_bindings/litellm/__init__.py,sha256=pNkwyRPeENvTM4CDh6Pj3kQfxHfhX2pvXhGJDjKjp30,12340
-lollms_client/llm_bindings/llamacpp/__init__.py,sha256=6LHrsSD_4Xdq57sJaGf8gSndmG4hHcOS9wlsfa62TPQ,58650
-lollms_client/llm_bindings/lollms/__init__.py,sha256=scGHEKzlGX5fw2XwefVicsf28GrwgN3wU5nl4EPJ_Sk,24424
-lollms_client/llm_bindings/lollms_webui/__init__.py,sha256=Thoq3PJR2e03Y2Kd_FBb-DULJK0zT5-2ID1YIJLcPlw,17864
-lollms_client/llm_bindings/mistral/__init__.py,sha256=DFQAcrUR69oF1FhtKi1glMXqCF4jhqN2ctRN19tmrcM,14193
-lollms_client/llm_bindings/ollama/__init__.py,sha256=dXKHIeQCS9pz5AS07GF1eWj3ieWiz3aFOtxOX7yojbs,41314
-lollms_client/llm_bindings/open_router/__init__.py,sha256=tcG8yWd4ULIhqrfMURqqJbnAEDw8H3zU500FSHkoXMM,14977
-lollms_client/llm_bindings/openai/__init__.py,sha256=Z0zNTfBgBGwkwArN375kBt4otrUTI_84pHgVuyuDy0c,26253
-lollms_client/llm_bindings/openllm/__init__.py,sha256=xv2XDhJNCYe6NPnWBboDs24AQ1VJBOzsTuMcmuQ6xYY,29864
-lollms_client/llm_bindings/pythonllamacpp/__init__.py,sha256=VNa6NXe7bY44Oev8r6q5XlQbxqWU2nBV62SFlsPPH78,31810
-lollms_client/llm_bindings/tensor_rt/__init__.py,sha256=nPaNhGRd-bsG0UlYwcEqjd_UagCMEf5VEbBUW-GWu6A,32203
-lollms_client/llm_bindings/transformers/__init__.py,sha256=9LkqEC5bp1zHgyeGEcPQ3_uqvEAEf_B4p9DztcBaC5w,37211
-lollms_client/llm_bindings/vllm/__init__.py,sha256=2NqeeqYWXNq1aNicdcAwN9DaoL4gq96GZ7hsKErfC6c,32187
-lollms_client/mcp_bindings/local_mcp/__init__.py,sha256=FxS-0PULS3TZScPLDbwHlfyW5joWucmMfn6b9ydyeM0,14341
-lollms_client/mcp_bindings/local_mcp/default_tools/file_writer/file_writer.py,sha256=2pkt1JcEKj61lIA5zuW3s6qkdpQN5rKfooo7bnebx24,3061
-lollms_client/mcp_bindings/local_mcp/default_tools/generate_image_from_prompt/generate_image_from_prompt.py,sha256=THtZsMxNnXZiBdkwoBlfbWY2C5hhDdmPtnM-8cSKN6s,9488
-lollms_client/mcp_bindings/local_mcp/default_tools/internet_search/internet_search.py,sha256=PLC31-D04QKTOTb1uuCHnrAlpysQjsk89yIJngK0VGc,4586
-lollms_client/mcp_bindings/local_mcp/default_tools/python_interpreter/python_interpreter.py,sha256=McDCBVoVrMDYgU7EYtyOY7mCk1uEeTea0PSD69QqDsQ,6228
-lollms_client/mcp_bindings/remote_mcp/__init__.py,sha256=6ebENOqO-oUk3IpitVyiMGRICSl_X5DKKaGG52BdiT8,20388
-lollms_client/mcp_bindings/standard_mcp/__init__.py,sha256=zpF4h8cTUxoERI-xcVjmS_V772LK0V4jegjz2k1PK98,31658
-lollms_client/stt_bindings/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-lollms_client/stt_bindings/lollms/__init__.py,sha256=jBz3285atdPRqQe9ZRrb-AvjqKRB4f8tjLXjma0DLfE,6082
-lollms_client/stt_bindings/whisper/__init__.py,sha256=vrua7fLwDId9_WiH4y2gXOE0hy3Gr2Ig-z5ScIT2bHI,15447
-lollms_client/stt_bindings/whispercpp/__init__.py,sha256=B45lOn5rSoHXJSG9duPzBEPBOoNTrDzpCUdOL8KHaDM,21874
-lollms_client/tti_bindings/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-lollms_client/tti_bindings/dalle/__init__.py,sha256=V-rODzcwaN9eg0wX8DGOQXXcDtnYYDWYD6t6pFNTo0w,23628
-lollms_client/tti_bindings/diffusers/__init__.py,sha256=Hcu3F3KU8h6b_T4Dpl98OsQrsUoecd_z7xOVIEyYoDc,37679
-lollms_client/tti_bindings/gemini/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-lollms_client/tti_bindings/lollms/__init__.py,sha256=GJShFW6Y8MrfM9PXaPrdAp8OUpD6rraSRFt8ZOnrauo,8735
-lollms_client/ttm_bindings/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-lollms_client/ttm_bindings/audiocraft/__init__.py,sha256=a0k6wTrHth6GaVOiNnVboeFY3oKVvCQPbQlqO38XEyc,14328
-lollms_client/ttm_bindings/bark/__init__.py,sha256=Pr3ou2a-7hNYDqbkxrAbghZpO5HvGUhz7e-7VGXIHHA,18976
-lollms_client/ttm_bindings/lollms/__init__.py,sha256=DU3WLmJaWNM1NAMtJsnaFo4Y9wlfc675M8aUiaLnojA,3143
-lollms_client/tts_bindings/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-lollms_client/tts_bindings/bark/__init__.py,sha256=cpnmr6rmXcNdy4ib_5UHAbUP5oGoMJwB931_vU6VI14,19480
-lollms_client/tts_bindings/lollms/__init__.py,sha256=8x2_T9XscvISw2TiaLoFxvrS7TIsVLdqbwSc04cX-wc,7164
-lollms_client/tts_bindings/piper_tts/__init__.py,sha256=0IEWG4zH3_sOkSb9WbZzkeV5Lvhgp5Gs2-2GN51MTjA,18930
-lollms_client/tts_bindings/xtts/__init__.py,sha256=FgcdUH06X6ZR806WQe5ixaYx0QoxtAcOgYo87a2qxYc,18266
-lollms_client/ttv_bindings/__init__.py,sha256=UZ8o2izQOJLQgtZ1D1cXoNST7rzqW22rL2Vufc7ddRc,3141
-lollms_client/ttv_bindings/lollms/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-lollms_client-0.33.0.dist-info/licenses/LICENSE,sha256=HrhfyXIkWY2tGFK11kg7vPCqhgh5DcxleloqdhrpyMY,11558
-test/test_lollms_discussion.py,sha256=KxTsV1bPdNz8QqZd7tIof9kTWkeXLUtAMU08BQmoY6U,16829
-lollms_client-0.33.0.dist-info/METADATA,sha256=kPveiEdFJW9h1_mj_DYJJ9Gq1-upw6YaFd2a70lLhWo,38717
-lollms_client-0.33.0.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-lollms_client-0.33.0.dist-info/top_level.txt,sha256=1jIpjTnOSGEGtIW2rGAFM6tVRzgsDdMOiox_SmDH_zw,28
-lollms_client-0.33.0.dist-info/RECORD,,

test/test_lollms_discussion.py

@@ -1,368 +0,0 @@
-# test_lollms_discussion.py
-
-import pytest
-from unittest.mock import MagicMock, patch
-from pathlib import Path
-import base64
-import json
-
-# Assuming your class definitions are in a file named 'lollms_discussion_module.py'
-# Adjust the import path as necessary.
-from lollms_client import LollmsDataManager, LollmsDiscussion
-
-# --- Test Fixtures and Helpers ---
-
-@pytest.fixture
-def mock_lollms_client():
-    """Creates a mock LollmsClient that simulates its core functions."""
-    client = MagicMock()
-    # Simulate token counting with a simple word count
-    client.count_tokens.side_effect = lambda text: len(str(text).split())
-    # Simulate image token counting with a fixed value
-    client.count_image_tokens.return_value = 75
-    # Simulate structured content generation
-    client.generate_structured_content.return_value = {"title": "A Mocked Title"}
-    # Simulate text generation for summary and memory
-    client.generate_text.return_value = "This is a mock summary."
-    # Simulate chat generation
-    client.chat.return_value = "This is a mock chat response."
-    client.remove_thinking_blocks.return_value = "This is a mock chat response."
-    return client
-
-@pytest.fixture
-def db_manager(tmp_path):
-    """Creates a LollmsDataManager with a temporary database."""
-    db_file = tmp_path / "test_discussions.db"
-    return LollmsDataManager(f'sqlite:///{db_file}')
-
-def create_dummy_image_b64(text="Test"):
-    """Generates a valid base64 encoded dummy PNG image string."""
-    try:
-        from PIL import Image, ImageDraw
-        img = Image.new('RGB', (60, 30), color = 'red')
-        d = ImageDraw.Draw(img)
-        d.text((10,10), text, fill='white')
-        
-        # In-memory saving to avoid disk I/O in test
-        from io import BytesIO
-        buffered = BytesIO()
-        img.save(buffered, format="PNG")
-        return base64.b64encode(buffered.getvalue()).decode('utf-8')
-    except ImportError:
-        # Fallback if Pillow is not installed
-        # This is a 1x1 transparent pixel
-        return "iVBORw0KGgoAAAANSUhEUgAAAAEAAAABCAQAAAC1HAwCAAAAC0lEQVR42mNkYAAAAAYAAjCB0C8AAAAASUVORK5CYII="
-
-@pytest.fixture
-def dummy_image_b64_1():
-    return create_dummy_image_b64("Image 1")
-
-@pytest.fixture
-def dummy_image_b64_2():
-    return create_dummy_image_b64("Image 2")
-
-
-# --- Test Class ---
-
-class TestLollmsDiscussion:
-
-    def test_creation_in_memory(self, mock_lollms_client):
-        """Tests that an in-memory discussion can be created."""
-        disc = LollmsDiscussion.create_new(lollms_client=mock_lollms_client)
-        assert disc.id is not None
-        assert not disc._is_db_backed
-
-    def test_creation_db_backed(self, mock_lollms_client, db_manager):
-        """Tests that a database-backed discussion can be created."""
-        metadata = {"title": "DB Test"}
-        disc = LollmsDiscussion.create_new(
-            lollms_client=mock_lollms_client,
-            db_manager=db_manager,
-            discussion_metadata=metadata
-        )
-        assert disc._is_db_backed
-        assert disc.metadata["title"] == "DB Test"
-
-    def test_add_and_get_message(self, mock_lollms_client):
-        """Tests basic message addition and retrieval."""
-        disc = LollmsDiscussion.create_new(lollms_client=mock_lollms_client)
-        disc.add_message(sender="user", content="Hello, world!")
-        
-        assert disc.active_branch_id is not None
-        messages = disc.get_messages()
-        assert len(messages) == 1
-        assert messages[0].sender == "user"
-        assert messages[0].content == "Hello, world!"
-
-    def test_data_zones_and_export(self, mock_lollms_client):
-        """Tests that data zones are correctly included in exports."""
-        disc = LollmsDiscussion.create_new(lollms_client=mock_lollms_client)
-        disc.system_prompt = "You are a tester."
-        disc.memory = "Remember this."
-        disc.user_data_zone = "User is a dev."
-        disc.add_message(sender="user", content="Test prompt.")
-
-        exported_md = disc.export("markdown")
-        assert "You are a tester." in exported_md
-        assert "-- Memory --" in exported_md
-        assert "Remember this." in exported_md
-        assert "-- User Data Zone --" in exported_md
-        assert "User is a dev." in exported_md
-        assert "**User**: Test prompt." in exported_md
-
-    def test_message_image_activation(self, mock_lollms_client, dummy_image_b64_1, dummy_image_b64_2):
-        """Tests activating and deactivating images on a specific message."""
-        disc = LollmsDiscussion.create_new(lollms_client=mock_lollms_client)
-        msg = disc.add_message(sender="user", content="Look", images=[dummy_image_b64_1, dummy_image_b64_2])
-
-        # Initially, both images should be active
-        assert len(msg.get_active_images()) == 2
-        
-        # Deactivate the first image
-        msg.toggle_image_activation(0, active=False)
-        assert len(msg.get_active_images()) == 1
-        all_imgs = msg.get_all_images()
-        assert not all_imgs[0]["active"]
-        assert all_imgs[1]["active"]
-
-        # Toggle the first image back on
-        msg.toggle_image_activation(0)
-        assert len(msg.get_active_images()) == 2
-        assert msg.get_all_images()[0]["active"]
-
-    def test_discussion_image_management(self, mock_lollms_client, dummy_image_b64_1, dummy_image_b64_2):
-        """Tests adding, toggling, and retrieving discussion-level images."""
-        disc = LollmsDiscussion.create_new(lollms_client=mock_lollms_client)
-        
-        # Add two images
-        disc.add_discussion_image(dummy_image_b64_1)
-        disc.add_discussion_image(dummy_image_b64_2)
-
-        # Check initial state
-        assert len(disc.get_discussion_images()) == 2
-        assert disc.get_discussion_images()[0]["active"]
-        assert disc.get_discussion_images()[1]["active"]
-
-        # Deactivate the second image
-        disc.toggle_discussion_image_activation(1, active=False)
-        
-        # Verify changes
-        all_disc_imgs = disc.get_discussion_images()
-        assert len(all_disc_imgs) == 2
-        assert all_disc_imgs[0]["active"]
-        assert not all_disc_imgs[1]["active"]
-
-    def test_export_with_multimodal_system_prompt(self, mock_lollms_client, dummy_image_b64_1):
-        """Tests that active discussion images are included in the system prompt for export."""
-        disc = LollmsDiscussion.create_new(lollms_client=mock_lollms_client)
-        disc.system_prompt = "Analyze this image."
-        disc.add_discussion_image(dummy_image_b64_1) # This one is active
-        disc.add_discussion_image(create_dummy_image_b64("Inactive"))
-        disc.toggle_discussion_image_activation(1, active=False) # This one is inactive
-
-        disc.add_message(sender="user", content="What do you see?")
-        
-        openai_export = disc.export("openai_chat")
-        
-        system_message = openai_export[0]
-        assert system_message["role"] == "system"
-        
-        content_parts = system_message["content"]
-        assert len(content_parts) == 2 
-        
-        text_part = next(p for p in content_parts if p["type"] == "text")
-        image_part = next(p for p in content_parts if p["type"] == "image_url")
-
-        assert text_part["text"] == "Analyze this image."
-        assert image_part["image_url"]["url"].endswith(dummy_image_b64_1)
-
-    def test_auto_title_updates_metadata(self, mock_lollms_client, db_manager):
-        """Tests that auto_title calls the LLM and updates metadata."""
-        disc = LollmsDiscussion.create_new(lollms_client=mock_lollms_client, db_manager=db_manager)
-        disc.add_message(sender="user", content="This is a test to generate a title.")
-
-        title = disc.auto_title()
-        
-        mock_lollms_client.generate_structured_content.assert_called_once()
-        assert title == "A Mocked Title"
-        assert disc.metadata['title'] == "A Mocked Title"
-
-    def test_delete_branch(self, mock_lollms_client, db_manager):
-        """Tests deleting a message and its descendants."""
-        disc = LollmsDiscussion.create_new(lollms_client=mock_lollms_client, db_manager=db_manager)
-        
-        msg_a = disc.add_message(sender="user", content="A")
-        msg_b = disc.add_message(sender="user", content="B", parent_id=msg_a.id)
-        msg_c = disc.add_message(sender="user", content="C", parent_id=msg_b.id)
-        
-        assert disc.active_branch_id == msg_c.id
-        assert len(disc.get_messages()) == 3
-        
-        disc.delete_branch(msg_b.id)
-
-        assert disc.active_branch_id == msg_a.id
-        
-        remaining_messages = disc.get_messages()
-        assert len(remaining_messages) == 1
-        assert remaining_messages[0].id == msg_a.id
-
-    def test_branching_and_switching(self, mock_lollms_client):
-        """Tests creating and switching between conversational branches."""
-        disc = LollmsDiscussion.create_new(lollms_client=mock_lollms_client)
-        
-        root_msg = disc.add_message(sender="user", content="Hello")
-        ai_msg_1 = disc.add_message(sender="assistant", content="Hi there!")
-        
-        disc.switch_to_branch(root_msg.id)
-        ai_msg_2 = disc.add_message(sender="assistant", content="Greetings!")
-
-        assert len(disc._message_index) == 3
-        assert disc.active_branch_id == ai_msg_2.id
-
-        branch_1 = disc.get_branch(ai_msg_1.id)
-        assert [m.id for m in branch_1] == [root_msg.id, ai_msg_1.id]
-
-        branch_2 = disc.get_branch(ai_msg_2.id)
-        assert [m.id for m in branch_2] == [root_msg.id, ai_msg_2.id]
-        
-    def test_regenerate_branch(self, mock_lollms_client, db_manager):
-        """Tests that regeneration deletes the old AI message and creates a new one."""
-        disc = LollmsDiscussion.create_new(lollms_client=mock_lollms_client, db_manager=db_manager)
-        
-        user_msg = disc.add_message(sender="user", content="Tell me a joke.")
-        ai_msg = disc.add_message(sender="assistant", content="Why did the scarecrow win an award?")
-
-        original_ai_id = ai_msg.id
-        
-        result = disc.regenerate_branch()
-        new_ai_msg = result["ai_message"]
-        
-        assert new_ai_msg.id != original_ai_id
-        assert new_ai_msg.content == "This is a mock chat response."
-        
-        messages = disc.get_messages(new_ai_msg.id)
-        assert len(messages) == 2
-        assert original_ai_id not in [m.id for m in messages]
-        
-        assert original_ai_id in disc._messages_to_delete_from_db
-
-    def test_summarize_and_prune(self, mock_lollms_client):
-        """Tests that the discussion is pruned when the token limit is exceeded."""
-        disc = LollmsDiscussion.create_new(lollms_client=mock_lollms_client)
-        disc.max_context_size = 20
-        
-        disc.add_message(sender="user", content="This is the first long message to establish history.")
-        disc.add_message(sender="assistant", content="I see. This is the second long message.")
-        m3 = disc.add_message(sender="user", content="Third message.")
-        disc.add_message(sender="assistant", content="Fourth message.")
-        disc.add_message(sender="user", content="Fifth message.")
-        disc.add_message(sender="assistant", content="Sixth message.")
-
-        disc.summarize_and_prune(max_tokens=20, preserve_last_n=4)
-        
-        mock_lollms_client.generate_text.assert_called_once()
-        assert "This is a mock summary." in disc.pruning_summary
-        assert disc.pruning_point_id == m3.id
-        
-    def test_memorize_updates_memory_zone(self, mock_lollms_client):
-        """Tests that memorize calls the LLM and appends to the memory data zone."""
-        disc = LollmsDiscussion.create_new(lollms_client=mock_lollms_client)
-        disc.add_message(sender="user", content="My favorite color is blue.")
-        
-        disc.memorize()
-        
-        mock_lollms_client.generate_text.assert_called_once()
-        assert "This is a mock summary." in disc.memory
-        assert "Memory entry from" in disc.memory
-        
-    def test_get_context_status(self, mock_lollms_client, dummy_image_b64_1, dummy_image_b64_2):
-        """Tests the token calculation in get_context_status."""
-        disc = LollmsDiscussion.create_new(lollms_client=mock_lollms_client)
-        disc.system_prompt = "You are a helpful AI."
-        disc.add_discussion_image(dummy_image_b64_1) # 1 active discussion image
-        
-        msg = disc.add_message(sender="user", content="Hello there!", images=[dummy_image_b64_1, dummy_image_b64_2])
-        msg.toggle_image_activation(1, active=False) # 1 active message image
-        
-        status = disc.get_context_status()
-
-        # System context: "!@>system:\nYou are a helpful AI.\n" -> 6 tokens
-        system_tokens = status["zones"]["system_context"]["tokens"]
-        assert system_tokens == 6
-
-        # Discussion Images: 1 active discussion image -> 75 tokens
-        discussion_image_zone = status["zones"]["discussion_images"]
-        assert discussion_image_zone["tokens"] == 75
-
-        # History:
-        #   - Text: "!@>user:\nHello there!\n(2 image(s) attached)\n" -> 6 tokens
-        #   - Image: 1 active message image -> 75 tokens
-        history_zone = status["zones"]["message_history"]
-        assert history_zone["breakdown"]["text_tokens"] == 6
-        assert history_zone["breakdown"]["image_tokens"] == 75
-        assert history_zone["tokens"] == 81 # 6 + 75
-        
-        # Total should be sum of system, discussion images, and history
-        assert status["current_tokens"] == 6 + 75 + 81 # 162
-
-    # --- NEW PERSISTENCE AND SEPARATION TESTS ---
-
-    def test_image_separation_and_context(self, mock_lollms_client, dummy_image_b64_1, dummy_image_b64_2):
-        """Tests that discussion and message images are distinct but aggregated correctly."""
-        disc = LollmsDiscussion.create_new(lollms_client=mock_lollms_client)
-        
-        # Add one image to the discussion, one to the message
-        disc.add_discussion_image(dummy_image_b64_1)
-        msg = disc.add_message(sender="user", content="Check these out", images=[dummy_image_b64_2])
-
-        # Test separation
-        assert len(disc.get_discussion_images()) == 1
-        assert disc.get_discussion_images()[0]['data'] == dummy_image_b64_1
-        assert len(msg.get_all_images()) == 1
-        assert msg.get_all_images()[0]['data'] == dummy_image_b64_2
-
-        # Test aggregation for context
-        all_active_images = disc.get_active_images()
-        assert len(all_active_images) == 2
-        assert dummy_image_b64_1 in all_active_images
-        assert dummy_image_b64_2 in all_active_images
-
-    def test_full_multimodal_persistence(self, mock_lollms_client, db_manager, dummy_image_b64_1, dummy_image_b64_2):
-        """End-to-end test for persisting both discussion and message images and their states."""
-        disc_id = None
-        
-        # --- Scope 1: Create and Save ---
-        with db_manager.get_session():
-            disc = LollmsDiscussion.create_new(lollms_client=mock_lollms_client, db_manager=db_manager)
-            disc_id = disc.id
-            
-            # Add and deactivate a discussion image
-            disc.add_discussion_image(dummy_image_b64_1)
-            disc.toggle_discussion_image_activation(0, active=False)
-            
-            # Add and deactivate a message image
-            msg = disc.add_message(sender="user", content="Multimodal test", images=[dummy_image_b64_2])
-            msg.toggle_image_activation(0, active=False)
-            
-            disc.commit()
-            disc.close()
-
-        # --- Scope 2: Reload and Verify ---
-        reloaded_disc = db_manager.get_discussion(mock_lollms_client, disc_id)
-        assert reloaded_disc is not None
-        
-        # Verify discussion image state
-        reloaded_disc_imgs = reloaded_disc.get_discussion_images()
-        assert len(reloaded_disc_imgs) == 1
-        assert reloaded_disc_imgs[0]['data'] == dummy_image_b64_1
-        assert not reloaded_disc_imgs[0]['active'] # Must be inactive
-
-        # Verify message image state
-        reloaded_msg = reloaded_disc.get_messages()[0]
-        reloaded_msg_imgs = reloaded_msg.get_all_images()
-        assert len(reloaded_msg_imgs) == 1
-        assert reloaded_msg_imgs[0]['data'] == dummy_image_b64_2
-        assert not reloaded_msg_imgs[0]['active'] # Must be inactive
-
-        # Verify that get_active_images returns nothing
-        assert len(reloaded_disc.get_active_images()) == 0