DEV Community

Cover image for Building a Golang Telegram Bot for Personal AI Assistance
Mukul Saini
Mukul Saini

Posted on

1

Building a Golang Telegram Bot for Personal AI Assistance

Creating a Telegram bot that integrates with an AI assistant is an exciting project that combines real-time messaging, AI processing, and Golang's powerful concurrency model. In this blog, I'll guide you through the architecture, features, and implementation of a Telegram bot using Golang, with a complete system design to boot!

telegram bot golang

Overview

This project builds a personal AI assistant accessible via two interfaces:

  1. Telegram Bot: A real-time conversational interface.

  2. Console Chat: A terminal-based chatbot for direct interaction.

Key Features:

  • AI model switching based on user input.

  • Persistent chat history across sessions.

  • Interactive bot responses with live editing.

  • Retry mechanism for robust API handling.


System Design

Architecture
The system comprises the following components:

  • Bot Interface: Handles incoming messages, processes user input, and sends responses.

  • Assistant Utilities: Contains AI model integration logic.

  • History Management: Manages conversation history for persistence.

  • Error Handling: Ensures graceful error recovery.

Flow Diagram

[User] <---> [Telegram API] <---> [Bot API Handler] <---> [AI Processing Logic]
                                  ^                      |
                                  |                      v
                            [History Management]  [Error Handler]

Enter fullscreen mode Exit fullscreen mode

Components

  • Telegram Bot API: Utilized for real-time messaging and updates.
  • Golang’s Concurrency: To handle multiple user conversations seamlessly.
  • History Module: Saves chat history in local files for reloadable sessions.
  • Assistant Utilities: Processes user input and integrates AI logic.

Image description

Key Code Walkthrough

  1. Main Application Entry The main() function provides users with two modes: Telegram Bot and Console Chat. This ensures the assistant is accessible both online and offline.
func main() {
    fmt.Println("Choose mode: [1] Telegram Bot, [2] Console Chat")
    var choice int
    fmt.Scan(&choice)
    switch choice {
        case 1:
            deploy.TelegramBot()
        case 2:
            runConsoleChat()
        default:
            fmt.Println("Invalid choice.")
    }
}
Enter fullscreen mode Exit fullscreen mode
  1. Telegram Bot Initialization The bot uses go-telegram-bot-api for interacting with Telegram.
func NewTelegramBot() {
    token := os.Getenv("TELEGRAM_BOT_TOKEN")
    bot, err := NewBot(token)
    if err != nil {
        log.Fatal("Failed to start bot:", err)
    }
    bot.Start(context.Background())
}
Enter fullscreen mode Exit fullscreen mode
  1. Handling User Messages

olama bot golang
This method manages user interactions, including history loading and AI response handling.

func handleTelegramMessageNew(ctx context.Context, b *tgbotapi.BotAPI, client *Client, update tgbotapi.Update) {
    userMessage := update.Message.Text
    response, err := assistantutils.ProcessUserMessage(userMessage, client.ModelType, client.History, "", nil, true)
    if err != nil {
        log.Println("Error processing message:", err)
        b.Send(tgbotapi.NewMessage(update.Message.Chat.ID, "An error occurred. Please try again."))
    }
    b.Send(tgbotapi.NewMessage(update.Message.Chat.ID, response))
}

Enter fullscreen mode Exit fullscreen mode
  1. Persistent History Management Chat history is managed through file-based persistence for continuity between sessions.

telegram golang

func selectSession(files []string) string {
    if len(files) == 0 {
        return history.GenerateFileName("conversation")
    }
}

Enter fullscreen mode Exit fullscreen mode
  1. AI Processing Logic The bot integrates with an AI assistant, handling retries and partial responses.
func processUserMessageWithRetry(ctx context.Context, userMessage, modelType string, hist *history.History, identifier string, sendChunk func(chunk string, isFinal bool)) (string, error) {
    for i := 0; i < 3; i++ {
        response, err := assistantutils.ProcessUserMessage(userMessage, modelType, hist, identifier, sendChunk, true)
        if err == nil {
            return response, nil
        }
        time.Sleep(time.Second * (i + 1))
    }
    return "", fmt.Errorf("failed to process message after retries")
}

Enter fullscreen mode Exit fullscreen mode

Want a demo ?



demo


Conclusion
With this bot, we leverage Golang's concurrency and efficient libraries to build a scalable and interactive AI assistant. The integration with Telegram API ensures a seamless real-time experience for users. Start building yours today and explore the power of AI-driven conversations!

API Trace View

How I Cut 22.3 Seconds Off an API Call with Sentry

Struggling with slow API calls? Dan Mindru walks through how he used Sentry's new Trace View feature to shave off 22.3 seconds from an API call.

Get a practical walkthrough of how to identify bottlenecks, split tasks into multiple parallel tasks, identify slow AI model calls, and more.

Read more →

Top comments (0)

A Workflow Copilot. Tailored to You.

Pieces.app image

Our desktop app, with its intelligent copilot, streamlines coding by generating snippets, extracting code from screenshots, and accelerating problem-solving.

Read the docs

👋 Kindness is contagious

Discover a treasure trove of wisdom within this insightful piece, highly respected in the nurturing DEV Community enviroment. Developers, whether novice or expert, are encouraged to participate and add to our shared knowledge basin.

A simple "thank you" can illuminate someone's day. Express your appreciation in the comments section!

On DEV, sharing ideas smoothens our journey and strengthens our community ties. Learn something useful? Offering a quick thanks to the author is deeply appreciated.

Okay