@codebam/cf-workers-telegram-bot 5.14.0 → 5.16.0

package/dist/main/src/handler.js

@@ -24,10 +24,10 @@ export default class Handler {
         POST: this.postResponse,
         default: () => new Promise(() => new Response()),
     };
-    getAccessKeys = async (configs) => Promise.all(configs.map((bot_config) => sha256(bot_config.webhook?.token ?? "").then((hash) => [
-        hash,
-        bot_config,
-    ]))).then((result) => Object.fromEntries(result));
+    getAccessKeys = async (configs) => Promise.all(configs.map((bot_config) => sha256(bot_config.webhook?.token ?? "").then((hash) => {
+        console.log(hash);
+        return [hash, bot_config];
+    }))).then((result) => Object.fromEntries(result));
     // handles the request
     handle = async (request) => this.getAccessKeys(this.configs).then((access_keys) => Object.keys(this.responses).includes(request.method)
         ? this.responses[request.method](request, ((key) => {

package/dist/main/src/telegram_api.js

@@ -13,14 +13,22 @@ export default class TelegramApi extends BotApi {
         message: this.messageUpdate,
         default: new Response(),
     };
-    update = async (update) => ((log({ update }) &&
-        update.message &&
-        this.updates.message(update)) ||
-        (update.inline_query &&
-            update.inline_query.query === "" &&
-            undefined) ||
-        this.updates.inline_query(update)) ??
-        this.updates.default;
+    update = async (update) => {
+        console.log({ update });
+        if (update) {
+            if (update.inline_query) {
+                if (update.inline_query.query !== "") {
+                    return this.updates.inline_query(update);
+                }
+            }
+            else {
+                if (update.message) {
+                    return this.updates.message(update);
+                }
+            }
+        }
+        return this.updates.default;
+    };
     // greet new users who join
     greetUsers = async (update) => update.message?.new_chat_members
         ? this.sendMessage(update.message.chat.id, `Welcome to ${update.message.chat.title}, ${update.message.from.username}`)

package/dist/main/src/telegram_bot.d.ts

@@ -9,7 +9,9 @@ export default class TelegramBot extends TelegramApi {
     db: D1Database;
     r2: R2Bucket;
     bot_name: string;
+    chat_model: string;
     constructor(config: Config);
+    translate: (update: TelegramUpdate, args: string[]) => Promise<Response>;
     clear: (update: TelegramUpdate) => Promise<Response>;
     image: (update: TelegramUpdate, args: string[]) => Promise<Response>;
     question: (update: TelegramUpdate, args: string[]) => Promise<Response>;

package/dist/main/src/telegram_bot.js

@@ -11,6 +11,7 @@ export default class TelegramBot extends TelegramApi {
     db;
     r2;
     bot_name;
+    chat_model;
     constructor(config) {
         super(config.commands, config.webhook, config.handler);
         this.url = config.url;
@@ -20,7 +21,35 @@ export default class TelegramBot extends TelegramApi {
         this.db = config.db;
         this.r2 = config.r2;
         this.bot_name = config.bot_name;
+        this.chat_model = config.chat_model;
     }
+    // bot command: /translate
+    translate = async (update, args) => {
+        if (this.ai === undefined) {
+            return new Response("ok");
+        }
+        const ai = new Ai(this.ai);
+        let _prompt;
+        if (args[0][0] === "/") {
+            _prompt = args.slice(1).join(" ");
+        }
+        else {
+            _prompt = args.join(" ");
+        }
+        if (_prompt === "") {
+            _prompt = "";
+        }
+        const langs = ["french", "arabic", "german", "spanish", "russian"];
+        const inline_articles = await Promise.all(langs.map(async (lang) => {
+            const response = await ai.run("@cf/meta/m2m100-1.2b", {
+                text: _prompt,
+                source_lang: lang,
+                target_lang: "english",
+            });
+            return new TelegramInlineQueryResultArticle(response.translated_text, `${lang}: ${response.translated_text}`);
+        }));
+        return this.answerInlineQuery(update.inline_query?.id ?? 0, inline_articles);
+    };
     // bot command: /clear
     // reset the llama2 session by deleting messages from d1
     clear = async (update) => {
@@ -110,9 +139,13 @@ export default class TelegramBot extends TelegramApi {
                 return acc + cur.content + "\n";
             }, "") +
             "</s>";
+        const p = system_prompt + "[INST]" + _prompt + "[/INST]";
+        const prompt = p.slice(p.length - 4096, p.length);
         const response = await ai
-            .run("@hf/thebloke/llama-2-13b-chat-awq", {
-            prompt: system_prompt + "[INST]" + _prompt + "[/INST]",
+            // @ts-expect-error ModelName doesn't need to be verified at build time
+            .run(this.chat_model, {
+            prompt,
+            max_tokens: 596,
         })
             .then(({ response }) => response
             .replace(/(\[|)(\/|)INST(S|)(s|)(\]|)/, "")
@@ -155,6 +188,26 @@ export default class TelegramBot extends TelegramApi {
         if (_prompt === "") {
             _prompt = "";
         }
+        const results = await (async () => {
+            if (this.db) {
+                const { results } = await this.db
+                    .prepare("SELECT * FROM Messages WHERE userId=?")
+                    .bind(update.inline_query
+                    ? update.inline_query.from.id
+                    : update.message?.from.id)
+                    .all();
+                return results;
+            }
+        })();
+        const old_messages = (() => {
+            if (results) {
+                return results.map((col) => ({
+                    role: "system",
+                    content: col.content,
+                }));
+            }
+            return [];
+        })();
         const system_prompt = "<s>" +
             [
                 `Your name is ${this.bot_name}.`,
@@ -167,14 +220,33 @@ export default class TelegramBot extends TelegramApi {
             ].reduce((acc, cur) => {
                 return acc + cur + "\n";
             }) +
+            old_messages.reduce((acc, cur) => {
+                return acc + cur.content + "\n";
+            }, "") +
             "</s>";
+        const p = system_prompt + "[INST]" + _prompt + "[/INST]";
+        const prompt = p.slice(p.length - 4096, p.length);
         const response = await ai
-            .run("@hf/thebloke/llama-2-13b-chat-awq", {
-            prompt: system_prompt + "[INST]" + _prompt + "[/INST]",
+            // @ts-expect-error ModelName doesn't need to be verified at build time
+            .run(this.chat_model, {
+            prompt,
+            max_tokens: 596,
         })
             .then(({ response }) => response
             .replace(/(\[|)(\/|)INST(S|)(s|)(\]|)/, "")
-            .replace(/<<(\/|)SYS>>/, ""));
+            .replace(/<<(\/|)SYS>>/, "")
+            .replace(/[OUT]/, ""));
+        if (this.db) {
+            const { success } = await this.db
+                .prepare("INSERT INTO Messages (id, userId, content) VALUES (?, ?, ?)")
+                .bind(crypto.randomUUID(), update.inline_query
+                ? update.inline_query.from.id
+                : update.message?.from.id, "[INST] " + _prompt + " [/INST]" + "\n" + response)
+                .run();
+            if (!success) {
+                console.log("failed to insert data into d1");
+            }
+        }
         if (update.inline_query) {
             return this.answerInlineQuery(update.inline_query.id, [
                 new TelegramInlineQueryResultArticle(response),
@@ -206,7 +278,7 @@ export default class TelegramBot extends TelegramApi {
                 ? [
                     new TelegramInlineQueryResultArticle(`${instant_answer_url}\n\n<a href="${addSearchParams(new URL(duckduckgo_url), {
                         q: args
-                            .slice(1)
+                            .slice(2)
                             .join(" ")
                             .replace(/^!\w* /, ""),
                     }).href}">Results From DuckDuckGo</a>`, instant_answer_url, "HTML", thumb_url),

package/dist/main/src/telegram_commands.d.ts

@@ -15,4 +15,5 @@ export default class TelegramCommands {
     static code: TelegramCommand;
     static commandList: TelegramCommand;
     static image: TelegramCommand;
+    static translate: TelegramCommand;
 }

package/dist/main/src/telegram_commands.js

@@ -14,4 +14,5 @@ export default class TelegramCommands {
     static code = async (bot, update) => bot.code(update);
     static commandList = async (bot, update) => bot.commandList(update);
     static image = async (bot, update, args) => bot.image(update, args);
+    static translate = async (bot, update, args) => bot.translate(update, args);
 }

package/dist/main/src/types.d.ts

@@ -19,6 +19,7 @@ export declare class Config {
     ai: any;
     db: any;
     r2: any;
+    chat_model: string;
     constructor(config?: Partial<Config>);
 }
 export declare const localhost: URL;

package/dist/main/src/types.js

@@ -16,6 +16,7 @@ export class Config {
     db;
     // eslint-disable-next-line @typescript-eslint/no-explicit-any
     r2;
+    chat_model;
     constructor(config = {}) {
         this.bot_name = config.bot_name || "";
         this.api = config.api || BotApi;
@@ -27,6 +28,7 @@ export class Config {
         this.ai = config.ai;
         this.db = config.db;
         this.r2 = config.r2;
+        this.chat_model = config.chat_model;
     }
 }
 export const localhost = new URL("http://localhost");

package/dist/worker/src/worker.d.ts

@@ -6,9 +6,11 @@ interface Environment {
     SECRET_TELEGRAM_API_TOKEN2: string;
     SECRET_TELEGRAM_API_TOKEN3: string;
     SECRET_TELEGRAM_API_TOKEN4: string;
+    SECRET_TELEGRAM_API_TOKEN5: string;
     AI: string;
     DB: D1Database;
     R2: R2Bucket;
+    CHAT_MODEL: string;
 }
 declare const _default: {
     fetch: (request: Request, env: Environment) => Promise<Response>;

package/dist/worker/src/worker.js

@@ -23,12 +23,13 @@ export default {
                 "/clear": TelegramCommands.clear,
                 "/help": TelegramCommands.commandList,
                 "/image": TelegramCommands.image,
-                "/start": TelegramCommands.commandList,
+                "/start": TelegramCommands.question,
             },
             kv: { get_set: env.KV_GET_SET, uid_data: env.KV_UID_DATA },
             ai: env.AI,
             db: env.DB,
             r2: env.R2,
+            chat_model: env.CHAT_MODEL,
         },
         {
             bot_name: "@duckduckbot",
@@ -76,5 +77,16 @@ export default {
             ai: env.AI,
             db: env.DB,
         },
+        {
+            bot_name: "@TranslatePartyBot",
+            api: TelegramBot,
+            webhook: new TelegramWebhook(new URL(`https://api.telegram.org/bot${env.SECRET_TELEGRAM_API_TOKEN5}`), env.SECRET_TELEGRAM_API_TOKEN5, new URL(new URL(request.url).origin)),
+            commands: {
+                default: TelegramCommands.translate,
+                inline: TelegramCommands.translate,
+                "/start": TelegramCommands.commandList,
+            },
+            ai: env.AI,
+        },
     ]).handle(request),
 };

package/package.json

@@ -1,45 +1,48 @@
 {
-	"name": "@codebam/cf-workers-telegram-bot",
-	"version": "5.14.0",
-	"description": "serverless telegram bot on cf workers",
-	"main": "./dist/main/src/main.js",
-	"module": "./dist/main/src/main.js",
-	"files": [
-		"dist",
-		"LICENSE",
-		"LICENSE_MIT",
-		"README.md"
-	],
-	"keywords": [
-		"cloudflare",
-		"telegram",
-		"worker",
-		"webhook"
-	],
-	"type": "module",
-	"private": false,
-	"scripts": {
-		"build": "tsc --project tsconfig.json",
-		"lint": "eslint src"
-	},
-	"author": "codebam",
-	"license": "Apache-2.0",
-	"repository": {
-		"type": "git",
-		"url": "https://github.com/codebam/cf-workers-telegram-bot.git"
-	},
-	"devDependencies": {
-		"@cloudflare/workers-types": "^4.20231121.0",
-		"@typescript-eslint/eslint-plugin": "^6.12.0",
-		"@typescript-eslint/parser": "^6.12.0",
-		"eslint": "^8.54.0",
-		"eslint-config-prettier": "^9.0.0",
-		"lerna": "^8.0.0",
-		"prettier": "^3.1.0",
-		"typescript": "^5.3.2"
-	},
-	"dependencies": {
-		"@cloudflare/ai": "^1.0.33"
-	},
-	"gitHead": "b61227dbb262c62ceb5c2401b71a353caa3e6346"
+  "name": "@codebam/cf-workers-telegram-bot",
+  "version": "5.16.0",
+  "description": "serverless telegram bot on cf workers",
+  "main": "./dist/main/src/main.js",
+  "module": "./dist/main/src/main.js",
+  "files": [
+    "dist",
+    "LICENSE",
+    "LICENSE_MIT",
+    "README.md"
+  ],
+  "keywords": [
+    "cloudflare",
+    "telegram",
+    "worker",
+    "webhook"
+  ],
+  "type": "module",
+  "private": false,
+  "scripts": {
+    "build": "tsc --project tsconfig.json",
+    "lint": "eslint src"
+  },
+  "author": "codebam",
+  "license": "Apache-2.0",
+  "repository": {
+    "type": "git",
+    "url": "https://github.com/codebam/cf-workers-telegram-bot.git"
+  },
+  "devDependencies": {
+    "@cloudflare/workers-types": "^4.20240502.0",
+    "@eslint/js": "^9.1.1",
+    "@typescript-eslint/eslint-plugin": "^7.8.0",
+    "@typescript-eslint/parser": "^7.8.0",
+    "eslint": "^8.56.0",
+    "eslint-config-prettier": "^9.1.0",
+    "globals": "^15.1.0",
+    "lerna": "^8.1.2",
+    "prettier": "^3.2.5",
+    "typescript": "^5.4.5",
+    "typescript-eslint": "^7.8.0"
+  },
+  "dependencies": {
+    "@cloudflare/ai": "1.0.53"
+  },
+  "gitHead": "fb3449a1162c24d725a1cde10ea9a5b3fde0fba0"
 }