mirror of
https://github.com/supermemoryai/supermemory.git
synced 2026-05-05 23:40:57 +00:00
made and documented the telegram bot (HYPE)
This commit is contained in:
parent
fff8b7abd7
commit
51dd5ec9dd
7 changed files with 307 additions and 7 deletions
|
|
@ -1,3 +1,7 @@
|
|||
import { db } from "@/server/db";
|
||||
import { storedContent, users } from "@/server/db/schema";
|
||||
import { cipher, decipher } from "@/server/encrypt";
|
||||
import { eq } from "drizzle-orm";
|
||||
import { Bot, webhookCallback } from "grammy";
|
||||
import { User } from "grammy/types";
|
||||
|
||||
|
|
@ -12,17 +16,96 @@ const token = process.env.TELEGRAM_BOT_TOKEN;
|
|||
|
||||
const bot = new Bot(token);
|
||||
|
||||
const getUserByTelegramId = async (telegramId: string) => {
|
||||
return await db.query.users
|
||||
.findFirst({
|
||||
where: eq(users.telegramId, telegramId),
|
||||
})
|
||||
.execute();
|
||||
};
|
||||
|
||||
bot.command("start", async (ctx) => {
|
||||
const user: User = (await ctx.getAuthor()).user;
|
||||
const cipherd = cipher(user.id.toString());
|
||||
await ctx.reply(
|
||||
`Welcome to Supermemory bot, ${user.first_name}. I am here to help you remember things better.`,
|
||||
`Welcome to Supermemory bot. I am here to help you remember things better. Click here to create and link your accont: http://localhost:3000/signin?telegramUser=${cipherd}`,
|
||||
);
|
||||
});
|
||||
|
||||
bot.on("message", async (ctx) => {
|
||||
await ctx.reply(
|
||||
"Hi there! This is Supermemory bot. I am here to help you remember things better.",
|
||||
const user: User = (await ctx.getAuthor()).user;
|
||||
const cipherd = cipher(user.id.toString());
|
||||
|
||||
const dbUser = await getUserByTelegramId(user.id.toString());
|
||||
|
||||
if (!dbUser) {
|
||||
await ctx.reply(
|
||||
`Welcome to Supermemory bot. I am here to help you remember things better. Click here to create and link your accont: http://localhost:3000/signin?telegramUser=${cipherd}`,
|
||||
);
|
||||
|
||||
return;
|
||||
}
|
||||
|
||||
const message = await ctx.reply("I'm thinking...");
|
||||
|
||||
const response = await fetch(
|
||||
`${process.env.BACKEND_BASE_URL}/api/autoChatOrAdd?query=${ctx.message.text}&user=${dbUser.id}`,
|
||||
{
|
||||
method: "POST",
|
||||
headers: {
|
||||
Authorization: "Bearer " + process.env.BACKEND_SECURITY_KEY,
|
||||
"Content-Type": "application/json",
|
||||
},
|
||||
body: JSON.stringify({
|
||||
// TODO: we can use the conversations API to get the last 5 messages
|
||||
// get chatHistory from this conversation.
|
||||
// Basically the last 5 messages between the user and the assistant.
|
||||
// In ths form of [{role: 'user' | 'assistant', content: string}]
|
||||
// https://grammy.dev/plugins/conversations
|
||||
chatHistory: [],
|
||||
}),
|
||||
},
|
||||
);
|
||||
|
||||
if (response.status !== 200) {
|
||||
console.log("Failed to get response from backend");
|
||||
console.log(response.status);
|
||||
console.log(await response.text());
|
||||
await ctx.reply(
|
||||
"Sorry, I am not able to process your request at the moment.",
|
||||
);
|
||||
return;
|
||||
}
|
||||
|
||||
const data = (await response.json()) as {
|
||||
status: string;
|
||||
response: string;
|
||||
contentAdded: {
|
||||
type: string;
|
||||
content: string;
|
||||
url: string;
|
||||
};
|
||||
};
|
||||
|
||||
// TODO: we might want to enrich this data with more information
|
||||
if (data.contentAdded) {
|
||||
await db
|
||||
.insert(storedContent)
|
||||
.values({
|
||||
content: data.contentAdded.content,
|
||||
title: `${data.contentAdded.content.slice(0, 30)}... (Added from chatbot)`,
|
||||
description: "",
|
||||
url: data.contentAdded.url,
|
||||
baseUrl: data.contentAdded.url,
|
||||
image: "",
|
||||
savedAt: new Date(),
|
||||
userId: dbUser.id,
|
||||
type: data.contentAdded.type,
|
||||
})
|
||||
.returning({ id: storedContent.id });
|
||||
}
|
||||
|
||||
await ctx.api.editMessageText(ctx.chat.id, message.message_id, data.response);
|
||||
});
|
||||
|
||||
export const POST = webhookCallback(bot, "std/http");
|
||||
|
|
|
|||
Loading…
Add table
Add a link
Reference in a new issue