try to reply to business messages

Author: codebam

packages/main/src/types/PartialTelegramUpdate.ts

@@ -1,3 +1,4 @@
+import TelegramBusinessMessage from './TelegramBusinessMessage.js';
 import TelegramInlineQuery from './TelegramInlineQuery.js';
 import TelegramMessage from './TelegramMessage.js';
 
@@ -8,5 +9,6 @@ interface PartialTelegramUpdate {
 	channel_post?: TelegramMessage;
 	edited_channel_post?: TelegramMessage;
 	inline_query?: TelegramInlineQuery;
+	business_message?: TelegramBusinessMessage;
 }
 export default PartialTelegramUpdate;

packages/main/src/types/TelegramBusinessMessage.ts

@@ -0,0 +1,63 @@
+import TelegramChat from './TelegramChat.js';
+import { TelegramDocument } from './TelegramDocument.js';
+import TelegramFrom from './TelegramFrom.js';
+import TelegramMessageEntity from './TelegramMessageEntity.js';
+import TelegramPhotoSize from './TelegramPhotoSize.js';
+import TelegramUser from './TelegramUser.js';
+import TelegramMessage from './TelegramMessage.js';
+
+interface TelegramBusinessMessage {
+	message_id: number;
+	business_connection_id: string;
+	from: TelegramFrom;
+	sender_chat?: TelegramChat;
+	date: number;
+	chat: TelegramChat;
+	forward_from?: TelegramUser;
+	forward_from_chat?: TelegramChat;
+	forward_from_message_id?: number;
+	forward_signature?: string;
+	forward_sender_name?: string;
+	forward_date?: number;
+	is_automatic_forward?: boolean;
+	reply_to_message?: TelegramMessage;
+	via_bot?: TelegramUser;
+	edit_date?: number;
+	has_protected_content?: boolean;
+	media_group_id?: string;
+	author_signature?: string;
+	text?: string;
+	entities?: TelegramMessageEntity[];
+	// animation?: TelegramAnimation;
+	// audio?: TelegramAudio;
+	document?: TelegramDocument;
+	photo?: TelegramPhotoSize[];
+	// sticker?: TelegramSticker;
+	// video?: TelegramVideo;
+	// video_note?: TelegramVideoNote;
+	// voice?: TelegramVoice;
+	caption?: string;
+	caption_entities?: TelegramMessageEntity[];
+	// contact?: TelegramContact;
+	// dice?: TelegramDice;
+	// poll?: TelegramPoll;
+	// venue?: TelegramVenue;
+	// location?: TelegramLocation;
+	new_chat_members?: TelegramUser[];
+	new_chat_member?: TelegramUser;
+	left_chat_member?: TelegramUser;
+	new_chat_title?: string;
+	// new_chat_photo?: TelegramPhotoSize[];
+	delete_chat_photo?: boolean;
+	group_chat_created?: boolean;
+	supergroup_chat_created?: boolean;
+	channel_chat_created?: boolean;
+	// message_auto_delete_timer_changed?: TelegramAutoDeleteTimerChanged;
+	migrate_to_chat_id?: number;
+	migrate_from_chat_id?: number;
+	pinned_message?: TelegramMessage;
+	// invoice?: TelegramInvoice;
+	// successful_payment?: TelegramSuccessfulPayment;
+	connected_website?: string;
+}
+export default TelegramBusinessMessage;

packages/main/src/types/TelegramUpdate.ts

@@ -2,6 +2,7 @@ import TelegramInlineQuery from './TelegramInlineQuery.js';
 import TelegramMessage from './TelegramMessage.js';
 import PartialTelegramUpdate from './PartialTelegramUpdate.js';
 import TelegramCallbackQuery from './TelegramCallbackQuery.js';
+import TelegramBusinessMessage from './TelegramBusinessMessage.js';
 
 export default class TelegramUpdate {
 	update_id: number;
@@ -12,6 +13,7 @@ export default class TelegramUpdate {
 	inline_query?: TelegramInlineQuery;
 	// chosen_inline_result?: TelegramChosenInlineResult;
 	callback_query?: TelegramCallbackQuery;
+	business_message?: TelegramBusinessMessage;
 	// shipping_query?: TelegramShippingQuery;
 	// pre_checkout_query?: TelegramPreCheckoutQuery;
 	// poll?: TelegramPoll;
@@ -26,6 +28,7 @@ export default class TelegramUpdate {
 		this.channel_post = update.channel_post;
 		this.edited_channel_post = update.edited_channel_post;
 		this.inline_query = update.inline_query;
+		this.business_message = update.business_message;
 		// chosen_inline_result = update.chosen_inline_result;
 		// callback_query = update.callback_query;
 		// shipping_query = update.shipping_query;

packages/worker/src/worker.ts

@@ -265,7 +265,43 @@ export default {
 							}
 							break;
 						}
-
+						case 'business_message': {
+							const prompt = bot.update.business_message?.text?.toString() ?? '';
+							const { results } = await env.DB.prepare('SELECT * FROM Messages WHERE userId=?')
+								.bind(bot.update.inline_query ? bot.update.inline_query.from.id : bot.update.message?.from.id)
+								.all();
+							const message_history = results.map((col) => ({ role: 'system', content: col.content as string }));
+							const messages = [
+								{ role: 'system', content: 'You are a friendly assistant named TuxRobot. Use lots of emojis in your responses.' },
+								...message_history,
+								{
+									role: 'user',
+									content: prompt,
+								},
+							];
+							let response: AiTextGenerationOutput;
+							try {
+								// @ts-expect-error broken bindings
+								response = await env.AI.run('@cf/meta/llama-3.1-8b-instruct', { messages, max_tokens: 150 });
+							} catch (e) {
+								console.log(e);
+								await bot.reply(`Error: ${e as string}`);
+								return new Response('ok');
+							}
+							if ('response' in response) {
+								if (response.response) {
+									await bot.reply(await markdown_to_html(response.response ?? ''), 'HTML');
+									await env.DB.prepare('INSERT INTO Messages (id, userId, content) VALUES (?, ?, ?)')
+										.bind(
+											crypto.randomUUID(),
+											bot.update.inline_query ? bot.update.inline_query.from.id : bot.update.message?.from.id,
+											`'[INST] ${prompt} [/INST] \n ${response.response}`,
+										)
+										.run();
+								}
+							}
+							break;
+						}
 						default:
 							break;
 					}