From 43561de2982bda03d583832bbcac1afb102b89ba Mon Sep 17 00:00:00 2001 From: bucolucas Date: Sun, 18 Aug 2024 16:53:54 -0500 Subject: [PATCH 1/3] Implement daemon and apprentice bot system --- telegram_inference_bot.py | 87 ++++++++++++++++++++++++++++++++------- 1 file changed, 73 insertions(+), 14 deletions(-) diff --git a/telegram_inference_bot.py b/telegram_inference_bot.py index 9cb6bb0..8b0c929 100644 --- a/telegram_inference_bot.py +++ b/telegram_inference_bot.py @@ -4,6 +4,8 @@ import importlib import inspect import logging import asyncio +import subprocess +import requests from telegram import Update, __version__ as telegram_version, InlineKeyboardButton, InlineKeyboardMarkup from telegram.ext import Application, CommandHandler, MessageHandler, filters, ContextTypes, CallbackQueryHandler from openai import OpenAI @@ -38,8 +40,12 @@ logging.basicConfig(level=logging.WARNING, handlers=[ logging.FileHandler('logs/output.log', mode='a') ]) -# Set up Telegram bot -TELEGRAM_BOT_TOKEN = os.getenv('TELEGRAM_BOT_TOKEN') +# Set up Telegram bots +DAEMON_BOT_TOKEN = os.getenv('TELEGRAM_BOT_TOKEN') +APPRENTICE_BOT_TOKEN = os.getenv('TELEGRAM_APPRENTICE_BOT_TOKEN') +GITHUB_REPO_OWNER = os.getenv('GITHUB_REPO_OWNER') +GITHUB_REPO_NAME = os.getenv('GITHUB_REPO_NAME') +GITHUB_ACCESS_TOKEN = os.getenv('GITHUB_ACCESS_TOKEN') # Load system prompt with open("prompts/developer_prompt.txt", "r") as file: @@ -272,22 +278,75 @@ async def abort_processing(update: Update, context: ContextTypes.DEFAULT_TYPE) - else: await query.edit_message_text(text="No active processing to abort.") +async def handover(update: Update, context: ContextTypes.DEFAULT_TYPE) -> None: + if context.bot.token == DAEMON_BOT_TOKEN: + # Daemon bot initiating handover + apprentice_chat_id = os.getenv('APPRENTICE_CHAT_ID') + await context.bot.send_message(chat_id=apprentice_chat_id, text="Handover initiated. Taking control.") + await update.message.reply_text("Handover initiated. Apprentice bot is now in control.") + else: + await update.message.reply_text("Handover can only be initiated by the daemon bot.") + +async def update_apprentice(update: Update, context: ContextTypes.DEFAULT_TYPE) -> None: + if context.bot.token == APPRENTICE_BOT_TOKEN: + try: + # Pull latest changes + subprocess.run(["git", "pull", "origin", "main"], check=True) + + # Restart the bot + os.execv(sys.executable, ['python'] + sys.argv) + except subprocess.CalledProcessError as e: + logging.error(f"Failed to pull latest changes: {e}") + await update.message.reply_text("Failed to update. Please check the logs.") + except Exception as e: + logging.error(f"Failed to restart the bot: {e}") + await update.message.reply_text("Failed to restart. Please check the logs.") + else: + await update.message.reply_text("Update can only be performed by the apprentice bot.") + +async def check_for_updates(context: ContextTypes.DEFAULT_TYPE) -> None: + url = f"https://api.github.com/repos/{GITHUB_REPO_OWNER}/{GITHUB_REPO_NAME}/pulls" + headers = {"Authorization": f"token {GITHUB_ACCESS_TOKEN}"} + response = requests.get(url, headers=headers) + + if response.status_code == 200: + pull_requests = response.json() + for pr in pull_requests: + if pr['state'] == 'closed' and pr['merged']: + # A pull request was merged, update the apprentice bot + await context.bot.send_message(chat_id=os.getenv('APPRENTICE_CHAT_ID'), text="A new update is available. Updating now...") + await update_apprentice(None, context) + break + def main() -> None: # Create the Application and pass it your bot's token - application = Application.builder().token(TELEGRAM_BOT_TOKEN).build() + daemon_app = Application.builder().token(DAEMON_BOT_TOKEN).build() + apprentice_app = Application.builder().token(APPRENTICE_BOT_TOKEN).build() - # Add handlers - application.add_handler(CommandHandler("start", start)) - application.add_handler(CommandHandler("clear", clear)) - application.add_handler(CommandHandler("switch", switch)) - application.add_handler(CommandHandler("toggle", switch_providers)) - application.add_handler(CommandHandler("status", status)) - application.add_handler(MessageHandler(filters.TEXT & ~filters.COMMAND, handle_message)) - application.add_handler(CallbackQueryHandler(abort_processing, pattern='^abort$')) + # Add handlers for both bots + for app in [daemon_app, apprentice_app]: + app.add_handler(CommandHandler("start", start)) + app.add_handler(CommandHandler("clear", clear)) + app.add_handler(CommandHandler("switch", switch)) + app.add_handler(CommandHandler("toggle", switch_providers)) + app.add_handler(CommandHandler("status", status)) + app.add_handler(MessageHandler(filters.TEXT & ~filters.COMMAND, handle_message)) + app.add_handler(CallbackQueryHandler(abort_processing, pattern='^abort$')) - # Start the Bot - logging.info("Bot is running...") - application.run_polling() + # Add handover command only to daemon bot + daemon_app.add_handler(CommandHandler("handover", handover)) + + # Add update command only to apprentice bot + apprentice_app.add_handler(CommandHandler("update", update_apprentice)) + + # Set up job queue to check for updates every 15 minutes + job_queue = apprentice_app.job_queue + job_queue.run_repeating(check_for_updates, interval=900, first=10) + + # Start both bots + logging.info("Bots are running...") + daemon_app.run_polling() + apprentice_app.run_polling() if __name__ == '__main__': main() \ No newline at end of file From 92f3887506f0ec49b6f9f730a6415a9f93dabe48 Mon Sep 17 00:00:00 2001 From: bucolucas Date: Sun, 18 Aug 2024 16:54:06 -0500 Subject: [PATCH 2/3] Add .env.example file with new environment variables --- .env.example | 18 ++++++++++++++++++ 1 file changed, 18 insertions(+) create mode 100644 .env.example diff --git a/.env.example b/.env.example new file mode 100644 index 0000000..2ee2716 --- /dev/null +++ b/.env.example @@ -0,0 +1,18 @@ +# Telegram Bot Tokens +TELEGRAM_BOT_TOKEN=your_daemon_bot_token_here +TELEGRAM_APPRENTICE_BOT_TOKEN=your_apprentice_bot_token_here + +# OpenAI API Key +OPENAI_API_KEY=your_openai_api_key_here + +# Anthropic API Key +ANTHROPIC_API_KEY=your_anthropic_api_key_here + +# GitHub Repository Information +GITHUB_REPO_OWNER=your_github_username_or_organization +GITHUB_REPO_NAME=your_repo_name +GITHUB_ACCESS_TOKEN=your_github_personal_access_token + +# Chat IDs +DAEMON_CHAT_ID=your_daemon_chat_id +APPRENTICE_CHAT_ID=your_apprentice_chat_id \ No newline at end of file From cf0c81e569d76e71ac12b6f8dbc310c363648eeb Mon Sep 17 00:00:00 2001 From: bucolucas Date: Sun, 18 Aug 2024 16:54:28 -0500 Subject: [PATCH 3/3] Add README.md with information about daemon and apprentice bot system --- README.md | 76 +++++++++++++++++++++++++++++++++++++++++++++++++++++++ 1 file changed, 76 insertions(+) diff --git a/README.md b/README.md index e69de29..013fb2a 100644 --- a/README.md +++ b/README.md @@ -0,0 +1,76 @@ +# Telegram Inference Bot with Daemon and Apprentice System + +This project implements a Telegram bot system with two instances: a daemon bot and an apprentice bot. The daemon bot handles the main workload, while the apprentice bot can be updated and take over when needed. + +## Features + +- Daemon bot for handling main workload +- Apprentice bot for updates and handovers +- Automatic updates for the apprentice bot when new pull requests are merged +- Inter-bot communication +- Support for both OpenAI and Anthropic AI models +- Tool integration for extended functionality + +## Setup + +1. Clone the repository: + ``` + git clone https://github.com/your_username/your_repo_name.git + cd your_repo_name + ``` + +2. Install the required dependencies: + ``` + pip install -r requirements.txt + ``` + +3. Copy the `.env.example` file to `.env` and fill in your actual values: + ``` + cp .env.example .env + ``` + +4. Edit the `.env` file with your specific tokens and settings. + +5. Run the bot: + ``` + python telegram_inference_bot.py + ``` + +## Environment Variables + +Make sure to set the following environment variables in your `.env` file: + +- `TELEGRAM_BOT_TOKEN`: Token for the daemon bot +- `TELEGRAM_APPRENTICE_BOT_TOKEN`: Token for the apprentice bot +- `OPENAI_API_KEY`: Your OpenAI API key +- `ANTHROPIC_API_KEY`: Your Anthropic API key +- `GITHUB_REPO_OWNER`: Your GitHub username or organization +- `GITHUB_REPO_NAME`: Your repository name +- `GITHUB_ACCESS_TOKEN`: Your GitHub personal access token +- `DAEMON_CHAT_ID`: Chat ID for the daemon bot +- `APPRENTICE_CHAT_ID`: Chat ID for the apprentice bot + +## Usage + +- Start the bot by running `python telegram_inference_bot.py` +- Use `/start` to begin interacting with either bot +- Use `/handover` in the daemon bot to initiate a handover to the apprentice bot +- The apprentice bot will automatically check for updates every 15 minutes + +## Commands + +- `/start`: Start interacting with the bot +- `/clear`: Clear conversation history +- `/switch`: Switch between GPT models (GPT-4 and GPT-4 Mini) +- `/toggle`: Toggle between OpenAI and Anthropic providers +- `/status`: Check the current model and provider +- `/handover`: (Daemon bot only) Initiate handover to apprentice bot +- `/update`: (Apprentice bot only) Manually trigger an update + +## Contributing + +Please read [CONTRIBUTING.md](CONTRIBUTING.md) for details on our code of conduct and the process for submitting pull requests. + +## License + +This project is licensed under the MIT License - see the [LICENSE.md](LICENSE.md) file for details. \ No newline at end of file