Skip to content

Commit 72f26b4

Browse files
committed
feat(upload): save files uploaded after chat created
1 parent f680188 commit 72f26b4

File tree

1 file changed

+10
-5
lines changed

1 file changed

+10
-5
lines changed

src/app/api/chat/route.ts

Lines changed: 10 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -1,11 +1,7 @@
1-
import prompts from '@/lib/prompts';
2-
import MetaSearchAgent from '@/lib/search/metaSearchAgent';
31
import crypto from 'crypto';
42
import { AIMessage, BaseMessage, HumanMessage } from '@langchain/core/messages';
53
import { EventEmitter } from 'stream';
64
import {
7-
chatModelProviders,
8-
embeddingModelProviders,
95
getAvailableChatModelProviders,
106
getAvailableEmbeddingModelProviders,
117
} from '@/lib/providers';
@@ -138,6 +134,8 @@ const handleHistorySave = async (
138134
where: eq(chats.id, message.chatId),
139135
});
140136

137+
const fileData = files.map(getFileDetails);
138+
141139
if (!chat) {
142140
await db
143141
.insert(chats)
@@ -146,9 +144,16 @@ const handleHistorySave = async (
146144
title: message.content,
147145
createdAt: new Date().toString(),
148146
focusMode: focusMode,
149-
files: files.map(getFileDetails),
147+
files: fileData,
150148
})
151149
.execute();
150+
} else if (JSON.stringify(chat.files ?? []) != JSON.stringify(fileData)) {
151+
db
152+
.update(chats)
153+
.set({
154+
files: files.map(getFileDetails),
155+
})
156+
.where(eq(chats.id, message.chatId));
152157
}
153158

154159
const messageExists = await db.query.messages.findFirst({

0 commit comments

Comments
 (0)