chat-console 0.4.2__py3-none-any.whl → 0.4.3__py3-none-any.whl

app/__init__.py

@@ -3,4 +3,4 @@ Chat CLI
 A command-line interface for chatting with various LLM providers like ChatGPT and Claude.
 """
 
-__version__ = "0.4.2"
+__version__ = "0.4.3"

app/api/ollama.py

@@ -165,7 +165,7 @@ class OllamaClient(BaseModelClient):
             user_msg = next((msg for msg in messages if msg.get("role") == "user"), None)
             if user_msg and "content" in user_msg:
                 # Create a direct prompt
-                prompt = "Generate a short descriptive title (maximum 40 characters) for this conversation. ONLY RESPOND WITH THE TITLE FOR THE FOLLOWING MESSAGE:\n\n" + user_msg["content"]
+                prompt = "You must generate a short, descriptive title (maximum 40 characters) for this conversation. ONLY output the title with no additional text, no quotes, and no explanation. Do not start with phrases like 'Here's a title' or 'Title:'. RESPOND ONLY WITH THE TITLE TEXT for the following message:\n\n" + user_msg["content"]
                 debug_log(f"Created title generation prompt: {prompt[:100]}...")
                 return prompt
             else:

app/main.py

@@ -708,21 +708,32 @@ class SimpleChatApp(App): # Keep SimpleChatApp class definition
                 if client_type == OllamaClient:
                     debug_log(f"Title generation with Ollama model detected: {selected_model_resolved}")
                     
-                    # Try common small/fast models first if they exist
+                    # Always try to use smalllm2:135m first, then fall back to other small models
                     try:
-                        # Check if we have any smaller models available for faster title generation
+                        # Check if we have smalllm2:135m or other smaller models available
                         ollama_client = await OllamaClient.create()
                         available_models = await ollama_client.get_available_models()
-                        small_model_options = ["gemma:2b", "phi3:mini", "llama3:8b", "orca-mini:3b", "phi2"]
                         
+                        # Use smalllm2:135m if available (extremely small and fast)
+                        preferred_model = "smalllm2:135m"
+                        fallback_models = ["tinyllama", "gemma:2b", "phi3:mini", "llama3:8b", "orca-mini:3b", "phi2"]
+                        
+                        # First check for our preferred smallest model
                         small_model_found = False
-                        for model_name in small_model_options:
-                            if any(model["id"] == model_name for model in available_models):
-                                debug_log(f"Found smaller Ollama model for title generation: {model_name}")
-                                title_model = model_name
-                                small_model_found = True
-                                break
-                                
+                        if any(model["id"] == preferred_model for model in available_models):
+                            debug_log(f"Found optimal small model for title generation: {preferred_model}")
+                            title_model = preferred_model
+                            small_model_found = True
+                        
+                        # If not found, try fallbacks in order
+                        if not small_model_found:
+                            for model_name in fallback_models:
+                                if any(model["id"] == model_name for model in available_models):
+                                    debug_log(f"Found alternative small model for title generation: {model_name}")
+                                    title_model = model_name
+                                    small_model_found = True
+                                    break
+                                    
                         if not small_model_found:
                             # Use the current model if no smaller models found
                             title_model = selected_model_resolved

app/utils.py

@@ -66,7 +66,7 @@ async def generate_conversation_title(message: str, model: str, client: Any) ->
         title_prompt = [
             {
                 "role": "system", 
-                "content": "Generate a brief, descriptive title (maximum 40 characters) for a conversation that starts with the following message. Return only the title text with no additional explanation or formatting."
+                "content": "Generate a brief, descriptive title (maximum 40 characters) for a conversation that starts with the following message. ONLY output the title text. DO NOT include phrases like 'Sure, here's a title' or any additional formatting, explanation, or quotes."
             },
             {
                 "role": "user",
@@ -96,12 +96,31 @@ async def generate_conversation_title(message: str, model: str, client: Any) ->
                 max_tokens=60
             )
         
-        # Sanitize the title
+        # Sanitize the title - remove quotes, extra spaces and unwanted prefixes
         title = title.strip().strip('"\'').strip()
+        
+        # Remove common LLM prefixes like "Title:", "Sure, here's a title:", etc.
+        prefixes_to_remove = [
+            "title:", "here's a title:", "here is a title:", 
+            "a title for this conversation:", "sure,", "certainly,",
+            "the title is:", "suggested title:"
+        ]
+        
+        # Case-insensitive prefix removal
+        title_lower = title.lower()
+        for prefix in prefixes_to_remove:
+            if title_lower.startswith(prefix):
+                title = title[len(prefix):].strip()
+                title_lower = title.lower()  # Update lowercase version after removal
+        
+        # Remove any remaining quotes
+        title = title.strip('"\'').strip()
+        
+        # Enforce length limit
         if len(title) > 40:
             title = title[:37] + "..."
             
-        debug_log(f"Generated title: {title}")
+        debug_log(f"Generated title (after sanitization): {title}")
         return title
         
     except Exception as e:

{chat_console-0.4.2.dist-info → chat_console-0.4.3.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: chat-console
-Version: 0.4.2
+Version: 0.4.3
 Summary: A command-line interface for chatting with LLMs, storing chats and (future) rag interactions
 Home-page: https://github.com/wazacraftrfid/chat-console
 Author: Johnathan Greenaway

{chat_console-0.4.2.dist-info → chat_console-0.4.3.dist-info}/RECORD

@@ -1,13 +1,13 @@
-app/__init__.py,sha256=aSLqazfzD7omcg24nSmExHXGmkwbhjI84zBzS-AL-To,130
+app/__init__.py,sha256=T3d41tTB1sDy6ix7bg43dp4zvNqqJku0JGlwMGKFGHo,130
 app/config.py,sha256=F-0hO3NT5kRJxZelGLxaeUmnwx8i0LPHzYtNftL6CwM,8468
 app/database.py,sha256=nt8CVuDpy6zw8mOYqDcfUmNw611t7Ln7pz22M0b6-MI,9967
-app/main.py,sha256=lOJiPpBE2F8PQlraVeRHGKS_C0Zka8n__lcE5LW3Gvo,85554
+app/main.py,sha256=8UU9GcPJINu_TmbKKKFBZXIgLHNDf6vabyupKjj3Img,86297
 app/models.py,sha256=4-y9Lytay2exWPFi0FDlVeRL3K2-I7E-jBqNzTfokqY,2644
-app/utils.py,sha256=RtyW_kK3M8WWSDfkuH6G2OuLcInIXfPG4Nv_xijf-gs,40877
+app/utils.py,sha256=-L38KGP8TlVl5vtZl5QgTiEAdhLcDsIXm7e62nnXgP8,41765
 app/api/__init__.py,sha256=A8UL84ldYlv8l7O-yKzraVFcfww86SgWfpl4p7R03-w,62
 app/api/anthropic.py,sha256=uInwNvGLJ_iPUs4BjdwaqXTU6NfmK1SzX7498Pt44fI,10667
 app/api/base.py,sha256=valBWV5So76r8tjrgU5-sLfY73WaViTrszdCy8Rimjo,10314
-app/api/ollama.py,sha256=bVG2UPkpA2arHKyRVLwHt2Q9Lth_qdZLfijiNXvPuYA,76771
+app/api/ollama.py,sha256=364PcXoPLJq9jLMF-HhPyQvaBp87U6FzNHDWx4g_Cvc,76925
 app/api/openai.py,sha256=XuHJHpD7tN_ZHLkRpNUcL1VxTtsXOVk1hDPXX8JnBxQ,15322
 app/ui/__init__.py,sha256=RndfbQ1Tv47qdSiuQzvWP96lPS547SDaGE-BgOtiP_w,55
 app/ui/chat_interface.py,sha256=oSDZi0Jgj_L8WnBh1RuJpIeIcN-RQ38CNejwsXiWTVg,18267
@@ -16,9 +16,9 @@ app/ui/model_browser.py,sha256=pdblLVkdyVF0_Bo02bqbErGAtieyH-y6IfhMOPEqIso,71124
 app/ui/model_selector.py,sha256=2G0TOXfcNodrXZOhLeaJJ2iG3Nck4c_NN1AvUAmaF3M,19172
 app/ui/search.py,sha256=b-m14kG3ovqW1-i0qDQ8KnAqFJbi5b1FLM9dOnbTyIs,9763
 app/ui/styles.py,sha256=04AhPuLrOd2yenfRySFRestPeuTPeMLzhmMB67NdGvw,5615
-chat_console-0.4.2.dist-info/licenses/LICENSE,sha256=srHZ3fvcAuZY1LHxE7P6XWju2njRCHyK6h_ftEbzxSE,1057
-chat_console-0.4.2.dist-info/METADATA,sha256=_-h1KJkkcpmExhmDOmF-4P2Qj8iEAm-RiziHBX-_11Q,3810
-chat_console-0.4.2.dist-info/WHEEL,sha256=7ciDxtlje1X8OhobNuGgi1t-ACdFSelPnSmDPrtlobY,91
-chat_console-0.4.2.dist-info/entry_points.txt,sha256=kkVdEc22U9PAi2AeruoKklfkng_a_aHAP6VRVwrAD7c,67
-chat_console-0.4.2.dist-info/top_level.txt,sha256=io9g7LCbfmTG1SFKgEOGXmCFB9uMP2H5lerm0HiHWQE,4
-chat_console-0.4.2.dist-info/RECORD,,
+chat_console-0.4.3.dist-info/licenses/LICENSE,sha256=srHZ3fvcAuZY1LHxE7P6XWju2njRCHyK6h_ftEbzxSE,1057
+chat_console-0.4.3.dist-info/METADATA,sha256=y01SPyzTsYIyCbc5gEpCx6dlQ_ROG16T1gTZPOScia8,3810
+chat_console-0.4.3.dist-info/WHEEL,sha256=GHB6lJx2juba1wDgXDNlMTyM13ckjBMKf-OnwgKOCtA,91
+chat_console-0.4.3.dist-info/entry_points.txt,sha256=kkVdEc22U9PAi2AeruoKklfkng_a_aHAP6VRVwrAD7c,67
+chat_console-0.4.3.dist-info/top_level.txt,sha256=io9g7LCbfmTG1SFKgEOGXmCFB9uMP2H5lerm0HiHWQE,4
+chat_console-0.4.3.dist-info/RECORD,,

{chat_console-0.4.2.dist-info → chat_console-0.4.3.dist-info}/WHEEL

@@ -1,5 +1,5 @@
 Wheel-Version: 1.0
-Generator: setuptools (80.2.0)
+Generator: setuptools (80.3.0)
 Root-Is-Purelib: true
 Tag: py3-none-any