indent 0.1.26__py3-none-any.whl

exponent/core/types/event_types.py

@@ -0,0 +1,89 @@
+from datetime import datetime
+from enum import Enum
+from typing import Generic, Protocol, TypeVar
+
+from pydantic import BaseModel, Field, JsonValue, ValidationInfo, field_validator
+
+from exponent.core.types.command_data import (
+    DEFAULT_CODE_BLOCK_TIMEOUT,
+    WRITE_STRATEGY_NATURAL_EDIT,
+    CommandDataType,
+    EditContent,
+    FileWriteStrategyName,
+    NaturalEditContent,
+)
+
+
+class FileWriteErrorType(str, Enum):
+    TERMINATION_REQUESTED = "TERMINATION_REQUESTED"
+    NO_OP = "NO_OP"
+    FAILED_APPLY = "FAILED_APPLY"
+    FAILED_GENERATION = "FAILED_GENERATION"
+    CLI_DISCONNECTED = "CLI_DISCONNECTED"
+
+
+class ExponentEvent(BaseModel):
+    chat_uuid: str
+    event_uuid: str
+    parent_uuid: str | None
+    turn_uuid: str
+
+    metadata: dict[str, JsonValue] = Field(default_factory=dict)
+
+
+class PersistedExponentEvent(ExponentEvent):
+    db_timestamp: datetime | None = None
+
+
+class CodeBlockEvent(PersistedExponentEvent):
+    language: str
+    content: str
+    timeout: int = DEFAULT_CODE_BLOCK_TIMEOUT
+    require_confirmation: bool = False
+
+
+class FileWriteEvent(PersistedExponentEvent):
+    file_path: str
+    language: str
+    write_strategy: FileWriteStrategyName
+    write_content: NaturalEditContent | EditContent
+    content: str
+    error_content: str | None
+    error_type: FileWriteErrorType | None
+    require_confirmation: bool = False
+
+    @field_validator("write_content")
+    def validate_write_content_type(
+        cls, v: NaturalEditContent | EditContent, info: ValidationInfo
+    ) -> NaturalEditContent | EditContent:
+        write_strategy = info.data.get("write_strategy")
+        if write_strategy == WRITE_STRATEGY_NATURAL_EDIT:
+            if not isinstance(v, NaturalEditContent):
+                raise ValueError(
+                    "When write_strategy is NATURAL_EDIT, write_content must be NaturalEditContent"
+                )
+        elif not isinstance(v, EditContent):
+            raise ValueError(
+                "For non-NATURAL_EDIT strategies, write_content must be EditContent"
+            )
+        return v
+
+
+T = TypeVar("T", bound=CommandDataType)
+
+
+class HoldsCommandData(Protocol, Generic[T]):
+    data: T
+
+
+class CommandEvent(PersistedExponentEvent):
+    data: CommandDataType
+    require_confirmation: bool = False
+
+
+class MultiCommandEvent(PersistedExponentEvent):
+    data: list[CommandDataType]
+    require_confirmation: bool = False
+
+
+LocalEventType = FileWriteEvent | CodeBlockEvent | CommandEvent

exponent/core/types/generated/strategy_info.py

@@ -0,0 +1,213 @@
+# Auto-generated, do not edit directly. Run `make generate_strategy_info` to update.
+
+import enum
+
+from pydantic import BaseModel
+
+
+class StrategyName(str, enum.Enum):
+    NATURAL_EDIT_CLAUDE_3_7_XML = "NATURAL_EDIT_CLAUDE_3_7_XML"
+    READ_ONLY = "READ_ONLY"
+    NATURAL_EDIT_CLAUDE_3_7_XML_WITH_STEPS = "NATURAL_EDIT_CLAUDE_3_7_XML_WITH_STEPS"
+    NATURAL_EDIT_CLAUDE_3_7_FUNCTION_CALLING = (
+        "NATURAL_EDIT_CLAUDE_3_7_FUNCTION_CALLING"
+    )
+    SEARCH_REPLACE_FUNCTION_CALLING = "SEARCH_REPLACE_FUNCTION_CALLING"
+    FULL_FILE_REWRITE = "FULL_FILE_REWRITE"
+    FUNCTION_CALLING = "FUNCTION_CALLING"
+    RAW_GPT = "RAW_GPT"
+    NATURAL_EDIT = "NATURAL_EDIT"
+    AGENT = "AGENT"
+    AGENT_STEP = "AGENT_STEP"
+    DATABASE = "DATABASE"
+    DATABASE_AGENT = "DATABASE_AGENT"
+    DATABASE_AGENT_STEP = "DATABASE_AGENT_STEP"
+    REVIEW_FILES = "REVIEW_FILES"
+    EXPLORE_CODEBASE = "EXPLORE_CODEBASE"
+    GENERATE_REVIEW = "GENERATE_REVIEW"
+    CUSTOM_STEP = "CUSTOM_STEP"
+
+
+class StrategyInfo(BaseModel):
+    strategy_name: StrategyName
+    display_name: str
+    description: str
+    disabled: bool
+    display_order: int
+    is_agentic: bool
+
+
+STRATEGY_INFO_LIST: list[StrategyInfo] = [
+    StrategyInfo(
+        strategy_name=StrategyName.NATURAL_EDIT_CLAUDE_3_7_XML,
+        display_name="Natural Edit Claude 3.7 XML",
+        description="A natural file editing strategy that uses Claude 3.7 XML.",
+        disabled=False,
+        display_order=-1,
+        is_agentic=False,
+    ),
+    StrategyInfo(
+        strategy_name=StrategyName.READ_ONLY,
+        display_name="Read Only",
+        description="A strategy that allows you to read the codebase.",
+        disabled=False,
+        display_order=-1,
+        is_agentic=False,
+    ),
+    StrategyInfo(
+        strategy_name=StrategyName.NATURAL_EDIT_CLAUDE_3_7_XML_WITH_STEPS,
+        display_name="Natural Edit Claude 3.7 XML with Steps",
+        description="A natural file editing strategy that uses Claude 3.7 XML with steps.",
+        disabled=True,
+        display_order=0,
+        is_agentic=False,
+    ),
+    StrategyInfo(
+        strategy_name=StrategyName.NATURAL_EDIT_CLAUDE_3_7_FUNCTION_CALLING,
+        display_name="Natural Edit Claude 3.7 Function Calling",
+        description="A natural file editing strategy that uses Claude 3.7 Function Calling.",
+        disabled=True,
+        display_order=0,
+        is_agentic=True,
+    ),
+    StrategyInfo(
+        strategy_name=StrategyName.SEARCH_REPLACE_FUNCTION_CALLING,
+        display_name="Search Replace Function Calling",
+        description="A search replace strategy that uses Function Calling.",
+        disabled=True,
+        display_order=0,
+        is_agentic=True,
+    ),
+    StrategyInfo(
+        strategy_name=StrategyName.FULL_FILE_REWRITE,
+        display_name="Full File Rewrites",
+        description="Rewrites the full file every time. Use this if your files are generally less than 300 lines.",
+        disabled=False,
+        display_order=2,
+        is_agentic=False,
+    ),
+    StrategyInfo(
+        strategy_name=StrategyName.FUNCTION_CALLING,
+        display_name="Function Calling",
+        description="Using native function calling.",
+        disabled=True,
+        display_order=99,
+        is_agentic=False,
+    ),
+    StrategyInfo(
+        strategy_name=StrategyName.RAW_GPT,
+        display_name="Raw GPT",
+        description="No description",
+        disabled=True,
+        display_order=99,
+        is_agentic=False,
+    ),
+    StrategyInfo(
+        strategy_name=StrategyName.NATURAL_EDIT,
+        display_name="Natural Edit",
+        description="Deprecated",
+        disabled=True,
+        display_order=99,
+        is_agentic=False,
+    ),
+    StrategyInfo(
+        strategy_name=StrategyName.AGENT,
+        display_name="Agent",
+        description="No description",
+        disabled=True,
+        display_order=99,
+        is_agentic=True,
+    ),
+    StrategyInfo(
+        strategy_name=StrategyName.AGENT_STEP,
+        display_name="Agent Step",
+        description="No description",
+        disabled=True,
+        display_order=99,
+        is_agentic=False,
+    ),
+    StrategyInfo(
+        strategy_name=StrategyName.DATABASE,
+        display_name="Database",
+        description="No description",
+        disabled=True,
+        display_order=99,
+        is_agentic=False,
+    ),
+    StrategyInfo(
+        strategy_name=StrategyName.DATABASE_AGENT,
+        display_name="Database Agent",
+        description="No description",
+        disabled=True,
+        display_order=99,
+        is_agentic=True,
+    ),
+    StrategyInfo(
+        strategy_name=StrategyName.DATABASE_AGENT_STEP,
+        display_name="Database Agent Step",
+        description="No description",
+        disabled=True,
+        display_order=99,
+        is_agentic=False,
+    ),
+    StrategyInfo(
+        strategy_name=StrategyName.REVIEW_FILES,
+        display_name="Review Files",
+        description="No description",
+        disabled=True,
+        display_order=99,
+        is_agentic=False,
+    ),
+    StrategyInfo(
+        strategy_name=StrategyName.EXPLORE_CODEBASE,
+        display_name="Explore Codebase",
+        description="No description",
+        disabled=True,
+        display_order=99,
+        is_agentic=False,
+    ),
+    StrategyInfo(
+        strategy_name=StrategyName.GENERATE_REVIEW,
+        display_name="Generate Review",
+        description="No description",
+        disabled=True,
+        display_order=99,
+        is_agentic=False,
+    ),
+    StrategyInfo(
+        strategy_name=StrategyName.CUSTOM_STEP,
+        display_name="Custom Step",
+        description="No description",
+        disabled=True,
+        display_order=99,
+        is_agentic=False,
+    ),
+]
+
+
+ENABLED_STRATEGY_INFO_LIST: list[StrategyInfo] = [
+    StrategyInfo(
+        strategy_name=StrategyName.NATURAL_EDIT_CLAUDE_3_7_XML,
+        display_name="Natural Edit Claude 3.7 XML",
+        description="A natural file editing strategy that uses Claude 3.7 XML.",
+        disabled=False,
+        display_order=-1,
+        is_agentic=False,
+    ),
+    StrategyInfo(
+        strategy_name=StrategyName.READ_ONLY,
+        display_name="Read Only",
+        description="A strategy that allows you to read the codebase.",
+        disabled=False,
+        display_order=-1,
+        is_agentic=False,
+    ),
+    StrategyInfo(
+        strategy_name=StrategyName.FULL_FILE_REWRITE,
+        display_name="Full File Rewrites",
+        description="Rewrites the full file every time. Use this if your files are generally less than 300 lines.",
+        disabled=False,
+        display_order=2,
+        is_agentic=False,
+    ),
+]

exponent/migration-docs/login.md

@@ -0,0 +1,112 @@
+# Authentication and Login System Documentation
+
+## Overview
+The authentication system in Exponent is built around API keys and GraphQL-based authentication. The system supports multiple environments (development, staging, production) with different API endpoints and authentication mechanisms.
+
+## Key Components
+
+### 1. Configuration and Settings
+**Location**: `/Users/dkzlv/Projects/exponent-wrapper/exponent/python_modules/exponent/exponent/core/config.py`
+
+The settings system manages:
+- API keys for different environments (development, staging, production)
+- Base URLs for different services
+- Configuration storage in `~/.config/exponent/config.json`
+
+### 2. Login Command
+**Location**: `/Users/dkzlv/Projects/exponent-wrapper/exponent/python_modules/exponent/exponent/commands/config_commands.py`
+
+The login command:
+- Accepts an API key via `indent login --key <API-KEY>`
+- Verifies the API key with the server
+- Stores the verified key in the config file
+
+### 3. Authentication Flow
+
+#### Initial Authentication
+1. When a command is run, the system checks for an API key
+2. If no API key is found:
+   - In SSH session: Shows message to run `indent login --key <API-KEY>`
+   - Otherwise: Redirects to `<base_url>/cli` for web-based login
+
+#### API Key Verification
+**Location**: `/Users/dkzlv/Projects/exponent-wrapper/exponent/python_modules/exponent/exponent/commands/common.py`
+
+The system verifies API keys through:
+1. GraphQL mutation `SET_LOGIN_COMPLETE_MUTATION`
+2. Verification of returned API key matching the provided key
+
+#### API Key Refresh
+**Location**: `/Users/dkzlv/Projects/exponent-wrapper/exponent/python_modules/exponent/exponent/commands/config_commands.py`
+
+Users can refresh their API key using:
+- Command: `exponent config refresh-key`
+- Uses `REFRESH_API_KEY_MUTATION` GraphQL mutation
+- Automatically updates the stored API key
+
+### 4. GraphQL Authentication
+
+#### Client Setup
+**Location**: `/Users/dkzlv/Projects/exponent-wrapper/exponent/python_modules/exponent/exponent/core/graphql/client.py`
+
+The GraphQL client:
+- Adds API key to HTTP headers for REST calls
+- Includes API key in WebSocket init payload
+- Handles both HTTP and WebSocket connections
+
+#### Key Mutations and Subscriptions
+**Location**: `/Users/dkzlv/Projects/exponent-wrapper/exponent/python_modules/exponent/exponent/core/graphql/mutations.py`
+
+Important GraphQL operations:
+```graphql
+SET_LOGIN_COMPLETE_MUTATION
+REFRESH_API_KEY_MUTATION
+AUTHENTICATED_USER_SUBSCRIPTION
+```
+
+### 5. Environment Support
+**Location**: `/Users/dkzlv/Projects/exponent-wrapper/exponent/python_modules/exponent/exponent/core/config.py`
+
+Supports multiple environments:
+- Development: `localhost:3000` (API), `localhost:8000` (WebSocket)
+- Staging: `staging.exponent.run`
+- Production: `exponent.run`
+
+## Security Considerations
+
+1. API Key Storage
+   - Keys stored in `~/.config/exponent/config.json`
+   - Separate keys for different environments
+
+2. SSL/TLS Security
+   - System checks for SSL certificates
+   - Can install certificates if missing (macOS specific)
+   - Uses `certifi` for certificate verification
+
+## Error Handling
+
+The system handles various authentication errors:
+- Invalid API keys
+- Network connection issues
+- SSL certificate problems
+- Environment-specific configuration issues
+
+## Related Files (Absolute Paths)
+
+1. `/Users/dkzlv/Projects/exponent-wrapper/exponent/python_modules/exponent/exponent/core/config.py`
+   - Core configuration and settings management
+
+2. `/Users/dkzlv/Projects/exponent-wrapper/exponent/python_modules/exponent/exponent/commands/config_commands.py`
+   - Login and API key management commands
+
+3. `/Users/dkzlv/Projects/exponent-wrapper/exponent/python_modules/exponent/exponent/commands/common.py`
+   - Authentication helpers and utilities
+
+4. `/Users/dkzlv/Projects/exponent-wrapper/exponent/python_modules/exponent/exponent/core/graphql/client.py`
+   - GraphQL client implementation
+
+5. `/Users/dkzlv/Projects/exponent-wrapper/exponent/python_modules/exponent/exponent/core/graphql/mutations.py`
+   - Authentication-related GraphQL mutations
+
+6. `/Users/dkzlv/Projects/exponent-wrapper/exponent/python_modules/exponent/exponent/core/graphql/subscriptions.py`
+   - Authentication-related GraphQL subscriptions

exponent/py.typed

@@ -0,0 +1,4 @@
+# This is a typing sentinel. It's required to make type checkers regard this as a typed library when the library is installed as a dependency.
+# since this code is installed inside the exponent_server package as well, via a dependency, we have to ensure this sentinel is in place.
+
+# further reading: https://peps.python.org/pep-0561/

exponent/utils/colors.py

@@ -0,0 +1,92 @@
+import math
+
+from colour import Color
+
+
+def relative_luminance(c: Color) -> float:
+    """Compute sRGB-based luminance, per the standard (WCAG-like) formula."""
+
+    r, g, b = c.rgb
+
+    def to_linear(channel: float) -> float:
+        return (
+            channel / 12.92
+            if channel <= 0.03928
+            else ((channel + 0.055) / 1.055) ** 2.4
+        )
+
+    R, G, B = map(to_linear, (r, g, b))
+
+    return 0.2126 * R + 0.7152 * G + 0.0722 * B
+
+
+def contrast_ratio(c1: Color, c2: Color) -> float:
+    """Compute ratio = (Llighter + 0.05) / (Ldarker + 0.05)."""
+
+    L1 = relative_luminance(c1)
+    L2 = relative_luminance(c2)
+    L_high, L_low = max(L1, L2), min(L1, L2)
+
+    return (L_high + 0.05) / (L_low + 0.05)
+
+
+def adjust_color_for_contrast(
+    base: Color, target: Color, min_contrast: float = 4.5, step: float = 0.005
+) -> Color:
+    """
+    Increments either upward or downward in HSL 'l'
+    (depending on whether target is lighter or darker than base)
+    until the desired contrast ratio is reached or we hit the boundary (0 or 1).
+    """
+
+    # Work on a copy so we don't mutate the original
+    new_color = Color(target.hex)
+
+    L_base = relative_luminance(base)
+    L_target = relative_luminance(new_color)
+
+    # Check which color is darker:
+    # if the target is darker, we'll push it darker; if it's lighter, we push it lighter.
+    h, s, l = new_color.hsl  # noqa: E741
+
+    if L_target < L_base:
+        # target color is darker => keep making it darker
+        while l >= 0.0:
+            if contrast_ratio(base, new_color) >= min_contrast:
+                return new_color
+            l -= step  # noqa: E741
+            new_color.hsl = (h, s, max(l, 0.0))
+    else:
+        # target color is lighter => keep making it lighter
+        while l <= 1.0:
+            if contrast_ratio(base, new_color) >= min_contrast:
+                return new_color
+            l += step  # noqa: E741
+            new_color.hsl = (h, s, min(l, 1.0))
+
+    # If we exhaust the channel range, just return whatever we ended with
+    return new_color
+
+
+def blend_colors_srgb(c1: Color, c2: Color, alpha: float) -> Color:
+    """
+    Blend c1 and c2 in sRGB space using alpha in [0..1],
+    returning a new Color object.
+    E.g. alpha=0.0 => c1, alpha=1.0 => c2.
+    """
+
+    # Interpolate each channel separately
+    r = (1 - alpha) * c1.red + alpha * c2.red
+    g = (1 - alpha) * c1.green + alpha * c2.green
+    b = (1 - alpha) * c1.blue + alpha * c2.blue
+
+    # Return a new Color with the blended channels
+    return Color(rgb=(r, g, b))
+
+
+def color_distance(c1: Color, c2: Color) -> float:
+    return math.sqrt(
+        math.pow(c1.red - c2.red, 2)
+        + math.pow(c1.green - c2.green, 2)
+        + math.pow(c1.blue - c2.blue, 2)
+    )