src/scribe/services/embeddings.py

@@ -16,6 +16,8 @@ import os
 
 from collections.abc import Sequence
 
+from typing import TYPE_CHECKING
+
 from sqlalchemy import delete, or_, select
 
 from scribe.models import async_session
@@ -23,6 +25,9 @@ from scribe.models.embedding import NoteEmbedding, RuleEmbedding
 from scribe.models.note import Note
 from scribe.services.access import notes_visibility_clause
 
+if TYPE_CHECKING:  # resolves the Rule forward ref without importing at runtime
+    from scribe.models.rulebook import Rule
+
 logger = logging.getLogger(__name__)
 
 # Minimum cosine similarity to include a note in context results.