Skip to content

Commit 11f8581

Browse files
committed
Load local personas from personas subdirectory of .jupyter directory
1 parent 3e66664 commit 11f8581

File tree

1 file changed

+9
-2
lines changed

1 file changed

+9
-2
lines changed

packages/jupyter-ai/jupyter_ai/personas/persona_manager.py

Lines changed: 9 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -209,8 +209,15 @@ def _init_local_persona_classes(self) -> None:
209209
dotjupyter_dir = self.get_dotjupyter_dir()
210210
if dotjupyter_dir is None:
211211
self.log.info("No .jupyter directory found for loading local personas.")
212-
else:
213-
self._local_persona_classes = load_from_dir(dotjupyter_dir, self.log)
212+
return
213+
214+
personas_subdir = os.path.join(dotjupyter_dir, "personas")
215+
if not os.path.exists(personas_subdir):
216+
self.log.info(
217+
"No 'personas' subdirectory found in .jupyter directory for loading local personas."
218+
)
219+
220+
self._local_persona_classes = load_from_dir(personas_subdir, self.log)
214221

215222
def _init_personas(self) -> dict[str, BasePersona]:
216223
"""

0 commit comments

Comments
 (0)