Merge pull request #40 from Dialogue-Bot/DIAL-34-Add-training-model

DIAL-34-Add-training-model

Author: khanh1902

server/model.nlp

@@ -56,6 +56,7 @@
     "moment": "^2.30.1",
     "morgan": "^1.10.0",
     "multer": "1.4.5-lts.1",
+    "node-nlp": "^4.27.0",
     "nodemailer": "^6.9.8",
     "pg": "^8.11.3",
     "react": "^18.2.0",
@@ -100,7 +101,7 @@
     "node-gyp": "^9.1.0",
     "nodemon": "^2.0.19",
     "pm2": "^5.2.0",
-    "prettier": "^2.8.8",
+    "prettier": "^2.7.1",
     "supertest": "^6.2.4",
     "ts-jest": "^28.0.7",
     "ts-node": "^10.9.1",

server/package-lock.json

@@ -32,4 +32,5 @@ export const {
   BOT_ENDPOINT,
   MAIL_USER,
   MAIL_PASS,
+  API_KEY,
 } = process.env

server/package.json

@@ -56,6 +56,10 @@ export const ENDPOINTS = {
     ADD_CHANNELS: '/flow/add-channels',
     SELECT_FLOWS_FOR_CHANNEL: '/flow/select',
   },
+  INTENT: {
+    INDEX: '/intent',
+    PREDICT: '/intent/predict',
+  },
 }
 
 export const LOCALE_KEY = 'lang'

server/src/config/index.ts

@@ -66,13 +66,14 @@ export class FlowController {
     res.status(StatusCodes.OK).json({ data })
   })
 
-  public getFlowsForSelect = catchAsync(async (req: RequestWithUser, res) => {
-    console.log('req.query.channelId', req.query.channelId)
-    const data = await this.flowService.getFlowsForSelect(
-      req.user?.id as string,
-    )
-    res.status(StatusCodes.OK).json({ data })
-  })
+  public selectFlowsForChannel = catchAsync(
+    async (req: RequestWithUser, res) => {
+      const data = await this.flowService.getFlowsForSelect(
+        req.user?.id as string,
+      )
+      res.status(StatusCodes.OK).json({ data })
+    },
+  )
 
   public getFlowByContactId = catchAsync(async (req: RequestWithUser, res) => {
     const data = await this.flowService.getFlowByContactId(req.params.contactId)

server/src/constants/index.ts

@@ -0,0 +1,69 @@
+import { LOCALE_KEY } from '@/constants'
+import { PagingDTO } from '@/dtos/paging.dto'
+import { LocaleService } from '@/i18n/ctx'
+import { RequestWithUser } from '@/interfaces/auth.interface'
+import { IntentService } from '@/services/intent.service'
+import { catchAsync } from '@/utils/catch-async'
+import { plainToClass } from 'class-transformer'
+import { StatusCodes } from 'http-status-codes'
+import Container from 'typedi'
+
+export class IntentController {
+  public intentService = Container.get(IntentService)
+  public localeService = Container.get<LocaleService>(LOCALE_KEY)
+
+  public createIntent = catchAsync(async (req: RequestWithUser, res) => {
+    const data = await this.intentService.create({
+      fields: req.body,
+      userId: req.user?.id as string,
+    })
+    res.status(StatusCodes.OK).json({
+      message: this.localeService.i18n().INTENT.CREATE_SUCCESS(),
+      data,
+    })
+  })
+
+  public predictIntent = catchAsync(async (req: RequestWithUser, res) => {
+    req.body.userId = req.user?.id
+    const data = await this.intentService.PredictTrainIntent(req.body)
+    res.status(StatusCodes.OK).json({
+      message: this.localeService.i18n().INTENT.PREDICT_SUCCESS(),
+      data,
+    })
+  })
+
+  public getById = catchAsync(async (req, res) => {
+    const data = await this.intentService.getById(req.params.id)
+    res.status(StatusCodes.OK).json({ data })
+  })
+
+  public updateIntent = catchAsync(async (req: RequestWithUser, res) => {
+    const data = await this.intentService.updateById({
+      id: req.params.id,
+      fields: req.body,
+      userId: req.user?.id as string,
+    })
+
+    res.status(StatusCodes.OK).json({
+      message: this.localeService.i18n().INTENT.UPDATE_SUCCESS(),
+      data,
+    })
+  })
+
+  public getAllIntents = catchAsync(async (req: RequestWithUser, res) => {
+    const paging = plainToClass(PagingDTO, req.query)
+    const data = await this.intentService.getAllIntents(
+      paging,
+      req.user?.id as string,
+    )
+    res.status(StatusCodes.OK).json({ data })
+  })
+
+  public deleteIntent = catchAsync(async (req, res) => {
+    const data = await this.intentService.deleteById(req.params.id)
+    res.status(StatusCodes.OK).json({
+      message: this.localeService.i18n().INTENT.DELETE_SUCCESS(),
+      data,
+    })
+  })
+}

server/src/controllers/flows.controller.ts

@@ -0,0 +1,19 @@
+CREATE TABLE IF NOT EXISTS "intents" (
+	"id" varchar(36) PRIMARY KEY NOT NULL,
+	"name" text NOT NULL,
+	"reference_id" text NOT NULL,
+	"intents" json DEFAULT '[]'::json NOT NULL,
+	"entities" json DEFAULT '[]'::json NOT NULL,
+	"user_id" varchar(36) NOT NULL,
+	"created_at" timestamp DEFAULT now(),
+	"updated_at" timestamp DEFAULT now(),
+	CONSTRAINT "intents_reference_id_unique" UNIQUE("reference_id")
+);
+--> statement-breakpoint
+ALTER TABLE "flows" ALTER COLUMN "variables" SET DEFAULT '[]'::json;--> statement-breakpoint
+ALTER TABLE "flows" DROP COLUMN IF EXISTS "diagrams";--> statement-breakpoint
+DO $$ BEGIN
+ ALTER TABLE "intents" ADD CONSTRAINT "intents_user_id_users_id_fk" FOREIGN KEY ("user_id") REFERENCES "users"("id") ON DELETE no action ON UPDATE no action;
+EXCEPTION
+ WHEN duplicate_object THEN null;
+END $$;

server/src/controllers/intent.controller.ts

@@ -0,0 +1,12 @@
+ALTER TABLE "intents" DROP CONSTRAINT "intents_reference_id_unique";--> statement-breakpoint
+ALTER TABLE "intents" DROP CONSTRAINT "intents_user_id_users_id_fk";
+--> statement-breakpoint
+ALTER TABLE "intents" ALTER COLUMN "intents" DROP NOT NULL;--> statement-breakpoint
+ALTER TABLE "intents" ADD COLUMN "deleted" boolean DEFAULT false;--> statement-breakpoint
+DO $$ BEGIN
+ ALTER TABLE "intents" ADD CONSTRAINT "intents_user_id_users_id_fk" FOREIGN KEY ("user_id") REFERENCES "users"("id") ON DELETE cascade ON UPDATE no action;
+EXCEPTION
+ WHEN duplicate_object THEN null;
+END $$;
+--> statement-breakpoint
+ALTER TABLE "intents" DROP COLUMN IF EXISTS "reference_id";

server/src/database/migrations/0004_loose_nightcrawler.sql

@@ -0,0 +1,25 @@
+CREATE TABLE IF NOT EXISTS "intents" (
+	"id" varchar(36) PRIMARY KEY NOT NULL,
+	"name" text NOT NULL,
+	"reference_id" text NOT NULL,
+	"intents" json DEFAULT '[]'::json,
+	"entities" json DEFAULT '[]'::json NOT NULL,
+	"user_id" varchar(36) NOT NULL,
+	"deleted" boolean DEFAULT false,
+	"created_at" timestamp DEFAULT now(),
+	"updated_at" timestamp DEFAULT now()
+);
+--> statement-breakpoint
+ALTER TABLE "flows" DROP CONSTRAINT "flows_user_id_users_id_fk";
+--> statement-breakpoint
+DO $$ BEGIN
+ ALTER TABLE "flows" ADD CONSTRAINT "flows_user_id_users_id_fk" FOREIGN KEY ("user_id") REFERENCES "users"("id") ON DELETE no action ON UPDATE no action;
+EXCEPTION
+ WHEN duplicate_object THEN null;
+END $$;
+--> statement-breakpoint
+DO $$ BEGIN
+ ALTER TABLE "intents" ADD CONSTRAINT "intents_user_id_users_id_fk" FOREIGN KEY ("user_id") REFERENCES "users"("id") ON DELETE cascade ON UPDATE no action;
+EXCEPTION
+ WHEN duplicate_object THEN null;
+END $$;