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

feat: add smooth stream (#722)

Browse source
This commit is contained in:
Jeremy 2025-01-23 01:53:41 +05:30 • committed by GitHub
parent 3f9d379a6a
commit 5e8cddc886
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
5 changed files with 52 additions and 20 deletions
Download patch file Download diff file Expand all files Collapse all files

2
lib/ai/tools/create-document.ts
View file

@ -2,6 +2,7 @@ import { generateUUID } from '@/lib/utils';
import {
DataStreamWriter,
experimental_generateImage,
smoothStream,
streamObject,
streamText,
tool,
@ -60,6 +61,7 @@ export const createDocument = ({
model: customModel(model.apiIdentifier),
system:
'Write about the given topic. Markdown is supported. Use headings wherever appropriate.',
experimental_transform: smoothStream({ chunking: 'word' }),
prompt: title,
});

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

@ -1,6 +1,7 @@
import {
DataStreamWriter,
experimental_generateImage,
smoothStream,
streamObject,
streamText,
tool,
@ -52,6 +53,7 @@ export const updateDocument = ({
const { fullStream } = streamText({
model: customModel(model.apiIdentifier),
system: updateDocumentPrompt(currentContent, 'text'),
experimental_transform: smoothStream({ chunking: 'word' }),
prompt: description,
experimental_providerMetadata: {
openai: {