Merge pull request #1716 from tommygonzaleza/development

Add user auth context to rigobot
breatheco-de · Nov 26, 2024 · c22ed8c · c22ed8c
2 parents d344c13 + 1325ef9
commit c22ed8c
Show file tree

Hide file tree

Showing 5 changed files with 24 additions and 5 deletions.
diff --git a/bun.lockb b/bun.lockb
diff --git a/package-lock.json b/package-lock.json
diff --git a/src/common/context/AuthContext.jsx b/src/common/context/AuthContext.jsx
@@ -15,6 +15,7 @@ import ModalInfo from '../../js_modules/moduleMap/modalInfo';
 import Text from '../components/Text';
 import { SILENT_CODE } from '../../lib/types';
 import { warn } from '../../utils/logging';
+import { generateUserContext } from '../../utils/rigobotContext';
 
 const initialState = {
   isLoading: true,
@@ -240,7 +241,10 @@ function AuthProvider({ children, pageProps }) {
   useEffect(() => {
     if (user && isRigoInitialized) {
       const token = getToken();
+      const context = generateUserContext(user);
+
       rigo.updateOptions({
+        context,
         user: {
           token,
           nickname: `${user.first_name} ${user.last_name}`,

diff --git a/src/pages/syllabus/[cohortSlug]/[lesson]/[lessonSlug]/index.jsx b/src/pages/syllabus/[cohortSlug]/[lesson]/[lessonSlug]/index.jsx
@@ -41,6 +41,7 @@ import { ORIGIN_HOST, BREATHECODE_HOST } from '../../../../../utils/variables';
 import useSession from '../../../../../common/hooks/useSession';
 import { log } from '../../../../../utils/logging';
 import completions from './completion-jobs.json';
+import { generateUserContext } from '../../../../../utils/rigobotContext';
 
 function SyllabusContent() {
   const { t, lang } = useTranslation('syllabus');
@@ -233,9 +234,11 @@ function SyllabusContent() {
       } else aiContext = cachedContext;
 
       if (aiContext) {
+        const userContext = generateUserContext(user);
+
         rigo.updateOptions({
           showBubble: false,
-          context: aiContext.ai_context,
+          context: `${userContext ? `Here is some information about this user: ${userContext}. \n` : ''}${aiContext.ai_context}`,
           completions,
         });
       }

diff --git a/src/utils/rigobotContext.js b/src/utils/rigobotContext.js
@@ -0,0 +1,12 @@
+export const generateUserContext = (user) => {
+    const { first_name, last_name, date_joined, roles, settings } = user;
+    let ai_context = '';
+
+    if (first_name) ai_context += "The user's first name is " + first_name + '. ';
+    if (last_name) ai_context += "The user's last name is " + last_name + '. ';
+    if (date_joined) ai_context += 'The user joined on ' + date_joined + '. ';
+    if (roles) roles.forEach((role) => ai_context += 'The user has the role of ' + role.role + ' in the academy of ' + role.academy.name + '. ');
+    if (settings) ai_context += 'The user has the following settings: Default language: ' + settings.lang + '.';
+
+    return ai_context;
+};