fix note ai provenance updates

Author: MomiJiSan

plugin/plugins/study_companion/store_notebook.py

@@ -378,6 +378,11 @@ def upsert_note(
             existing.title if title is None else title,
             content_plain,
         )
+        next_is_ai_generated = (
+            existing.is_ai_generated
+            if is_ai_generated is None
+            else bool(is_ai_generated)
+        )
         topic_values = existing.topic_ids if topic_ids is _UNSET else topic_ids
         tag_values = existing.tags if tags is _UNSET else tags
         with self.store._lock:
@@ -392,6 +397,7 @@ def upsert_note(
                     snippet = ?,
                     topic_ids = ?,
                     tags = ?,
+                    is_ai_generated = ?,
                     word_count = ?,
                     updated_at = datetime('now'),
                     edited_at = datetime('now')
@@ -405,6 +411,7 @@ def upsert_note(
                     _snippet(content_plain),
                     self.store._json_dumps(_normalize_string_list(topic_values)),
                     self.store._json_dumps(_normalize_string_list(tag_values)),
+                    1 if next_is_ai_generated else 0,
                     _word_count(content_plain),
                     note_key,
                 ),

plugin/tests/unit/plugins/test_study_notebook.py

@@ -103,6 +103,7 @@ def test_notebook_store_crud_search_and_topic_counts(tmp_path) -> None:
             notebook_id=notebook.id,
             title="Closure updated",
             content="closure update",
+            is_ai_generated=True,
             source_type="topic",
             source_ref="closure",
             topic_ids=["closure"],
@@ -115,13 +116,18 @@ def test_notebook_store_crud_search_and_topic_counts(tmp_path) -> None:
             "source_type update ignored" in str(args[0])
             for args, _kwargs in logger.warnings
         )
+        assert updated.is_ai_generated is True
 
         partial = notebooks.upsert_note(note_id=note.id, title="Closure final")
         assert partial.title == "Closure final"
         assert partial.notebook_id == notebook.id
         assert partial.content == "closure update"
         assert partial.topic_ids == ["closure"]
         assert partial.tags == ["updated"]
+        assert partial.is_ai_generated is True
+
+        manual = notebooks.upsert_note(note_id=note.id, is_ai_generated=False)
+        assert manual.is_ai_generated is False
 
         deleted = notebooks.delete_notebook(notebook.id)
         assert deleted == {"deleted": 1, "notes_unlinked": 1}