ai module

2026-07-23 10:52:29 +03:30
parent 549da39473
commit 27c45974f4
15 changed files with 538 additions and 10 deletions
@@ -0,0 +1,59 @@
+import { Migration } from '@mikro-orm/migrations';
+
+export class Migration20260723090000_addAiChats extends Migration {
+  override async up(): Promise<void> {
+    this.addSql(`
+      create table "ai_chats" (
+        "id" char(26) not null,
+        "created_at" timestamptz not null default now(),
+        "updated_at" timestamptz not null default now(),
+        "deleted_at" timestamptz null,
+        "restaurant_id" char(26) not null,
+        "user_id" char(26) null,
+        "question" text not null,
+        "consumed_tokens" int not null,
+        "cost" numeric(10,0) not null,
+        constraint "ai_chats_pkey" primary key ("id")
+      );
+    `);
+    this.addSql(`create index "ai_chats_created_at_index" on "ai_chats" ("created_at");`);
+    this.addSql(`create index "ai_chats_deleted_at_index" on "ai_chats" ("deleted_at");`);
+    this.addSql(`create index "ai_chats_restaurant_id_index" on "ai_chats" ("restaurant_id");`);
+    this.addSql(`
+      alter table "ai_chats"
+      add constraint "ai_chats_restaurant_id_foreign"
+      foreign key ("restaurant_id") references "restaurants" ("id") on update cascade;
+    `);
+    this.addSql(`
+      alter table "ai_chats"
+      add constraint "ai_chats_user_id_foreign"
+      foreign key ("user_id") references "users" ("id") on update cascade on delete set null;
+    `);
+
+    this.addSql(`
+      alter table "restaurant_credit_transactions"
+      drop constraint if exists "restaurant_credit_transactions_reason_check";
+    `);
+    this.addSql(`
+      alter table "restaurant_credit_transactions"
+      add constraint "restaurant_credit_transactions_reason_check"
+      check ("reason" in ('sms_send', 'deposit', 'adjustment', 'ai_chat'));
+    `);
+  }
+
+  override async down(): Promise<void> {
+    this.addSql(`
+      alter table "restaurant_credit_transactions"
+      drop constraint if exists "restaurant_credit_transactions_reason_check";
+    `);
+    this.addSql(`
+      alter table "restaurant_credit_transactions"
+      add constraint "restaurant_credit_transactions_reason_check"
+      check ("reason" in ('sms_send', 'deposit', 'adjustment'));
+    `);
+
+    this.addSql(`alter table "ai_chats" drop constraint if exists "ai_chats_user_id_foreign";`);
+    this.addSql(`alter table "ai_chats" drop constraint if exists "ai_chats_restaurant_id_foreign";`);
+    this.addSql(`drop table if exists "ai_chats" cascade;`);
+  }
+}