TeLLMgramBot 3.0.2__tar.gz → 3.0.4__tar.gz

{tellmgrambot-3.0.2 → tellmgrambot-3.0.4}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: TeLLMgramBot
-Version: 3.0.2
+Version: 3.0.4
 Summary: LLM-powered Telegram bot (OpenAI + Anthropic)
 Home-page: https://github.com/Digital-Heresy/TeLLMgramBot
 Author: Digital Heresy

tellmgrambot-3.0.4/TeLLMgramBot/providers/__init__.py

@@ -0,0 +1 @@
+

{tellmgrambot-3.0.2 → tellmgrambot-3.0.4}/TeLLMgramBot/utils.py

@@ -85,15 +85,44 @@ def generate_file_path(directory: str, file: str, name="", text="") -> str:
     return path
 
 def generate_error_path(file='tellmgrambot_error.log') -> str:
-    """Generate a location of error log file by env variable vs. system temporary location."""
+    """
+    Generate the path to the error log file, respecting the TELLMGRAMBOT_ERRORLOGS_PATH environment variable.
+
+    If TELLMGRAMBOT_ERRORLOGS_PATH is not set, defaults to the system temporary directory and prints
+    a notice. Returns the full file path (directory + filename).
+
+    Args:
+        file (str): Error log filename. Defaults to 'tellmgrambot_error.log'.
+
+    Returns:
+        str: Full path to the error log file.
+    """
     env_var = 'TELLMGRAMBOT_ERRORLOGS_PATH'
     if not os.environ.get(env_var):
         os.environ[env_var] = gettempdir()
-        print(f"{env_var} initialized with: {os.environ[env_var]}")
+        print(f"{env_var} not set, using system temp: {os.environ[env_var]}")
     return generate_file_path(os.environ[env_var], file, f"empty {env_var}")
 
-def log_error(error: Exception or str, error_type: str, error_filename=generate_error_path()):
-    """Log an error message to a file including the timestamp and error type."""
+def log_error(error: Exception or str, error_type: str, error_filename=None):
+    """
+    Log an error message to both console and file with timestamp and error type.
+
+    The error filename is lazily evaluated: if not provided, the path is resolved by calling
+    `generate_error_path()`, which respects the TELLMGRAMBOT_ERRORLOGS_PATH environment variable.
+    This deferred resolution allows the environment variable to be set at any point before the
+    error is logged, rather than at import/definition time.
+
+    Args:
+        error (Exception or str): The error object or message to log.
+        error_type (str): A label for the error type (e.g., 'ValueError', 'ConnectionError').
+        error_filename (str, optional): Full path to the error log file. If None, `generate_error_path()`
+            is called to determine the path. Defaults to None.
+
+    Returns:
+        None: Logs to both console (via print) and the error file.
+    """
+    if error_filename is None:
+        error_filename = generate_error_path()
     text = f"{get_timestamp()} {error_type}: {error}"
     print(text) # Also log into console
     append_text_file_path(error_filename, text)

tellmgrambot-3.0.4/TeLLMgramBot.egg-info/PKG-INFO

@@ -0,0 +1,174 @@
+Metadata-Version: 2.4
+Name: TeLLMgramBot
+Version: 3.0.4
+Summary: LLM-powered Telegram bot (OpenAI + Anthropic)
+Home-page: https://github.com/Digital-Heresy/TeLLMgramBot
+Author: Digital Heresy
+Author-email: ronin.atx@gmail.com
+License: MIT
+Requires-Python: >=3.10
+Description-Content-Type: text/markdown
+License-File: LICENSE
+Requires-Dist: openai>2.0
+Requires-Dist: anthropic>=0.40
+Requires-Dist: PyYAML
+Requires-Dist: httpx
+Requires-Dist: beautifulsoup4
+Requires-Dist: validators
+Requires-Dist: tiktoken>=0.12
+Requires-Dist: python-telegram-bot>20.0
+Requires-Dist: aiosqlite>=0.19
+Dynamic: author
+Dynamic: author-email
+Dynamic: description
+Dynamic: description-content-type
+Dynamic: home-page
+Dynamic: license
+Dynamic: license-file
+Dynamic: requires-dist
+Dynamic: requires-python
+Dynamic: summary
+
+# TeLLMgramBot
+The basic goal of this project is to create a bridge between a Telegram Bot and a Large Language Model (LLM), supporting both OpenAI's GPT models and Anthropic's Claude models.
+* To use this library, you must have a Telegram account **with a user name**, not just a phone number. If you don't have one, [create one online](https://telegram.org/).
+* If added to a Telegram group, the bot must be [administrator](https://www.alphr.com/add-admin-telegram/) in order to respond to a user calling out its name, initials, or nickname.
+<img src="assets/TeLLMgramBot_Logo.png" width=200 align=center />
+
+## Telegram Bot + LLM Encapsulation
+* The Telegram interface handles special commands, especially on some basic "chatty" prompts and responses that don't require LLM, like "Hello".
+* The more dynamic conversation gets handed off to the LLM to manage prompts and responses, and Telegram acts as the interaction broker.
+* Pass the URL in [square brackets] and mention how the bot should interpret it.
+  * Example: "What do you think of this article? [https://some_site/article]"
+  * This uses a separate model (configurable via `url_model`) to support more URL content with its higher token limit.
+* Tokens are used to measure the length of all conversation messages between the Telegram bot assistant and the user. This is useful to:
+  * Ensure the length does not go over the model limit. If it does, prune oldest messages to fit within the limit.
+  * Remember past conversations when restarting: loads the user's full history across all chats (private and groups) plus all other participants' messages in the current chat, up to 50% of the token budget. This eliminates amnesia when users switch between contexts.
+* Users can manage privacy via two commands:
+  * `/forget` — In private chats, clears the full conversation (including bot replies). In group chats, removes only your messages; other participants' messages remain.
+  * `/private` — Toggles per-user private mode (private chats only). When ON, messages are excluded from group conversation contexts, enabling selective privacy even in shared groups.
+
+## Why Telegram?
+Using Telegram as the interface not only solves "exposing" the interface, but gives you boatloads of interactivity over a standard Command Line interface, or trying to create a website with input boxes and submit buttons to try to handle everything:
+1. Telegram already lets you paste in verbose, multiline messages.
+2. Telegram already lets you paste in pictures, videos, links, etc.
+3. Telegram already lets you react with emojis, stickers, etc.
+
+## Supported LLM Providers
+TeLLMgramBot selects the LLM provider automatically based on the model name:
+
+| Model prefix | Provider | Example models |
+|---|---|---|
+| `gpt-` | OpenAI | `gpt-4o`, `gpt-4o-mini`, `gpt-5-mini` |
+| `claude-` | Anthropic | `claude-sonnet-4-6`, `claude-haiku-4-5` |
+
+Simply set `chat_model` (and optionally `url_model`) in your `config.yaml` to any supported model and supply the corresponding API key — no other changes needed.
+
+## Directories
+When initializing TeLLMgramBot, the following directories get created:
+* `configs` - Contains bot configuration files.
+  * `config.yaml` (can be a different name)
+    * This file sets main bot parameters like naming and the LLM models to use.
+    * `chat_model` — the model used for normal conversation (e.g. `gpt-5-mini` or `claude-sonnet-4-6`).
+    * `url_model` — the model used to read and summarize URL content, can differ from `chat_model`.
+    * An empty `token_limit` will use the maximum tokens supported by the `chat_model`.
+  * `models.yaml`
+    * Contains token size parameters for all supported models.
+    * On first run, GPT and Claude model families are pre-populated. Additional models can be added manually.
+* `prompts` - Contains prompt files for how the bot interacts with any user.
+    * `test_personality.prmpt` (can be a different name)
+      * A sample prompt file defining the bot's personality: generic, helpful, and multi-provider-aware.
+      * The prompt emphasizes the bot's ability to fetch and analyze URLs passed in square brackets `[]`.
+      * The user can create more prompt files as needed for different personalities.
+    * `url_analysis.prmpt`
+      * Prompt template used to analyze URL content passed in brackets `[]`.
+* `errorlogs`
+   * Contains a `tellmgrambot_error.log` file to investigate if there are problems during the interaction.
+   * User will also get notified to contact the owner.
+* `data`
+  * Contains `conversations.db` — a SQLite database storing all conversations between the bot and users across all chats.
+  * When a user messages in any chat, their full history is available for context: private messages appear in group contexts, group messages appear in private contexts. This creates seamless cross-context awareness.
+  * Users can manage their context via `/forget` (private chat: clears full conversation; group chat: removes only your messages) or `/private` (toggles per-user privacy for group contexts).
+
+### Environment Variables
+TeLLMgramBot also creates or utilizes the following environment variables that can be pre-loaded, especially in containerized environments like Home Assistant with different persistent storage locations:
+1. `TELLMGRAMBOT_CONFIGS_PATH` — Directory containing `config.yaml` and `models.yaml`
+2. `TELLMGRAMBOT_PROMPTS_PATH` — Directory containing prompt files
+3. `TELLMGRAMBOT_ERRORLOGS_PATH` — Directory for error logs
+4. `TELLMGRAMBOT_DATA_PATH` — Directory containing `conversations.db` (e.g. `/data`). Defaults to `data/` in the execution directory.
+
+If none are defined, all paths default to subdirectories of the execution directory (the directory containing the entry-point script).
+
+## API Keys
+TeLLMgramBot supports the following API keys. Each can be supplied via environment variable or `.key` file:
+
+* **[OpenAI](https://platform.openai.com/overview)** — required when using a `gpt-*` model. Missing: chat and URL analysis disabled.
+* **[Anthropic](https://console.anthropic.com/)** — required when using a `claude-*` model. Missing: chat and URL analysis disabled.
+* **[Telegram](https://core.telegram.org/api)** — always required; available through BotFather. Missing: bot will not start.
+* **[VirusTotal](https://www.virustotal.com/gui/home/)** — optional; performs safety checks on URLs. Missing: URL analysis disabled.
+
+If a provider API key matching your configured model is missing, the bot will start but disable chat and URL analysis features. A startup summary shows which features are enabled.
+
+### Environment Variables
+TeLLMgramBot uses the following environment variables for API keys:
+1. `TELLMGRAMBOT_OPENAI_API_KEY` *(OpenAI models)*
+2. `TELLMGRAMBOT_ANTHROPIC_API_KEY` *(Anthropic models)*
+3. `TELLMGRAMBOT_TELEGRAM_API_KEY`
+4. `TELLMGRAMBOT_VIRUSTOTAL_API_KEY`
+
+During spin-up time, a user can call out `os.environ[env_var]` to set those variables, like the following example:
+```
+my_keys = Some_Vault_Fetch_Function()
+
+os.environ['TELLMGRAMBOT_OPENAI_API_KEY']     = my_keys['OpenAIKey']
+os.environ['TELLMGRAMBOT_ANTHROPIC_API_KEY']  = my_keys['AnthropicKey']
+os.environ['TELLMGRAMBOT_TELEGRAM_API_KEY']   = my_keys['BotFatherToken']
+os.environ['TELLMGRAMBOT_VIRUSTOTAL_API_KEY'] = my_keys['VirusTotalToken']
+```
+
+This means the user can implement whatever key vault they want to fetch the keys at runtime, without needing files stored in the directory.
+
+### API Key Files
+By default, API key files are created in the execution directory (or the directory specified by `TELLMGRAMBOT_KEYS_PATH` for legacy deployments):
+1. `openai.key` — OpenAI API key for GPT models
+2. `anthropic.key` — Anthropic API key for Claude models
+3. `telegram.key` — Telegram Bot API key
+4. `virustotal.key` — VirusTotal API key for URL safety checks
+
+Each file with the associated API key will update its respective environment variable if not defined. Missing provider keys (OpenAI or Anthropic) will disable chat and URL analysis but allow the bot to start. Missing VirusTotal keys will disable URL analysis.
+
+## Bot Setup
+This library includes an example script `test_local.py`, which uses files from the folders `configs` and `prompts` for TeLLMgramBot to process.
+1. Ensure the previous sections are followed with the proper API keys and your Telegram bot set.
+2. Install this library via PIP (`pip install TeLLMgramBot`) and then import into your project.
+3. Instantiate the bot by passing in various configuration pieces needed below.
+   Note the Telegram bot's full name and username auto-populate before startup.
+   ```
+   telegram_bot = TeLLMgramBot.TelegramBot(
+       bot_owner      = <Bot owner's Telegram username>,
+       bot_nickname   = <Bot nickname like 'Botty'>,
+       bot_initials   = <Bot initials like 'FB'>,
+       chat_model     = <Conversation model like 'gpt-4o-mini' or 'claude-sonnet-4-6'>,
+       url_model      = <URL analysis model like 'gpt-4o' or 'claude-haiku-4-5'>,
+       token_limit    = <Maximum token count set, by default chat_model max>,
+       persona_temp   = <LLM factual to creative value [0-2], by default 1.0>,
+       persona_prompt = <System prompt summarizing bot personality>
+   )
+   ```
+4. Turn on TeLLMgramBot by calling:
+   ```
+   telegram_bot.start_polling()
+   ```
+   Once you see `TeLLMgramBot polling...`, the bot is online in Telegram.
+5. Converse! Type `/help` for all available commands.
+
+## Resources
+* GitHub repository [python-telegram-bot](https://github.com/python-telegram-bot/python-telegram-bot) has guides to create a Telegram bot.
+* For more information on OpenAI models and token limits:
+  * [OpenAI model overview and maximum tokens](https://platform.openai.com/docs/models)
+  * [OpenAI message conversion to tokens](https://github.com/openai/openai-python)
+  * [OpenAI custom fine-tuning](https://platform.openai.com/docs/guides/model-optimization)
+  * [OpenAI's tiktoken library](https://github.com/openai/tiktoken/tree/main)
+* For more information on Anthropic Claude models:
+  * [Anthropic model overview and context windows](https://docs.anthropic.com/en/docs/about-claude/models)
+  * [Anthropic Python SDK](https://github.com/anthropic/anthropic-sdk-python)

{tellmgrambot-3.0.2 → tellmgrambot-3.0.4}/TeLLMgramBot.egg-info/SOURCES.txt

@@ -1,7 +1,5 @@
 LICENSE
-MANIFEST.in
 README.md
-requirements.txt
 setup.py
 TeLLMgramBot/TeLLMgramBot.py
 TeLLMgramBot/__init__.py
@@ -12,12 +10,13 @@ TeLLMgramBot/message_handlers.py
 TeLLMgramBot/models.py
 TeLLMgramBot/utils.py
 TeLLMgramBot/web_utils.py
+TeLLMgramBot.egg-info/PKG-INFO
+TeLLMgramBot.egg-info/SOURCES.txt
+TeLLMgramBot.egg-info/dependency_links.txt
+TeLLMgramBot.egg-info/requires.txt
+TeLLMgramBot.egg-info/top_level.txt
 TeLLMgramBot/providers/__init__.py
 TeLLMgramBot/providers/anthropic_provider.py
 TeLLMgramBot/providers/base.py
 TeLLMgramBot/providers/factory.py
-TeLLMgramBot/providers/openai_provider.py
-Tests/test_api_key_status.py
-Tests/test_conversation_db.py
-Tests/test_providers.py
-Tests/test_web_utils.py
+TeLLMgramBot/providers/openai_provider.py

tellmgrambot-3.0.4/TeLLMgramBot.egg-info/requires.txt

@@ -0,0 +1,9 @@
+openai>2.0
+anthropic>=0.40
+PyYAML
+httpx
+beautifulsoup4
+validators
+tiktoken>=0.12
+python-telegram-bot>20.0
+aiosqlite>=0.19

tellmgrambot-3.0.4/TeLLMgramBot.egg-info/top_level.txt

@@ -0,0 +1 @@
+TeLLMgramBot

{tellmgrambot-3.0.2 → tellmgrambot-3.0.4}/setup.py

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

tellmgrambot-3.0.2/MANIFEST.in

@@ -1,21 +0,0 @@
-# Include top-level docs
-include README.md
-include LICENSE
-include requirements.txt
-
-# Exclude development/CI-only files and directories
-exclude CLAUDE.md
-exclude RELEASE_NOTES.md
-exclude test_local.py
-exclude deploy_local.py
-recursive-exclude .claude *
-recursive-exclude .beans *
-recursive-include Tests *.py
-recursive-exclude configs *
-recursive-exclude prompts *
-recursive-exclude errorLogs *
-recursive-exclude keys *
-recursive-exclude data *
-recursive-exclude assets *
-recursive-exclude build *
-recursive-exclude *.egg-info *

tellmgrambot-3.0.2/Tests/test_api_key_status.py

@@ -1,234 +0,0 @@
-import os
-import unittest
-from unittest.mock import patch, MagicMock
-from TeLLMgramBot.initialize import ApiKeyStatus, init_keys, print_api_key_status, _model_provider
-
-# ---------------------------------------------------------------------------
-# Helpers
-# ---------------------------------------------------------------------------
-
-TELEGRAM_KEY = 'TELLMGRAMBOT_TELEGRAM_API_KEY'
-VIRUSTOTAL_KEY = 'TELLMGRAMBOT_VIRUSTOTAL_API_KEY'
-OPENAI_KEY = 'TELLMGRAMBOT_OPENAI_API_KEY'
-ANTHROPIC_KEY = 'TELLMGRAMBOT_ANTHROPIC_API_KEY'
-
-BASE_ENV = {
-    TELEGRAM_KEY: 'tg-key',
-    VIRUSTOTAL_KEY: 'vt-key',
-    OPENAI_KEY: 'openai-key',
-    ANTHROPIC_KEY: 'anthropic-key',
-}
-
-OPENAI_CONFIG = {'chat_model': 'gpt-5-mini', 'url_model': 'gpt-5'}
-ANTHROPIC_CONFIG = {'chat_model': 'claude-sonnet-4-6', 'url_model': 'claude-haiku-4-5'}
-MIXED_CONFIG = {'chat_model': 'gpt-5-mini', 'url_model': 'claude-haiku-4-5'}
-
-def _make_requires_provider(config: dict):
-    """Build a _requires_provider return value from a config dict."""
-    needs = {'openai': False, 'anthropic': False}
-    for field in ('chat_model', 'url_model'):
-        model = config.get(field, '')
-        if model.startswith('claude-'):
-            needs['anthropic'] = True
-        elif model:
-            needs['openai'] = True
-    return needs
-
-# ---------------------------------------------------------------------------
-# _model_provider helper
-# ---------------------------------------------------------------------------
-
-class TestModelProvider(unittest.TestCase):
-    """Tests for _model_provider() — maps a model name string to its provider name."""
-
-    def test_openai_model(self):
-        """A gpt-* model name must return 'openai'."""
-        self.assertEqual(_model_provider('gpt-5-mini'), 'openai')
-
-    def test_anthropic_model(self):
-        """A claude-* model name must return 'anthropic'."""
-        self.assertEqual(_model_provider('claude-sonnet-4-6'), 'anthropic')
-
-    def test_empty_model(self):
-        """An empty string must return None (no provider required)."""
-        self.assertIsNone(_model_provider(''))
-
-    def test_none_model(self):
-        """None must return None (no provider required)."""
-        self.assertIsNone(_model_provider(None))
-
-# ---------------------------------------------------------------------------
-# ApiKeyStatus defaults
-# ---------------------------------------------------------------------------
-
-class TestApiKeyStatusDefaults(unittest.TestCase):
-    """Tests for ApiKeyStatus dataclass default values."""
-
-    def test_all_features_enabled_by_default(self):
-        """A freshly constructed ApiKeyStatus must have all features enabled and no disabled reasons."""
-        fm = ApiKeyStatus()
-        self.assertTrue(fm.chat_enabled)
-        self.assertTrue(fm.url_analysis_enabled)
-        self.assertEqual(fm.disabled_reasons, {})
-
-# ---------------------------------------------------------------------------
-# init_keys — all keys present
-# ---------------------------------------------------------------------------
-
-class TestInitKeysAllPresent(unittest.TestCase):
-    """Tests for init_keys() when all required API keys are present."""
-
-    @patch('TeLLMgramBot.initialize.read_yaml', return_value=OPENAI_CONFIG)
-    @patch('TeLLMgramBot.initialize._requires_provider', return_value=_make_requires_provider(OPENAI_CONFIG))
-    @patch('TeLLMgramBot.initialize.read_text', return_value='fake-key')
-    @patch('TeLLMgramBot.initialize.generate_file_path')
-    @patch('os.path.exists', return_value=True)
-    def test_all_features_enabled_when_all_keys_present(self, mock_exists, mock_gen, mock_read, mock_req, mock_yaml):
-        """All features must be enabled when all API keys are available."""
-        with patch.dict(os.environ, BASE_ENV, clear=False):
-            fm = init_keys()
-        self.assertTrue(fm.chat_enabled)
-        self.assertTrue(fm.url_analysis_enabled)
-        self.assertEqual(fm.disabled_reasons, {})
-
-# ---------------------------------------------------------------------------
-# init_keys — Telegram key missing → sys.exit
-# ---------------------------------------------------------------------------
-
-class TestInitKeysTelegramMissing(unittest.TestCase):
-    """Tests for init_keys() hard exit when the Telegram key is missing."""
-
-    @patch('TeLLMgramBot.initialize.read_yaml', return_value=OPENAI_CONFIG)
-    @patch('TeLLMgramBot.initialize._requires_provider', return_value=_make_requires_provider(OPENAI_CONFIG))
-    @patch('TeLLMgramBot.initialize.generate_file_path')
-    @patch('os.path.exists', return_value=False)
-    def test_exits_when_telegram_key_missing(self, mock_exists, mock_gen, mock_req, mock_yaml):
-        """init_keys() must call sys.exit() when the Telegram API key is absent — it is always required."""
-        env = {k: v for k, v in BASE_ENV.items() if k != TELEGRAM_KEY}
-        with patch.dict(os.environ, env, clear=False):
-            os.environ.pop(TELEGRAM_KEY, None)
-            with self.assertRaises(SystemExit):
-                init_keys()
-
-# ---------------------------------------------------------------------------
-# init_keys — provider key missing → feature disabled
-# ---------------------------------------------------------------------------
-
-class TestInitKeysProviderMissing(unittest.TestCase):
-    """Tests for init_keys() graceful degradation when a provider key (OpenAI/Anthropic) is missing."""
-
-    @patch('TeLLMgramBot.initialize.read_yaml', return_value=OPENAI_CONFIG)
-    @patch('TeLLMgramBot.initialize._requires_provider', return_value=_make_requires_provider(OPENAI_CONFIG))
-    @patch('TeLLMgramBot.initialize.generate_file_path')
-    @patch('os.path.exists', return_value=False)
-    def test_chat_disabled_when_openai_key_missing(self, mock_exists, mock_gen, mock_req, mock_yaml):
-        """Missing OpenAI key must disable both chat and URL analysis when an OpenAI model is configured."""
-        env = {k: v for k, v in BASE_ENV.items() if k != OPENAI_KEY}
-        with patch.dict(os.environ, env, clear=False):
-            os.environ.pop(OPENAI_KEY, None)
-            fm = init_keys()
-        self.assertFalse(fm.chat_enabled)
-        self.assertFalse(fm.url_analysis_enabled)
-        self.assertIn('chat', fm.disabled_reasons)
-
-    @patch('TeLLMgramBot.initialize.read_yaml', return_value=ANTHROPIC_CONFIG)
-    @patch('TeLLMgramBot.initialize._requires_provider', return_value=_make_requires_provider(ANTHROPIC_CONFIG))
-    @patch('TeLLMgramBot.initialize.generate_file_path')
-    @patch('os.path.exists', return_value=False)
-    def test_chat_disabled_when_anthropic_key_missing(self, mock_exists, mock_gen, mock_req, mock_yaml):
-        """Missing Anthropic key must disable both chat and URL analysis when a Claude model is configured."""
-        env = {k: v for k, v in BASE_ENV.items() if k != ANTHROPIC_KEY}
-        with patch.dict(os.environ, env, clear=False):
-            os.environ.pop(ANTHROPIC_KEY, None)
-            fm = init_keys()
-        self.assertFalse(fm.chat_enabled)
-        self.assertFalse(fm.url_analysis_enabled)
-        self.assertIn('chat', fm.disabled_reasons)
-
-    @patch('TeLLMgramBot.initialize.read_yaml', return_value=OPENAI_CONFIG)
-    @patch('TeLLMgramBot.initialize._requires_provider', return_value=_make_requires_provider(OPENAI_CONFIG))
-    def test_does_not_exit_when_only_provider_key_missing(self, mock_req, mock_yaml):
-        """init_keys() must not call sys.exit() when only a provider key is missing — graceful degradation."""
-        env = {k: v for k, v in BASE_ENV.items() if k != OPENAI_KEY}
-        with patch.dict(os.environ, env, clear=False):
-            os.environ.pop(OPENAI_KEY, None)
-            with patch('TeLLMgramBot.initialize.generate_file_path'):
-                with patch('os.path.exists', return_value=False):
-                    try:
-                        fm = init_keys()
-                    except SystemExit:
-                        self.fail("init_keys() should not exit when only a provider key is missing")
-
-# ---------------------------------------------------------------------------
-# init_keys — VirusTotal key missing → virustotal + url_analysis disabled
-# ---------------------------------------------------------------------------
-
-class TestInitKeysVirusTotalMissing(unittest.TestCase):
-    """Tests for init_keys() graceful degradation when the VirusTotal key is missing."""
-
-    @patch('TeLLMgramBot.initialize.read_yaml', return_value=OPENAI_CONFIG)
-    @patch('TeLLMgramBot.initialize._requires_provider', return_value=_make_requires_provider(OPENAI_CONFIG))
-    @patch('TeLLMgramBot.initialize.generate_file_path')
-    @patch('os.path.exists', return_value=False)
-    def test_virustotal_and_url_analysis_disabled_when_vt_key_missing(self, mock_exists, mock_gen, mock_req, mock_yaml):
-        """Missing VirusTotal key must disable url_analysis while leaving chat enabled."""
-        env = {k: v for k, v in BASE_ENV.items() if k != VIRUSTOTAL_KEY}
-        with patch.dict(os.environ, env, clear=False):
-            os.environ.pop(VIRUSTOTAL_KEY, None)
-            fm = init_keys()
-        self.assertTrue(fm.chat_enabled)
-        self.assertFalse(fm.url_analysis_enabled)
-        self.assertIn('url_analysis', fm.disabled_reasons)
-
-# ---------------------------------------------------------------------------
-# init_keys — both provider and VirusTotal keys missing
-# ---------------------------------------------------------------------------
-
-class TestInitKeysBothProviderAndVTMissing(unittest.TestCase):
-    """Tests for init_keys() when both the provider key and VirusTotal key are missing."""
-
-    @patch('TeLLMgramBot.initialize.read_yaml', return_value=OPENAI_CONFIG)
-    @patch('TeLLMgramBot.initialize._requires_provider', return_value=_make_requires_provider(OPENAI_CONFIG))
-    @patch('TeLLMgramBot.initialize.generate_file_path')
-    @patch('os.path.exists', return_value=False)
-    def test_url_analysis_disabled_when_both_provider_and_vt_keys_missing(self, mock_exists, mock_gen, mock_req, mock_yaml):
-        """When both the provider key and VirusTotal key are missing, url_analysis must be disabled with the provider key cited as the reason."""
-        env = {k: v for k, v in BASE_ENV.items() if k not in (OPENAI_KEY, VIRUSTOTAL_KEY)}
-        with patch.dict(os.environ, env, clear=False):
-            os.environ.pop(OPENAI_KEY, None)
-            os.environ.pop(VIRUSTOTAL_KEY, None)
-            fm = init_keys()
-        self.assertFalse(fm.chat_enabled)
-        self.assertFalse(fm.url_analysis_enabled)
-        self.assertIn('openai', fm.disabled_reasons.get('url_analysis', ''))
-
-# ---------------------------------------------------------------------------
-# print_api_key_status output
-# ---------------------------------------------------------------------------
-
-class TestPrintApiKeyStatus(unittest.TestCase):
-    """Tests for print_api_key_status() startup output formatting."""
-
-    def test_all_enabled_output(self):
-        """When all features are enabled, output must contain ENABLED and no DISABLED lines."""
-        fm = ApiKeyStatus()
-        with patch('builtins.print') as mock_print:
-            print_api_key_status(fm)
-        output = ' '.join(str(c) for c in [call.args[0] for call in mock_print.call_args_list])
-        self.assertIn('ENABLED', output)
-        self.assertNotIn('DISABLED', output)
-
-    def test_disabled_feature_shows_reason(self):
-        """A disabled feature must appear as DISABLED with its reason string in the output."""
-        fm = ApiKeyStatus(
-            chat_enabled=False,
-            disabled_reasons={'chat': 'missing openai API key'}
-        )
-        with patch('builtins.print') as mock_print:
-            print_api_key_status(fm)
-        output = ' '.join(str(call.args[0]) for call in mock_print.call_args_list)
-        self.assertIn('DISABLED', output)
-        self.assertIn('missing openai API key', output)
-
-if __name__ == '__main__':
-    unittest.main()