Skip to main content

AI Assistant Framework with CLI and Telegram Bot

Project description

Assistants Framework

A flexible framework for creating AI assistants with multiple frontend interfaces.

Features

  • Multi-Front-End Support: CLI and Telegram interfaces built on the same core framework
  • CLI Features: Code highlighting, thread management, editor integration, file input, image generation
  • Multiple LLM Support: OpenAI (gpt-*, o*), Anthropic (claude-*), MistralAI (mistral-*, codestral-*), and image generation (DALL-E)
  • New Universal Assistant Interface: See MIGRATION_GUIDE.md for details

Installation

Requires Python 3.11+

pip install assistants-framework

For Telegram bot functionality:

pip install assistants-framework[telegram]

Add commands to your PATH:

ai-cli install

Usage

Command Line Interface

ai-cli --help

Key CLI commands (prefixed with /):

  • /help - Show help message
  • /editor - Open editor for prompt composition
  • /image <prompt> - Generate an image
  • /copy - Copy response to clipboard
  • /new - Start new thread
  • /threads - List and select threads
  • /thinking <level> - Toggle thinking mode (for reasoning models)
  • /last - Retrieve last message

Use the claude command for Anthropic models (Now defaults to Claude 4):

claude -e  # Open editor for Claude

There's also a chatgpt command that uses the default ChatGPT model:

chatgpt -t  # Continue the last thread with ChatGPT (`gpt-4.1-mini`)

Run migrations in case of breaking changes:

ai-cli migrate

Rebuild the database:

ai-cli rebuild

Telegram Interface

The framework includes a Telegram bot interface with the following features:

  • User Management: Authorise/deauthorise users and chats, promote/demote users
  • Thread Management: Start new conversation threads
  • Auto-Reply Toggle: Enable/disable automatic responses
  • Media Generation: Generate images from text prompts
  • Voice Responses: Generate audio responses with the /voice command

Key Telegram commands:

  • /new_thread - Clear conversation history and start a new thread
  • /auto_reply - Toggle automatic responses on/off
  • /image <prompt> - Generate an image from a text prompt
  • /voice <text> - Generate an audio response.

Environment Variables

  • ASSISTANT_INSTRUCTIONS - System message (default: "You are a helpful assistant")
  • ASSISTANTS_API_KEY_NAME - API key variable name (default: OPENAI_API_KEY)
  • ANTHROPIC_API_KEY_NAME - Anthropic API key variable (default: ANTHROPIC_API_KEY)
  • MISTRAL_API_KEY_NAME - Mistral API key variable (default: MISTRAL_API_KEY)
  • DEFAULT_MODEL - Default model (default: gpt-5-mini)
  • DEFAULT_CLAUDE_SONNET_MODEL - Default Claude model (default: claude-sonnet-4-20250514)
  • DEFAULT_CLAUDE_OPUS_MODEL - Default Claude Opus model (default: claude-opus-4-1-20250805)
  • DEFAULT_CHATGPT_MODEL - Default ChatGPT model (default: gpt-5-mini)
  • DEFAULT_GPT_REASONING_MODEL - Default GPT reasoning model (default: o4-mini)
  • CODE_MODEL - Reasoning model (default: o4-mini)
  • IMAGE_MODEL - Image model (default: gpt-image-1)
  • ASSISTANTS_DATA_DIR - Data directory (default: ~/.local/share/assistants)
  • ASSISTANTS_CONFIG_DIR - Config directory (default: ~/.config/assistants)
  • TG_BOT_TOKEN - Telegram bot token
  • OPEN_IMAGES_IN_BROWSER - Open images automatically (default: true)
  • DEFAULT_MAX_RESPONSE_TOKENS - Default max response tokens (default: 4096)
  • DEFAULT_MAX_HISTORY_TOKENS - Default max history tokens (default: 10000)

File Tagging in Prompts

You can include the contents of files directly in your prompt by tagging them with an @ followed by the file path. Both absolute and relative paths are supported. For example:

Hi Claude, can you check my @~/.zshrc and tell me what you think?
Hi Claude, can you check my @./my_local_config.txt and tell me what you think?

When you use a tag like @/path/to/file.txt or @relative/path/to/file.txt in your prompt, the assistant will automatically append the contents of that file to the end of your input, like this:

Hi Claude, can you check my @./my_local_config.txt and tell me what you think?

===./my_local_config.txt===
// file content
===EOF===
  • You can tag multiple files in a single prompt; each will be appended in the same format.
  • If a file cannot be read, an error message will be shown in place of its content.
  • Both absolute (starting with /) and relative paths (like ./file.txt or subdir/file.txt) are supported for tagging.

Contributing

Contributions welcome! Fork the repository, make changes, and submit a pull request.

Useful Make Commands

  • make help – Show all available make commands
  • make install – Install package for production
  • make install-dev – Install package with development dependencies
  • make dev-setup – Complete development environment setup
  • make lint – Run all pre-commit hooks on all files
  • make format – Format code with ruff
  • make mypy – Run mypy type checks (baseline)
  • make mypy-generate – Generate a new mypy baseline
  • make test – Run all tests with pytest
  • make clean – Remove build artifacts and cache files
  • make build – Build distribution packages
  • make version – Show current version

TODOs:

  • Improved conversation handling/truncation for token limits - currently uses tiktoken for all models
  • Additional model/API support
  • Additional database support

License

MIT License

Project details


Release history Release notifications | RSS feed

Download files

Download the file for your platform. If you're not sure which to choose, learn more about installing packages.

Source Distribution

assistants_framework-0.8.4.tar.gz (58.3 kB view details)

Uploaded Source

Built Distribution

If you're not sure about the file name format, learn more about wheel file names.

assistants_framework-0.8.4-py3-none-any.whl (74.8 kB view details)

Uploaded Python 3

File details

Details for the file assistants_framework-0.8.4.tar.gz.

File metadata

  • Download URL: assistants_framework-0.8.4.tar.gz
  • Upload date:
  • Size: 58.3 kB
  • Tags: Source
  • Uploaded using Trusted Publishing? No
  • Uploaded via: twine/6.2.0 CPython/3.12.11

File hashes

Hashes for assistants_framework-0.8.4.tar.gz
Algorithm Hash digest
SHA256 ce725a3a46211fb86143d9ded0c0084c96053a49db2806e9d8318a28f074ec79
MD5 327507f132f9da8cceb8a75adcce9797
BLAKE2b-256 feea621c984c47bf7910f03f17ab3e0663b9b4c30c31ee8c4012b5d2bf6bacf7

See more details on using hashes here.

File details

Details for the file assistants_framework-0.8.4-py3-none-any.whl.

File metadata

File hashes

Hashes for assistants_framework-0.8.4-py3-none-any.whl
Algorithm Hash digest
SHA256 766b357d3718ad4d36442f4de5a099aa6c44fa4f680239d8f0147047ee10c19c
MD5 4b124f182ce63dcab8023313d2a1554c
BLAKE2b-256 0ef5b1f2a12c70cd8785c140057964d51f75ed3123fb9f711e911904d640d347

See more details on using hashes here.

Supported by

AWS Cloud computing and Security Sponsor Datadog Monitoring Depot Continuous Integration Fastly CDN Google Download Analytics Pingdom Monitoring Sentry Error logging StatusPage Status page