mirror of https://github.com/buster-so/buster.git
51 lines
1.3 KiB
TypeScript
51 lines
1.3 KiB
TypeScript
import { db, messages } from '@buster/database';
|
|
import { v4 as uuidv4 } from 'uuid';
|
|
|
|
/**
|
|
* Creates a test message record in the database
|
|
* @param chatId - The chat ID to associate the message with
|
|
* @param createdBy - The user ID who created the message
|
|
* @returns The ID of the newly created message
|
|
*/
|
|
export async function createTestMessage(chatId: string, createdBy: string): Promise<string> {
|
|
const messageId = uuidv4();
|
|
|
|
await db.insert(messages).values({
|
|
id: messageId,
|
|
chatId,
|
|
createdBy,
|
|
title: 'Test Message',
|
|
requestMessage: 'This is a test message request',
|
|
responseMessages: [{ content: 'This is a test response' }],
|
|
reasoning: { steps: ['Test reasoning step 1', 'Test reasoning step 2'] },
|
|
rawLlmMessages: [],
|
|
finalReasoningMessage: 'Test final reasoning',
|
|
isCompleted: true,
|
|
});
|
|
|
|
return messageId;
|
|
}
|
|
|
|
/**
|
|
* Creates a test message with minimal required fields
|
|
* @returns The ID of the newly created message
|
|
*/
|
|
export async function createMinimalTestMessage(): Promise<string> {
|
|
const messageId = uuidv4();
|
|
const chatId = uuidv4();
|
|
const userId = uuidv4();
|
|
|
|
await db.insert(messages).values({
|
|
id: messageId,
|
|
chatId,
|
|
createdBy: userId,
|
|
title: 'Test Message',
|
|
responseMessages: [],
|
|
reasoning: {},
|
|
rawLlmMessages: [],
|
|
isCompleted: false,
|
|
});
|
|
|
|
return messageId;
|
|
}
|