aline-ai 0.6.0__py3-none-any.whl → 0.6.2__py3-none-any.whl

realign/db/sqlite_db.py

@@ -442,6 +442,21 @@ class SQLiteDatabase(DatabaseInterface):
         )
         conn.commit()
 
+    def update_session_metadata_flag(self, session_id: str, key: str, value: Any) -> None:
+        """Update a single key in session metadata JSON."""
+        conn = self._get_connection()
+        row = conn.execute(
+            "SELECT metadata FROM sessions WHERE id = ?", (session_id,)
+        ).fetchone()
+        if row:
+            meta = json.loads(row[0] or "{}")
+            meta[key] = value
+            conn.execute(
+                "UPDATE sessions SET metadata = ?, updated_at = datetime('now') WHERE id = ?",
+                (json.dumps(meta), session_id),
+            )
+            conn.commit()
+
     def backfill_session_total_turns(self) -> int:
         """Backfill total_turns for all sessions from turns table (V10 migration).
 
@@ -957,6 +972,7 @@ class SQLiteDatabase(DatabaseInterface):
         skip_session_summary: bool = False,
         expected_turns: Optional[int] = None,
         skip_dedup: bool = False,
+        no_track: bool = False,
     ) -> str:
         session_id = session_file_path.stem
         dedupe_key = f"turn:{session_id}:{int(turn_number)}"
@@ -973,6 +989,8 @@ class SQLiteDatabase(DatabaseInterface):
             payload["expected_turns"] = int(expected_turns)
         if skip_dedup:
             payload["skip_dedup"] = True
+        if no_track:
+            payload["no_track"] = True
 
         # For append-only session formats (Claude/Codex/Gemini), a turn is immutable once completed.
         # Avoid re-running already-done turn jobs on repeated enqueue attempts.

realign/events/session_summarizer.py

@@ -97,9 +97,24 @@ def update_session_summary_now(db: SQLiteDatabase, session_id: str) -> bool:
                 pass
             return True
 
+        # Check session metadata for no_track mode
+        is_no_track = False
         try:
-            # Generate title and summary using LLM
-            title, summary = _generate_session_summary_llm(turns)
+            session = db.get_session_by_id(session_id)
+            if session:
+                session_meta = getattr(session, "metadata", None) or {}
+                is_no_track = bool(session_meta.get("no_track", False))
+        except Exception:
+            pass
+
+        try:
+            # Skip LLM call for no-track mode
+            if is_no_track:
+                title, summary = "No Track", "No Track"
+                logger.info(f"No-track mode: skipping LLM for session summary {session_id}")
+            else:
+                # Generate title and summary using LLM
+                title, summary = _generate_session_summary_llm(turns)
 
             # Update database
             db.update_session_summary(session_id, title, summary)

realign/watcher_core.py

@@ -368,6 +368,7 @@ class DialogueWatcher:
                     session_id = signal_data.get("session_id", "")
                     project_dir = signal_data.get("project_dir", "")
                     transcript_path = signal_data.get("transcript_path", "")
+                    no_track = bool(signal_data.get("no_track", False))
 
                     logger.info(f"Stop signal received for session {session_id}")
                     print(f"[Watcher] Stop signal received for {session_id}", file=sys.stderr)
@@ -402,6 +403,7 @@ class DialogueWatcher:
                                     workspace_path=project_path,
                                     turn_number=target_turn,
                                     session_type=self._detect_session_type(session_file),
+                                    no_track=no_track,
                                 )
                             except Exception as e:
                                 logger.warning(
@@ -462,6 +464,7 @@ class DialogueWatcher:
                 prompt = str(signal_data.get("prompt") or "")
                 transcript_path = str(signal_data.get("transcript_path") or "")
                 project_dir = str(signal_data.get("project_dir") or "")
+                no_track = bool(signal_data.get("no_track", False))
 
                 session_file = None
                 if transcript_path and Path(transcript_path).exists():
@@ -488,6 +491,7 @@ class DialogueWatcher:
                     session_id,
                     prompt,
                     project_dir,
+                    no_track,
                 )
 
                 signal_file.unlink(missing_ok=True)
@@ -1318,6 +1322,7 @@ class DialogueWatcher:
         debug_callback: Optional[Callable[[Dict[str, Any]], None]] = None,
         skip_dedup: bool = False,
         skip_session_summary: bool = False,
+        no_track: bool = False,
     ) -> bool:
         """
         Execute commit with DB-backed lease locking to prevent cross-process races.
@@ -1365,6 +1370,7 @@ class DialogueWatcher:
                     debug_callback=debug_callback,
                     skip_dedup=skip_dedup,
                     skip_session_summary=skip_session_summary,
+                    no_track=no_track,
                 )
         except Exception as e:
             print(f"[Watcher] Commit error: {e}", file=sys.stderr)
@@ -1381,6 +1387,7 @@ class DialogueWatcher:
         debug_callback: Optional[Callable[[Dict[str, Any]], None]] = None,
         skip_dedup: bool = False,
         skip_session_summary: bool = False,
+        no_track: bool = False,
     ) -> bool:
         """
         Perform the actual commit operation to SQLite database.
@@ -1443,7 +1450,7 @@ class DialogueWatcher:
             file_created = datetime.fromtimestamp(
                 getattr(file_stat, "st_birthtime", file_stat.st_ctime)
             )
-            db.get_or_create_session(
+            session = db.get_or_create_session(
                 session_id=session_id,
                 session_file_path=session_file,
                 session_type=self._detect_session_type(session_file),
@@ -1451,6 +1458,19 @@ class DialogueWatcher:
                 workspace_path=str(project_path) if project_path else None,
             )
 
+            # Check no_track from parameter or existing session metadata (polling path)
+            is_no_track = no_track
+            if not is_no_track and session:
+                session_meta = getattr(session, "metadata", None) or {}
+                is_no_track = bool(session_meta.get("no_track", False))
+
+            # Store no_track flag in session metadata if applicable
+            if is_no_track:
+                try:
+                    db.update_session_metadata_flag(session_id, "no_track", True)
+                except Exception:
+                    pass
+
             takeover_attempt = False
             existing_turn = db.get_turn_by_number(session_id, turn_number)
             if existing_turn and not skip_dedup:
@@ -1517,14 +1537,19 @@ class DialogueWatcher:
                 logger.debug(f"Failed to write processing placeholder: {e}")
 
             try:
-                # Generate LLM summary with fallback for errors
-                llm_result = self._generate_llm_summary(
-                    session_file,
-                    turn_number=turn_number,
-                    turn_content=turn_content,
-                    user_message=user_message,
-                    debug_callback=debug_callback,
-                )
+                # Skip LLM call for no-track mode
+                if is_no_track:
+                    llm_result = ("No Track", None, "No Track", "no", "fine")
+                    logger.info(f"No-track mode: skipping LLM for {session_id} turn {turn_number}")
+                else:
+                    # Generate LLM summary with fallback for errors
+                    llm_result = self._generate_llm_summary(
+                        session_file,
+                        turn_number=turn_number,
+                        turn_content=turn_content,
+                        user_message=user_message,
+                        debug_callback=debug_callback,
+                    )
 
                 if not llm_result:
                     # LLM summary failed, use error marker to continue commit
@@ -1774,6 +1799,7 @@ class DialogueWatcher:
         session_id: str,
         prompt: str,
         project_dir: str,
+        no_track: bool = False,
     ) -> None:
         """Generate and store a temporary turn title for a newly submitted user prompt."""
         try:
@@ -1802,20 +1828,28 @@ class DialogueWatcher:
             if not user_message:
                 user_message = str(group.get("user_message") or "")
 
-            turn_content = self._extract_turn_content_by_number(session_file, turn_number)
-            result = self._generate_llm_summary(
-                session_file,
-                turn_number=turn_number,
-                turn_content=turn_content,
-                user_message=user_message or None,
-                session_id=session_id or session_file.stem,
-            )
-            if not result:
-                return
+            # Skip LLM call for no-track mode
+            if no_track:
+                title = "No Track"
+                model_name = None
+                description = "No Track"
+                if_last_task = "no"
+                satisfaction = "fine"
+            else:
+                turn_content = self._extract_turn_content_by_number(session_file, turn_number)
+                result = self._generate_llm_summary(
+                    session_file,
+                    turn_number=turn_number,
+                    turn_content=turn_content,
+                    user_message=user_message or None,
+                    session_id=session_id or session_file.stem,
+                )
+                if not result:
+                    return
 
-            title, model_name, description, if_last_task, satisfaction = result
-            if not title:
-                return
+                title, model_name, description, if_last_task, satisfaction = result
+                if not title:
+                    return
 
             from .db import get_database
             from .db.base import TurnRecord

realign/worker_core.py

@@ -186,6 +186,7 @@ class AlineWorker:
         expected_turns_raw = payload.get("expected_turns")
         expected_turns = int(expected_turns_raw) if expected_turns_raw is not None else None
         skip_dedup = bool(payload.get("skip_dedup") or False)
+        no_track = bool(payload.get("no_track") or False)
 
         if not session_id or turn_number <= 0 or not session_file_path:
             raise ValueError(f"Invalid turn_summary payload: {payload}")
@@ -212,6 +213,7 @@ class AlineWorker:
             quiet=True,
             skip_session_summary=skip_session_summary,
             skip_dedup=skip_dedup,
+            no_track=no_track,
         )
 
         if created: