Skip to content

Commit 0ec5051

Browse files
committed
add docstrings to persona_manager methods
1 parent f394f3f commit 0ec5051

File tree

1 file changed

+8
-0
lines changed

1 file changed

+8
-0
lines changed

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

Lines changed: 8 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -143,9 +143,17 @@ def _init_personas(self) -> dict[str, BasePersona]:
143143

144144
@property
145145
def personas(self) -> dict[str, BasePersona]:
146+
"""
147+
Returns a dictionary of all available persona instances, keyed by
148+
persona ID.
149+
"""
146150
return self._personas
147151

148152
def get_mentioned_personas(self, new_message: Message) -> list[BasePersona]:
153+
"""
154+
Returns a list of all personas `@`-mentioned in a chat message, given a
155+
reference to the chat message.
156+
"""
149157
mentioned_ids = set(new_message.mentions)
150158
persona_list: list[BasePersona] = []
151159

0 commit comments

Comments
 (0)