feat: daily report by ai (#87)

Author: hmbanan666

apps/web-app/.env.example

@@ -18,12 +18,14 @@ NUXT_PUBLIC_MEDIA_URL=
 NUXT_AI_API_KEY=
 NUXT_AI_BASE_URL=
 NUXT_AI_MODEL=
+NUXT_AI_MODEL_PRO=
 NUXT_AI_SERVICE_TOKEN=
 
 # Telegram
 NUXT_TELEGRAM_ADMIN_ID=
 NUXT_TELEGRAM_WASABI_BOT_ID=
 NUXT_TELEGRAM_ATRIUM_BOT_ID=
+NUXT_TELEGRAM_TEAM_GROUP_ID=
 
 # App version
 VERSION=

apps/web-app/nuxt.config.ts

@@ -14,6 +14,7 @@ export default defineNuxtConfig({
     },
     ai: {
       model: '',
+      modelPro: '',
       baseUrl: '',
       apiKey: '',
       serviceToken: '',
@@ -22,6 +23,7 @@ export default defineNuxtConfig({
       wasabiBotId: '',
       atriumBotId: '',
       adminId: '',
+      teamGroupId: '',
     },
     public: {
       mediaUrl: '',
@@ -45,6 +47,7 @@ export default defineNuxtConfig({
       '* * * * *': ['task:auto-create'], // Every minute
       '0 * * * *': ['kitchen:revenue-update'], // Every hour
       '0 0 * * *': ['kitchen:rating-update'], // Every day
+      '0 17 * * 1-5': ['ai:daily-report'], // Mon-Fri 17:00
     },
   },
   tiptap: {

apps/web-app/server/tasks/ai/daily-report.ts

@@ -0,0 +1,53 @@
+import { repository } from '@roll-stack/database'
+import OpenAI from 'openai'
+import { useAtriumBot } from '~~/server/services/telegram/atrium-bot'
+
+export default defineTask({
+  meta: {
+    name: 'ai:daily-report',
+    description: 'Prepare and post daily report to Telegram group',
+  },
+  async run() {
+    try {
+      const { ai, telegram } = useRuntimeConfig()
+
+      const tasks = await repository.task.listCompletedToday()
+      const preparedTasks = tasks.map((task) => ({
+        completedAt: task.completedAt,
+        name: task.name,
+        description: task.description,
+        report: task.report,
+      }))
+
+      const client = new OpenAI({
+        apiKey: ai.apiKey,
+        baseURL: ai.baseUrl,
+      })
+
+      const response = await client.chat.completions.create({
+        model: ai.modelPro,
+        messages: [
+          {
+            role: 'system',
+            content: 'Ты работаешь в компании "Суши Love", которая является сетью доставки суши. В 3-5 абзацах расскажи что сегодня было сделано, лучшие моменты. Отвечай на русском. Не бойся использовать emoji',
+          },
+          {
+            role: 'user',
+            content: JSON.stringify(preparedTasks),
+          },
+        ],
+      })
+
+      const finalMessage = response.choices[0].message.content
+      if (!finalMessage) {
+        return { result: true }
+      }
+
+      await useAtriumBot().api.sendMessage(telegram.teamGroupId, finalMessage)
+    } catch (error) {
+      errorResolver(error)
+    }
+
+    return { result: true }
+  },
+})

packages/database/src/repository/task.ts

@@ -67,6 +67,13 @@ export class Task {
     })
   }
 
+  static async listCompletedToday() {
+    return useDatabase().query.tasks.findMany({
+      where: (tasks, { gte }) => gte(tasks.completedAt, sql`now() - interval '12 hour'`),
+      orderBy: (tasks, { desc }) => desc(tasks.updatedAt),
+    })
+  }
+
   static async autoCreatorsList() {
     return useDatabase().query.taskAutoCreators.findMany()
   }