diff --git a/stream_app/chat.py b/stream_app/chat.py
index 00d8bd5..c883780 100644
--- a/stream_app/chat.py
+++ b/stream_app/chat.py
@@ -111,5 +111,6 @@ def chat_sidebar(session_id):
                             make_note_from_chat(
                                 content=msg.content,
                                 notebook_id=st.session_state[session_id]["notebook"].id,
+                                type="ai",
                             )
                             st.rerun()