@codebam/cf-workers-telegram-bot 12.5.0 → 12.6.1

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_bot.js

@@ -129,6 +129,9 @@ export default class TelegramBot {
         // Then check if there's a command starting with / anywhere in the message
         const commandIndex = args.findIndex((arg) => arg.startsWith('/'));
         if (commandIndex !== -1) {
+            if (ctx.update_type === 'message' && ctx.update.message?.chat.type !== 'private') {
+                return this.defaultCommand;
+            }
             const fullCommand = args[commandIndex];
             const command = fullCommand.substring(1, fullCommand.lastIndexOf('@') > -1 ? fullCommand.lastIndexOf('@') : fullCommand.length);
             if (command in this.commands) {

package/package.json

@@ -1,6 +1,6 @@
 {
 	"name": "@codebam/cf-workers-telegram-bot",
-	"version": "12.5.0",
+	"version": "12.6.1",
 	"description": "serverless telegram bot on cf workers",
 	"main": "./dist/main.js",
 	"module": "./dist/main.js",
@@ -61,4 +61,4 @@
 			"./src/main.ts"
 		]
 	}
-}
+}