indent 0.1.20__py3-none-any.whl → 0.1.28__py3-none-any.whl

exponent/__init__.py

@@ -28,7 +28,7 @@ version_tuple: VERSION_TUPLE
 commit_id: COMMIT_ID
 __commit_id__: COMMIT_ID
 
-__version__ = version = '0.1.20'
-__version_tuple__ = version_tuple = (0, 1, 20)
+__version__ = version = '0.1.28'
+__version_tuple__ = version_tuple = (0, 1, 28)
 
 __commit_id__ = commit_id = None

exponent/commands/cloud_commands.py

@@ -23,7 +23,6 @@ from exponent.core.graphql.client import GraphQLClient
 from exponent.core.graphql.mutations import (
     CREATE_CLOUD_CHAT_FROM_REPOSITORY_MUTATION,
     ENABLE_CLOUD_REPOSITORY_MUTATION,
-    INCREMENTAL_BUILD_CLOUD_REPOSITORY_MUTATION,
     REBUILD_CLOUD_REPOSITORY_MUTATION,
     START_CHAT_TURN_MUTATION,
 )
@@ -62,32 +61,6 @@ async def enable_cloud_repository(
     return cast(dict[str, Any], result["enableCloudRepository"])
 
 
-async def incremental_build_cloud_repository(
-    api_key: str,
-    base_api_url: str,
-    base_ws_url: str,
-    org_name: str,
-    repo_name: str,
-) -> dict[str, Any]:
-    graphql_client = GraphQLClient(
-        api_key=api_key, base_api_url=base_api_url, base_ws_url=base_ws_url
-    )
-
-    variables = {
-        "orgName": org_name,
-        "repoName": repo_name,
-    }
-
-    result = await graphql_client.execute(
-        INCREMENTAL_BUILD_CLOUD_REPOSITORY_MUTATION,
-        variables,
-        "IncrementalBuildCloudRepository",
-        timeout=120,
-    )
-
-    return cast(dict[str, Any], result["incrementalBuildCloudRepository"])
-
-
 async def rebuild_cloud_repository(
     api_key: str,
     base_api_url: str,
@@ -138,6 +111,7 @@ async def create_cloud_chat_from_repository(
     base_api_url: str,
     base_ws_url: str,
     repository_id: str,
+    provider: str | None = None,
 ) -> dict[str, Any]:
     graphql_client = GraphQLClient(
         api_key=api_key, base_api_url=base_api_url, base_ws_url=base_ws_url
@@ -145,6 +119,7 @@ async def create_cloud_chat_from_repository(
 
     variables = {
         "repositoryId": repository_id,
+        "provider": provider,
     }
 
     result = await graphql_client.execute(
@@ -243,63 +218,6 @@ def enable_repo(
         sys.exit(1)
 
 
-@cloud_cli.command(hidden=True)
-@click.option(
-    "--org-name",
-    help="GitHub organization name",
-    required=True,
-)
-@click.option(
-    "--repo-name",
-    help="GitHub repository name",
-    required=True,
-)
-@use_settings
-def incremental_build(
-    settings: Settings,
-    org_name: str,
-    repo_name: str,
-) -> None:
-    """Test utility for incremental build of cloud repository."""
-    check_exponent_version_and_upgrade(settings)
-
-    if not settings.api_key:
-        redirect_to_login(settings)
-        return
-
-    loop = asyncio.get_event_loop()
-
-    api_key = settings.api_key
-    base_api_url = settings.get_base_api_url()
-    base_ws_url = settings.get_base_ws_url()
-
-    try:
-        result = loop.run_until_complete(
-            incremental_build_cloud_repository(
-                api_key, base_api_url, base_ws_url, org_name, repo_name
-            )
-        )
-
-        if result["__typename"] == "ContainerImage":
-            click.secho(
-                f"✓ Successfully triggered incremental build for {org_name}/{repo_name}",
-                fg="green",
-            )
-            click.echo(f"  Build ref: {result.get('buildRef', 'N/A')}")
-            click.echo(f"  Created at: {result.get('createdAt', 'N/A')}")
-            click.echo(f"  Updated at: {result.get('updatedAt', 'N/A')}")
-        else:
-            click.secho(
-                f"✗ Failed to trigger incremental build: {result.get('message', 'Unknown error')}",
-                fg="red",
-            )
-            click.echo(f"  Error type: {result['__typename']}")
-
-    except Exception as e:
-        click.secho(f"✗ Error triggering incremental build: {e!s}", fg="red")
-        sys.exit(1)
-
-
 @cloud_cli.command(hidden=True)
 @click.option(
     "--org-name",
@@ -654,9 +572,6 @@ def cloud(
         create_cloud_chat(api_key, base_api_url, base_ws_url, cloud_config_id)
     )
 
-    if chat_uuid is None:
-        sys.exit(1)
-
     loop.run_until_complete(
         start_chat_turn(api_key, base_api_url, base_ws_url, chat_uuid, prompt)
     )

exponent/commands/common.py

@@ -207,11 +207,14 @@ def run_until_complete(coro: Coroutine[Any, Any, Any]) -> Any:
     try:
         loop.run_until_complete(task)
     except KeyboardInterrupt:
+        click.echo("\nReceived interrupt signal, shutting down gracefully...")
         task.cancel()
         try:
             loop.run_until_complete(task)
         except asyncio.CancelledError:
             pass
+        click.secho("Exited due to keyboard interrupt", fg="yellow")
+        sys.exit(130)  # Standard exit code for SIGINT (128 + 2)
     except ExponentError as e:
         try:
             settings = get_settings()

exponent/core/config.py

@@ -162,7 +162,7 @@ def get_settings(use_prod: bool = False, use_staging: bool = False) -> Settings:
         base_ws_url = "ws://localhost:8000"
         environment = Environment.development
     elif use_staging:
-        base_url = "https://staging.indent.com"
+        base_url = "https://app.staging.indent.com"
         base_api_url = "https://staging-api.indent.com"
         base_ws_url = "wss://ws-staging-api.indent.com"
         environment = Environment.staging

exponent/core/container_build/types.py

@@ -0,0 +1,25 @@
+"""Types for container build log streaming."""
+
+from typing import Literal
+
+import msgspec
+
+
+class BuildLogOutput(msgspec.Struct, tag="container_build_log_output"):
+    """A single log line from a container build."""
+
+    container_image_uuid: str
+    data: str
+    timestamp: float
+    level: Literal["info", "error", "warning"] = "info"
+
+
+class BuildLogStatus(msgspec.Struct, tag="container_build_log_status"):
+    """Status update for a container build."""
+
+    container_image_uuid: str
+    status: Literal["started", "completed", "failed"]
+    message: str | None = None
+
+
+BuildLogMessage = BuildLogOutput | BuildLogStatus

exponent/core/graphql/mutations.py

@@ -76,8 +76,8 @@ mutation CreateCloudChat($configId: String!) {
 
 
 CREATE_CLOUD_CHAT_FROM_REPOSITORY_MUTATION = """
-mutation CreateCloudChatFromRepository($repositoryId: String!) {
-  createCloudChat(repositoryId: $repositoryId) {
+mutation CreateCloudChatFromRepository($repositoryId: String!, $provider: SandboxProvider) {
+  createCloudChat(repositoryId: $repositoryId, provider: $provider) {
     __typename
     ...on Chat {
       chatUuid
@@ -131,35 +131,6 @@ mutation EnableCloudRepository($orgName: String!, $repoName: String!) {
 """
 
 
-INCREMENTAL_BUILD_CLOUD_REPOSITORY_MUTATION = """
-mutation IncrementalBuildCloudRepository($orgName: String!, $repoName: String!) {
-  incrementalBuildCloudRepository(orgName: $orgName, repoName: $repoName) {
-    __typename
-    ...on ContainerImage {
-      buildRef
-      createdAt
-      updatedAt
-    }
-    ...on UnauthenticatedError {
-      message
-    }
-    ...on CloudConfigNotFoundError {
-      message
-    }
-    ...on GithubConfigNotFoundError {
-      message
-    }
-    ...on CloudSessionError {
-      message
-    }
-    ...on Error {
-      message
-    }
-  }
-}
-"""
-
-
 REBUILD_CLOUD_REPOSITORY_MUTATION = """
 mutation RebuildCloudRepository($orgName: String!, $repoName: String!) {
   rebuildCloudRepository(orgName: $orgName, repoName: $repoName) {

exponent/core/graphql/queries.py

@@ -28,7 +28,6 @@ EVENTS_FOR_CHAT_QUERY: str = """query EventsForChat($chatUuid: UUID!) {
         ... on UserEvent {
           uuid
           parentUuid
-          chatId
           isSidechain
           version
           createdAt
@@ -70,7 +69,6 @@ EVENTS_FOR_CHAT_QUERY: str = """query EventsForChat($chatUuid: UUID!) {
         ... on AssistantEvent {
           uuid
           parentUuid
-          chatId
           isSidechain
           version
           createdAt
@@ -98,7 +96,6 @@ EVENTS_FOR_CHAT_QUERY: str = """query EventsForChat($chatUuid: UUID!) {
         ... on SystemEvent {
           uuid
           parentUuid
-          chatId
           isSidechain
           version
           createdAt

exponent/core/remote_execution/cli_rpc_types.py

@@ -66,14 +66,28 @@ class FileMetadata(msgspec.Struct):
     file_mode: str
 
 
+class ReadToolArtifactResult(ToolResult, tag=READ_TOOL_ARTIFACT_NAME):
+    s3_uri: str
+    file_path: str
+    media_type: str
+
+    def to_text(self) -> str:
+        return f"[Image artifact uploaded to {self.s3_uri}]"
+
+
 class ReadToolResult(ToolResult, tag=READ_TOOL_NAME):
-    content: str
-    num_lines: int
-    start_line: int
-    total_lines: int
+    content: str | None = None
+    num_lines: int | None = None
+    start_line: int | None = None
+    total_lines: int | None = None
     metadata: FileMetadata | None = None
+    artifact: ReadToolArtifactResult | None = None
 
     def to_text(self) -> str:
+        if self.artifact:
+            return self.artifact.to_text()
+        assert self.content is not None
+        assert self.start_line is not None
         lines = self.content.splitlines()
         lines = [
             f"{str(i).rjust(6)}→{line}"
@@ -82,15 +96,6 @@ class ReadToolResult(ToolResult, tag=READ_TOOL_NAME):
         return "\n".join(lines)
 
 
-class ReadToolArtifactResult(ToolResult, tag=READ_TOOL_ARTIFACT_NAME):
-    s3_uri: str
-    file_path: str
-    media_type: str
-
-    def to_text(self) -> str:
-        return f"[Image artifact uploaded to {self.s3_uri}]"
-
-
 LIST_TOOL_NAME = "ls"
 
 
@@ -200,6 +205,42 @@ class BashToolResult(ToolResult, tag=BASH_TOOL_NAME):
     stopped_by_user: bool
 
 
+DOWNLOAD_ARTIFACT_TOOL_NAME = "download_artifact"
+
+
+class DownloadArtifactToolInput(ToolInput, tag=DOWNLOAD_ARTIFACT_TOOL_NAME):
+    presigned_url: str
+    file_path: str
+    artifact_id: str
+    overwrite: bool = True
+
+
+class DownloadArtifactToolResult(ToolResult, tag=DOWNLOAD_ARTIFACT_TOOL_NAME):
+    file_path: str
+    artifact_id: str
+    file_size_bytes: int
+    content_preview: str | None = None
+    num_lines: int | None = None
+    total_lines: int | None = None
+    truncated: bool = False
+
+
+UPLOAD_ARTIFACT_TOOL_NAME = "upload_artifact"
+
+
+class UploadArtifactToolInput(ToolInput, tag=UPLOAD_ARTIFACT_TOOL_NAME):
+    presigned_url: str
+    file_path: str
+    artifact_id: str
+    content_type: str
+
+
+class UploadArtifactToolResult(ToolResult, tag=UPLOAD_ARTIFACT_TOOL_NAME):
+    artifact_id: str
+    file_size_bytes: int
+    content_type: str
+
+
 class HttpRequest(msgspec.Struct, tag="http_fetch_cli"):
     url: str
     method: str = "GET"
@@ -223,6 +264,8 @@ ToolInputType = (
     | GrepToolInput
     | EditToolInput
     | BashToolInput
+    | DownloadArtifactToolInput
+    | UploadArtifactToolInput
 )
 PartialToolResultType = PartialBashToolResult
 
@@ -235,6 +278,8 @@ ToolResultType = (
     | GrepToolResult
     | EditToolResult
     | BashToolResult
+    | DownloadArtifactToolResult
+    | UploadArtifactToolResult
     | ErrorToolResult
 )
 
@@ -273,6 +318,10 @@ class TerminateResponse(msgspec.Struct, tag="terminate"):
     pass
 
 
+class TimeoutResponse(msgspec.Struct, tag="timeout"):
+    pass
+
+
 class BatchToolExecutionResponse(msgspec.Struct, tag="batch_tool_execution"):
     tool_results: list[ToolResultType]
 
@@ -295,6 +344,80 @@ class GenerateUploadUrlResponse(msgspec.Struct, tag="generate_upload_url"):
     s3_uri: str
 
 
+# Terminal session management
+class StartTerminalRequest(msgspec.Struct, tag="start_terminal"):
+    """Start a new terminal session with PTY"""
+
+    session_id: str
+    command: list[str] | None = None  # None = default shell, or specific command
+    cols: int = 80
+    rows: int = 24
+    env: dict[str, str] | None = None  # Additional environment variables
+
+
+class StartTerminalResponse(msgspec.Struct, tag="start_terminal"):
+    """Response after starting terminal"""
+
+    session_id: str
+    success: bool
+    error_message: str | None = None
+
+
+# Terminal input (user typing)
+class TerminalInputRequest(msgspec.Struct, tag="terminal_input"):
+    """Send user input to terminal"""
+
+    session_id: str
+    data: str  # Raw input data from user
+
+
+class TerminalInputResponse(msgspec.Struct, tag="terminal_input"):
+    """Acknowledge input received"""
+
+    session_id: str
+    success: bool
+    error_message: str | None = None
+
+
+# Terminal resize
+class TerminalResizeRequest(msgspec.Struct, tag="terminal_resize"):
+    """Resize terminal dimensions"""
+
+    session_id: str
+    cols: int
+    rows: int
+
+
+class TerminalResizeResponse(msgspec.Struct, tag="terminal_resize"):
+    """Acknowledge resize"""
+
+    session_id: str
+    success: bool
+    error_message: str | None = None
+
+
+# Terminal stop
+class StopTerminalRequest(msgspec.Struct, tag="stop_terminal"):
+    """Stop a terminal session"""
+
+    session_id: str
+
+
+class StopTerminalResponse(msgspec.Struct, tag="stop_terminal"):
+    """Acknowledge terminal stopped"""
+
+    session_id: str
+    success: bool
+    error_message: str | None = None
+
+
+class StreamingCodeExecutionRequest(msgspec.Struct, tag="streaming_code_execution"):
+    correlation_id: str
+    language: str  # "python" or "shell"
+    content: str
+    timeout: int
+
+
 class CliRpcRequest(msgspec.Struct):
     request_id: str
     request: (
@@ -306,6 +429,11 @@ class CliRpcRequest(msgspec.Struct):
         | SwitchCLIChatRequest
         | KeepAliveCliChatRequest
         | GenerateUploadUrlRequest
+        | StartTerminalRequest
+        | TerminalInputRequest
+        | TerminalResizeRequest
+        | StopTerminalRequest
+        | StreamingCodeExecutionRequest
     )
 
 
@@ -317,6 +445,40 @@ class ErrorResponse(msgspec.Struct, tag="error"):
     error_message: str
 
 
+class StreamingCodeExecutionResponseChunk(
+    msgspec.Struct, tag="streaming_code_execution_chunk"
+):
+    correlation_id: str
+    content: str
+    truncated: bool = False
+
+    def add(
+        self, new_chunk: "StreamingCodeExecutionResponseChunk"
+    ) -> "StreamingCodeExecutionResponseChunk":
+        """Aggregates content of this and a new chunk."""
+        assert self.correlation_id == new_chunk.correlation_id
+        return StreamingCodeExecutionResponseChunk(
+            correlation_id=self.correlation_id,
+            content=self.content + new_chunk.content,
+            truncated=self.truncated or new_chunk.truncated,
+        )
+
+
+class StreamingCodeExecutionResponse(msgspec.Struct, tag="streaming_code_execution"):
+    correlation_id: str
+    content: str
+    truncated: bool = False
+    # Only present for shell code execution
+    cancelled_for_timeout: bool = False
+    exit_code: int | None = None
+    halted: bool = False
+
+
+class StreamingErrorResponse(msgspec.Struct, tag="streaming_error"):
+    correlation_id: str
+    error_message: str
+
+
 class CliRpcResponse(msgspec.Struct):
     request_id: str
     response: (
@@ -324,9 +486,17 @@ class CliRpcResponse(msgspec.Struct):
         | GetAllFilesResponse
         | ErrorResponse
         | TerminateResponse
+        | TimeoutResponse
         | BatchToolExecutionResponse
         | HttpResponse
         | SwitchCLIChatResponse
         | KeepAliveCliChatResponse
         | GenerateUploadUrlResponse
+        | StartTerminalResponse
+        | TerminalInputResponse
+        | TerminalResizeResponse
+        | StopTerminalResponse
+        | StreamingCodeExecutionResponseChunk
+        | StreamingCodeExecutionResponse
+        | StreamingErrorResponse
     )