test / src /lib /types.ts
aiioe's picture
Automated build & deploy
e454383 verified
Raw History Blame Contribute Delete
1.28 kB
type Chat = {
id: string;
title: string;
messages: Array<Message>;
date: string;
};
type TempChat = {
messages: Array<Message>;
};
type Message = {
id: string;
content: string;
transcripts: Array<any>;
role: "user" | "assistant" | "system";
timestamp: Date;
};
type Transcript = {
index: number;
text: string;
timestamp: string;
timestamp_seconds: number;
filename: string;
speaker: string;
};
export type { Chat, Message, Transcript, TempChat };
function chatToJson(chat: Chat): string {
return JSON.stringify(chat, null, 2);
}
function messageToJson(message: Message): Object {
// remove the timestamp, id and the transcripts from the message
const { content, role } = message;
return { content, role };
}
function transcriptToJson(transcript: Transcript): string {
return JSON.stringify(transcript, null, 2);
}
function tempChatToJson(tempChat: TempChat): string {
return JSON.stringify(tempChat, null, 2);
}
function tempChatToChat(tempChat: TempChat, title: string = "New Chat"): Chat {
return {
id: crypto.randomUUID(),
title: title,
messages: tempChat.messages,
date: new Date().toISOString(),
};
}
export {
chatToJson,
messageToJson,
transcriptToJson,
tempChatToJson,
tempChatToChat,
};