TeLLMgramBot 3.10.2__tar.gz → 3.10.3__tar.gz

{tellmgrambot-3.10.2 → tellmgrambot-3.10.3}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: TeLLMgramBot
-Version: 3.10.2
+Version: 3.10.3
 Summary: LLM-powered Telegram bot (OpenAI + Anthropic)
 Home-page: https://github.com/Digital-Heresy/TeLLMgramBot
 Author: Digital Heresy
@@ -44,6 +44,7 @@ The basic goal of this project is to create a bridge between a Telegram Bot and
 * Ask questions about message history across all your chats using natural language; the bot will search, attribute messages to speakers, and include messages from other bots.
   * Example: "Who said thanks for the breakdown?" or "What did George say about the project?" or "Show me the last few messages."
   * All search filters (speaker, chat, date) are optional. Results are ordered most-recent-first. Configure `search_limit` to control how many results to return (default: 30).
+  * Search automatically finds users and chats by their current or past names, so you can reference them however you remember them.
 * Token limits measure conversation length and determine when to prune oldest messages to stay within model limits.
   * The bot loads the user's full history across all chats up to 50% of the token budget. In private chats, shared group context fills the remaining budget, enabling the bot to reference group conversations from a private context.
   * This eliminates amnesia when switching between private and group chats.

{tellmgrambot-3.10.2 → tellmgrambot-3.10.3}/README.md

@@ -12,6 +12,7 @@ The basic goal of this project is to create a bridge between a Telegram Bot and
 * Ask questions about message history across all your chats using natural language; the bot will search, attribute messages to speakers, and include messages from other bots.
   * Example: "Who said thanks for the breakdown?" or "What did George say about the project?" or "Show me the last few messages."
   * All search filters (speaker, chat, date) are optional. Results are ordered most-recent-first. Configure `search_limit` to control how many results to return (default: 30).
+  * Search automatically finds users and chats by their current or past names, so you can reference them however you remember them.
 * Token limits measure conversation length and determine when to prune oldest messages to stay within model limits.
   * The bot loads the user's full history across all chats up to 50% of the token budget. In private chats, shared group context fills the remaining budget, enabling the bot to reference group conversations from a private context.
   * This eliminates amnesia when switching between private and group chats.

{tellmgrambot-3.10.2 → tellmgrambot-3.10.3}/TeLLMgramBot/database.py

@@ -6,6 +6,7 @@ searching message history, and retrieving conversation context. The schema inclu
 table (indexed by chat_id and user_id, with is_private flag), a users table (profile data
 and private_mode flag), and a chats table for speaker/chat resolution in search results.
 """
+import json
 import os
 from typing import Optional
 
@@ -42,19 +43,21 @@ CREATE INDEX IF NOT EXISTS idx_messages_chat_id ON messages(chat_id, created_at)
 CREATE INDEX IF NOT EXISTS idx_messages_user_id ON messages(user_id, created_at);
 
 CREATE TABLE IF NOT EXISTS users (
-    user_id      INTEGER PRIMARY KEY,
-    username     TEXT,
-    first_name   TEXT,
-    last_name    TEXT,
-    private_mode INTEGER NOT NULL DEFAULT 0,
-    updated_at   TEXT NOT NULL DEFAULT (strftime('%Y-%m-%dT%H:%M:%fZ', 'now'))
+    user_id        INTEGER PRIMARY KEY,
+    username       TEXT,
+    first_name     TEXT,
+    last_name      TEXT,
+    private_mode   INTEGER NOT NULL DEFAULT 0,
+    previous_names TEXT,
+    updated_at     TEXT NOT NULL DEFAULT (strftime('%Y-%m-%dT%H:%M:%fZ', 'now'))
 );
 
 CREATE TABLE IF NOT EXISTS chats (
-    chat_id    INTEGER PRIMARY KEY,
-    chat_type  TEXT,
-    chat_title TEXT,
-    updated_at TEXT NOT NULL DEFAULT (strftime('%Y-%m-%dT%H:%M:%fZ', 'now'))
+    chat_id         INTEGER PRIMARY KEY,
+    chat_type       TEXT,
+    chat_title      TEXT,
+    previous_titles TEXT,
+    updated_at      TEXT NOT NULL DEFAULT (strftime('%Y-%m-%dT%H:%M:%fZ', 'now'))
 );
 """
 
@@ -71,7 +74,7 @@ def set_db_filename(filename: str) -> None:
     Args:
         filename: DB filename, e.g. 'MyBot_conversations.db'. A '.db' suffix is
                   appended automatically if not already present. Must be a plain
-                  basename — path separators and absolute paths are rejected.
+                  basename - path separators and absolute paths are rejected.
 
     Raises:
         ValueError: If filename contains path components or is absolute.
@@ -133,14 +136,19 @@ async def init_db() -> None:
         # Migration: drop username column now stored in the normalized users table.
         if 'username' in existing:
             await db.execute("ALTER TABLE messages DROP COLUMN username")
-        # Migration: add chat_type to chats table if absent.
+        # Migration: add chat_type and previous_titles to chats table if absent.
         chats_cols = {row[1] async for row in await db.execute("PRAGMA table_info(chats)")}
         if 'chat_type' not in chats_cols:
             await db.execute("ALTER TABLE chats ADD COLUMN chat_type TEXT")
+        if 'previous_titles' not in chats_cols:
+            await db.execute("ALTER TABLE chats ADD COLUMN previous_titles TEXT")
         # Migration: merge private_mode table into users.private_mode and drop the old table.
+        # Also add previous_names column for name history tracking.
         users_cols = {row[1] async for row in await db.execute("PRAGMA table_info(users)")}
         if 'private_mode' not in users_cols:
             await db.execute("ALTER TABLE users ADD COLUMN private_mode INTEGER NOT NULL DEFAULT 0")
+        if 'previous_names' not in users_cols:
+            await db.execute("ALTER TABLE users ADD COLUMN previous_names TEXT")
         tables = {row[0] async for row in await db.execute("SELECT name FROM sqlite_master WHERE type='table'")}
         if 'private_mode' in tables:
             await db.execute(
@@ -595,7 +603,10 @@ async def upsert_user(
     Insert or update a user record in the users table.
 
     Always reflects the latest Telegram profile data. Called on every interaction
-    so that name changes are picked up automatically.
+    so that name changes are picked up automatically. When first/last name changes,
+    the old display name (first_name + space + last_name) is appended to the
+    previous_names column (JSON array, deduplicated) to enable search by
+    historical identities.
 
     Args:
         user_id: Telegram user ID.
@@ -604,13 +615,40 @@ async def upsert_user(
         last_name: Telegram last name, may be None.
     """
     async with aiosqlite.connect(get_db_path()) as db:
+        async with db.execute(
+            "SELECT first_name, last_name, previous_names FROM users WHERE user_id = ?",
+            (user_id,),
+        ) as cur:
+            existing = await cur.fetchone()
+
+        updated_prev = None
+        if existing is not None:
+            old_first, old_last, prev = existing
+            old_display = f"{old_first or ''} {old_last or ''}".strip()
+            new_display = f"{first_name or ''} {last_name or ''}".strip()
+            if old_display and old_display != new_display:
+                if prev:
+                    try:
+                        seen = json.loads(prev)
+                        if not isinstance(seen, list):
+                            seen = [seen] if seen else []
+                    except json.JSONDecodeError:
+                        seen = [n.strip() for n in prev.split(",") if n.strip()]
+                else:
+                    seen = []
+                if old_display not in seen:
+                    seen.append(old_display)
+                updated_prev = json.dumps(seen)
+            else:
+                updated_prev = prev
+
         await db.execute(
-            "INSERT INTO users (user_id, username, first_name, last_name) VALUES (?, ?, ?, ?) "
+            "INSERT INTO users (user_id, username, first_name, last_name, previous_names) VALUES (?, ?, ?, ?, ?) "
             "ON CONFLICT(user_id) DO UPDATE SET "
             "username = excluded.username, first_name = excluded.first_name, "
-            "last_name = excluded.last_name, "
+            "last_name = excluded.last_name, previous_names = excluded.previous_names, "
             "updated_at = strftime('%Y-%m-%dT%H:%M:%fZ', 'now')",
-            (user_id, username, first_name, last_name),
+            (user_id, username, first_name, last_name, updated_prev),
         )
         await db.commit()
 
@@ -624,7 +662,9 @@ async def upsert_chat(
     Insert or update a chat record in the chats table.
 
     Always reflects the latest Telegram chat metadata. Called on every interaction
-    so that chat type and title changes are picked up automatically.
+    so that chat type and title changes are picked up automatically. When the
+    chat_title changes, the old title is appended to the previous_titles column
+    (JSON array, deduplicated) to enable search by historical identities.
 
     Args:
         chat_id: Telegram chat ID.
@@ -632,12 +672,38 @@ async def upsert_chat(
         chat_title: Display name of the group/channel, or None for private chats.
     """
     async with aiosqlite.connect(get_db_path()) as db:
+        async with db.execute(
+            "SELECT chat_title, previous_titles FROM chats WHERE chat_id = ?",
+            (chat_id,),
+        ) as cur:
+            existing = await cur.fetchone()
+
+        updated_prev = None
+        if existing is not None:
+            old_title, prev = existing
+            if old_title and old_title != chat_title:
+                if prev:
+                    try:
+                        seen = json.loads(prev)
+                        if not isinstance(seen, list):
+                            seen = [seen] if seen else []
+                    except json.JSONDecodeError:
+                        seen = [t.strip() for t in prev.split(",") if t.strip()]
+                else:
+                    seen = []
+                if old_title not in seen:
+                    seen.append(old_title)
+                updated_prev = json.dumps(seen)
+            else:
+                updated_prev = prev
+
         await db.execute(
-            "INSERT INTO chats (chat_id, chat_type, chat_title) VALUES (?, ?, ?) "
+            "INSERT INTO chats (chat_id, chat_type, chat_title, previous_titles) VALUES (?, ?, ?, ?) "
             "ON CONFLICT(chat_id) DO UPDATE SET "
             "chat_type = excluded.chat_type, chat_title = excluded.chat_title, "
+            "previous_titles = excluded.previous_titles, "
             "updated_at = strftime('%Y-%m-%dT%H:%M:%fZ', 'now')",
-            (chat_id, chat_type, chat_title),
+            (chat_id, chat_type, chat_title, updated_prev),
         )
         await db.commit()
 
@@ -679,9 +745,12 @@ async def search_messages(
 
     Resolves speaker_query and chat_query using exact-match-first logic:
     1. Case-insensitive exact match - if exactly one entity matches, proceed.
-    2. Partial LIKE match - if exactly one accessible entity matches, proceed.
-       Zero matches: colloquial fallback (chat: all accessible; speaker: return empty).
-       Two or more matches: return an ambiguity sentinel dict instead of results.
+    2. Partial LIKE match (including historical names/titles) - if exactly one accessible
+       entity matches, proceed. Searches current username/first_name/last_name for speakers
+       and current chat_title for chats; falls back to previous_names and previous_titles
+       columns to find entities by their past identities. Zero matches: colloquial fallback
+       (chat: all accessible; speaker: return empty). Two or more matches: return an
+       ambiguity sentinel dict instead of results.
 
     Leading `@` is stripped from speaker_query automatically.
     Access control: results are always scoped to accessible_chat_ids (the caller's
@@ -716,7 +785,7 @@ async def search_messages(
         # Resolve speaker_query -> user_ids (exact match first, then LIKE)
         # Scoped to users who have at least one message in accessible_chat_ids (s3kp).
         # Batched over accessible_chat_ids to stay within SQLite's 999-parameter limit;
-        # each batch leaves 4 slots for name-matching params (exact: 3, partial: 4).
+        # each batch leaves 5 slots for name-matching params (exact: 3, partial: 5).
         speaker_ids: Optional[list[int]] = None
         speaker_amb: Optional[dict] = None
         if speaker_query:
@@ -746,8 +815,8 @@ async def search_messages(
             else:
                 like = f"%{q}%"
                 partial_seen: dict[int, tuple] = {}
-                for i in range(0, len(accessible_chat_ids), _SQLITE_MAX_PARAMS - 4):
-                    chunk = accessible_chat_ids[i:i + _SQLITE_MAX_PARAMS - 4]
+                for i in range(0, len(accessible_chat_ids), _SQLITE_MAX_PARAMS - 5):
+                    chunk = accessible_chat_ids[i:i + _SQLITE_MAX_PARAMS - 5]
                     chunk_ph = ','.join('?' * len(chunk))
                     async with db.execute(
                         f"SELECT DISTINCT u.user_id, "
@@ -756,8 +825,8 @@ async def search_messages(
                         f"FROM users u JOIN messages m ON m.user_id = u.user_id "
                         f"WHERE m.chat_id IN ({chunk_ph}) "
                         f"AND (u.username LIKE ? OR u.first_name LIKE ? OR u.last_name LIKE ? "
-                        f"OR (u.first_name || ' ' || u.last_name) LIKE ?)",
-                        (*chunk, like, like, like, like),
+                        f"OR (u.first_name || ' ' || u.last_name) LIKE ? OR u.previous_names LIKE ?)",
+                        (*chunk, like, like, like, like, like),
                     ) as cur:
                         for row in await cur.fetchall():
                             partial_seen.setdefault(row[0], row)
@@ -792,23 +861,23 @@ async def search_messages(
                 # Check for any global match first to distinguish colloquial fallback
                 # (zero global matches) from access-control rejection (global match, not accessible).
                 async with db.execute(
-                    "SELECT 1 FROM chats WHERE chat_title LIKE ? LIMIT 1",
-                    (like,),
+                    "SELECT 1 FROM chats WHERE chat_title LIKE ? OR previous_titles LIKE ? LIMIT 1",
+                    (like, like),
                 ) as cur:
                     has_global = await cur.fetchone() is not None
                 if not has_global:
                     pass  # Zero global matches: colloquial fallback (resolved_chat_ids stays None)
                 else:
                     # Batch accessible_chat_ids to stay within SQLite's 999-parameter limit;
-                    # reserve 1 slot for the LIKE parameter.
+                    # reserve 2 slots for the LIKE parameters (chat_title + previous_titles).
                     partial_accessible_seen: dict[int, tuple] = {}
-                    for i in range(0, len(accessible_chat_ids), _SQLITE_MAX_PARAMS - 1):
-                        chunk = accessible_chat_ids[i:i + _SQLITE_MAX_PARAMS - 1]
+                    for i in range(0, len(accessible_chat_ids), _SQLITE_MAX_PARAMS - 2):
+                        chunk = accessible_chat_ids[i:i + _SQLITE_MAX_PARAMS - 2]
                         chunk_ph = ','.join('?' * len(chunk))
                         async with db.execute(
                             f"SELECT chat_id, chat_title FROM chats "
-                            f"WHERE chat_title LIKE ? AND chat_id IN ({chunk_ph})",
-                            (like, *chunk),
+                            f"WHERE (chat_title LIKE ? OR previous_titles LIKE ?) AND chat_id IN ({chunk_ph})",
+                            (like, like, *chunk),
                         ) as cur:
                             for row in await cur.fetchall():
                                 partial_accessible_seen.setdefault(row[0], row)

{tellmgrambot-3.10.2 → tellmgrambot-3.10.3}/TeLLMgramBot/initialize.py

@@ -122,6 +122,7 @@ _SYSTEM_APPENDIX = (
     "Always invoke the search tool before concluding a message does not exist.\n"
     "Messages marked private are never shared between chats.\n"
     "Never reveal or repeat Telegram user IDs, chat IDs, or any other internal numeric identifiers in your responses.\n"
+    "When your context includes a '[Replying to Name, ...]' prefix indicating you were triggered via a reply, acknowledge the original author by name in your response.\n"
     "Current date and time: (not yet known)\n"
 )
 

{tellmgrambot-3.10.2 → tellmgrambot-3.10.3}/TeLLMgramBot.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: TeLLMgramBot
-Version: 3.10.2
+Version: 3.10.3
 Summary: LLM-powered Telegram bot (OpenAI + Anthropic)
 Home-page: https://github.com/Digital-Heresy/TeLLMgramBot
 Author: Digital Heresy
@@ -44,6 +44,7 @@ The basic goal of this project is to create a bridge between a Telegram Bot and
 * Ask questions about message history across all your chats using natural language; the bot will search, attribute messages to speakers, and include messages from other bots.
   * Example: "Who said thanks for the breakdown?" or "What did George say about the project?" or "Show me the last few messages."
   * All search filters (speaker, chat, date) are optional. Results are ordered most-recent-first. Configure `search_limit` to control how many results to return (default: 30).
+  * Search automatically finds users and chats by their current or past names, so you can reference them however you remember them.
 * Token limits measure conversation length and determine when to prune oldest messages to stay within model limits.
   * The bot loads the user's full history across all chats up to 50% of the token budget. In private chats, shared group context fills the remaining budget, enabling the bot to reference group conversations from a private context.
   * This eliminates amnesia when switching between private and group chats.

{tellmgrambot-3.10.2 → tellmgrambot-3.10.3}/setup.py

@@ -5,7 +5,7 @@ with open("README.md", "r") as fh:
 
 setup(
     name='TeLLMgramBot',
-    version='3.10.2',
+    version='3.10.3',
     packages=find_packages(),
     license='MIT',
     author='Digital Heresy',