Minor doc set fix (#4480)

* update

* update

* update

* k

Author: pablonyx

Diff for: backend/onyx/db/persona.py

@@ -564,6 +564,7 @@ def upsert_persona(
             if is_default_persona is not None
             else existing_persona.is_default_persona
         )
+
         # Do not delete any associations manually added unless
         # a new updated list is provided
         if document_sets is not None:

Diff for: backend/onyx/server/features/persona/models.py

@@ -194,6 +194,10 @@ def from_model(
                 if persona.user
                 else None
             ),
+            document_sets=[
+                DocumentSet.from_model(document_set_model)
+                for document_set_model in persona.document_sets
+            ],
             search_start_date=persona.search_start_date,
             prompts=[PromptSnapshot.from_model(prompt) for prompt in persona.prompts],
             llm_relevance_filter=persona.llm_relevance_filter,

Diff for: web/src/app/admin/assistants/AssistantEditor.tsx

@@ -281,7 +281,11 @@ export function AssistantEditor({
     selectedGroups: existingPersona?.groups ?? [],
     user_file_ids: existingPersona?.user_file_ids ?? [],
     user_folder_ids: existingPersona?.user_folder_ids ?? [],
-    knowledge_source: "user_files",
+    knowledge_source:
+      (existingPersona?.user_file_ids?.length ?? 0) > 0 ||
+      (existingPersona?.user_folder_ids?.length ?? 0) > 0
+        ? "user_files"
+        : "team_knowledge",
     is_default_persona: existingPersona?.is_default_persona ?? false,
   };
 
@@ -375,6 +379,7 @@ export function AssistantEditor({
       }
     }
   };
+
   const canShowKnowledgeSource =
     ccPairs.length > 0 &&
     searchTool &&
@@ -891,6 +896,7 @@ export function AssistantEditor({
                       </div>
                     </>
                   )}
+
                   {searchTool && values.enabled_tools_map[searchTool.id] && (
                     <div>
                       {canShowKnowledgeSource && (
@@ -899,43 +905,43 @@ export function AssistantEditor({
                             <div className="flex gap-2.5">
                               <div
                                 className={`w-[150px] h-[110px] rounded-lg border flex flex-col items-center justify-center cursor-pointer transition-all ${
-                                  values.knowledge_source === "user_files"
+                                  values.knowledge_source === "team_knowledge"
                                     ? "border-2 border-blue-500 bg-blue-50 dark:bg-blue-950/20"
                                     : "border-gray-200 hover:border-gray-300 dark:border-gray-700 dark:hover:border-gray-600"
                                 }`}
                                 onClick={() =>
                                   setFieldValue(
                                     "knowledge_source",
-                                    "user_files"
+                                    "team_knowledge"
                                   )
                                 }
                               >
                                 <div className="text-blue-500 mb-2">
-                                  <FileIcon size={24} />
+                                  <BookIcon size={24} />
                                 </div>
                                 <p className="font-medium text-xs">
-                                  User Knowledge
+                                  Team Knowledge
                                 </p>
                               </div>
 
                               <div
                                 className={`w-[150px] h-[110px] rounded-lg border flex flex-col items-center justify-center cursor-pointer transition-all ${
-                                  values.knowledge_source === "team_knowledge"
+                                  values.knowledge_source === "user_files"
                                     ? "border-2 border-blue-500 bg-blue-50 dark:bg-blue-950/20"
                                     : "border-gray-200 hover:border-gray-300 dark:border-gray-700 dark:hover:border-gray-600"
                                 }`}
                                 onClick={() =>
                                   setFieldValue(
                                     "knowledge_source",
-                                    "team_knowledge"
+                                    "user_files"
                                   )
                                 }
                               >
                                 <div className="text-blue-500 mb-2">
-                                  <BookIcon size={24} />
+                                  <FileIcon size={24} />
                                 </div>
                                 <p className="font-medium text-xs">
-                                  Team Knowledge
+                                  User Knowledge
                                 </p>
                               </div>
                             </div>