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

realign/logging_config.py

@@ -22,15 +22,15 @@ def get_log_level() -> int:
     Returns:
         int: Logging level constant from logging module
     """
-    level_name = os.getenv('REALIGN_LOG_LEVEL', 'INFO').upper()
+    level_name = os.getenv("REALIGN_LOG_LEVEL", "INFO").upper()
 
     # Map string to logging constant
     level_map = {
-        'DEBUG': logging.DEBUG,
-        'INFO': logging.INFO,
-        'WARNING': logging.WARNING,
-        'ERROR': logging.ERROR,
-        'CRITICAL': logging.CRITICAL,
+        "DEBUG": logging.DEBUG,
+        "INFO": logging.INFO,
+        "WARNING": logging.WARNING,
+        "ERROR": logging.ERROR,
+        "CRITICAL": logging.CRITICAL,
     }
 
     return level_map.get(level_name, logging.INFO)
@@ -46,12 +46,12 @@ def get_log_directory() -> Path:
     Returns:
         Path: Log directory path
     """
-    log_dir_str = os.getenv('REALIGN_LOG_DIR')
+    log_dir_str = os.getenv("REALIGN_LOG_DIR")
 
     if log_dir_str:
         log_dir = Path(log_dir_str).expanduser()
     else:
-        log_dir = Path.home() / '.aline' / '.logs'
+        log_dir = Path.home() / ".aline" / ".logs"
 
     # Create directory if it doesn't exist
     log_dir.mkdir(parents=True, exist_ok=True)
@@ -64,7 +64,7 @@ def setup_logger(
     log_file: Optional[str] = None,
     max_bytes: int = 10 * 1024 * 1024,  # 10MB
     backup_count: int = 5,
-    console_output: bool = False
+    console_output: bool = False,
 ) -> logging.Logger:
     """
     Set up a logger with file rotation and optional console output.
@@ -95,8 +95,7 @@ def setup_logger(
 
     # Standard formatter with timestamp, level, name, and message
     formatter = logging.Formatter(
-        fmt='[%(asctime)s] [%(levelname)s] [%(name)s] %(message)s',
-        datefmt='%Y-%m-%d %H:%M:%S'
+        fmt="[%(asctime)s] [%(levelname)s] [%(name)s] %(message)s", datefmt="%Y-%m-%d %H:%M:%S"
     )
 
     # File handler with rotation
@@ -106,10 +105,7 @@ def setup_logger(
             log_path = log_dir / log_file
 
             file_handler = RotatingFileHandler(
-                log_path,
-                maxBytes=max_bytes,
-                backupCount=backup_count,
-                encoding='utf-8'
+                log_path, maxBytes=max_bytes, backupCount=backup_count, encoding="utf-8"
             )
             file_handler.setLevel(logging.DEBUG)  # Capture all levels to file
             file_handler.setFormatter(formatter)
@@ -119,6 +115,7 @@ def setup_logger(
             # If file logging fails, fall back to stderr only
             # Don't let logging setup break the application
             import sys
+
             print(f"Warning: Failed to set up file logging: {e}", file=sys.stderr)
 
     # Optional console handler (stderr)

realign/mcp_server.py

@@ -10,6 +10,7 @@ from urllib.parse import urlparse
 
 try:
     import httpx
+
     HTTPX_AVAILABLE = True
 except ImportError:
     HTTPX_AVAILABLE = False
@@ -33,15 +34,15 @@ def extract_share_id(share_url: str) -> str:
         https://example.com/share/xyz789/chat -> xyz789
     """
     # Remove trailing slash
-    url = share_url.rstrip('/')
+    url = share_url.rstrip("/")
 
     # Extract path components
     parsed = urlparse(url)
-    path_parts = [p for p in parsed.path.split('/') if p]
+    path_parts = [p for p in parsed.path.split("/") if p]
 
     # Find 'share' in path and get next component
-    if 'share' in path_parts:
-        share_idx = path_parts.index('share')
+    if "share" in path_parts:
+        share_idx = path_parts.index("share")
         if share_idx + 1 < len(path_parts):
             return path_parts[share_idx + 1]
 
@@ -59,7 +60,9 @@ def extract_base_url(share_url: str) -> str:
     return f"{parsed.scheme}://{parsed.netloc}"
 
 
-async def authenticate_share(share_url: str, password: Optional[str] = None) -> tuple[str, str, str]:
+async def authenticate_share(
+    share_url: str, password: Optional[str] = None
+) -> tuple[str, str, str]:
     """
     Authenticate with share and get session token.
 
@@ -97,16 +100,13 @@ async def authenticate_share(share_url: str, password: Optional[str] = None) ->
             password_hash = hashlib.sha256(password.encode()).hexdigest()
 
             auth_resp = await client.post(
-                f"{base_url}/api/share/{share_id}/auth",
-                json={"password_hash": password_hash}
+                f"{base_url}/api/share/{share_id}/auth", json={"password_hash": password_hash}
             )
             auth_resp.raise_for_status()
             data = auth_resp.json()
         else:
             # No password needed - create session directly
-            session_resp = await client.post(
-                f"{base_url}/api/share/{share_id}/session"
-            )
+            session_resp = await client.post(f"{base_url}/api/share/{share_id}/session")
             session_resp.raise_for_status()
             data = session_resp.json()
 
@@ -117,12 +117,7 @@ async def authenticate_share(share_url: str, password: Optional[str] = None) ->
         return base_url, share_id, session_token
 
 
-async def ask_conversation(
-    base_url: str,
-    share_id: str,
-    session_token: str,
-    question: str
-) -> str:
+async def ask_conversation(base_url: str, share_id: str, session_token: str, question: str) -> str:
     """
     Send a question to the remote AI agent and receive the answer.
 
@@ -147,7 +142,7 @@ async def ask_conversation(
         resp = await client.post(
             f"{base_url}/api/chat/{share_id}",
             headers={"x-session-token": session_token},
-            json={"messages": [{"role": "user", "parts": [{"type": "text", "text": question}]}]}
+            json={"messages": [{"role": "user", "parts": [{"type": "text", "text": question}]}]},
         )
         resp.raise_for_status()
 
@@ -158,13 +153,13 @@ async def ask_conversation(
 
         async for chunk in resp.aiter_text():
             # Split into lines and process each
-            for line in chunk.split('\n'):
+            for line in chunk.split("\n"):
                 line = line.strip()
                 if not line:
                     continue
 
                 # Remove "data: " prefix if present
-                if line.startswith('data: '):
+                if line.startswith("data: "):
                     line = line[6:]
 
                 # Try to parse as JSON
@@ -176,8 +171,8 @@ async def ask_conversation(
                     if isinstance(data, dict):
                         # Extract text from text-delta chunks
                         # The field name is 'delta' not 'textDelta'
-                        if data.get('type') == 'text-delta':
-                            delta = data.get('delta', '')
+                        if data.get("type") == "text-delta":
+                            delta = data.get("delta", "")
                             if delta:
                                 text_chunks.append(delta)
 
@@ -186,21 +181,22 @@ async def ask_conversation(
                     continue
 
         # Combine all text chunks to get the final answer
-        answer = ''.join(text_chunks)
+        answer = "".join(text_chunks)
 
         # Apply reasonable length limit to prevent overwhelming the MCP client
         # If answer is too long, truncate and add notice
         MAX_RESPONSE_LENGTH = 50000  # ~50KB of text
         if len(answer) > MAX_RESPONSE_LENGTH:
-            answer = answer[:MAX_RESPONSE_LENGTH] + "\n\n[Response truncated due to length. Please ask more specific questions to get complete answers.]"
+            answer = (
+                answer[:MAX_RESPONSE_LENGTH]
+                + "\n\n[Response truncated due to length. Please ask more specific questions to get complete answers.]"
+            )
 
         return answer if answer else "No response received from the agent."
 
 
 async def handle_ask_tool(
-    share_url: str,
-    question: str,
-    password: Optional[str] = None
+    share_url: str, question: str, password: Optional[str] = None
 ) -> list[TextContent]:
     """
     Handle the ask_shared_conversation tool.
@@ -221,21 +217,12 @@ async def handle_ask_tool(
         answer = await ask_conversation(base_url, share_id, token, question)
 
         # Return the answer
-        return [TextContent(
-            type="text",
-            text=answer
-        )]
+        return [TextContent(type="text", text=answer)]
 
     except ValueError as e:
-        return [TextContent(
-            type="text",
-            text=f"Authentication error: {str(e)}"
-        )]
+        return [TextContent(type="text", text=f"Authentication error: {str(e)}")]
     except Exception as e:
-        return [TextContent(
-            type="text",
-            text=f"Error querying shared conversation: {str(e)}"
-        )]
+        return [TextContent(type="text", text=f"Error querying shared conversation: {str(e)}")]
 
 
 @app.list_tools()
@@ -288,24 +275,17 @@ async def call_tool(name: str, arguments: Any) -> list[TextContent]:
         return await handle_ask_tool(
             share_url=arguments.get("share_url", ""),
             question=arguments.get("question", ""),
-            password=arguments.get("password")
+            password=arguments.get("password"),
         )
     else:
-        return [TextContent(
-            type="text",
-            text=f"Unknown tool: {name}"
-        )]
+        return [TextContent(type="text", text=f"Unknown tool: {name}")]
 
 
 async def async_main():
     """Main async entry point for the MCP server."""
     # Start stdio server
     async with stdio_server() as (read_stream, write_stream):
-        await app.run(
-            read_stream,
-            write_stream,
-            app.create_initialization_options()
-        )
+        await app.run(read_stream, write_stream, app.create_initialization_options())
 
 
 def main():
@@ -313,9 +293,8 @@ def main():
     # Check if httpx is available
     if not HTTPX_AVAILABLE:
         print(
-            "Error: httpx package is required for aline-mcp.\n"
-            "Install with: pip install httpx",
-            file=sys.stderr
+            "Error: httpx package is required for aline-mcp.\n" "Install with: pip install httpx",
+            file=sys.stderr,
         )
         sys.exit(1)
 

realign/mcp_watcher.py

@@ -32,4 +32,3 @@ _watcher_core.find_all_active_sessions = _find_all_active_sessions_proxy
 ReAlignConfig = _watcher_core.ReAlignConfig
 DialogueWatcher = _watcher_core.DialogueWatcher
 is_path_blacklisted = _watcher_core.is_path_blacklisted
-

realign/models/event.py

@@ -8,9 +8,10 @@ from enum import Enum
 
 class EventSource(Enum):
     """How the event was generated."""
-    USER = "user"                  # Manually created via `aline watcher event generate`
-    PRESET_DAY = "preset_day"      # Auto-generated daily aggregation
-    PRESET_WEEK = "preset_week"    # Auto-generated weekly aggregation
+
+    USER = "user"  # Manually created via `aline watcher event generate`
+    PRESET_DAY = "preset_day"  # Auto-generated daily aggregation
+    PRESET_WEEK = "preset_week"  # Auto-generated weekly aggregation
 
 
 @dataclass
@@ -21,17 +22,18 @@ class Event:
     An Event is a semantic abstraction that groups related sessions
     together based on user selection or time-based presets.
     """
-    id: str                                    # Unique event ID (UUID)
-    title: str                                 # Human-readable event title
-    description: Optional[str] = None          # Detailed description
-    source: EventSource = EventSource.USER     # How the event was generated
+
+    id: str  # Unique event ID (UUID)
+    title: str  # Human-readable event title
+    description: Optional[str] = None  # Detailed description
+    source: EventSource = EventSource.USER  # How the event was generated
 
     # Commit references
     commit_hashes: List[str] = field(default_factory=list)  # Full commit hashes
 
     # Temporal metadata
-    start_timestamp: Optional[datetime] = None # Earliest commit timestamp
-    end_timestamp: Optional[datetime] = None   # Latest commit timestamp
+    start_timestamp: Optional[datetime] = None  # Earliest commit timestamp
+    end_timestamp: Optional[datetime] = None  # Latest commit timestamp
     created_at: datetime = field(default_factory=datetime.now)
     updated_at: datetime = field(default_factory=datetime.now)
 
@@ -41,9 +43,9 @@ class Event:
     session_ids: List[str] = field(default_factory=list)  # Related session IDs
 
     # Generation metadata
-    auto_generated: bool = True                # Whether auto-generated or manual
-    generation_method: Optional[str] = None    # e.g., "llm_clustering", "time_window"
-    confidence_score: Optional[float] = None   # Clustering confidence (0-1)
+    auto_generated: bool = True  # Whether auto-generated or manual
+    generation_method: Optional[str] = None  # e.g., "llm_clustering", "time_window"
+    confidence_score: Optional[float] = None  # Clustering confidence (0-1)
 
     # UI metadata (optional, for shares)
     ui_metadata: Optional[Dict[str, Any]] = None
@@ -66,11 +68,11 @@ class Event:
             "auto_generated": self.auto_generated,
             "generation_method": self.generation_method,
             "confidence_score": self.confidence_score,
-            "ui_metadata": self.ui_metadata
+            "ui_metadata": self.ui_metadata,
         }
 
     @classmethod
-    def from_dict(cls, data: Dict[str, Any]) -> 'Event':
+    def from_dict(cls, data: Dict[str, Any]) -> "Event":
         """Create Event from dict."""
         # Handle legacy data with event_type/status
         source_value = data.get("source") or data.get("event_type", "user")
@@ -83,8 +85,14 @@ class Event:
             description=data.get("description"),
             source=EventSource(source_value),
             commit_hashes=data.get("commit_hashes", []),
-            start_timestamp=datetime.fromisoformat(data["start_timestamp"]) if data.get("start_timestamp") else None,
-            end_timestamp=datetime.fromisoformat(data["end_timestamp"]) if data.get("end_timestamp") else None,
+            start_timestamp=(
+                datetime.fromisoformat(data["start_timestamp"])
+                if data.get("start_timestamp")
+                else None
+            ),
+            end_timestamp=(
+                datetime.fromisoformat(data["end_timestamp"]) if data.get("end_timestamp") else None
+            ),
             created_at=datetime.fromisoformat(data["created_at"]),
             updated_at=datetime.fromisoformat(data["updated_at"]),
             tags=data.get("tags", []),
@@ -93,13 +101,14 @@ class Event:
             auto_generated=data.get("auto_generated", True),
             generation_method=data.get("generation_method"),
             confidence_score=data.get("confidence_score"),
-            ui_metadata=data.get("ui_metadata")
+            ui_metadata=data.get("ui_metadata"),
         )
 
 
 @dataclass
 class EventCollection:
     """Container for all events in a project."""
+
     version: int = 1  # Schema version for migration
     events: List[Event] = field(default_factory=list)
     metadata: Dict[str, Any] = field(default_factory=dict)
@@ -117,14 +126,14 @@ class EventCollection:
         return {
             "version": self.version,
             "metadata": self.metadata,
-            "events": [e.to_dict() for e in self.events]
+            "events": [e.to_dict() for e in self.events],
         }
 
     @classmethod
-    def from_dict(cls, data: Dict[str, Any]) -> 'EventCollection':
+    def from_dict(cls, data: Dict[str, Any]) -> "EventCollection":
         """Create EventCollection from dict."""
         return cls(
             version=data.get("version", 1),
             events=[Event.from_dict(e) for e in data.get("events", [])],
-            metadata=data.get("metadata", {})
+            metadata=data.get("metadata", {}),
         )

realign/prompts/__init__.py

@@ -11,11 +11,11 @@ from .presets import (
 )
 
 __all__ = [
-    'PromptPreset',
-    'get_all_presets',
-    'get_preset_by_id',
-    'get_preset_by_index',
-    'load_custom_presets',
-    'display_preset_menu',
-    'prompt_for_custom_instructions',
+    "PromptPreset",
+    "get_all_presets",
+    "get_preset_by_id",
+    "get_preset_by_index",
+    "load_custom_presets",
+    "display_preset_menu",
+    "prompt_for_custom_instructions",
 ]

realign/prompts/presets.py

@@ -13,7 +13,7 @@ import yaml
 
 from ..logging_config import setup_logger
 
-logger = setup_logger('realign.prompts.presets', 'presets.log')
+logger = setup_logger("realign.prompts.presets", "presets.log")
 
 
 @dataclass
@@ -139,29 +139,31 @@ def load_custom_presets() -> List[PromptPreset]:
     Returns:
         List of custom PromptPreset objects
     """
-    config_path = Path.home() / '.aline' / 'prompt_presets.yaml'
+    config_path = Path.home() / ".aline" / "prompt_presets.yaml"
 
     if not config_path.exists():
         logger.debug(f"No custom presets file found at {config_path}")
         return []
 
     try:
-        with open(config_path, 'r', encoding='utf-8') as f:
+        with open(config_path, "r", encoding="utf-8") as f:
             config = yaml.safe_load(f)
 
-        if not config or 'custom_presets' not in config:
+        if not config or "custom_presets" not in config:
             logger.warning("Custom presets file exists but has no 'custom_presets' key")
             return []
 
         custom_presets = []
-        for preset_data in config['custom_presets']:
+        for preset_data in config["custom_presets"]:
             try:
                 preset = PromptPreset(
-                    id=preset_data['id'],
-                    name=preset_data['name'],
-                    description=preset_data['description'],
-                    allow_custom_instructions=preset_data.get('allow_custom_instructions', True),
-                    custom_instructions_placeholder=preset_data.get('custom_instructions_placeholder', ''),
+                    id=preset_data["id"],
+                    name=preset_data["name"],
+                    description=preset_data["description"],
+                    allow_custom_instructions=preset_data.get("allow_custom_instructions", True),
+                    custom_instructions_placeholder=preset_data.get(
+                        "custom_instructions_placeholder", ""
+                    ),
                 )
                 custom_presets.append(preset)
                 logger.debug(f"Loaded custom preset: {preset.id}")
@@ -231,7 +233,9 @@ def prompt_for_custom_instructions(preset: PromptPreset) -> str:
     instructions = input("Instructions: ").strip()
 
     if instructions:
-        print(f"\n✓ Custom instructions added: {instructions[:50]}{'...' if len(instructions) > 50 else ''}")
+        print(
+            f"\n✓ Custom instructions added: {instructions[:50]}{'...' if len(instructions) > 50 else ''}"
+        )
     else:
         print("\n✓ No custom instructions")