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

chore: cleanup context window ui, adjust for reasoning + added db migration

Browse source
This commit is contained in:
Nicklas Scharpff 2025-09-08 23:47:13 +02:00 • committed by GitHub
parent 431eb919b5
commit 02dde68ea3
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
13 changed files with 875 additions and 144 deletions
Download patch file Download diff file Expand all files Collapse all files

12
app/(chat)/api/chat/route.ts
View file

@ -18,6 +18,7 @@ import {
saveChat,
saveMessages,
} from '@/lib/db/queries';
import { updateChatLastContextById } from '@/lib/db/queries';
import { convertToUIMessages, generateUUID } from '@/lib/utils';
import { generateTitleFromUserMessage } from '../../actions';
import { createDocument } from '@/lib/ai/tools/create-document';
@ -208,6 +209,17 @@ export async function POST(request: Request) {
chatId: id,
})),
});
if (finalUsage) {
try {
await updateChatLastContextById({
chatId: id,
context: finalUsage,
});
} catch (err) {
console.warn('Unable to persist last usage for chat', id, err);
}
}
},
onError: () => {
return 'Oops, an error occurred!';

3
app/(chat)/chat/[id]/page.tsx
View file

@ -7,6 +7,7 @@ import { getChatById, getMessagesByChatId } from '@/lib/db/queries';
import { DataStreamHandler } from '@/components/data-stream-handler';
import { DEFAULT_CHAT_MODEL } from '@/lib/ai/models';
import { convertToUIMessages } from '@/lib/utils';
import { LanguageModelV2Usage } from '@ai-sdk/provider';
export default async function Page(props: { params: Promise<{ id: string }> }) {
const params = await props.params;
@ -53,6 +54,7 @@ export default async function Page(props: { params: Promise<{ id: string }> }) {
isReadonly={session?.user?.id !== chat.userId}
session={session}
autoResume={true}
initialLastContext={chat.lastContext}
/>
<DataStreamHandler />
</>
@ -69,6 +71,7 @@ export default async function Page(props: { params: Promise<{ id: string }> }) {
isReadonly={session?.user?.id !== chat.userId}
session={session}
autoResume={true}
initialLastContext={chat.lastContext}
/>
<DataStreamHandler />
</>