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

chore: update to ai sdk v5 beta (#1074)

Browse source
This commit is contained in:
Jeremy 2025-07-03 02:26:34 -07:00 • committed by GitHub
parent 7d8e71383f
commit 4c281fe09d
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
54 changed files with 1372 additions and 1060 deletions
Download patch file Download diff file Expand all files Collapse all files

20
lib/ai/tools/update-document.ts
View file

@ -1,18 +1,19 @@
import { DataStreamWriter, tool } from 'ai';
import { Session } from 'next-auth';
import { tool, type UIMessageStreamWriter } from 'ai';
import type { Session } from 'next-auth';
import { z } from 'zod';
import { getDocumentById, saveDocument } from '@/lib/db/queries';
import { getDocumentById } from '@/lib/db/queries';
import { documentHandlersByArtifactKind } from '@/lib/artifacts/server';
import type { ChatMessage } from '@/lib/types';
interface UpdateDocumentProps {
session: Session;
dataStream: DataStreamWriter;
dataStream: UIMessageStreamWriter<ChatMessage>;
}
export const updateDocument = ({ session, dataStream }: UpdateDocumentProps) =>
tool({
description: 'Update a document with the given description.',
parameters: z.object({
inputSchema: z.object({
id: z.string().describe('The ID of the document to update'),
description: z
.string()
@ -27,9 +28,10 @@ export const updateDocument = ({ session, dataStream }: UpdateDocumentProps) =>
};
}
dataStream.writeData({
type: 'clear',
content: document.title,
dataStream.write({
type: 'data-clear',
data: null,
transient: true,
});
const documentHandler = documentHandlersByArtifactKind.find(
@ -48,7 +50,7 @@ export const updateDocument = ({ session, dataStream }: UpdateDocumentProps) =>
session,
});
dataStream.writeData({ type: 'finish', content: '' });
dataStream.write({ type: 'data-finish', data: null, transient: true });
return {
id,