dmitry.galkin/chatbot-template
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Save to db on final ai state update (#429)

Browse source 
Co-authored-by: jb-dev1 <10735400+jb-dev1@users.noreply.github.com>
This commit is contained in:
Jeremy 2024-09-24 20:11:50 +05:30 • committed by GitHub
parent 2a6bf4b00b
commit 124efca9a1
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
1 changed files with 22 additions and 23 deletions
Download patch file Download diff file Expand all files Collapse all files

39
lib/chat/actions.tsx
View file

@ -517,34 +517,33 @@ export const AI = createAI<AIState, UIState>({
return return
} }
}, },
onSetAIState: async ({ state }) => { onSetAIState: async ({ state, done }) => {
'use server' 'use server'
if (!done) return
const session = await auth() const session = await auth()
if (!session || !session.user) return
if (session && session.user) { const { chatId, messages } = state
const { chatId, messages } = state
const createdAt = new Date() const createdAt = new Date()
const userId = session.user.id as string const userId = session.user.id as string
const path = `/chat/${chatId}` const path = `/chat/${chatId}`
const firstMessageContent = messages[0].content as string const firstMessageContent = messages[0].content as string
const title = firstMessageContent.substring(0, 100) const title = firstMessageContent.substring(0, 100)
const chat: Chat = { const chat: Chat = {
id: chatId, id: chatId,
title, title,
userId, userId,
createdAt, createdAt,
messages, messages,
path path
}
await saveChat(chat)
} else {
return
} }
await saveChat(chat)
} }
}) })