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

lollms_client-1.0.0.dist-info/RECORD

@@ -0,0 +1,69 @@
+lollms_client/__init__.py,sha256=TB277pr7BJ2eH1ezWZ-B0_8lxMBqjQYPhc2jUJLOASg,1146
+lollms_client/lollms_config.py,sha256=goEseDwDxYJf3WkYJ4IrLXwg3Tfw73CXV2Avg45M_hE,21876
+lollms_client/lollms_core.py,sha256=68EFgqGyw0cpJQOhJZrRzogd20j1zSghwLBie6kyJGI,167221
+lollms_client/lollms_discussion.py,sha256=wkadV6qiegxOzukMVn5vukdeJivnlyygSzZBkzOi9Gc,106714
+lollms_client/lollms_js_analyzer.py,sha256=01zUvuO2F_lnUe_0NLxe1MF5aHE1hO8RZi48mNPv-aw,8361
+lollms_client/lollms_llm_binding.py,sha256=TBdFNNktpIUSUd4mlUHeNUPUQeLRWKBj80UTJ-YdwBg,24940
+lollms_client/lollms_mcp_binding.py,sha256=psb27A23VFWDfZsR2WUbQXQxiZDW5yfOak6ZtbMfszI,10222
+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=XBDwct0nkvWfpo1J9J9lTOszH_c_4IiCYxEsG6aJLo0,16501
+lollms_client/llm_bindings/claude/__init__.py,sha256=tzt9sR-9WlkgTgDBOtV708ZmuBjMm55fEYhurMnfXO4,24669
+lollms_client/llm_bindings/gemini/__init__.py,sha256=bMzjVo_LU5uC493q4DVjq-WVotzgTZN1T_w0Qqv5aDg,21386
+lollms_client/llm_bindings/grok/__init__.py,sha256=tVIIl2uXpBYD7ia3k8JqYM8uvAVYlp-mtG-8D3LFVS8,22929
+lollms_client/llm_bindings/groq/__init__.py,sha256=EGrMh9vuCoM4pskDw8ydfsAWYgEb423e9HBwqdO2JQc,12120
+lollms_client/llm_bindings/hugging_face_inference_api/__init__.py,sha256=SFcj5XQTDmN9eR4of82IgQa9iRYZaGlF6rMlF5S5wWg,13938
+lollms_client/llm_bindings/litellm/__init__.py,sha256=lRH4VfZMUG5JCCj6a7hk2PTfSyDowAu-ujLOM-XPl-8,12756
+lollms_client/llm_bindings/llamacpp/__init__.py,sha256=4CbNYpfquVEgfsxuLsxQta_dZRSpbSBL-VWhyDMdBAc,59379
+lollms_client/llm_bindings/lollms/__init__.py,sha256=a4gNH4axiDgsri8NGAcq0OitgYdnzBDLNkzUMhkFArA,24781
+lollms_client/llm_bindings/lollms_webui/__init__.py,sha256=iuDfhZZoLC-PDEPLHrcjk5-962S5c7OeCI7PMdJxI_A,17753
+lollms_client/llm_bindings/mistral/__init__.py,sha256=cddz9xIj8NRFLKHe2JMxzstpUrNIu5s9juci3mhiHfo,14133
+lollms_client/llm_bindings/ollama/__init__.py,sha256=d61pSEWlJ2KOvnaztji2wblvadu0oTelEJeHG4IcL9I,41193
+lollms_client/llm_bindings/open_router/__init__.py,sha256=cAFWtCWJx0WjIe1w2JReCf6WlAZjrXYA4jZ8l3zqxMs,14915
+lollms_client/llm_bindings/openai/__init__.py,sha256=J8v7XU9TrvXJd1ffwhYkya5YeXxWnNiFuNBAwRfoHDk,26066
+lollms_client/llm_bindings/openllm/__init__.py,sha256=RC9dVeopslS-zXTsSJ7VC4iVsKgZCBwfmccmr_LCHA0,29971
+lollms_client/llm_bindings/pythonllamacpp/__init__.py,sha256=ZTuVa5ngu9GPVImjs_g8ArV7Bx7a1Rze518Tz8AFJ3U,31807
+lollms_client/llm_bindings/tensor_rt/__init__.py,sha256=xiT-JAyNI_jo6CE0nle9Xoc7U8-UHAfEHrnCwmDTiOE,32023
+lollms_client/llm_bindings/transformers/__init__.py,sha256=hEonNvmrVSc9YWg_1uVwxe31rC-fsjVGh6QvyBc0TEE,37598
+lollms_client/llm_bindings/vllm/__init__.py,sha256=pZGzCuBos1dzkbqfkAnSAIC2XUX-2BT25c15Yeh8Nwo,32721
+lollms_client/mcp_bindings/local_mcp/__init__.py,sha256=S_hEilM3WLt0u2uluhuJXMNAzTAnWC8FWteBP92V_2g,14336
+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=YpSclbNJDYVUe2W0H5Xki4gs_qqAqE95uCl_RYZsxPA,20406
+lollms_client/mcp_bindings/standard_mcp/__init__.py,sha256=wJQofr4zS5RIS9V5_WuMMFsJxSDJgXDW3PQPX1hlx6g,31519
+lollms_client/stt_bindings/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+lollms_client/stt_bindings/lollms/__init__.py,sha256=9Vmn1sQQZKLGLe7nZnc-0LnNeSY8r9xw3pYZF-wVtPo,5889
+lollms_client/stt_bindings/whisper/__init__.py,sha256=1Ej67GdRKBy1bba14jMaYDYHiZkxJASkWm5eF07ztDQ,15363
+lollms_client/stt_bindings/whispercpp/__init__.py,sha256=xSAQRjAhljak3vWCpkP0Vmdb6WmwTzPjXyaIB85KLGU,21439
+lollms_client/tti_bindings/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+lollms_client/tti_bindings/dalle/__init__.py,sha256=KWUow9z-xR_am_gHg3kGZ_5u_nnF88BB_0JvyfuOG_s,23456
+lollms_client/tti_bindings/diffusers/__init__.py,sha256=vjzuJfOAIhoiwomuIZvWpxmcM5UxpTwqTbD04E-zi3Y,37786
+lollms_client/tti_bindings/gemini/__init__.py,sha256=iFJMAR944nRDhXA7SdFjsCOqJ1dPSqyTG4eMFNg6Vwc,9734
+lollms_client/tti_bindings/lollms/__init__.py,sha256=5Tnsn4b17djvieQkcjtIDBm3qf0pg5ZWWov-4_2wmo0,8762
+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-1.0.0.dist-info/licenses/LICENSE,sha256=HrhfyXIkWY2tGFK11kg7vPCqhgh5DcxleloqdhrpyMY,11558
+lollms_client-1.0.0.dist-info/METADATA,sha256=qqfmnRtnMRYdgvQTzwjlh8v_lAaOGPJ5H0Sf-X3j1dk,58549
+lollms_client-1.0.0.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+lollms_client-1.0.0.dist-info/top_level.txt,sha256=Bk_kz-ri6Arwsk7YG-T5VsRorV66uVhcHGvb_g2WqgE,14
+lollms_client-1.0.0.dist-info/RECORD,,

{lollms_client-0.33.0.dist-info → lollms_client-1.0.0.dist-info}/top_level.txt

@@ -1,3 +1 @@
-examples
 lollms_client
-test

examples/article_summary/article_summary.py

@@ -1,58 +0,0 @@
-from lollms_client import LollmsClient
-import pipmaster as pm
-from ascii_colors import ASCIIColors
-if not pm.is_installed("docling"):
-    pm.install("docling")
-from docling.document_converter import DocumentConverter
-
-ASCIIColors.set_log_file("log.log")
-
-lc = LollmsClient()
-# Create prompts for each section
-article_url = "https://arxiv.org/pdf/2109.09572"
-converter = DocumentConverter()
-result = converter.convert(article_url)
-article_text = result.document.export_to_markdown()
-
-ASCIIColors.info("Article loaded successfully")
-
-# Use the sequential_summarize method from lollms
-summary = lc.sequential_summarize(
-                    article_text,
-                    """
-Extract the following information if present in the chunk:
-
-1. **Title**: 
-   - Found in text chunk number 1 at the beginning. It should be followed by # or ##
-   - Copy exactly as presented; do not interpret.
-   - Never alter this if already in the memory. This is important
-
-2. **Authors**: 
-   - Listed in text chunk number 1 at the beginning.
-   - If you fail to find the authors keep this empty.
-   - Copy exactly as presented; do not interpret.
-   - Never alter this if already in the memory. This is important
-
-3. **Summary**: 
-   - Provide a concise but detailed summary of the article by adding ned information from the text chunk to the memory content.
-
-4. **Results**: 
-   - Extract quantified results if available.
-
-Ensure that any information already in memory is retained unless explicitly updated by the current chunk.
-""",
-                    "markdown",
-                    """Write a final markdown with these sections:
-## Title
-## Authors
-## Summary
-## Results
-                    """,
-                    ctx_size=128000,
-                    chunk_size=4096,
-                    bootstrap_chunk_size=1024,
-                    bootstrap_steps=1,
-                    debug = True
-                )
-
-ASCIIColors.yellow(summary)

examples/console_discussion/console_app.py

@@ -1,266 +0,0 @@
-import os
-import shutil
-import subprocess
-import sys
-import json
-from pathlib import Path
-from typing import Optional, List, Dict, Any
-
-# Correctly import all necessary classes from the lollms_client package
-from lollms_client import LollmsClient, LollmsDataManager, LollmsDiscussion, LollmsPersonality
-from ascii_colors import ASCIIColors, trace_exception
-
-# --- Configuration ---
-MAX_CONTEXT_SIZE_FOR_TEST = 2048 # Increased for agentic turns
-
-# Database and workspace configuration
-WORKSPACE_DIR = Path("./test_workspace_agentic")
-DATABASE_PATH = f"sqlite:///{WORKSPACE_DIR / 'test_discussion_agentic.db'}"
-DISCUSSION_ID = "console-agentic-test-1" # Use a fixed ID for easy resumption
-
-# --- MOCK KNOWLEDGE BASE for RAG ---
-MOCK_KNOWLEDGE_BASE = {
-    "python_basics.md": [
-        {"chunk_id": 1, "text": "Python is a high-level, interpreted programming language known for its readability. It was created by Guido van Rossum and released in 1991."},
-        {"chunk_id": 2, "text": "Key features of Python include dynamic typing, garbage collection, and a large standard library. It supports procedural, object-oriented, and functional programming."},
-    ],
-    "javascript_info.js": [
-        {"chunk_id": 1, "text": "JavaScript is a scripting language for front-end web development. It is also used in back-end development (Node.js)."},
-        {"chunk_id": 2, "text": "Popular JavaScript frameworks include React, Angular, and Vue.js."},
-    ],
-    "ai_concepts.txt": [
-        {"chunk_id": 1, "text": "Retrieval Augmented Generation (RAG) is an AI framework for improving LLM responses by grounding the model on external knowledge sources."},
-    ]
-}
-
-# --- Dummy MCP Server Scripts ---
-TIME_SERVER_PY = """
-import asyncio
-from datetime import datetime
-from mcp.server.fastmcp import FastMCP
-mcp_server = FastMCP("TimeMCP", description="A server that provides the current time.", host="localhost",
-            port=9624,
-            log_level="DEBUG")
-@mcp_server.tool()
-def get_current_time(user_id: str = "unknown"):
-    return {"time": datetime.now().isoformat(), "user_id": user_id}
-if __name__ == "__main__": mcp_server.run(transport="streamable-http")
-"""
-CALCULATOR_SERVER_PY = """
-import asyncio
-from typing import List, Union
-from mcp.server.fastmcp import FastMCP
-mcp_server = FastMCP("TimeMCP", description="A server that provides the current time.", host="localhost",
-            port=9625,
-            log_level="DEBUG")
-@mcp_server.tool()
-def add_numbers(numbers: List[Union[int, float]]):
-    if not isinstance(numbers, list): return {"error": "Input must be a list"}
-    return {"sum": sum(numbers)}
-if __name__ == "__main__": mcp_server.run(transport="streamable-http")
-"""
-
-# --- RAG Mock Function ---
-def mock_rag_query_function(query_text: str, top_k: int = 3, **kwargs) -> List[Dict[str, Any]]:
-    ASCIIColors.magenta(f"\n  [MOCK RAG] Querying knowledge base for: '{query_text}'")
-    results = []
-    query_lower = query_text.lower()
-    for file_path, chunks in MOCK_KNOWLEDGE_BASE.items():
-        for chunk in chunks:
-            if any(word in chunk["text"].lower() for word in query_lower.split() if len(word) > 2):
-                results.append({"file_path": file_path, "chunk_text": chunk["text"]})
-    ASCIIColors.magenta(f"  [MOCK RAG] Found {len(results[:top_k])} relevant chunks.")
-    return results[:top_k]
-
-def start_mcp_servers():
-    """Starts the dummy MCP servers in the background."""
-    ASCIIColors.yellow("--- Starting background MCP servers ---")
-    server_dir = WORKSPACE_DIR / "mcp_servers"
-    server_dir.mkdir(exist_ok=True, parents=True)
-    
-    (server_dir / "time_server.py").write_text(TIME_SERVER_PY)
-    (server_dir / "calculator_server.py").write_text(CALCULATOR_SERVER_PY)
-
-    procs = []
-    procs.append(subprocess.Popen([sys.executable, str(server_dir / "time_server.py")], stdout=subprocess.DEVNULL, stderr=subprocess.DEVNULL))
-    procs.append(subprocess.Popen([sys.executable, str(server_dir / "calculator_server.py")], stdout=subprocess.DEVNULL, stderr=subprocess.DEVNULL))
-    ASCIIColors.yellow("--- MCP servers launched ---")
-    return procs
-
-def setup_client_and_discussion() -> LollmsDiscussion:
-    """Sets up the LollmsClient with MCP, the DB manager, and the discussion."""
-    print("--- Setting up Lollms Environment ---")
-    WORKSPACE_DIR.mkdir(exist_ok=True)
-
-    mcp_config = {
-        "servers_infos": {
-            "time_machine": {"server_url": "http://localhost:9624/mcp"},
-            "calc_unit": {"server_url": "http://localhost:9625/mcp"},
-        }
-    }
-
-    try:
-        client = LollmsClient(
-            "ollama",
-            model_name="mistral-nemo:latest",
-            mcp_binding_name="remote_mcp",
-            mcp_binding_config=mcp_config
-        )
-    except Exception as e:
-        trace_exception(e)
-        print("\n---FATAL ERROR---")
-        print("Could not initialize LollmsClient. Ensure Ollama is running and mcp is installed.")
-        exit()
-
-    print(f"-> Using model: {client.binding.model_name}")
-    print(f"-> Using MCP binding: {client.mcp.binding_name}")
-
-    db_manager = LollmsDataManager(db_path=DATABASE_PATH)
-    discussion = db_manager.get_discussion(client, DISCUSSION_ID)
-    
-    if discussion:
-        print(f"-> Resuming discussion (ID: {DISCUSSION_ID})")
-        discussion.max_context_size = MAX_CONTEXT_SIZE_FOR_TEST
-    else:
-        print(f"-> Creating new discussion (ID: {DISCUSSION_ID})")
-        discussion = LollmsDiscussion.create_new(
-            lollms_client=client,
-            db_manager=db_manager,
-            id=DISCUSSION_ID,
-            title="Console Agentic Test",
-            max_context_size=MAX_CONTEXT_SIZE_FOR_TEST
-        )
-
-    print("--- Setup Complete. Ready to chat! ---\n")
-    return discussion
-
-def print_help():
-    print("\n--- Commands ---")
-    print("!agent <prompt>  - Run a prompt using all available tools (MCP).")
-    print("!rag <prompt>    - Run a prompt using the mock knowledge base (RAG).")
-    print("!both <prompt>   - Run a prompt using both MCP tools and RAG.")
-    print("!status          - Show current discussion state (pruning, message count).")
-    print("!regen           - Regenerate the last AI response.")
-    print("!exit            - Exit the application.")
-    print("----------------\n")
-
-def print_agentic_results(response_dict):
-    """Renders a beautiful report of the agent's turn."""
-    ai_message = response_dict.get('ai_message')
-    if not ai_message:
-        return
-
-    ASCIIColors.cyan("\n" + "="*22 + " Agentic Turn Report " + "="*22)
-
-    # --- Final Answer ---
-    ASCIIColors.blue("\nFinal Answer:")
-    ASCIIColors.green(f"  {ai_message.content}")
-
-    # --- Agent's Internal Monologue (The Scratchpad) ---
-    if ai_message.scratchpad:
-        ASCIIColors.blue("\nAgent's Reasoning Log (Scratchpad):")
-        # Print scratchpad line by line for better color coding
-        for line in ai_message.scratchpad.split('\n'):
-            if line.startswith("### Step"):
-                ASCIIColors.yellow(line)
-            elif line.startswith("- **Action**:") or line.startswith("- **Result**:") or line.startswith("- **Error**:") :
-                ASCIIColors.magenta(line)
-            else:
-                print(line)
-
-    # --- Sources Used (from metadata) ---
-    if ai_message.metadata and "sources" in ai_message.metadata:
-        sources = ai_message.metadata.get("sources", [])
-        if sources:
-            ASCIIColors.blue("\nSources Consulted (RAG):")
-            for i, source in enumerate(sources):
-                print(f"  [{i+1}] Path: {source.get('file_path', 'N/A')}")
-                # Indent the content for readability
-                content = source.get('chunk_text', 'N/A').replace('\n', '\n      ')
-                print(f"      Content: \"{content}\"")
-    
-    ASCIIColors.cyan("\n" + "="*61 + "\n")
-
-def run_chat_console(discussion: LollmsDiscussion):
-    print_help()
-    while True:
-        user_input = input("You: ")
-        if not user_input:
-            continue
-
-        use_mcps_flag = False
-        use_data_store_flag = False
-        prompt = user_input
-
-        # --- Command Handling ---
-        if user_input.lower().startswith("!exit"):
-            break
-        elif user_input.lower().startswith("!help"):
-            print_help()
-            continue
-        elif user_input.lower().startswith("!status"):
-            # Assuming a print_status function exists
-            # print_status(discussion)
-            continue
-        elif user_input.lower().startswith("!regen"):
-            # Assuming a regenerate_branch method exists
-            # discussion.regenerate_branch(...)
-            continue
-        elif user_input.lower().startswith("!agent "):
-            use_mcps_flag = True
-            prompt = user_input[7:].strip()
-            ASCIIColors.yellow(f"Agentic MCP turn initiated for: '{prompt}'")
-        elif user_input.lower().startswith("!rag "):
-            use_data_store_flag = True
-            prompt = user_input[5:].strip()
-            ASCIIColors.yellow(f"Agentic RAG turn initiated for: '{prompt}'")
-        elif user_input.lower().startswith("!both "):
-            use_mcps_flag = True
-            use_data_store_flag = True
-            prompt = user_input[6:].strip()
-            ASCIIColors.yellow(f"Agentic MCP+RAG turn initiated for: '{prompt}'")
-        
-        # --- Streaming Callback ---
-        def stream_callback(chunk, msg_type, metadata={}, **kwargs):
-            # Render steps and thoughts in real-time
-            if msg_type == 12: # MSG_TYPE.MSG_TYPE_STEP_START
-                 ASCIIColors.cyan(f"\n> Starting: {chunk}")
-            elif msg_type == 13: # MSG_TYPE.MSG_TYPE_STEP_END
-                 ASCIIColors.cyan(f"> Finished: {chunk}")
-            elif msg_type == 2: # MSG_TYPE.MSG_TYPE_INFO (for thoughts)
-                 ASCIIColors.yellow(f"\n  (Thought): {chunk}")
-            else: # Final answer chunks are printed by the main loop
-                pass # The final answer is printed after the full report
-            return True
-
-        # --- Main Chat Logic ---
-        try:
-            #print("\nAI: ", end="", flush=True)
-            
-            response_dict = discussion.chat(
-                user_message=prompt,
-                use_mcps=use_mcps_flag,
-                use_data_store={"coding_store": mock_rag_query_function} if use_data_store_flag else None,
-                streaming_callback=stream_callback
-            )
-            if use_mcps_flag or use_data_store_flag:
-                print_agentic_results(response_dict)
-            print("\nAI: ", end="")
-            ASCIIColors.green(response_dict['ai_message'].content)            
-
-        except Exception as e:
-            trace_exception(e)
-            print(f"\nAn error occurred during generation.")
-
-if __name__ == "__main__":
-    mcp_procs = start_mcp_servers()
-    try:
-        discussion_session = setup_client_and_discussion()
-        run_chat_console(discussion_session)
-    finally:
-        ASCIIColors.red("\n--- Shutting down MCP servers ---")
-        for proc in mcp_procs:
-            proc.terminate()
-            proc.wait()
-        shutil.rmtree(WORKSPACE_DIR, ignore_errors=True)
-        print("Cleanup complete. Goodbye!")