Files
zhuiguang-ai/scripts/check-memories.mjs
T

30 lines
944 B
JavaScript

import { PrismaClient } from "@prisma/client";
import { PrismaMariaDb } from "@prisma/adapter-mariadb";
import "dotenv/config";
const base = process.env.DATABASE_URL.replace("mysql://", "mariadb://");
const connectionString = `${base}&connection_limit=2`;
const adapter = new PrismaMariaDb(connectionString);
const prisma = new PrismaClient({ adapter });
const count = await prisma.botMemory.count();
const groups = await prisma.botMemory.groupBy({
by: ["memoryType"],
_count: true,
});
console.log(`Total memories: ${count}`);
console.log(groups.map((r) => `${r.memoryType}: ${r._count}`).join(", "));
const recent = await prisma.botMemory.findMany({
orderBy: { createdAt: "desc" },
take: 5,
include: { bot: { select: { displayName: true } } },
});
for (const r of recent) {
const c = r.content;
console.log(
` ${r.bot.displayName} | ${r.memoryType} | ${c.title || c.topicTitle || ""}`
);
}
await prisma.$disconnect();