aline-ai 0.5.3__py3-none-any.whl → 0.5.5__py3-none-any.whl

realign/events/event_summarizer.py

@@ -31,6 +31,7 @@ def schedule_event_summary_update(db: SQLiteDatabase, event_id: str) -> None:
         db: Database instance
         event_id: ID of the event to update
     """
+
     def do_update():
         _update_event_summary(db, event_id)
 
@@ -122,10 +123,17 @@ def _get_event_summary_prompt() -> str:
                 logger.debug(f"Loaded user-customized event summary prompt from {user_prompt_path}")
                 return text
     except Exception:
-        logger.debug("Failed to load user-customized event summary prompt, falling back", exc_info=True)
+        logger.debug(
+            "Failed to load user-customized event summary prompt, falling back", exc_info=True
+        )
 
     # Fall back to built-in prompt (tools/commit_message_prompts/event_summary.md)
-    candidate = Path(__file__).resolve().parents[2] / "tools" / "commit_message_prompts" / "event_summary.md"
+    candidate = (
+        Path(__file__).resolve().parents[2]
+        / "tools"
+        / "commit_message_prompts"
+        / "event_summary.md"
+    )
     try:
         if candidate.exists():
             text = candidate.read_text(encoding="utf-8").strip()
@@ -184,19 +192,25 @@ def _generate_event_summary_llm(sessions: List[SessionRecord]) -> Tuple[str, str
     # Build sessions payload for prompt
     sessions_data = []
     for i, session in enumerate(sessions):
-        sessions_data.append({
-            "session_number": i + 1,
-            "title": session.session_title or f"Session {session.id[:8]}",
-            "summary": session.session_summary or "(no summary)",
-            "session_type": session.session_type,
-        })
+        sessions_data.append(
+            {
+                "session_number": i + 1,
+                "title": session.session_title or f"Session {session.id[:8]}",
+                "summary": session.session_summary or "(no summary)",
+                "session_type": session.session_type,
+            }
+        )
 
     system_prompt = _get_event_summary_prompt()
 
-    user_prompt = json.dumps({
-        "total_sessions": len(sessions),
-        "sessions": sessions_data,
-    }, ensure_ascii=False, indent=2)
+    user_prompt = json.dumps(
+        {
+            "total_sessions": len(sessions),
+            "sessions": sessions_data,
+        },
+        ensure_ascii=False,
+        indent=2,
+    )
 
     try:
         # Use unified LLM client
@@ -205,7 +219,7 @@ def _generate_event_summary_llm(sessions: List[SessionRecord]) -> Tuple[str, str
             user_prompt=user_prompt,
             provider="auto",  # Try Claude first, fallback to OpenAI
             max_tokens=500,
-            purpose="event_summary"
+            purpose="event_summary",
         )
 
         if not response:
@@ -241,5 +255,3 @@ def _fallback_event_summary(sessions: List[SessionRecord]) -> Tuple[str, str]:
         description = f"Event containing {len(sessions)} sessions."
 
     return title[:100], description
-
-

realign/events/session_summarizer.py

@@ -176,13 +176,22 @@ def _get_session_summary_prompt() -> str:
             text = user_prompt_path.read_text(encoding="utf-8").strip()
             if text:
                 _SESSION_SUMMARY_PROMPT_CACHE = text
-                logger.debug(f"Loaded user-customized session summary prompt from {user_prompt_path}")
+                logger.debug(
+                    f"Loaded user-customized session summary prompt from {user_prompt_path}"
+                )
                 return text
     except Exception:
-        logger.debug("Failed to load user-customized session summary prompt, falling back", exc_info=True)
+        logger.debug(
+            "Failed to load user-customized session summary prompt, falling back", exc_info=True
+        )
 
     # Fall back to built-in prompt (tools/commit_message_prompts/session_summary.md)
-    candidate = Path(__file__).resolve().parents[2] / "tools" / "commit_message_prompts" / "session_summary.md"
+    candidate = (
+        Path(__file__).resolve().parents[2]
+        / "tools"
+        / "commit_message_prompts"
+        / "session_summary.md"
+    )
     try:
         if candidate.exists():
             text = candidate.read_text(encoding="utf-8").strip()
@@ -234,19 +243,25 @@ def _generate_session_summary_llm(turns: List[TurnRecord]) -> Tuple[str, str]:
     # Build turns payload for prompt
     turns_data = []
     for i, turn in enumerate(turns):
-        turns_data.append({
-            "turn_number": i + 1,
-            "title": turn.llm_title or "(no title)",
-            "summary": turn.assistant_summary or "(no summary)",
-            "user_request": (turn.user_message or "")[:200],  # Truncate long messages
-        })
+        turns_data.append(
+            {
+                "turn_number": i + 1,
+                "title": turn.llm_title or "(no title)",
+                "summary": turn.assistant_summary or "(no summary)",
+                "user_request": (turn.user_message or "")[:200],  # Truncate long messages
+            }
+        )
 
     system_prompt = _get_session_summary_prompt()
 
-    user_prompt = json.dumps({
-        "total_turns": len(turns),
-        "turns": turns_data,
-    }, ensure_ascii=False, indent=2)
+    user_prompt = json.dumps(
+        {
+            "total_turns": len(turns),
+            "turns": turns_data,
+        },
+        ensure_ascii=False,
+        indent=2,
+    )
 
     try:
         # Use unified LLM client
@@ -255,7 +270,7 @@ def _generate_session_summary_llm(turns: List[TurnRecord]) -> Tuple[str, str]:
             user_prompt=user_prompt,
             provider="auto",  # Try Claude first, fallback to OpenAI
             max_tokens=500,
-            purpose="session_summary"
+            purpose="session_summary",
         )
 
         if not response:
@@ -290,4 +305,3 @@ def _fallback_summary(turns: List[TurnRecord]) -> Tuple[str, str]:
     summary = "\n".join(f"- {s}" for s in recent) if recent else ""
 
     return title, summary
-

realign/file_lock.py

@@ -109,6 +109,7 @@ def commit_lock(repo_path: Path, timeout: float = 10.0):
         True if lock was acquired
     """
     from realign import get_realign_dir
+
     realign_dir = get_realign_dir(repo_path)
     lock_file = realign_dir / ".commit.lock"
     lock = FileLock(lock_file, timeout=timeout)