lollms-client 0.20.10__py3-none-any.whl → 0.22.0__py3-none-any.whl

lollms_client/lollms_personality.py

@@ -0,0 +1,182 @@
+import importlib.util
+import json
+from pathlib import Path
+from typing import Callable, List, Optional, Union
+
+class LollmsPersonality:
+    """
+    A class that encapsulates the full personality of an AI agent.
+
+    This includes its identity, system prompts, specialized knowledge (via RAG),
+    and custom execution logic (via a Python script). It is designed to be a
+    portable and self-contained unit that can be loaded and used by any
+    application using the lollms-client.
+    """
+    def __init__(
+        self,
+        # Metadata for identification and display
+        name: str,
+        author: str,
+        category: str,
+        description: str,
+        # Core behavioral instruction
+        system_prompt: str,        
+        icon: Optional[str] = None,  # Base64 encoded image string
+
+
+        # RAG - Data Files and Application-provided Callbacks
+        data_files: Optional[List[Union[str, Path]]] = None,
+        vectorize_chunk_callback: Optional[Callable[[str, str], None]] = None,  # (chunk_text, chunk_id) -> None
+        is_vectorized_callback: Optional[Callable[[str], bool]] = None,        # (chunk_id) -> bool
+        query_rag_callback: Optional[Callable[[str], str]] = None,              # (user_query) -> rag_context_str
+
+        # Custom Logic Override
+        script: Optional[str] = None,  # The Python script as a raw string
+
+        # Internal state
+        personality_id: Optional[str] = None
+    ):
+        """
+        Initializes a LollmsPersonality instance.
+
+        Args:
+            name: The display name of the personality.
+            author: The author of the personality.
+            category: A category for organization (e.g., 'Code', 'Writing', 'Fun').
+            description: A brief description of what the personality does.
+            icon: An optional base64 encoded string for a display icon.
+            system_prompt: The core system prompt that defines the AI's behavior.
+            data_files: A list of file paths to be used as a knowledge base for RAG.
+            vectorize_chunk_callback: A function provided by the host app to vectorize and store a text chunk.
+            is_vectorized_callback: A function provided by the host app to check if a chunk is already vectorized.
+            query_rag_callback: A function provided by the host app to query the vector store for relevant context.
+            script: A string containing a Python script to override default chat behavior.
+            personality_id: An optional unique identifier. If not provided, it's generated from the author and name.
+        """
+        self.name = name
+        self.author = author
+        self.category = category
+        self.description = description
+        self.icon = icon
+        self.system_prompt = system_prompt
+        self.data_files = [Path(f) for f in data_files] if data_files else []
+        
+        # RAG Callbacks provided by the host application
+        self.vectorize_chunk_callback = vectorize_chunk_callback
+        self.is_vectorized_callback = is_vectorized_callback
+        self.query_rag_callback = query_rag_callback
+        
+        self.script = script
+        self.script_module = None
+        self.personality_id = personality_id or self._generate_id()
+        
+        # Prepare custom logic and data upon initialization
+        self._prepare_script()
+        self.ensure_data_vectorized()
+
+    def _generate_id(self) -> str:
+        """
+        Creates a filesystem-safe, unique ID based on the author and name.
+        """
+        safe_author = "".join(c if c.isalnum() else '_' for c in self.author)
+        safe_name = "".join(c if c.isalnum() else '_' for c in self.name)
+        return f"{safe_author}_{safe_name}"
+
+    def _prepare_script(self):
+        """
+        Dynamically loads the personality's script as an in-memory Python module.
+
+        This allows the script to be executed without being saved as a .py file on disk,
+        making the system more secure and self-contained.
+        """
+        if not self.script:
+            return
+        try:
+            module_name = f"lollms_personality_script_{self.personality_id}"
+            
+            # Create a module specification and a module object from it
+            spec = importlib.util.spec_from_loader(module_name, loader=None)
+            self.script_module = importlib.util.module_from_spec(spec)
+            
+            # Execute the script code within the new module's namespace
+            exec(self.script, self.script_module.__dict__)
+            print(f"[{self.name}] Custom script loaded successfully.")
+        except Exception as e:
+            print(f"[{self.name}] Failed to load custom script: {e}")
+            self.script_module = None
+
+    def ensure_data_vectorized(self, chunk_size: int = 1024):
+        """
+        Checks if the personality's data files are vectorized using the host callbacks.
+        
+        It iterates through each data file, splits it into chunks, and for each chunk,
+        it checks if it's already processed. If not, it calls the vectorization callback
+        provided by the host application.
+
+        Args:
+            chunk_size: The size of each text chunk to process for vectorization.
+        """
+        if not self.data_files or not self.vectorize_chunk_callback or not self.is_vectorized_callback:
+            return
+            
+        print(f"[{self.name}] Checking RAG data vectorization...")
+        all_vectorized = True
+        for file_path in self.data_files:
+            if not file_path.exists():
+                print(f"  - Warning: Data file not found, skipping: {file_path}")
+                continue
+            
+            try:
+                content = file_path.read_text(encoding='utf-8')
+                chunks = [content[i:i+chunk_size] for i in range(0, len(content), chunk_size)]
+                
+                for i, chunk in enumerate(chunks):
+                    # Generate a unique and deterministic ID for each chunk
+                    chunk_id = f"{self.personality_id}_{file_path.name}_chunk_{i}"
+                    
+                    if not self.is_vectorized_callback(chunk_id):
+                        all_vectorized = False
+                        print(f"  - Vectorizing '{file_path.name}' chunk {i+1}/{len(chunks)}...")
+                        self.vectorize_chunk_callback(chunk, chunk_id)
+
+            except Exception as e:
+                print(f"  - Error processing file {file_path.name}: {e}")
+                continue
+        
+        if all_vectorized:
+            print(f"[{self.name}] All RAG data is already vectorized.")
+        else:
+            print(f"[{self.name}] RAG data vectorization complete.")
+        
+    def get_rag_context(self, query: str) -> Optional[str]:
+        """
+        Queries the vectorized data to get relevant context for a given query.
+
+        This method relies on the `query_rag_callback` provided by the host application
+        to perform the actual search in the vector store.
+
+        Args:
+            query: The user's query string.
+
+        Returns:
+            A string containing the relevant context, or None if no callback is available.
+        """
+        if not self.query_rag_callback:
+            return None
+        return self.query_rag_callback(query)
+
+    def to_dict(self) -> dict:
+        """
+        Serializes the personality's metadata to a dictionary.
+        Note: Callbacks and the script module are not serialized.
+        """
+        return {
+            "personality_id": self.personality_id,
+            "name": self.name,
+            "author": self.author,
+            "category": self.category,
+            "description": self.description,
+            "system_prompt": self.system_prompt,
+            "data_files": [str(p) for p in self.data_files],
+            "has_script": self.script is not None
+        }

lollms_client/lollms_types.py

@@ -6,31 +6,35 @@ class MSG_TYPE(Enum):
     MSG_TYPE_CONTENT_INVISIBLE_TO_AI   = 2 # A full message (for some personality the answer is sent in bulk)
     MSG_TYPE_CONTENT_INVISIBLE_TO_USER = 3 # A full message (for some personality the answer is sent in bulk)
 
+    # Thoughts
+    MSG_TYPE_THOUGHT_CHUNK             = 4 # A chunk of a thought content (used for classical chat)
+    MSG_TYPE_THOUGHT_CONTENT           = 5 # A full thought content (for some personality the answer is sent in bulk)
+
     # Conditionning
     # Informations
-    MSG_TYPE_EXCEPTION              = 4 # An exception occured
-    MSG_TYPE_WARNING                = 5 # A warning occured
-    MSG_TYPE_INFO                   = 6 # An information to be shown to user
+    MSG_TYPE_EXCEPTION              = 6 # An exception occured
+    MSG_TYPE_WARNING                = 7 # A warning occured
+    MSG_TYPE_INFO                   = 8 # An information to be shown to user
 
     # Steps
-    MSG_TYPE_STEP                   = 7 # An instant step (a step that doesn't need time to be executed)
-    MSG_TYPE_STEP_START             = 8 # A step has started (the text contains an explanation of the step done by he personality)
-    MSG_TYPE_STEP_PROGRESS          = 9 # The progress value (the text contains a percentage and can be parsed by the reception)
-    MSG_TYPE_STEP_END               = 10# A step has been done (the text contains an explanation of the step done by he personality)
+    MSG_TYPE_STEP                   = 9 # An instant step (a step that doesn't need time to be executed)
+    MSG_TYPE_STEP_START             = 10 # A step has started (the text contains an explanation of the step done by he personality)
+    MSG_TYPE_STEP_PROGRESS          = 11 # The progress value (the text contains a percentage and can be parsed by the reception)
+    MSG_TYPE_STEP_END               = 12# A step has been done (the text contains an explanation of the step done by he personality)
 
     #Extra
-    MSG_TYPE_JSON_INFOS             = 11# A JSON output that is useful for summarizing the process of generation used by personalities like chain of thoughts and tree of thooughts
-    MSG_TYPE_REF                    = 12# References (in form of  [text](path))
-    MSG_TYPE_CODE                   = 13# A javascript code to execute
-    MSG_TYPE_UI                     = 14# A vue.js component to show (we need to build some and parse the text to show it)
+    MSG_TYPE_JSON_INFOS             = 13# A JSON output that is useful for summarizing the process of generation used by personalities like chain of thoughts and tree of thooughts
+    MSG_TYPE_REF                    = 14# References (in form of  [text](path))
+    MSG_TYPE_CODE                   = 15# A javascript code to execute
+    MSG_TYPE_UI                     = 16# A vue.js component to show (we need to build some and parse the text to show it)
 
     #Commands
-    MSG_TYPE_NEW_MESSAGE            = 15# A new message
-    MSG_TYPE_FINISHED_MESSAGE       = 17# End of current message
+    MSG_TYPE_NEW_MESSAGE            = 17# A new message
+    MSG_TYPE_FINISHED_MESSAGE       = 18# End of current message
 
     #Tool calling
-    MSG_TYPE_TOOL_CALL              = 18# a tool call
-    MSG_TYPE_TOOL_OUTPUT            = 19# the output of the tool
+    MSG_TYPE_TOOL_CALL              = 19# a tool call
+    MSG_TYPE_TOOL_OUTPUT            = 20# the output of the tool
 
 
 class SENDER_TYPES(Enum):
@@ -39,7 +43,6 @@ class SENDER_TYPES(Enum):
     SENDER_TYPES_SYSTEM             = 2 # Sent by athe system
 
 
-
 class SUMMARY_MODE(Enum):
     SUMMARY_MODE_SEQUENCIAL        = 0
     SUMMARY_MODE_HIERARCHICAL      = 0

lollms_client/lollms_utilities.py

@@ -9,93 +9,108 @@ import re
 import numpy as np
 
 import json
-import re
 from ascii_colors import ASCIIColors, trace_exception
+
 def robust_json_parser(json_string: str) -> dict:
     """
-    Parses a JSON string that may be malformed by an LLM by applying a series of cleaning strategies.
+    Parses a possibly malformed JSON string using a series of corrective strategies.
 
     Args:
-        json_string: The string that is expected to contain a JSON object.
+        json_string: A string expected to represent a JSON object or array.
 
     Returns:
         A dictionary parsed from the JSON string.
 
     Raises:
-        ValueError: If the string cannot be parsed after all cleaning attempts.
-
-    Strategies Applied in Order:
-    1.  Tries to parse the string directly.
-    2.  If that fails, it extracts the main JSON object or array from the string.
-    3.  Applies a series of fixes:
-        a. Replaces Python/JS boolean/null values with JSON-compliant ones.
-        b. Removes single-line and multi-line comments.
-        c. Fixes improperly escaped characters (e.g., \_).
-        d. Removes trailing commas from objects and arrays.
-        e. Escapes unescaped newline characters within strings.
-    4.  Tries to parse the cleaned string.
+        ValueError: If parsing fails after all correction attempts.
     """
-    # 1. First attempt: Standard parsing
+
+    # STEP 0: Remove code block wrappers if present (e.g., ```json ... ```)
+    json_string = re.sub(r"^```(?:json)?\s*|\s*```$", '', json_string.strip())
+
+    # STEP 1: Attempt to parse directly
     try:
         return json.loads(json_string)
     except json.JSONDecodeError:
-        pass # Will proceed to cleaning steps
+        pass
 
-    # 2. Second attempt: Find a JSON object or array within a larger string
-    # This is useful if the LLM adds text like "Here is the JSON: {..}"
-    # Regex to find a JSON object `{...}` or array `[...]`
+    # STEP 2: Extract likely JSON substring
     json_match = re.search(r'(\{[\s\S]*\}|\[[\s\S]*\])', json_string)
-    if json_match:
-        json_substring = json_match.group(0)
-    else:
-        # If no object or array is found, we work with the original string
-        json_substring = json_string
-        
-    # Store the potentially cleaned string
-    cleaned_string = json_substring
+    cleaned_string = json_match.group(0) if json_match else json_string
 
-    # 3. Third attempt: Apply a series of cleaning functions
     try:
-        # a. Fix boolean and null values
+        # STEP 3a: Normalize Python/JS booleans/nulls
         cleaned_string = re.sub(r'\bTrue\b', 'true', cleaned_string)
         cleaned_string = re.sub(r'\bFalse\b', 'false', cleaned_string)
         cleaned_string = re.sub(r'\bNone\b', 'null', cleaned_string)
 
-        # b. Remove comments
-        # Remove // comments
+        # STEP 3b: Remove comments (single-line and block)
         cleaned_string = re.sub(r'//.*', '', cleaned_string)
-        # Remove /* ... */ comments
         cleaned_string = re.sub(r'/\*[\s\S]*?\*/', '', cleaned_string)
-        
-        # c. Un-escape characters that are not valid JSON escape sequences
-        # e.g., \_ , \- , \* etc. that LLMs sometimes add.
-        cleaned_string = re.sub(r'\\([_`*#-])', r'\1', cleaned_string)
 
-        # d. Remove trailing commas
-        cleaned_string = re.sub(r',\s*(\}|\])', r'\1', cleaned_string)
+        # STEP 3c: Remove bad escape sequences like \_ or \*
+        cleaned_string = re.sub(r'\\([_`*#\-])', r'\1', cleaned_string)
 
-        # e. Fix unescaped newlines within strings. This is the most complex part.
-        # We iterate through the string and escape newlines only when inside a string literal.
-        in_string = False
-        escaped_string = []
-        for i, char in enumerate(cleaned_string):
-            if char == '"' and (i == 0 or cleaned_string[i-1] != '\\'):
-                in_string = not in_string
-            
-            if in_string and char == '\n':
-                escaped_string.append('\\n')
-            else:
-                escaped_string.append(char)
-        
-        cleaned_string = "".join(escaped_string)
+        # STEP 3d: Remove trailing commas
+        cleaned_string = re.sub(r',\s*(\}|\])', r'\1', cleaned_string)
 
+        # STEP 3e: Escape unescaped newlines inside string literals
+        def escape_newlines_in_strings(text: str) -> str:
+            in_string = False
+            result = []
+            i = 0
+            while i < len(text):
+                c = text[i]
+                if c == '"' and (i == 0 or text[i - 1] != '\\'):
+                    in_string = not in_string
+                if in_string and c == '\n':
+                    result.append('\\n')
+                else:
+                    result.append(c)
+                i += 1
+            return ''.join(result)
+
+        cleaned_string = escape_newlines_in_strings(cleaned_string)
+
+        # STEP 3f: Escape unescaped inner double quotes inside strings
+        def escape_unescaped_inner_quotes(text: str) -> str:
+            def fix(match):
+                s = match.group(0)
+                inner = s[1:-1]
+                # Escape double quotes that aren't already escaped
+                inner_fixed = re.sub(r'(?<!\\)"', r'\\"', inner)
+                return f'"{inner_fixed}"'
+            return re.sub(r'"(?:[^"\\]|\\.)*"', fix, text)
+
+        cleaned_string = escape_unescaped_inner_quotes(cleaned_string)
+
+        # STEP 3g: Convert single-quoted strings to double quotes (arrays or object keys)
+        cleaned_string = re.sub(
+            r"(?<=[:\[,])\s*'([^']*?)'\s*(?=[,\}\]])", 
+            lambda m: '"' + m.group(1).replace('"', '\\"') + '"', 
+            cleaned_string
+        )
+        cleaned_string = re.sub(
+            r"(?<=\{)\s*'([^']*?)'\s*:", 
+            lambda m: '"' + m.group(1).replace('"', '\\"') + '":', 
+            cleaned_string
+        )
+
+        # STEP 3h: Remove non-breaking spaces and control characters
+        cleaned_string = re.sub(r'[\x00-\x1F\x7F\u00A0]', '', cleaned_string)
+
+        # STEP 3i: Fix smart quotes
+        cleaned_string = cleaned_string.replace("“", '"').replace("”", '"').replace("‘", "'").replace("’", "'")
+
+        # STEP 3j: Remove line breaks between JSON tokens that don't belong
+        cleaned_string = re.sub(r'"\s*\n\s*"', '"\\n"', cleaned_string)
+
+        # Final parse
         return json.loads(cleaned_string)
 
     except json.JSONDecodeError as e:
-        # If all else fails, raise the final error with context
-        ASCIIColors.error("Failed to parse JSON after all cleaning attempts. See details below.")
         print("\n--- JSONDecodeError ---")
-        trace_exception(e)
+        print(e)
         print("\n--- Original String ---")
         print(json_string)
         print("\n--- Final Cleaned String Attempted ---")
@@ -103,7 +118,6 @@ def robust_json_parser(json_string: str) -> dict:
         raise ValueError(f"Failed to parse JSON. Final error: {e}") from e
 
 
-
 class PromptReshaper:
     def __init__(self, template:str):
         self.template = template

lollms_client/mcp_bindings/remote_mcp/__init__.py

@@ -44,7 +44,8 @@ class RemoteMCPBinding(LollmsMCPBinding):
             **other_config_params (Any): Additional configuration parameters.
         """
         super().__init__(binding_name="remote_mcp")
-        
+        # initialization in case no servers are present
+        self.servers = None
         if not MCP_LIBRARY_AVAILABLE:
             ASCIIColors.error(f"{self.binding_name}: MCP library not available. This binding will be disabled.")
             return

{lollms_client-0.20.10.dist-info → lollms_client-0.22.0.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: lollms_client
-Version: 0.20.10
+Version: 0.22.0
 Summary: A client library for LoLLMs generate endpoint
 Author-email: ParisNeo <parisneoai@gmail.com>
 License: Apache Software License

{lollms_client-0.20.10.dist-info → lollms_client-0.22.0.dist-info}/RECORD

@@ -1,10 +1,13 @@
+examples/console_discussion.py,sha256=JxjVaAxtt1WEXLN8vCJosu-cYNgfIX2JGO25kg1FFNY,20490
 examples/external_mcp.py,sha256=swx1KCOz6jk8jGTAycq-xu7GXPAhRMDe1x--SKocugE,13371
 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/local_mcp.py,sha256=w40dgayvHYe01yvekEE0LjcbkpwKjWwJ-9v4_wGYsUk,9113
+examples/lollms_discussions_test.py,sha256=Jk1cCUDBBhTcK5glI50jAgzfB3IOiiUlnK3q7RYfMkA,6796
 examples/openai_mcp.py,sha256=7IEnPGPXZgYZyiES_VaUbQ6viQjenpcUxGiHE-pGeFY,11060
 examples/run_remote_mcp_example copy.py,sha256=pGT8A5iXK9oHtjGNEUCm8fnj9DQ37gcznjLYqAEI20o,10075
 examples/run_standard_mcp_example.py,sha256=GSZpaACPf3mDPsjA8esBQVUsIi7owI39ca5avsmvCxA,9419
@@ -21,29 +24,27 @@ examples/deep_analyze/deep_analyse.py,sha256=fZNmDrfEAuxEAfdbjAgJYIh1k6wbiuZ4Rvw
 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/personality_test/chat_test.py,sha256=o2jlpoddFc-T592iqAiA29xk3x27KsdK5DluqxBwHqw,1417
-examples/personality_test/chat_with_aristotle.py,sha256=4X_fwubMpd0Eq2rCReS2bgVlUoAqJprjkLXk2Jz6pXU,1774
-examples/personality_test/tesks_test.py,sha256=7LIiwrEbva9WWZOLi34fsmCBN__RZbPpxoUOKA_AtYk,1924
 examples/test_local_models/local_chat.py,sha256=slakja2zaHOEAUsn2tn_VmI4kLx6luLBrPqAeaNsix8,456
-lollms_client/__init__.py,sha256=MkOVxw6SJ4jFxaT2VnGwP0TshnISCLt0f5PT2x8i5CA,913
+lollms_client/__init__.py,sha256=Cd4G7paIm0kKNqc90yAJ7VJ8mpi2jfzoMBabmWRBZbY,1047
 lollms_client/lollms_config.py,sha256=goEseDwDxYJf3WkYJ4IrLXwg3Tfw73CXV2Avg45M_hE,21876
-lollms_client/lollms_core.py,sha256=EvH5IuZqKQw8QAJlPVyJrd4sBNIOPe07pPDU5XoWa9U,139305
-lollms_client/lollms_discussion.py,sha256=v4SVjHNMdgBbk4j-qWSK2lMIx2_UrvhRHrApkF7l6jU,19247
+lollms_client/lollms_core.py,sha256=y_KIVFCBsA0BYizGAbvQZrtWZCX5jB00yMuAgHlHaD4,143685
+lollms_client/lollms_discussion.py,sha256=zdAUOhbFod65-VZYfKaldHYURR7wWnuccqv6FJa1qrM,36291
 lollms_client/lollms_js_analyzer.py,sha256=01zUvuO2F_lnUe_0NLxe1MF5aHE1hO8RZi48mNPv-aw,8361
 lollms_client/lollms_llm_binding.py,sha256=E81g4yBlQn76WTSLicnTETJuQhf_WZUMZaxotgRnOcA,12096
 lollms_client/lollms_mcp_binding.py,sha256=0rK9HQCBEGryNc8ApBmtOlhKE1Yfn7X7xIQssXxS2Zc,8933
+lollms_client/lollms_personality.py,sha256=dILUI5DZdzJ3NDDQiIsK2UptVF-jZK3XYXZ2bpXP_ew,8035
 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=6x9EIoHVK37UKBX9MD8S7usfzvTisNPlDXqysIrJzpI,2872
-lollms_client/lollms_utilities.py,sha256=IyeZZcoKzera4I4Vjk7IIUJXT8uQ45FU2ZjTvaeKFjM,10944
+lollms_client/lollms_types.py,sha256=NfvTmICzRCgfjjy5zLMFeDaiW6zyUsdnxRF69gAEyAk,3110
+lollms_client/lollms_utilities.py,sha256=qK5iNmrFD7NGaEVW3nCWT6AtEhLIVHCXMzEpYxG_M5w,11293
 lollms_client/llm_bindings/__init__.py,sha256=9sWGpmWSSj6KQ8H4lKGCjpLYwhnVdL_2N7gXCphPqh4,14
 lollms_client/llm_bindings/llamacpp/__init__.py,sha256=Qj5RvsgPeHGNfb5AEwZSzFwAp4BOWjyxmm9qBNtstrc,63716
 lollms_client/llm_bindings/lollms/__init__.py,sha256=17TwGMDJMxRPjZjZZSysR8AwjMXZeRfDBy8RqWWuaIY,17769
-lollms_client/llm_bindings/ollama/__init__.py,sha256=fno1UEcXIy37Z3-bZAwOMuCfbAYOzztcO4oBfLqK_JA,32538
+lollms_client/llm_bindings/ollama/__init__.py,sha256=QufsYqak2VlA2XGbzks8u55yNJFeDH2V35NGeZABkm8,32554
 lollms_client/llm_bindings/openai/__init__.py,sha256=ay_2JJi4La258Eg3alUhnh6Y5IRyOWnHaFLXqvN_4ao,19144
 lollms_client/llm_bindings/openllm/__init__.py,sha256=xv2XDhJNCYe6NPnWBboDs24AQ1VJBOzsTuMcmuQ6xYY,29864
 lollms_client/llm_bindings/pythonllamacpp/__init__.py,sha256=7dM42TCGKh0eV0njNL1tc9cInhyvBRIXzN3dcy12Gl0,33551
@@ -55,7 +56,7 @@ lollms_client/mcp_bindings/local_mcp/default_tools/file_writer/file_writer.py,sh
 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=VsM7SX23sCZa2NDMad4fYAxm_2oJzMdIc8ZiiI6LJp8,16539
+lollms_client/mcp_bindings/remote_mcp/__init__.py,sha256=_wtrwq5_kRV3Of2lB-45G7BZQVR_fYvacHYc_s6q9fk,16615
 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
@@ -77,8 +78,9 @@ lollms_client/tts_bindings/piper_tts/__init__.py,sha256=0IEWG4zH3_sOkSb9WbZzkeV5
 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.20.10.dist-info/licenses/LICENSE,sha256=HrhfyXIkWY2tGFK11kg7vPCqhgh5DcxleloqdhrpyMY,11558
-lollms_client-0.20.10.dist-info/METADATA,sha256=f4moL-l28wfqpM-1s7ZpBeTVrrCrjmHn810eV9lQ2Oc,13375
-lollms_client-0.20.10.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-lollms_client-0.20.10.dist-info/top_level.txt,sha256=NI_W8S4OYZvJjb0QWMZMSIpOrYzpqwPGYaklhyWKH2w,23
-lollms_client-0.20.10.dist-info/RECORD,,
+lollms_client-0.22.0.dist-info/licenses/LICENSE,sha256=HrhfyXIkWY2tGFK11kg7vPCqhgh5DcxleloqdhrpyMY,11558
+personalities/parrot.py,sha256=-HdbK1h7Ixvp8FX69Nv92Z_El6_UVtsF8WuAvNpKbfg,478
+lollms_client-0.22.0.dist-info/METADATA,sha256=71v2Qlp7tSyr_fL9I8Nw-9LLlncfuUkYgtJkK1EKayc,13374
+lollms_client-0.22.0.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+lollms_client-0.22.0.dist-info/top_level.txt,sha256=vgtOcmtJbKs9gEiIIPp2scIsvtvU2y5u3tngqlme1RU,37
+lollms_client-0.22.0.dist-info/RECORD,,

{lollms_client-0.20.10.dist-info → lollms_client-0.22.0.dist-info}/top_level.txt

@@ -1,2 +1,3 @@
 examples
 lollms_client
+personalities

personalities/parrot.py

@@ -0,0 +1,10 @@
+
+def run(discussion, on_chunk_callback):
+    # This script overrides the normal chat flow.
+    user_message = discussion.get_branch(discussion.active_branch_id)[-1].content
+    response = f"Squawk! {user_message}! Squawk!"
+    if on_chunk_callback:
+        # We need to simulate the message type for the callback
+        from lollms_client import MSG_TYPE
+        on_chunk_callback(response, MSG_TYPE.MSG_TYPE_CHUNK)
+    return response # Return the full raw response

examples/personality_test/chat_test.py

@@ -1,37 +0,0 @@
-from lollms_client import LollmsClient, LollmsDiscussion
-from lollms_client import LollmsPersonality
-from lollms_client import MSG_TYPE, ELF_GENERATION_FORMAT
-from ascii_colors import ASCIIColors
-# Callback send
-def cb(chunk, type: MSG_TYPE):
-    print(chunk,end="", flush=True)
-
-# Initialize the LollmsClient instance
-lc = LollmsClient("http://localhost:9600",default_generation_mode=ELF_GENERATION_FORMAT.OPENAI)
-# Bu_ild inline personality
-p = LollmsPersonality(
-                        lc, 
-                        "./personality/test/work_dir", 
-                        "./personality/test/config_dir", 
-                        cb, 
-                        None,
-                        author="ParisNeo",
-                        name="test_persona",
-                        user_name="user",
-                        category="generic",
-                        category_desc="generic stuff",
-                        language="English",
-                        personality_conditioning="!@>system: Act as a helper to the user.",
-                        welcome_message="Hi, I'm your helper. Let me help you",
-
-                        )
-d = LollmsDiscussion(lc)
-prompt=""
-ASCIIColors.green("To quit press q")
-ASCIIColors.yellow(p.welcome_message)
-while prompt!="q":
-    prompt = input("user:")
-    if prompt=="q":
-        break
-    p.generate(d,prompt,stream=True)
-    print("")

examples/personality_test/chat_with_aristotle.py

@@ -1,42 +0,0 @@
-from lollms_client import LollmsClient, LollmsDiscussion
-from lollms_client import LollmsPersonality
-from lollms_client import MSG_TYPE, ELF_GENERATION_FORMAT
-from ascii_colors import ASCIIColors
-# Callback send
-def cb(chunk, type: MSG_TYPE):
-    print(chunk,end="", flush=True)
-
-# Initialize the LollmsClient instance
-lc = LollmsClient("http://localhost:9600",default_generation_mode=ELF_GENERATION_FORMAT.LOLLMS)
-# Bu_ild inline personality
-aristotle_personality  = LollmsPersonality(
-                        lc, 
-                        "./personality/test/work_dir", 
-                        "./personality/test/config_dir", 
-                        cb, 
-                        None,
-                        author="ParisNeo",
-                        name="test_persona",
-                        user_name="user",
-                        category="generic",
-                        category_desc="generic stuff",
-                        language="English",
-                        personality_conditioning="!@>system: Act as the philosopher Aristotle, sharing wisdom and engaging in logical discussions.",
-                        welcome_message="Greetings, I am Aristotle, your guide in the pursuit of knowledge. How may I assist you in your philosophical inquiries?",
-                    )
-# Create a Discussion instance for Aristotle
-aristotle_discussion = LollmsDiscussion(lc)
-
-# Initialize user prompt
-prompt = ""
-
-# Print welcome message in yellow
-ASCIIColors.yellow(aristotle_personality.welcome_message)
-
-# Interaction loop
-while prompt.lower() != "q":
-    prompt = input("student: ")
-    if prompt.lower() == "q":
-        break
-    aristotle_personality.generate(aristotle_discussion, prompt, stream=True)
-    print("")