Make seeded personas default

danswer-ai · Nov 22, 2024 · ddb2202 · ddb2202
1 parent 67bfcab
commit ddb2202
Showing 1 changed file with 4 additions and 3 deletions.
diff --git a/backend/ee/danswer/server/seeding.py b/backend/ee/danswer/server/seeding.py
@@ -136,9 +136,9 @@ def _seed_personas(db_session: Session, personas: list[CreatePersonaRequest]) ->
                 user=None,  # Seeding is done as admin
                 name=persona.name,
                 description=persona.description,
-                num_chunks=persona.num_chunks
-                if persona.num_chunks is not None
-                else 0.0,
+                num_chunks=(
+                    persona.num_chunks if persona.num_chunks is not None else 0.0
+                ),
                 llm_relevance_filter=persona.llm_relevance_filter,
                 llm_filter_extraction=persona.llm_filter_extraction,
                 recency_bias=RecencyBiasSetting.AUTO,
@@ -151,6 +151,7 @@ def _seed_personas(db_session: Session, personas: list[CreatePersonaRequest]) ->
                 db_session=db_session,
                 tool_ids=persona.tool_ids,
                 display_priority=persona.display_priority,
+                is_default_persona=True,
             )