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!

AWS Security LIVE!

Tune in for AWS Security LIVE!

Join AWS Security LIVE! for expert insights and actionable tips to protect your organization and keep security teams prepared.

Learn 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

Engage with a sea of insights in this enlightening article, highly esteemed within the encouraging DEV Community. Programmers of every skill level are invited to participate and enrich our shared knowledge.

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

On DEV, sharing knowledge smooths our journey and strengthens our community bonds. Found this useful? A brief thank you to the author can mean a lot.

Okay