From 72f26b4370bd2b6206cbc5db37273920203aea8c Mon Sep 17 00:00:00 2001
From: ItzCrazyKns <95534749+ItzCrazyKns@users.noreply.github.com>
Date: Thu, 21 Aug 2025 17:47:49 +0530
Subject: [PATCH] feat(upload): save files uploaded after chat created

---
 src/app/api/chat/route.ts | 15 ++++++++++-----
 1 file changed, 10 insertions(+), 5 deletions(-)

diff --git a/src/app/api/chat/route.ts b/src/app/api/chat/route.ts
index 2d53b75..8798075 100644
--- a/src/app/api/chat/route.ts
+++ b/src/app/api/chat/route.ts
@@ -1,11 +1,7 @@
-import prompts from '@/lib/prompts';
-import MetaSearchAgent from '@/lib/search/metaSearchAgent';
 import crypto from 'crypto';
 import { AIMessage, BaseMessage, HumanMessage } from '@langchain/core/messages';
 import { EventEmitter } from 'stream';
 import {
-  chatModelProviders,
-  embeddingModelProviders,
   getAvailableChatModelProviders,
   getAvailableEmbeddingModelProviders,
 } from '@/lib/providers';
@@ -138,6 +134,8 @@ const handleHistorySave = async (
     where: eq(chats.id, message.chatId),
   });
 
+  const fileData = files.map(getFileDetails);
+
   if (!chat) {
     await db
       .insert(chats)
@@ -146,9 +144,16 @@ const handleHistorySave = async (
         title: message.content,
         createdAt: new Date().toString(),
         focusMode: focusMode,
-        files: files.map(getFileDetails),
+        files: fileData,
       })
       .execute();
+  } else if (JSON.stringify(chat.files ?? []) != JSON.stringify(fileData)) {
+    db
+      .update(chats)
+      .set({
+        files: files.map(getFileDetails),
+      })
+      .where(eq(chats.id, message.chatId));
   }
 
   const messageExists = await db.query.messages.findFirst({