@codebam/cf-workers-telegram-bot 12.4.0 → 12.6.0

package/dist/history_manager.d.ts

@@ -0,0 +1,37 @@
+/** Class representing a manager for conversation history stored in KV */
+export declare class HistoryManager {
+    private kv;
+    constructor(kv: KVNamespace);
+    private getKey;
+    /**
+     * Get the conversation history for a user
+     * @param userId - the telegram user ID
+     * @param threadId - optional thread ID
+     * @returns array of messages
+     */
+    getHistory(userId: number, threadId?: number): Promise<{
+        role: string;
+        content: string;
+    }[]>;
+    /**
+     * Add a message and its response to the history
+     * @param userId - the telegram user ID
+     * @param prompt - the user message
+     * @param response - the bot response
+     * @param threadId - optional thread ID
+     */
+    addMessage(userId: number, prompt: string, response: string, threadId?: number): Promise<void>;
+    /**
+     * Clear the conversation history for a user
+     * @param userId - the telegram user ID
+     * @param threadId - optional thread ID
+     */
+    clearHistory(userId: number, threadId?: number): Promise<void>;
+}
+/**
+ * Get the balance for a user, initializing it if it doesn't exist
+ * @param userId - the telegram user ID
+ * @param kv - the KV namespace
+ * @returns the user's balance
+ */
+export declare function getBalance(userId: number, kv: KVNamespace): Promise<number>;

package/dist/history_manager.js

@@ -0,0 +1,69 @@
+/** Class representing a manager for conversation history stored in KV */
+export class HistoryManager {
+    kv;
+    constructor(kv) {
+        this.kv = kv;
+    }
+    getKey(userId, threadId) {
+        return threadId ? `history:${String(userId)}:${String(threadId)}` : `history:${String(userId)}`;
+    }
+    /**
+     * Get the conversation history for a user
+     * @param userId - the telegram user ID
+     * @param threadId - optional thread ID
+     * @returns array of messages
+     */
+    async getHistory(userId, threadId) {
+        if (!this.kv) {
+            return [];
+        }
+        const history = await this.kv.get(this.getKey(userId, threadId), 'json');
+        return history ?? [];
+    }
+    /**
+     * Add a message and its response to the history
+     * @param userId - the telegram user ID
+     * @param prompt - the user message
+     * @param response - the bot response
+     * @param threadId - optional thread ID
+     */
+    async addMessage(userId, prompt, response, threadId) {
+        if (!this.kv) {
+            return;
+        }
+        const history = await this.getHistory(userId, threadId);
+        history.push({ role: 'user', content: prompt });
+        history.push({ role: 'assistant', content: response });
+        const trimmedHistory = history.slice(-20);
+        await this.kv.put(this.getKey(userId, threadId), JSON.stringify(trimmedHistory), {
+            expirationTtl: 86400
+        });
+    }
+    /**
+     * Clear the conversation history for a user
+     * @param userId - the telegram user ID
+     * @param threadId - optional thread ID
+     */
+    async clearHistory(userId, threadId) {
+        if (!this.kv) {
+            return;
+        }
+        await this.kv.delete(this.getKey(userId, threadId));
+    }
+}
+/**
+ * Get the balance for a user, initializing it if it doesn't exist
+ * @param userId - the telegram user ID
+ * @param kv - the KV namespace
+ * @returns the user's balance
+ */
+export async function getBalance(userId, kv) {
+    const balanceKey = `balance:${String(userId)}`;
+    const balance = await kv.get(balanceKey, 'json');
+    if (balance === null) {
+        const defaultBalance = 200;
+        await kv.put(balanceKey, JSON.stringify(defaultBalance));
+        return defaultBalance;
+    }
+    return balance;
+}

package/dist/main.d.ts

@@ -8,5 +8,6 @@ import TelegramGuestMessage from './types/TelegramGuestMessage.js';
 import PartialTelegramUpdate from './types/PartialTelegramUpdate.js';
 import TelegramInlineQueryType from './types/TelegramInlineQueryType.js';
 import { markdownToHtml, fetchTool } from './utils.js';
+import { HistoryManager, getBalance } from './history_manager.js';
 export default TelegramBot;
-export { TelegramBot, TelegramExecutionContext, Webhook, TelegramApi, TelegramApiBaseParams, SendMessageParams, SendMessageDraftParams, SendPhotoParams, SendVideoParams, SendVoiceParams, SendChatActionParams, AnswerCallbackParams, AnswerInlineParams, AnswerGuestParams, SendInvoiceParams, AnswerPreCheckoutParams, TelegramApiParams, TelegramCommand, TelegramFrom, TelegramChat, TelegramUser, TelegramMessageEntity, TelegramPhotoSize, TelegramMessage, TelegramVoice, TelegramGuestMessage, TelegramInputMessageContent, TelegramInlineQuery, TelegramUpdate, PartialTelegramUpdate, TelegramInlineQueryType, TelegramInlineQueryResult, TelegramInlineQueryResultPhoto, TelegramInlineQueryResultArticle, TelegramInlineQueryResultVideo, TelegramInlineQueryResultVoice, ChatPermissions, TelegramBusinessMessage, TelegramCallbackQuery, TelegramPreCheckoutQuery, TelegramDocument, TelegramSuccessfulPayment, markdownToHtml, fetchTool, };
+export { TelegramBot, TelegramExecutionContext, Webhook, TelegramApi, TelegramApiBaseParams, SendMessageParams, SendMessageDraftParams, SendPhotoParams, SendVideoParams, SendVoiceParams, SendChatActionParams, AnswerCallbackParams, AnswerInlineParams, AnswerGuestParams, SendInvoiceParams, AnswerPreCheckoutParams, TelegramApiParams, TelegramCommand, TelegramFrom, TelegramChat, TelegramUser, TelegramMessageEntity, TelegramPhotoSize, TelegramMessage, TelegramVoice, TelegramGuestMessage, TelegramInputMessageContent, TelegramInlineQuery, TelegramUpdate, PartialTelegramUpdate, TelegramInlineQueryType, TelegramInlineQueryResult, TelegramInlineQueryResultPhoto, TelegramInlineQueryResultArticle, TelegramInlineQueryResultVideo, TelegramInlineQueryResultVoice, ChatPermissions, TelegramBusinessMessage, TelegramCallbackQuery, TelegramPreCheckoutQuery, TelegramDocument, TelegramSuccessfulPayment, markdownToHtml, fetchTool, HistoryManager, getBalance, };

package/dist/main.js

@@ -3,5 +3,6 @@ import TelegramExecutionContext from './telegram_execution_context.js';
 import Webhook from './webhook.js';
 import TelegramApi from './telegram_api.js';
 import { markdownToHtml, fetchTool } from './utils.js';
+import { HistoryManager, getBalance } from './history_manager.js';
 export default TelegramBot;
-export { TelegramBot, TelegramExecutionContext, Webhook, TelegramApi, markdownToHtml, fetchTool, };
+export { TelegramBot, TelegramExecutionContext, Webhook, TelegramApi, markdownToHtml, fetchTool, HistoryManager, getBalance, };

package/dist/telegram_execution_context.d.ts

@@ -7,8 +7,6 @@ export default class TelegramExecutionContext {
     private static businessOwners;
     /** Cache for dead business connections */
     private static poisonedConnections;
-    /** Cache for self-response counts to prevent infinite loops */
-    private static selfResponseCount;
     /** an instance of the telegram bot */
     bot: TelegramBot;
     /** an instance of the telegram update */

package/dist/telegram_execution_context.js

@@ -5,8 +5,6 @@ export default class TelegramExecutionContext {
     static businessOwners = new Map();
     /** Cache for dead business connections */
     static poisonedConnections = new Set();
-    /** Cache for self-response counts to prevent infinite loops */
-    static selfResponseCount = new Map();
     /** an instance of the telegram bot */
     bot;
     /** an instance of the telegram update */
@@ -85,16 +83,6 @@ export default class TelegramExecutionContext {
      */
     async shouldProcess() {
         if (this.update_type !== 'business_message') {
-            if (this.userId === this.bot.botId) {
-                const chatId = this.getChatId();
-                const count = TelegramExecutionContext.selfResponseCount.get(chatId) || 0;
-                if (count < this.bot.ttl) {
-                    TelegramExecutionContext.selfResponseCount.set(chatId, count + 1);
-                    return true;
-                }
-                return false;
-            }
-            TelegramExecutionContext.selfResponseCount.delete(this.getChatId());
             return true;
         }
         const connectionId = this.update.business_message?.business_connection_id?.toString();
@@ -132,16 +120,6 @@ export default class TelegramExecutionContext {
                 console.warn('Failed to fetch business connection info:', e);
             }
         }
-        if (ownerId !== undefined && (this.getChatId() === ownerId.toString() || this.userId === ownerId)) {
-            const chatId = this.getChatId();
-            const count = TelegramExecutionContext.selfResponseCount.get(chatId) || 0;
-            if (count < this.bot.ttl) {
-                TelegramExecutionContext.selfResponseCount.set(chatId, count + 1);
-                return true;
-            }
-            return false;
-        }
-        TelegramExecutionContext.selfResponseCount.delete(this.getChatId());
         return true;
     }
     determineUpdateType() {

package/package.json

@@ -1,6 +1,6 @@
 {
 	"name": "@codebam/cf-workers-telegram-bot",
-	"version": "12.4.0",
+	"version": "12.6.0",
 	"description": "serverless telegram bot on cf workers",
 	"main": "./dist/main.js",
 	"module": "./dist/main.js",