Remapping persona_name to name

NeonGeckoCom · Mar 10, 2024 · d4dcaf6 · d4dcaf6
1 parent 2f9d464
commit d4dcaf6
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/neon_llm_core/utils/personas/provider.py b/neon_llm_core/utils/personas/provider.py
@@ -81,6 +81,7 @@ def _fetch_persona_config(self):
                                    target_queue=PersonasProvider.GET_CONFIGURED_PERSONAS_QUEUE)
         self.personas = response.get('items', [])
         for persona in self.personas:
+            persona.setdefault('name', persona.pop('persona_name', None))
             persona = PersonaModel.parse_obj(obj=persona)
             self._persona_handlers_state.add_persona_handler(persona=persona)