feat: add sheet block (#743)

Co-authored-by: Nicolas <nicolascamara29@gmail.com>
This commit is contained in:
Jeremy 2025-02-03 13:33:48 +05:30 committed by GitHub
parent 93b02a85e1
commit 76804269c4
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
14 changed files with 419 additions and 6 deletions

View file

@ -123,6 +123,35 @@ export const updateDocument = ({
content: image.base64,
});
dataStream.writeData({ type: 'finish', content: '' });
} else if (document.kind === 'sheet') {
const { fullStream } = streamObject({
model: customModel(model.apiIdentifier),
system: updateDocumentPrompt(currentContent, 'sheet'),
prompt: description,
schema: z.object({
csv: z.string(),
}),
});
for await (const delta of fullStream) {
const { type } = delta;
if (type === 'object') {
const { object } = delta;
const { csv } = object;
if (csv) {
dataStream.writeData({
type: 'sheet-delta',
content: csv,
});
draftText = csv;
}
}
}
dataStream.writeData({ type: 'finish', content: '' });
}