feat: migrate chatbot prompt utils from apps to apps-query-client (#219)

src/index.ts

@@ -7,3 +7,4 @@ export { default as mockApi, MockSolution } from './mockServer/mockServer';
 export { buildMockLocalContext, buildDatabase } from './mockServer/fixtures';
 export * from './components';
 export * from './types';
+export * from './utils';

src/types.ts

@@ -111,3 +111,10 @@ export type ChatBotCompletion = {
   completion: string;
   model: string;
 };
+
+export interface ChatbotThreadMessage {
+  // represent the appDataTypes of the chatbot in the current app
+  botDataType: string;
+  msgType: string;
+  data: string;
+}

src/utils/chatbot.ts

@@ -0,0 +1,25 @@
+import { ChatBotMessage, ChatbotRole } from '@graasp/sdk';
+
+import { ChatbotThreadMessage } from 'src/types';
+
+export const buildPrompt = (
+  initialPrompt: string | undefined,
+  threadMessages: ChatbotThreadMessage[],
+  userMessage: string,
+): Array<ChatBotMessage> => {
+  // define the message to send to OpenAI with the initial prompt first if needed (role system).
+  // Each call to OpenAI must contain the whole history of the messages.
+  const finalPrompt: Array<ChatBotMessage> = initialPrompt
+    ? [{ role: ChatbotRole.System, content: initialPrompt }]
+    : [];
+
+  threadMessages.forEach((msg) => {
+    const msgRole = msg.msgType === msg.botDataType ? ChatbotRole.Assistant : ChatbotRole.User;
+    finalPrompt.push({ role: msgRole, content: msg.data });
+  });
+
+  // add the last user's message in the prompt
+  finalPrompt.push({ role: ChatbotRole.User, content: userMessage });
+
+  return finalPrompt;
+};

src/utils/index.ts

@@ -0,0 +1 @@
+export * from './chatbot';