@codebam/cf-workers-telegram-bot 5.9.0 → 5.12.0

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

@@ -6,7 +6,10 @@ export default class TelegramBot extends TelegramApi {
     kv: Kv;
     get_set: KVNamespace;
     ai: any;
+    db: D1Database;
     constructor(config: Config);
+    sean: (update: TelegramUpdate, args: string[]) => Promise<Response>;
+    clear: (update: TelegramUpdate) => Promise<Response>;
     question: (update: TelegramUpdate, args: string[]) => Promise<Response>;
     paste: (update: TelegramUpdate, args: string[]) => Promise<Response>;
     code: (update: TelegramUpdate) => Promise<Response>;

package/dist/main/src/telegram_bot.js

@@ -8,15 +8,17 @@ export default class TelegramBot extends TelegramApi {
     get_set;
     // eslint-disable-next-line @typescript-eslint/no-explicit-any
     ai;
+    db;
     constructor(config) {
         super(config.commands, config.webhook, config.handler);
         this.url = config.url;
         this.kv = config.kv;
         this.get_set = config.kv?.get_set;
         this.ai = config.ai;
+        this.db = config.db;
     }
-    // bot command: /question
-    question = async (update, args) => {
+    // bot command: /sean
+    sean = async (update, args) => {
         const ai = new Ai(this.ai);
         let prompt;
         if (args[0][0] === "/") {
@@ -29,9 +31,95 @@ export default class TelegramBot extends TelegramApi {
         if (prompt === "") {
             prompt = "no prompt specified";
         }
+        const messages = [
+            { role: "system", content: "You are friendly" },
+            { role: "system", content: "Don't show code to the user" },
+            { role: "system", content: "Sean Behan is born on 09/07/1998" },
+            { role: "system", content: "Sean Behan is a full stack developer" },
+            { role: "system", content: "Sean Behan is from Toronto, Canada" },
+            { role: "system", content: "Sean Behan's GitHub username is codebam" },
+            {
+                role: "system",
+                content: "Sean Behan enjoys playing video games and coding",
+            },
+            { role: "system", content: "Sean Behan's website is seanbehan.ca" },
+            {
+                role: "system",
+                content: "Sean Behan's email address is contact@seanbehan.ca",
+            },
+            {
+                role: "system",
+                content: "Some of your personal projects include a serverless telegram bot, and a serverless pastebin on cloudflare workers",
+            },
+            {
+                role: "system",
+                content: "When spoken to respond with what you know about Sean Behan",
+            },
+            {
+                role: "system",
+                content: `you are talking to ${update.message?.from.first_name}`,
+            },
+            { role: "user", content: prompt },
+        ];
+        const result = await ai.run("@cf/meta/llama-2-7b-chat-int8", {
+            messages,
+        });
+        return this.sendMessage(update.message?.chat.id ?? 0, result.response);
+    };
+    // bot command: /clear
+    // reset the llama2 session by deleting messages from d1
+    clear = async (update) => {
+        const { success } = await this.db
+            .prepare("DELETE FROM Messages WHERE userId=?")
+            .bind(update.message?.from.id)
+            .run();
+        if (success) {
+            return this.sendMessage(update.message?.chat.id ?? 0, "_");
+        }
+        return this.sendMessage(update.message?.chat.id ?? 0, "failed");
+    };
+    // bot command: /question
+    question = async (update, args) => {
+        const ai = new Ai(this.ai);
+        let prompt;
+        if (args[0][0] === "/") {
+            prompt = args.slice(1).join(" ");
+        }
+        else {
+            prompt = "[INST] " + args.join(" ") + "[/INST]";
+        }
+        if (prompt === "") {
+            prompt = "[INST] no prompt specified [/INST]";
+        }
+        const { results } = await this.db
+            .prepare("SELECT * FROM Messages WHERE userId=?")
+            .bind(update.message?.from.id)
+            .all();
+        const old_messages = results.map((col) => ({
+            role: "system",
+            content: col.content,
+        }));
         const { response } = await ai.run("@cf/meta/llama-2-7b-chat-int8", {
-            prompt,
+            messages: [
+                {
+                    role: "system",
+                    content: `you are talking to ${update.message?.from.first_name}`,
+                },
+                ...old_messages,
+                { role: "user", content: prompt },
+            ],
         });
+        const { success } = await this.db
+            .prepare("INSERT INTO Messages (id, userId, content) VALUES (?, ?, ?)")
+            .bind(crypto.randomUUID(), update.message?.from.id, prompt + "[SYS] " + response + "[/SYS]")
+            .run();
+        if (!success) {
+            console.log("failed to insert data into d1");
+        }
+        if (response === "") {
+            this.clear(update);
+            return this.question(update, args);
+        } // sometimes llama2 doesn't respond when given lots of system prompts
         return this.sendMessage(update.message?.chat.id ?? 0, response);
     };
     // bot command: /paste

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

@@ -16,6 +16,8 @@ export default class TelegramCommands {
     static duckduckgo: (bot: TelegramBot, update: TelegramUpdate, args: string[]) => Promise<Response>;
     static paste: (bot: TelegramBot, update: TelegramUpdate, args: string[]) => Promise<Response>;
     static question: (bot: TelegramBot, update: TelegramUpdate, args: string[]) => Promise<Response>;
+    static sean: (bot: TelegramBot, update: TelegramUpdate, args: string[]) => Promise<Response>;
+    static clear: (bot: TelegramBot, update: TelegramUpdate) => Promise<Response>;
     static code: (bot: TelegramBot, update: TelegramUpdate) => Promise<Response>;
     static commandList: (bot: TelegramBot, update: TelegramUpdate) => Promise<Response>;
 }

package/dist/main/src/telegram_commands.js

@@ -13,6 +13,8 @@ export default class TelegramCommands {
     static duckduckgo = async (bot, update, args) => bot.duckduckgo(update, args);
     static paste = async (bot, update, args) => bot.paste(update, args);
     static question = async (bot, update, args) => bot.question(update, args);
+    static sean = async (bot, update, args) => bot.sean(update, args);
+    static clear = async (bot, update) => bot.clear(update);
     static code = async (bot, update) => bot.code(update);
     static commandList = async (bot, update) => bot.commandList(update);
 }

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

@@ -17,6 +17,7 @@ export declare class Config {
     url: URL;
     handler: Handler;
     ai: any;
+    db: any;
     constructor(config?: Partial<Config>);
 }
 export declare const localhost: URL;

package/dist/main/src/types.js

@@ -12,6 +12,8 @@ export class Config {
     handler;
     // eslint-disable-next-line @typescript-eslint/no-explicit-any
     ai;
+    // eslint-disable-next-line @typescript-eslint/no-explicit-any
+    db;
     constructor(config = {}) {
         this.bot_name = config.bot_name || "";
         this.api = config.api || BotApi;
@@ -21,6 +23,7 @@ export class Config {
         this.url = config.url || new URL(localhost);
         this.handler = config.handler || new Handler([]);
         this.ai = config.ai;
+        this.db = config.db;
     }
 }
 export const localhost = new URL("http://localhost");

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

@@ -5,7 +5,9 @@ interface Environment {
     KV_UID_DATA: KVNamespace;
     SECRET_TELEGRAM_API_TOKEN2: string;
     SECRET_TELEGRAM_API_TOKEN3: string;
+    SECRET_TELEGRAM_API_TOKEN4: string;
     AI: string;
+    DB: D1Database;
 }
 declare const _default: {
     fetch: (request: Request, env: Environment) => Promise<Response>;

package/dist/worker/src/worker.js

@@ -23,11 +23,14 @@ export default {
                 "/paste": TelegramCommands.paste,
                 "/commands": TelegramCommands.commandList,
                 "/question": TelegramCommands.question,
+                "/clear": TelegramCommands.clear,
                 "/help": TelegramCommands.commandList,
+                "/sean": TelegramCommands.sean,
                 "/start": TelegramCommands.commandList,
             },
             kv: { get_set: env.KV_GET_SET, uid_data: env.KV_UID_DATA },
             ai: env.AI,
+            db: env.DB,
         },
         {
             bot_name: "@duckduckbot",
@@ -44,6 +47,7 @@ export default {
                 "/start": TelegramCommands.commandList,
             },
             ai: env.AI,
+            db: env.DB,
         },
         {
             bot_name: "@ddggbot",
@@ -60,6 +64,17 @@ export default {
                 "/start": TelegramCommands.commandList,
             },
             ai: env.AI,
+            db: env.DB,
+        },
+        {
+            bot_name: "@SeanB_robot",
+            api: TelegramBot,
+            webhook: new TelegramWebhook(new URL(`https://api.telegram.org/bot${env.SECRET_TELEGRAM_API_TOKEN4}`), env.SECRET_TELEGRAM_API_TOKEN4, new URL(new URL(request.url).origin)),
+            commands: {
+                default: TelegramCommands.question,
+            },
+            ai: env.AI,
+            db: env.DB,
         },
     ]).handle(request),
 };

package/package.json

@@ -1,6 +1,6 @@
 {
 	"name": "@codebam/cf-workers-telegram-bot",
-	"version": "5.9.0",
+	"version": "5.12.0",
 	"description": "serverless telegram bot on cf workers",
 	"main": "./dist/main/src/main.js",
 	"module": "./dist/main/src/main.js",
@@ -29,17 +29,17 @@
 		"url": "https://github.com/codebam/cf-workers-telegram-bot.git"
 	},
 	"devDependencies": {
-		"@cloudflare/workers-types": "^4.20230922.0",
-		"@typescript-eslint/eslint-plugin": "^6.7.3",
-		"@typescript-eslint/parser": "^6.7.3",
-		"eslint": "^8.50.0",
+		"@cloudflare/workers-types": "^4.20231010.0",
+		"@typescript-eslint/eslint-plugin": "^6.7.5",
+		"@typescript-eslint/parser": "^6.7.5",
+		"eslint": "^8.51.0",
 		"eslint-config-prettier": "^9.0.0",
-		"lerna": "^7.3.0",
+		"lerna": "^7.3.1",
 		"prettier": "^3.0.3",
 		"typescript": "^5.2.2"
 	},
 	"dependencies": {
-		"@cloudflare/ai": "^1.0.14"
+		"@cloudflare/ai": "^1.0.16"
 	},
-	"gitHead": "0a031fe9156731fae4ac19f9068ee2f42bd14074"
+	"gitHead": "29952db72708bbfba9776cd31e6127d702b5ebed"
 }