Merge pull request #1717 from Jaredude/feature/chatmessage-query-messageId

Allows query chatmessage API endpoint by messageId
This commit is contained in:
Henry Heng
2024-02-13 12:12:51 +08:00
committed by GitHub
+7 -3
View File
@@ -511,6 +511,7 @@ export class App {
const chatId = req.query?.chatId as string | undefined const chatId = req.query?.chatId as string | undefined
const memoryType = req.query?.memoryType as string | undefined const memoryType = req.query?.memoryType as string | undefined
const sessionId = req.query?.sessionId as string | undefined const sessionId = req.query?.sessionId as string | undefined
const messageId = req.query?.messageId as string | undefined
const startDate = req.query?.startDate as string | undefined const startDate = req.query?.startDate as string | undefined
const endDate = req.query?.endDate as string | undefined const endDate = req.query?.endDate as string | undefined
let chatTypeFilter = req.query?.chatType as chatType | undefined let chatTypeFilter = req.query?.chatType as chatType | undefined
@@ -538,7 +539,8 @@ export class App {
memoryType, memoryType,
sessionId, sessionId,
startDate, startDate,
endDate endDate,
messageId
) )
return res.json(chatmessages) return res.json(chatmessages)
}) })
@@ -1440,7 +1442,8 @@ export class App {
memoryType?: string, memoryType?: string,
sessionId?: string, sessionId?: string,
startDate?: string, startDate?: string,
endDate?: string endDate?: string,
messageId?: string
): Promise<ChatMessage[]> { ): Promise<ChatMessage[]> {
let fromDate let fromDate
if (startDate) fromDate = new Date(startDate) if (startDate) fromDate = new Date(startDate)
@@ -1455,7 +1458,8 @@ export class App {
chatId, chatId,
memoryType: memoryType ?? (chatId ? IsNull() : undefined), memoryType: memoryType ?? (chatId ? IsNull() : undefined),
sessionId: sessionId ?? undefined, sessionId: sessionId ?? undefined,
createdDate: toDate && fromDate ? Between(fromDate, toDate) : undefined createdDate: toDate && fromDate ? Between(fromDate, toDate) : undefined,
id: messageId ?? undefined
}, },
order: { order: {
createdDate: sortOrder === 'DESC' ? 'DESC' : 'ASC' createdDate: sortOrder === 'DESC' ? 'DESC' : 'ASC'