feat(kb): Link to personas

Libertai · Aug 6, 2024 · 37b589e · 37b589e
1 parent 4eb1a8e
commit 37b589e
Show file tree

Hide file tree

Showing 4 changed files with 24 additions and 116 deletions.
diff --git a/src/components/KnowledgeStoreUploader.ts b/src/components/KnowledgeStoreUploader.ts
diff --git a/src/components/dialog/PersonaDialog.vue b/src/components/dialog/PersonaDialog.vue
@@ -1,7 +1,16 @@
 <template>
   <ltai-dialog
     :title="title"
-    @save="emit('savePersona', { ...basePersona, name, role, description, avatar: toRaw(avatar) })"
+    @save="
+      emit('savePersona', {
+        ...basePersona,
+        name,
+        role,
+        description,
+        avatar: toRaw(avatar),
+        knowledgeBases: toRaw(selectedKnowledgeBases),
+      })
+    "
   >
     <q-card-section horizontal>
       <q-card-section class="tw-my-auto">
@@ -21,6 +30,10 @@
       <span>Persona Description</span>
       <q-input v-model="description" autogrow bg-color="secondary" input-class="text-light" outlined type="textarea" />
     </q-card-section>
+    <q-card-section>
+      <span>Knowledge bases</span>
+      <knowledge-bases-selector v-model="selectedKnowledgeBases" />
+    </q-card-section>
   </ltai-dialog>
 </template>
 
@@ -29,6 +42,7 @@ import { PropType, ref, toRaw, watch } from 'vue';
 import AlephAvatar from 'components/AlephAvatar.vue';
 import { BasePersonaEdition, defaultBasePersona } from 'src/types/personas';
 import LtaiDialog from 'components/libertai/LtaiDialog.vue';
+import KnowledgeBasesSelector from 'components/select/KnowledgeBasesSelector.vue';
 
 const props = defineProps({
   title: {
@@ -48,6 +62,7 @@ const name = ref(props.basePersona.name);
 const role = ref(props.basePersona.role);
 const description = ref(props.basePersona.description);
 const avatar = ref(props.basePersona.avatar);
+const selectedKnowledgeBases = ref<string[]>([]);
 
 watch(
   () => props.basePersona.name,
@@ -65,4 +80,8 @@ watch(
   () => props.basePersona.avatar,
   () => (avatar.value = props.basePersona.avatar),
 );
+watch(
+  () => props.basePersona.knowledgeBases,
+  () => (selectedKnowledgeBases.value = JSON.parse(JSON.stringify(props.basePersona.knowledgeBases))),
+);
 </script>
diff --git a/src/pages/PersonaManagement.vue b/src/pages/PersonaManagement.vue
@@ -18,7 +18,6 @@
               allowEdit: true,
               hidden: false,
               id: uuidv4(),
-              knowledgeBases: [],
             });
           }
         "
@@ -166,6 +165,7 @@ const personaExportImportSchema = z.object({
       item_hash: z.string(),
       ipfs_hash: z.string(),
     }),
+    knowledgeBases: z.array(z.string()),
   }),
 });
 type PersonaExportImportSchema = z.infer<typeof personaExportImportSchema>;
@@ -178,6 +178,7 @@ const exportPersona = (persona: UIPersona) => {
       name: persona.name,
       role: persona.role,
       avatar: persona.avatar,
+      knowledgeBases: persona.knowledgeBases,
     },
   };
 

diff --git a/src/types/personas.ts b/src/types/personas.ts
@@ -12,7 +12,7 @@ export type UIPersona = Persona & {
   knowledgeBases: string[]; // IDs of linked knowledge bases
 };
 
-export type BasePersonaEdition = Pick<UIPersona, 'name' | 'role' | 'description' | 'avatar'>;
+export type BasePersonaEdition = Pick<UIPersona, 'name' | 'role' | 'description' | 'avatar' | 'knowledgeBases'>;
 
 export const defaultBasePersona: BasePersonaEdition = {
   name: '',
@@ -22,6 +22,7 @@ export const defaultBasePersona: BasePersonaEdition = {
     item_hash: '90db3237796d27118e0b9e21dae10a4b1179878f869cb6c0058d0d7c00b0440d',
     ipfs_hash: 'QmQMBfgnmuxcQ4kptR1oPE9guYxG13GpASjYVeFQSxNxjE',
   },
+  knowledgeBases: [],
 };
 
 // eslint-disable-next-line no-unused-vars