aline-ai 0.5.4__py3-none-any.whl → 0.5.6__py3-none-any.whl

realign/commands/export_shares.py

@@ -154,7 +154,7 @@ def _find_events_by_uuid(selector: str, events: List) -> List[int]:
 
         for i, event in enumerate(events, 1):
             # Support both 'id' and 'event_id' attributes
-            eid = getattr(event, 'event_id', None) or getattr(event, 'id', None) or ""
+            eid = getattr(event, "event_id", None) or getattr(event, "id", None) or ""
             eid = eid.lower()
             if eid.startswith(part) or eid == part:
                 if i not in indices:
@@ -452,9 +452,7 @@ def _compact_message_content(content: Any, config: ExportCompactionConfig) -> An
                     cmd = input_obj.get("command")
                     if isinstance(cmd, str):
                         input_obj = dict(input_obj)
-                        input_obj["command"] = _truncate_string(
-                            cmd, config.max_tool_command_chars
-                        )
+                        input_obj["command"] = _truncate_string(cmd, config.max_tool_command_chars)
                     block = dict(block)
                     block["input"] = input_obj
                 compacted.append(block)
@@ -527,9 +525,7 @@ def _compact_jsonl_record(
                         if not isinstance(item, dict):
                             continue
                         t = item.get("type")
-                        if t in ("input_text", "output_text") and isinstance(
-                            item.get("text"), str
-                        ):
+                        if t in ("input_text", "output_text") and isinstance(item.get("text"), str):
                             texts.append(item["text"])
                 compact_message: Dict[str, Any] = {}
                 if isinstance(role, str):
@@ -573,7 +569,11 @@ def _compact_jsonl_record(
     if isinstance(role, str):
         compact_message["role"] = role
     compacted_content = _compact_message_content(content, config)
-    if isinstance(compacted_content, str) and config.omit_thinking_blocks and compacted_content.startswith("[Thinking]"):
+    if (
+        isinstance(compacted_content, str)
+        and config.omit_thinking_blocks
+        and compacted_content.startswith("[Thinking]")
+    ):
         return None
     compact_message["content"] = compacted_content
 
@@ -651,9 +651,7 @@ def _build_export_size_report(
                         content = msg.get("content")
                         if isinstance(content, list):
                             for block in content:
-                                if isinstance(block, dict) and isinstance(
-                                    block.get("type"), str
-                                ):
+                                if isinstance(block, dict) and isinstance(block.get("type"), str):
                                     block_types.append(block["type"])
                                     if block.get("type") == "tool_use":
                                         tool_use_id = block.get("id")
@@ -669,7 +667,11 @@ def _build_export_size_report(
                                                 command = input_obj["command"]
                                             if isinstance(input_obj, dict):
                                                 input_keys = sorted(
-                                                    [k for k in input_obj.keys() if isinstance(k, str)]
+                                                    [
+                                                        k
+                                                        for k in input_obj.keys()
+                                                        if isinstance(k, str)
+                                                    ]
                                                 )
                                                 for k, v in input_obj.items():
                                                     if isinstance(k, str) and isinstance(v, str):
@@ -681,20 +683,24 @@ def _build_export_size_report(
                                                         break
                                             tool_use_index[tool_use_id] = {
                                                 "tool_name": block.get("name"),
-                                                "description": input_obj.get("description")
-                                                if isinstance(input_obj, dict)
-                                                else None,
+                                                "description": (
+                                                    input_obj.get("description")
+                                                    if isinstance(input_obj, dict)
+                                                    else None
+                                                ),
                                                 "input_keys": input_keys,
                                                 "input_string_lengths": input_string_lengths,
                                                 "path_hint": input_path_hint,
-                                                "command_snippet": _truncate_string(
-                                                    command, 200
-                                                )
-                                                if isinstance(command, str)
-                                                else None,
-                                                "command_chars": len(command)
-                                                if isinstance(command, str)
-                                                else None,
+                                                "command_snippet": (
+                                                    _truncate_string(command, 200)
+                                                    if isinstance(command, str)
+                                                    else None
+                                                ),
+                                                "command_chars": (
+                                                    len(command)
+                                                    if isinstance(command, str)
+                                                    else None
+                                                ),
                                             }
                                     if block.get("type") == "tool_result":
                                         tool_use_id = block.get("tool_use_id")
@@ -718,11 +724,7 @@ def _build_export_size_report(
                             "session_id": session.session_id,
                             "turn_id": turn.id,
                             "turn_number": turn.turn_number,
-                            "timestamp": (
-                                turn.timestamp.isoformat()
-                                if turn.timestamp
-                                else None
-                            ),
+                            "timestamp": (turn.timestamp.isoformat() if turn.timestamp else None),
                             "record_type": kind,
                             "role": role,
                             "content_block_types": sorted(set(block_types))[:6],
@@ -780,9 +782,7 @@ def build_enhanced_conversation_data(
         Dictionary with v2.0 structure including event metadata, session metadata,
         and structured turns with messages
     """
-    logger.info(
-        f"Building enhanced conversation data (v2.0) for event: {selected_event.title}"
-    )
+    logger.info(f"Building enhanced conversation data (v2.0) for event: {selected_event.title}")
 
     # Build event data
     event_data = {
@@ -792,9 +792,7 @@ def build_enhanced_conversation_data(
         "event_type": selected_event.event_type,
         "status": selected_event.status,
         "created_at": None,  # Will be populated if available from EventRecord
-        "updated_at": selected_event.updated_at.isoformat()
-        if selected_event.updated_at
-        else None,
+        "updated_at": selected_event.updated_at.isoformat() if selected_event.updated_at else None,
         "metadata": {},
     }
 
@@ -834,9 +832,7 @@ def build_enhanced_conversation_data(
                                     continue
                             messages.append(msg)
                         except json.JSONDecodeError:
-                            logger.warning(
-                                f"Failed to parse JSONL line in turn {turn.id}"
-                            )
+                            logger.warning(f"Failed to parse JSONL line in turn {turn.id}")
                             continue
 
             # Build turn data (V9: includes creator fields)
@@ -844,9 +840,9 @@ def build_enhanced_conversation_data(
                 "turn_id": turn.id,
                 "turn_number": turn.turn_number,
                 "content_hash": turn.content_hash,
-                "timestamp": turn.timestamp.isoformat()
-                if turn.timestamp
-                else datetime.now().isoformat(),
+                "timestamp": (
+                    turn.timestamp.isoformat() if turn.timestamp else datetime.now().isoformat()
+                ),
                 "llm_title": turn.llm_title or "",
                 "llm_description": turn.llm_description,
                 "user_message": (
@@ -862,7 +858,11 @@ def build_enhanced_conversation_data(
                         _collapse_high_entropy_runs(turn.assistant_summary, compaction),
                         compaction.max_turn_assistant_summary_chars,
                     )
-                    if (compaction and compaction.enabled and isinstance(turn.assistant_summary, str))
+                    if (
+                        compaction
+                        and compaction.enabled
+                        and isinstance(turn.assistant_summary, str)
+                    )
                     else turn.assistant_summary
                 ),
                 "model_name": turn.model_name,
@@ -880,12 +880,10 @@ def build_enhanced_conversation_data(
             "workspace_path": session.workspace_path,
             "session_title": session.session_title,
             "session_summary": session.session_summary,
-            "started_at": session.started_at.isoformat()
-            if session.started_at
-            else None,
-            "last_activity_at": session.last_activity_at.isoformat()
-            if session.last_activity_at
-            else None,
+            "started_at": session.started_at.isoformat() if session.started_at else None,
+            "last_activity_at": (
+                session.last_activity_at.isoformat() if session.last_activity_at else None
+            ),
             "creator_name": session.creator_name,
             "creator_id": session.creator_id,
             "turns": turns_data,
@@ -938,9 +936,7 @@ def get_unpushed_commits(repo_root: Path) -> List[UnpushedCommit]:
     Returns empty list since git tracking is no longer used.
     Use get_sessions_for_export() instead.
     """
-    logger.warning(
-        "get_unpushed_commits is deprecated, use get_sessions_for_export instead"
-    )
+    logger.warning("get_unpushed_commits is deprecated, use get_sessions_for_export instead")
     return []
 
 
@@ -1212,11 +1208,7 @@ def display_sessions_for_selection(sessions: List[ExportableSession]) -> None:
         if title:
             title = title[:80] + ("..." if len(title) > 80 else "")
 
-        last_ts = (
-            session.last_activity_at.isoformat()
-            if session.last_activity_at
-            else "unknown"
-        )
+        last_ts = session.last_activity_at.isoformat() if session.last_activity_at else "unknown"
         turns = session.turn_count
         meta = f"{session.session_type} | turns={turns} | last={last_ts}"
         if title:
@@ -1254,11 +1246,7 @@ def display_events_for_selection(events: List[ExportableEvent]) -> None:
         # First line: [索引] event_id... | 会话数 | 更新时间
         event_id_short = event.event_id[:8]
         session_count = len(event.sessions)
-        updated = (
-            event.updated_at.strftime("%Y-%m-%d %H:%M")
-            if event.updated_at
-            else "unknown"
-        )
+        updated = event.updated_at.strftime("%Y-%m-%d %H:%M") if event.updated_at else "unknown"
 
         first_line = f"  [{event.index}] {event_id_short}... | {session_count} sessions | {updated}"
         if use_rich:
@@ -1451,7 +1439,10 @@ def export_shares_command(
                 indices_list = parse_indices(indices_input)
     except ValueError as e:
         print(f"Error: Invalid selector format: {e}", file=sys.stderr)
-        print(f"Valid formats: 1-{len(events)}, abc123de (UUID prefix), or abc123de,def456gh (multiple UUIDs)", file=sys.stderr)
+        print(
+            f"Valid formats: 1-{len(events)}, abc123de (UUID prefix), or abc123de,def456gh (multiple UUIDs)",
+            file=sys.stderr,
+        )
         logger.error(f"Invalid selector format: {e}")
         return 1
 
@@ -1540,9 +1531,7 @@ def encrypt_conversation_data(data: dict, password: str) -> dict:
         包含加密数据的字典: {encrypted_data, salt, nonce, password_hash}
     """
     if not CRYPTO_AVAILABLE:
-        raise RuntimeError(
-            "cryptography package not installed. Run: pip install cryptography"
-        )
+        raise RuntimeError("cryptography package not installed. Run: pip install cryptography")
 
     # 生成盐值和随机数
     salt = os.urandom(32)
@@ -1627,7 +1616,9 @@ def upload_to_backend(
     encrypted_data_size = len(encrypted_data.encode("utf-8"))
 
     # Always print payload size for debugging
-    print(f"📊 Payload size: {payload_size / 1024 / 1024:.2f}MB (threshold: {CHUNKED_UPLOAD_THRESHOLD / 1024 / 1024:.2f}MB)")
+    print(
+        f"📊 Payload size: {payload_size / 1024 / 1024:.2f}MB (threshold: {CHUNKED_UPLOAD_THRESHOLD / 1024 / 1024:.2f}MB)"
+    )
 
     # Decide upload method based on size
     # Use chunked upload if either the full payload or encrypted_data exceeds threshold
@@ -1724,9 +1715,7 @@ def _upload_chunks_and_complete(
     # Upload each chunk
     for i, chunk in enumerate(chunks):
         if progress_callback:
-            progress_callback(
-                i + 1, total_chunks + 2, f"Uploading chunk {i + 1}/{total_chunks}..."
-            )
+            progress_callback(i + 1, total_chunks + 2, f"Uploading chunk {i + 1}/{total_chunks}...")
 
         try:
             chunk_payload = {
@@ -1912,7 +1901,9 @@ def upload_to_backend_unencrypted(
     payload_size = len(payload_json.encode("utf-8"))
 
     # Always print payload size for debugging
-    print(f"📊 Payload size: {payload_size / 1024 / 1024:.2f}MB (threshold: {CHUNKED_UPLOAD_THRESHOLD / 1024 / 1024:.2f}MB)")
+    print(
+        f"📊 Payload size: {payload_size / 1024 / 1024:.2f}MB (threshold: {CHUNKED_UPLOAD_THRESHOLD / 1024 / 1024:.2f}MB)"
+    )
 
     # Decide upload method based on size
     if payload_size > CHUNKED_UPLOAD_THRESHOLD:
@@ -2398,9 +2389,7 @@ Return the content in JSON format."""
 
     except Exception as e:
         logger.error(f"LLM UI metadata generation failed: {e}", exc_info=True)
-        print(
-            f"   ⚠️  LLM generation failed: {e}, using default UI text", file=sys.stderr
-        )
+        print(f"   ⚠️  LLM generation failed: {e}, using default UI text", file=sys.stderr)
         return None, None
 
 
@@ -2629,9 +2618,7 @@ def _export_by_events_interactive(
 
     if not events:
         print("No events available yet.")
-        print(
-            "Tip: events are auto-generated by session_summarizer during watcher sessions"
-        )
+        print("Tip: events are auto-generated by session_summarizer during watcher sessions")
         return 1
 
     # Default: show 3-7 most recent events, but allow showing all (tree view).
@@ -2642,15 +2629,11 @@ def _export_by_events_interactive(
     )
     default_limit = 7
     show_choice = (
-        input(
-            f"Show [r]ecent events ({min(default_limit, len(events_sorted))}) or [a]ll? [r]: "
-        )
+        input(f"Show [r]ecent events ({min(default_limit, len(events_sorted))}) or [a]ll? [r]: ")
         .strip()
         .lower()
     )
-    events_to_show = (
-        events_sorted if show_choice.startswith("a") else events_sorted[:default_limit]
-    )
+    events_to_show = events_sorted if show_choice.startswith("a") else events_sorted[:default_limit]
 
     def _event_path(ev) -> List[str]:
         ui = ev.ui_metadata or {}
@@ -2714,9 +2697,7 @@ def _export_by_events_interactive(
                         "idx": idx,
                         "label": key,
                         "commit_count": commit_count,
-                        "desc": desc
-                        if (desc and not desc.startswith("Auto-added"))
-                        else "",
+                        "desc": desc if (desc and not desc.startswith("Auto-added")) else "",
                     }
                 )
             else:
@@ -2773,15 +2754,9 @@ def _export_by_events_interactive(
             console.print(Padding(_render_markdown_lite(line), (0, 0, 0, 2 * indent)))
             desc = (item.get("desc") or "").strip()
             if desc:
-                console.print(
-                    Padding(_render_markdown_lite(desc), (0, 0, 1, 2 * (indent + 1)))
-                )
+                console.print(Padding(_render_markdown_lite(desc), (0, 0, 1, 2 * (indent + 1))))
         else:
-            console.print(
-                Padding(
-                    _render_markdown_lite(str(item["label"])), (0, 0, 0, 2 * indent)
-                )
-            )
+            console.print(Padding(_render_markdown_lite(str(item["label"])), (0, 0, 0, 2 * indent)))
     console.print()
 
     # Get user selection
@@ -2807,9 +2782,7 @@ def _export_by_events_interactive(
             else:
                 # Fall back to numeric indices
                 indices_list = parse_commit_indices(selection)
-            selected_events = [
-                index_to_event[i] for i in indices_list if i in index_to_event
-            ]
+            selected_events = [index_to_event[i] for i in indices_list if i in index_to_event]
 
         if not selected_events:
             print("No valid events selected.")
@@ -2817,7 +2790,9 @@ def _export_by_events_interactive(
 
     except ValueError as e:
         print(f"Error: Invalid selector format: {e}", file=sys.stderr)
-        print("Valid formats: 1,3,5-7, all, abc123de (UUID prefix), or abc123de,def456gh (multiple UUIDs)")
+        print(
+            "Valid formats: 1,3,5-7, all, abc123de (UUID prefix), or abc123de,def456gh (multiple UUIDs)"
+        )
         return 1
 
     all_commit_hashes = set()
@@ -2833,9 +2808,7 @@ def _export_by_events_interactive(
             loaded = get_commits_by_hashes(shadow_git, missing_hashes)
             selected_commits.extend(loaded)
         except Exception as e:
-            logger.error(
-                f"Failed to load commits by hashes for events: {e}", exc_info=True
-            )
+            logger.error(f"Failed to load commits by hashes for events: {e}", exc_info=True)
             print(
                 f"Error: Failed to load commits for selected events: {e}",
                 file=sys.stderr,
@@ -2868,11 +2841,7 @@ def _export_by_events_interactive(
     selected_commits = sorted(selected_commits, key=lambda c: c.timestamp, reverse=True)
     print("\n📋 Commits included:\n")
     for i, c in enumerate(selected_commits, 1):
-        summary = (
-            c.llm_summary
-            if c.llm_summary and c.llm_summary != "(No summary)"
-            else c.message
-        )
+        summary = c.llm_summary if c.llm_summary and c.llm_summary != "(No summary)" else c.message
         print(f"  [{i}] {c.hash} - {summary}")
     print()
 
@@ -2888,9 +2857,7 @@ def _export_by_events_interactive(
             try:
                 keep = parse_commit_indices(subset)
                 selected_commits = [
-                    selected_commits[i - 1]
-                    for i in keep
-                    if 0 < i <= len(selected_commits)
+                    selected_commits[i - 1] for i in keep if 0 < i <= len(selected_commits)
                 ]
             except Exception as e:
                 print(f"Error: Invalid indices format: {e}", file=sys.stderr)
@@ -3101,9 +3068,7 @@ def export_shares_interactive_command(
 
     from .. import get_realign_dir
 
-    shadow_git = get_realign_dir(
-        repo_root
-    )  # used for log/output placement (not a git repo)
+    shadow_git = get_realign_dir(repo_root)  # used for log/output placement (not a git repo)
 
     # Always show all events, don't filter by workspace
     events = get_events_for_export(workspace_path=None, limit=200)
@@ -3122,9 +3087,7 @@ def export_shares_interactive_command(
             from rich.console import Console
 
             console = Console()
-            console.print(
-                "[yellow]Enter event number to export (single event only):[/yellow]"
-            )
+            console.print("[yellow]Enter event number to export (single event only):[/yellow]")
         except ImportError:
             print("Enter event number to export (single event only):")
         indices = input("Event #: ").strip()
@@ -3236,7 +3199,9 @@ def export_shares_interactive_command(
         and full_event.share_url
     ):
         reuse_share_url = full_event.share_url
-        reuse_share_id = getattr(full_event, "share_id", None) or _extract_share_id_from_url(reuse_share_url)
+        reuse_share_id = getattr(full_event, "share_id", None) or _extract_share_id_from_url(
+            reuse_share_url
+        )
         reuse_admin_token = getattr(full_event, "share_admin_token", None)
 
     # Fast path: if we're not debugging, don't rebuild payload—just extend expiry.
@@ -3312,9 +3277,7 @@ def export_shares_interactive_command(
                 compaction=None,
             )
             conversation_data_full_for_debug = conversation_data_full
-            full_bytes = len(
-                json.dumps(conversation_data_full, ensure_ascii=False).encode("utf-8")
-            )
+            full_bytes = len(json.dumps(conversation_data_full, ensure_ascii=False).encode("utf-8"))
 
             conversation_data_compact = build_enhanced_conversation_data(
                 selected_event=selected_events[0],
@@ -3334,9 +3297,7 @@ def export_shares_interactive_command(
 
             full_mb = full_bytes / 1024 / 1024
             compact_mb = compact_bytes / 1024 / 1024
-            reduction = (
-                (1 - (compact_bytes / full_bytes)) * 100 if full_bytes else 0.0
-            )
+            reduction = (1 - (compact_bytes / full_bytes)) * 100 if full_bytes else 0.0
             print("\n=== Export Payload Size (pre-encryption) ===")
             print(f"No compact: {full_mb:.2f}MB")
             print(f"Compact:    {compact_mb:.2f}MB ({reduction:.1f}% smaller)")
@@ -3351,7 +3312,9 @@ def export_shares_interactive_command(
             print("\nLargest lines (no content):")
             for item in report["top_lines"]:
                 size_kb = item["size_bytes"] / 1024
-                blocks = ",".join(item["content_block_types"]) if item["content_block_types"] else "-"
+                blocks = (
+                    ",".join(item["content_block_types"]) if item["content_block_types"] else "-"
+                )
                 tools = item.get("tools") or []
                 tool_hint = ""
                 if tools:
@@ -3663,9 +3626,7 @@ def export_shares_interactive_command(
         db = get_database()
         db.update_event_share_metadata(
             event_id=selected_event.event_id,
-            preset_questions=ui_metadata.get("preset_questions")
-            if ui_metadata
-            else None,
+            preset_questions=ui_metadata.get("preset_questions") if ui_metadata else None,
             slack_message=ui_metadata.get("slack_message") if ui_metadata else None,
             share_url=result.get("share_url"),
             share_id=result.get("share_id")
@@ -3706,7 +3667,5 @@ def export_shares_interactive_command(
         )
 
     if not json_output:
-        logger.info(
-            f"======== Interactive export completed: {result['share_url']} ========"
-        )
+        logger.info(f"======== Interactive export completed: {result['share_url']} ========")
     return 0