diff --git a/core/src/node/api/restful/common.ts b/core/src/node/api/restful/common.ts index b87bc946d..433632989 100644 --- a/core/src/node/api/restful/common.ts +++ b/core/src/node/api/restful/common.ts @@ -8,7 +8,7 @@ import { createMessage, createThread, getMessages, - retrieveMesasge, + retrieveMessage, updateThread, } from './helper/builder' @@ -17,10 +17,18 @@ import { startModel, stopModel } from './helper/startStopModel' import { ModelSettingParams } from '../../../types' export const commonRouter = async (app: HttpServer) => { + const normalizeData = (data: any) => { + return { + object: 'list', + data, + } + } // Common Routes // Read & Delete :: Threads | Models | Assistants Object.keys(JanApiRouteConfiguration).forEach((key) => { - app.get(`/${key}`, async (_request) => getBuilder(JanApiRouteConfiguration[key])) + app.get(`/${key}`, async (_request) => + getBuilder(JanApiRouteConfiguration[key]).then(normalizeData) + ) app.get(`/${key}/:id`, async (request: any) => retrieveBuilder(JanApiRouteConfiguration[key], request.params.id) @@ -34,10 +42,12 @@ export const commonRouter = async (app: HttpServer) => { // Threads app.post(`/threads/`, async (req, res) => createThread(req.body)) - app.get(`/threads/:threadId/messages`, async (req, res) => getMessages(req.params.threadId)) + app.get(`/threads/:threadId/messages`, async (req, res) => + getMessages(req.params.threadId).then(normalizeData) + ) app.get(`/threads/:threadId/messages/:messageId`, async (req, res) => - retrieveMesasge(req.params.threadId, req.params.messageId) + retrieveMessage(req.params.threadId, req.params.messageId) ) app.post(`/threads/:threadId/messages`, async (req, res) => diff --git a/core/src/node/api/restful/helper/builder.ts b/core/src/node/api/restful/helper/builder.ts index b7753fcb0..7001c0c76 100644 --- a/core/src/node/api/restful/helper/builder.ts +++ b/core/src/node/api/restful/helper/builder.ts @@ -125,7 +125,7 @@ export const getMessages = async (threadId: string): Promise => } } -export const retrieveMesasge = async (threadId: string, messageId: string) => { +export const retrieveMessage = async (threadId: string, messageId: string) => { const messages = await getMessages(threadId) const filteredMessages = messages.filter((m) => m.id === messageId) if (!filteredMessages || filteredMessages.length === 0) {