Add BotApiHistory model and create records in chatRequestAPIHandler

n4ze3m · Mar 5, 2024 · 4daf9ba · 4daf9ba
1 parent 5f6a0ed
commit 4daf9ba
Show file tree

Hide file tree

Showing 3 changed files with 39 additions and 0 deletions.
diff --git a/server/prisma/migrations/q_19/migration.sql b/server/prisma/migrations/q_19/migration.sql
@@ -0,0 +1,11 @@
+-- CreateTable
+CREATE TABLE "BotApiHistory" (
+    "id" SERIAL NOT NULL,
+    "api_key" TEXT NOT NULL,
+    "bot_id" TEXT,
+    "human" TEXT,
+    "bot" TEXT,
+    "createdAt" TIMESTAMP(3) DEFAULT CURRENT_TIMESTAMP,
+
+    CONSTRAINT "BotApiHistory_pkey" PRIMARY KEY ("id")
+);
diff --git a/server/prisma/schema.prisma b/server/prisma/schema.prisma
@@ -195,3 +195,12 @@ model DialoqbaseModels {
   deleted          Boolean  @default(false)
   createdAt        DateTime @default(now())
 }
+
+model BotApiHistory {
+  id         Int       @id @default(autoincrement())
+  api_key    String
+  bot_id     String?
+  human      String?
+  bot        String?
+  createdAt  DateTime? @default(now())
+}
diff --git a/server/src/handlers/bot/api.handler.ts b/server/src/handlers/bot/api.handler.ts
@@ -172,6 +172,16 @@ export const chatRequestAPIHandler = async (
       });
       const documents = await documentPromise;
 
+
+      await prisma.botApiHistory.create({
+        data: {
+          api_key: request.headers["x-api-key"],
+          bot_id: bot.id,
+          human: message,
+          bot: response,
+        }
+      })
+
       reply.sse({
         event: "result",
         id: "",
@@ -359,6 +369,15 @@ export const chatRequestAPIHandler = async (
 
       const documents = await documentPromise;
 
+      await prisma.botApiHistory.create({
+        data: {
+          api_key: request.headers["x-api-key"],
+          bot_id: bot.id,
+          human: message,
+          bot: botResponse,
+        }
+      })
+
       return {
         bot: {
           text: botResponse,