feat: Remove unused history field

The `history` field in the chat request body was no longer used. This change removes the unnecessary field, simplifying the API and improving efficiency. This change also introduces a mechanism to retrieve and utilize existing conversation history from the database based on the provided `history_id`. This allows for more seamless and interactive conversation experiences.
n4ze3m · Oct 15, 2024 · cd796bc · cd796bc
1 parent 7bff567
commit cd796bc
Show file tree

Hide file tree

Showing 3 changed files with 54 additions and 6 deletions.
diff --git a/app/ui/src/hooks/useMessage.tsx b/app/ui/src/hooks/useMessage.tsx
@@ -107,7 +107,6 @@ export const useMessage = () => {
       method: "POST",
       body: JSON.stringify({
         message,
-        history,
         history_id: historyId,
       }),
       headers: {

diff --git a/server/src/handlers/api/v1/bot/playground/chat.handler.ts b/server/src/handlers/api/v1/bot/playground/chat.handler.ts
@@ -17,7 +17,8 @@ export const chatRequestHandler = async (
   reply: FastifyReply
 ) => {
   const { id: bot_id } = request.params;
-  const { message, history, history_id } = request.body;
+  const { message, history_id } = request.body;
+  let history = [];
 
   try {
     const prisma = request.server.prisma;
@@ -33,6 +34,30 @@ export const chatRequestHandler = async (
       );
     }
 
+
+    if (history_id) {
+      const details = await prisma.botPlayground.findFirst({
+        where: {
+          id: history_id,
+          botId: bot_id,
+        },
+        include: {
+          BotPlaygroundMessage: {
+            orderBy: {
+              createdAt: "asc",
+            },
+          },
+        },
+      });
+
+      const botMessages = details?.BotPlaygroundMessage.map((message) => ({
+        type: message.type,
+        text: message.message,
+      }));
+
+      history = botMessages || [];
+    }
+
     const embeddingInfo = await getModelInfo({
       model: bot.embedding,
       prisma,
@@ -117,7 +142,8 @@ export const chatRequestStreamHandler = async (
   reply: FastifyReply
 ) => {
   const { id: bot_id } = request.params;
-  const { message, history, history_id } = request.body;
+  const { message, history_id } = request.body;
+  let history = [];
 
   try {
     const prisma = request.server.prisma;
@@ -133,6 +159,32 @@ export const chatRequestStreamHandler = async (
       );
     }
 
+
+    if (history_id) {
+      const details = await prisma.botPlayground.findFirst({
+        where: {
+          id: history_id,
+          botId: bot_id,
+        },
+        include: {
+          BotPlaygroundMessage: {
+            orderBy: {
+              createdAt: "asc",
+            },
+          },
+        },
+      });
+
+      const botMessages = details?.BotPlaygroundMessage.map((message) => ({
+        type: message.type,
+        text: message.message,
+      }));
+
+      history = botMessages || [];
+    }
+
+
+
     const embeddingInfo = await getModelInfo({
       model: bot.embedding,
       prisma,

diff --git a/server/src/schema/api/v1/bot/playground/index.ts b/server/src/schema/api/v1/bot/playground/index.ts
@@ -26,9 +26,6 @@ export const chatRequestSchema: FastifySchema = {
       message: {
         type: "string",
       },
-      history: {
-        type: "array",
-      },
       history_id: {
         type: "string",
       },