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

refactor: use generateId() for id creation (#736)

Browse source
This commit is contained in:
Jeremy 2025-01-28 18:51:04 +05:30 • committed by GitHub
parent 085f4a8ac4
commit 2d47ffbd77
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
8 changed files with 20 additions and 75 deletions
Download patch file Download diff file Expand all files Collapse all files

19
lib/utils.ts
View file

@ -126,9 +126,12 @@ export function convertToUIMessages(
}, []);
}
type ResponseMessageWithoutId = CoreToolMessage | CoreAssistantMessage;
type ResponseMessage = ResponseMessageWithoutId & { id: string };
export function sanitizeResponseMessages(
messages: Array<CoreToolMessage | CoreAssistantMessage>,
): Array<CoreToolMessage | CoreAssistantMessage> {
messages: Array<ResponseMessage>,
): Array<ResponseMessage> {
const toolResultIds: Array<string> = [];
for (const message of messages) {
@ -198,7 +201,7 @@ export function sanitizeUIMessages(messages: Array<Message>): Array<Message> {
);
}
export function getMostRecentUserMessage(messages: Array<CoreMessage>) {
export function getMostRecentUserMessage(messages: Array<Message>) {
const userMessages = messages.filter((message) => message.role === 'user');
return userMessages.at(-1);
}
@ -212,13 +215,3 @@ export function getDocumentTimestampByIndex(
return documents[index].createdAt;
}
export function getMessageIdFromAnnotations(message: Message) {
if (!message.annotations) return message.id;
const [annotation] = message.annotations;
if (!annotation) return message.id;
// @ts-expect-error messageIdFromServer is not defined in MessageAnnotation
return annotation.messageIdFromServer;
}