indent 0.0.8__py3-none-any.whl

exponent/core/types/command_data.py

@@ -0,0 +1,206 @@
+from abc import ABC
+from enum import Enum
+from typing import Annotated, Any, ClassVar, Literal
+
+from pydantic import BaseModel, Field
+
+WRITE_STRATEGY_FULL_FILE_REWRITE: Literal["FULL_FILE_REWRITE"] = "FULL_FILE_REWRITE"
+DEFAULT_CODE_BLOCK_TIMEOUT = 30
+WRITE_STRATEGY_NATURAL_EDIT: Literal["NATURAL_EDIT"] = "NATURAL_EDIT"
+WRITE_STRATEGY_SEARCH_REPLACE: Literal["SEARCH_REPLACE"] = "SEARCH_REPLACE"
+WRITE_STRATEGY_UDIFF: Literal["UDIFF"] = "UDIFF"
+
+FileWriteStrategyName = Literal[
+    "FULL_FILE_REWRITE", "UDIFF", "SEARCH_REPLACE", "NATURAL_EDIT"
+]
+
+
+class CommandType(str, Enum):
+    THINKING = "thinking"
+    FILE_READ = "file_read"
+    SUMMARIZE = "summarize"
+    STEP_OUTPUT = "step_output"
+    PROTOTYPE = "prototype"
+    DB_QUERY = "db_query"
+    DB_GET_TABLE_NAMES = "db_get_table_names"
+    DB_GET_TABLE_SCHEMA = "db_get_table_schema"
+    ANSWER = "answer"
+    ASK = "ask"
+    SHELL = "shell"
+    PYTHON = "python"
+    FILE_WRITE = "file_write"
+
+
+class CommandData(BaseModel):
+    executable: ClassVar[bool]
+
+
+class FileReadCommandData(CommandData):
+    executable: ClassVar[bool] = True
+    type: Literal[CommandType.FILE_READ] = CommandType.FILE_READ
+
+    file_path: str
+    language: str
+    limit: int | None = None
+    offset: int | None = None
+
+
+class ThinkingCommandData(CommandData):
+    executable: ClassVar[bool] = False
+    type: Literal[CommandType.THINKING] = CommandType.THINKING
+
+    content: str
+    signature: str | None = None
+
+
+class PrototypeCommandData(CommandData):
+    executable: ClassVar[bool] = True
+    type: Literal[CommandType.PROTOTYPE] = CommandType.PROTOTYPE
+
+    command_name: str
+    # Structured data extracted from LLM output
+    content_json: dict[str, Any]
+    # Raw text extracted from LLM output
+    content_raw: str
+    # Rendered LLM output for frontend display
+    content_rendered: str
+
+    llm_command_name_override: str | None = None
+
+    @property
+    def llm_command_name(self) -> str:
+        return self.llm_command_name_override or self.command_name
+
+
+# deprecated, use StepOutputCommandData instead
+class SummarizeCommandData(CommandData):
+    executable: ClassVar[bool] = True
+    type: Literal[CommandType.SUMMARIZE] = CommandType.SUMMARIZE
+
+    summary: str
+
+
+class StepOutputCommandData(CommandData):
+    executable: ClassVar[bool] = True
+    type: Literal[CommandType.STEP_OUTPUT] = CommandType.STEP_OUTPUT
+
+    step_output_raw: str
+
+
+class DBQueryCommandData(CommandData):
+    def __init__(
+        self,
+        **kwargs: Any,
+    ) -> None:
+        super().__init__(**kwargs)
+
+    executable: ClassVar[bool] = True
+    type: Literal[CommandType.DB_QUERY] = CommandType.DB_QUERY
+
+    query: str
+    max_gigabytes_billed: float | None = None  # BigQuery only
+
+
+class DBGetTableNamesCommandData(CommandData):
+    executable: ClassVar[bool] = True
+    type: Literal[CommandType.DB_GET_TABLE_NAMES] = CommandType.DB_GET_TABLE_NAMES
+
+
+class DBGetTableSchemaCommandData(CommandData):
+    executable: ClassVar[bool] = True
+    type: Literal[CommandType.DB_GET_TABLE_SCHEMA] = CommandType.DB_GET_TABLE_SCHEMA
+
+    table_name: str
+
+
+class AnswerCommandData(CommandData):
+    executable: ClassVar[bool] = False
+    type: Literal[CommandType.ANSWER] = CommandType.ANSWER
+
+    answer_raw: str
+
+
+class AskCommandData(CommandData):
+    executable: ClassVar[bool] = False
+    type: Literal[CommandType.ASK] = CommandType.ASK
+
+    ask_raw: str
+
+
+class ShellCommandData(CommandData):
+    exclude_from_schema_gen: ClassVar[bool] = True
+
+    executable: ClassVar[bool] = True
+    type: Literal[CommandType.SHELL] = CommandType.SHELL
+
+    timeout: int = DEFAULT_CODE_BLOCK_TIMEOUT
+    content: str
+
+
+class PythonCommandData(CommandData):
+    exclude_from_schema_gen: ClassVar[bool] = True
+
+    executable: ClassVar[bool] = True
+    type: Literal[CommandType.PYTHON] = CommandType.PYTHON
+
+    content: str
+
+
+class EditContent(BaseModel):
+    content: str
+    original_file: str | None = None
+
+
+class NaturalEditContent(BaseModel):
+    natural_edit: str
+    intermediate_edit: str | None
+    original_file: str | None
+    new_file: str | None
+    error_content: str | None
+
+    @property
+    def is_resolved(self) -> bool:
+        return self.new_file is not None or self.error_content is not None
+
+    @property
+    def is_noop(self) -> bool:
+        return bool(
+            self.new_file is not None
+            and self.original_file is not None
+            and self.new_file == self.original_file
+        )
+
+
+class FileWriteCommandData(CommandData):
+    exclude_from_schema_gen: ClassVar[bool] = True
+
+    executable: ClassVar[bool] = True
+    type: Literal[CommandType.FILE_WRITE] = CommandType.FILE_WRITE
+
+    file_path: str
+    language: str
+    write_strategy: FileWriteStrategyName
+    write_content: NaturalEditContent | EditContent
+    content: str
+
+
+CommandDataType = Annotated[
+    FileReadCommandData
+    | ThinkingCommandData
+    | PrototypeCommandData
+    | SummarizeCommandData
+    | DBQueryCommandData
+    | DBGetTableNamesCommandData
+    | DBGetTableSchemaCommandData
+    | StepOutputCommandData
+    | AnswerCommandData
+    | AskCommandData
+    | ShellCommandData
+    | PythonCommandData
+    | FileWriteCommandData,
+    Field(discriminator="type"),
+]
+
+
+class CommandImpl(ABC):
+    command_data_type: ClassVar[type[CommandData]]

exponent/core/types/event_types.py

@@ -0,0 +1,89 @@
+from datetime import datetime
+from enum import Enum
+from typing import Generic, Protocol, TypeVar, Union
+
+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 = Union[FileWriteEvent, CodeBlockEvent, CommandEvent]

exponent/core/types/generated/strategy_info.py

@@ -0,0 +1,225 @@
+# Auto-generated, do not edit directly. Run `make generate_strategy_info` to update.
+
+import enum
+
+from pydantic import BaseModel
+
+from exponent.core.remote_execution.types import ChatMode
+
+
+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
+
+
+CHAT_MODE_DEFAULTS: dict[ChatMode, StrategyName] = {
+    ChatMode.DEFAULT: StrategyName.RAW_GPT,
+    ChatMode.CLI: StrategyName.NATURAL_EDIT_CLAUDE_3_7_XML,
+    ChatMode.CLOUD: StrategyName.NATURAL_EDIT_CLAUDE_3_7_XML,
+    ChatMode.DATABASE: StrategyName.DATABASE,
+    ChatMode.PYTHON_INTERPRETER: StrategyName.NATURAL_EDIT_CLAUDE_3_7_XML,
+    ChatMode.WORKFLOW: StrategyName.NATURAL_EDIT_CLAUDE_3_7_XML,
+    ChatMode.CODEBASE: StrategyName.READ_ONLY,
+}
+
+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/