add send_message() method to BasePersona

Author: dlqqq

packages/jupyter-ai-test/jupyter_ai_test/debug_persona.py

@@ -1,5 +1,5 @@
 from jupyter_ai.personas.base_persona import BasePersona, PersonaDefaults
-from jupyterlab_chat.models import Message
+from jupyterlab_chat.models import Message, NewMessage
 
 
 class DebugPersona(BasePersona):
@@ -20,5 +20,8 @@ def defaults(self):
         )
 
     async def process_message(self, message: Message):
-        self.log.info("HI IM DEBUGPERSONA AND IDK WHAT TO DO")
+        self.ychat.add_message(NewMessage(
+            body="Hello!",
+            sender=self.id
+        ))
         return

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

@@ -250,3 +250,13 @@ async def stream_message(self, reply_stream: "AsyncIterator") -> None:
             self.log.exception(e)
         finally:
             self.awareness.set_local_state_field("isWriting", False)
+    
+
+    def send_message(self, body: str) -> None:
+        """
+        Sends a new message to the chat from this persona.
+        """
+        self.ychat.add_message(NewMessage(
+            body=body,
+            sender=self.id
+        ))