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

exponent/core/remote_execution/tool_execution.py

@@ -12,6 +12,8 @@ from exponent.core.remote_execution import files
 from exponent.core.remote_execution.cli_rpc_types import (
     BashToolInput,
     BashToolResult,
+    DownloadArtifactToolInput,
+    DownloadArtifactToolResult,
     EditToolInput,
     EditToolResult,
     ErrorToolResult,
@@ -26,21 +28,27 @@ from exponent.core.remote_execution.cli_rpc_types import (
     ReadToolResult,
     ToolInputType,
     ToolResultType,
+    UploadArtifactToolInput,
+    UploadArtifactToolResult,
     WriteToolInput,
     WriteToolResult,
 )
 
 if TYPE_CHECKING:
     from exponent.core.remote_execution.client import RemoteExecutionClient
+import urllib.request
+
+import aiohttp
+
+from exponent.core.remote_execution.cli_rpc_types import (
+    StreamingCodeExecutionRequest,
+    StreamingCodeExecutionResponse,
+)
 from exponent.core.remote_execution.code_execution import (
     execute_code_streaming,
 )
 from exponent.core.remote_execution.file_write import execute_full_file_rewrite
 from exponent.core.remote_execution.truncation import truncate_tool_result
-from exponent.core.remote_execution.types import (
-    StreamingCodeExecutionRequest,
-    StreamingCodeExecutionResponse,
-)
 from exponent.core.remote_execution.utils import (
     assert_unreachable,
     safe_get_file_metadata,
@@ -67,6 +75,10 @@ async def execute_tool(
         return await execute_grep_files(tool_input, working_directory)
     elif isinstance(tool_input, EditToolInput):
         return await execute_edit_file(tool_input, working_directory)
+    elif isinstance(tool_input, DownloadArtifactToolInput):
+        return await execute_download_artifact(tool_input, working_directory)
+    elif isinstance(tool_input, UploadArtifactToolInput):
+        return await execute_upload_artifact(tool_input, working_directory)
     elif isinstance(tool_input, BashToolInput):
         raise ValueError("Bash tool input should be handled by execute_bash_tool")
     else:
@@ -90,7 +102,7 @@ async def execute_read_file(  # noqa: PLR0911, PLR0915
     tool_input: ReadToolInput,
     working_directory: str,
     upload_client: "RemoteExecutionClient | None" = None,
-) -> ReadToolResult | ReadToolArtifactResult | ErrorToolResult:
+) -> ReadToolResult | ErrorToolResult:
     # Validate absolute path requirement
     if not tool_input.file_path.startswith("/"):
         return ErrorToolResult(
@@ -110,8 +122,6 @@ async def execute_read_file(  # noqa: PLR0911, PLR0915
     is_image, media_type = is_image_file(tool_input.file_path)
     if is_image and media_type and upload_client is not None:
         try:
-            import urllib.request
-
             file_name = Path(tool_input.file_path).name
             s3_key = f"images/{uuid.uuid4()}/{file_name}"
 
@@ -136,10 +146,12 @@ async def execute_read_file(  # noqa: PLR0911, PLR0915
                 if status != 200:
                     raise RuntimeError(f"Upload failed with status {status}")
 
-            return ReadToolArtifactResult(
-                s3_uri=upload_response.s3_uri,
-                file_path=tool_input.file_path,
-                media_type=media_type,
+            return ReadToolResult(
+                artifact=ReadToolArtifactResult(
+                    s3_uri=upload_response.s3_uri,
+                    file_path=tool_input.file_path,
+                    media_type=media_type,
+                )
             )
         except Exception as e:
             return ErrorToolResult(error_message=f"Failed to upload image to S3: {e!s}")
@@ -453,3 +465,131 @@ async def execute_bash_tool(
         timed_out=result.cancelled_for_timeout,
         stopped_by_user=result.halted,
     )
+
+
+async def execute_download_artifact(
+    tool_input: DownloadArtifactToolInput, working_directory: str
+) -> DownloadArtifactToolResult | ErrorToolResult:
+    """Download an artifact from S3 using a pre-signed URL."""
+
+    # Validate absolute path
+    if not tool_input.file_path.startswith("/"):
+        return ErrorToolResult(
+            error_message=f"File path must be absolute, got relative path: {tool_input.file_path}"
+        )
+
+    # Check if file exists and overwrite is False
+    file_path = Path(tool_input.file_path)
+    if file_path.exists() and not tool_input.overwrite:
+        return ErrorToolResult(
+            error_message=f"File already exists: {tool_input.file_path}. Set overwrite=True to replace it."
+        )
+
+    try:
+        # Download from pre-signed URL
+        async with aiohttp.ClientSession() as session:
+            async with session.get(tool_input.presigned_url) as response:
+                if response.status != 200:
+                    error_text = await response.text()
+                    return ErrorToolResult(
+                        error_message=f"Failed to download artifact: HTTP {response.status} - {error_text}"
+                    )
+
+                # Create parent directory if needed
+                file_path.parent.mkdir(parents=True, exist_ok=True)
+
+                # Write file
+                content = await response.read()
+                file_path.write_bytes(content)
+
+        file_size = len(content)
+
+        # Try to generate content preview for text files
+        # Attempt to decode as UTF-8 to determine if it's a text file
+        content_preview = None
+        num_lines = None
+        total_lines = None
+        truncated = False
+
+        try:
+            text_content = content.decode("utf-8")
+            lines = text_content.splitlines()
+            total_lines = len(lines)
+
+            # Show first 50 lines
+            preview_limit = 50
+            if len(lines) > preview_limit:
+                preview_lines = lines[:preview_limit]
+                truncated = True
+                num_lines = preview_limit
+            else:
+                preview_lines = lines
+                num_lines = len(lines)
+
+            content_preview = "\n".join(preview_lines)
+        except UnicodeDecodeError:
+            # Binary file, skip preview
+            pass
+
+        return DownloadArtifactToolResult(
+            file_path=tool_input.file_path,
+            artifact_id=tool_input.artifact_id,
+            file_size_bytes=file_size,
+            content_preview=content_preview,
+            num_lines=num_lines,
+            total_lines=total_lines,
+            truncated=truncated,
+        )
+
+    except Exception as e:
+        logger.exception("Failed to download artifact")
+        return ErrorToolResult(error_message=f"Failed to download artifact: {e!s}")
+
+
+async def execute_upload_artifact(
+    tool_input: UploadArtifactToolInput, working_directory: str
+) -> UploadArtifactToolResult | ErrorToolResult:
+    """Upload an artifact to S3 using a pre-signed URL."""
+
+    # Validate absolute path
+    if not tool_input.file_path.startswith("/"):
+        return ErrorToolResult(
+            error_message=f"File path must be absolute, got relative path: {tool_input.file_path}"
+        )
+
+    # Check if file exists
+    file_path = Path(tool_input.file_path)
+    if not file_path.exists():
+        return ErrorToolResult(error_message=f"File not found: {tool_input.file_path}")
+
+    if not file_path.is_file():
+        return ErrorToolResult(
+            error_message=f"Path is not a file: {tool_input.file_path}"
+        )
+
+    try:
+        # Read file
+        content = file_path.read_bytes()
+        file_size = len(content)
+
+        # Upload to pre-signed URL
+        async with aiohttp.ClientSession() as session:
+            headers = {"Content-Type": tool_input.content_type}
+            async with session.put(
+                tool_input.presigned_url, data=content, headers=headers
+            ) as response:
+                if response.status not in (200, 204):
+                    error_text = await response.text()
+                    return ErrorToolResult(
+                        error_message=f"Failed to upload artifact: HTTP {response.status} - {error_text}"
+                    )
+
+        return UploadArtifactToolResult(
+            artifact_id=tool_input.artifact_id,
+            file_size_bytes=file_size,
+            content_type=tool_input.content_type,
+        )
+
+    except Exception as e:
+        logger.exception("Failed to upload artifact")
+        return ErrorToolResult(error_message=f"Failed to upload artifact: {e!s}")

exponent/core/remote_execution/truncation.py

@@ -11,14 +11,13 @@ from exponent.core.remote_execution.cli_rpc_types import (
     GlobToolResult,
     GrepToolResult,
     ListToolResult,
-    ReadToolArtifactResult,
     ReadToolResult,
     ToolResult,
     WriteToolResult,
 )
 from exponent.core.remote_execution.utils import truncate_output
 
-DEFAULT_CHARACTER_LIMIT = 90_000
+DEFAULT_CHARACTER_LIMIT = 50_000
 DEFAULT_LIST_ITEM_LIMIT = 1000
 DEFAULT_LIST_PREVIEW_ITEMS = 10
 
@@ -273,7 +272,6 @@ class StringListTruncation(TruncationStrategy):
 
 TRUNCATION_REGISTRY: dict[type[ToolResult], TruncationStrategy] = {
     ReadToolResult: StringFieldTruncation("content"),
-    ReadToolArtifactResult: NoOpTruncation(),
     WriteToolResult: StringFieldTruncation("message"),
     BashToolResult: TailTruncation("shell_output"),
     GrepToolResult: StringListTruncation("matches"),

exponent/core/remote_execution/types.py

@@ -50,6 +50,7 @@ class PrReviewWorkflowInput(BaseModel):
 
 
 class SlackWorkflowInput(BaseModel):
+    discriminator: Literal["slack_workflow"] = "slack_workflow"
     channel_id: str
     thread_ts: str
     slack_url: str | None = None
@@ -57,13 +58,34 @@ class SlackWorkflowInput(BaseModel):
     message_ts: str | None = None
 
 
+class SlackPlanApprovalWorkflowInput(BaseModel):
+    discriminator: Literal["slack_plan_approval"] = "slack_plan_approval"
+    channel_id: str
+    thread_ts: str
+    slack_url: str
+    channel_name: str
+    message_ts: str
+
+
 class SentryWorkflowInput(BaseModel):
     title: str
     issue_id: str
     permalink: str
 
 
-WorkflowInput = PrReviewWorkflowInput | SlackWorkflowInput | SentryWorkflowInput
+class GenericCloudWorkflowInput(BaseModel):
+    initial_prompt: str
+    system_prompt_override: str | None = None
+    reasoning_level: str = "LOW"
+
+
+WorkflowInput = (
+    PrReviewWorkflowInput
+    | SlackWorkflowInput
+    | SentryWorkflowInput
+    | GenericCloudWorkflowInput
+    | SlackPlanApprovalWorkflowInput
+)
 
 
 class WorkflowTriggerRequest(BaseModel):
@@ -98,6 +120,14 @@ class PythonEnvInfo(BaseModel):
     provider: Literal["venv", "pyenv", "pipenv", "conda"] | None = "pyenv"
 
 
+class PortInfo(BaseModel):
+    process_name: str
+    port: int
+    protocol: str
+    pid: int | None
+    uptime_seconds: float | None
+
+
 class SystemInfo(BaseModel):
     name: str
     cwd: str
@@ -105,6 +135,7 @@ class SystemInfo(BaseModel):
     shell: str
     git: GitInfo | None
     python_env: PythonEnvInfo | None
+    port_usage: list[PortInfo] | None = None
 
 
 class HeartbeatInfo(BaseModel):

exponent/utils/version.py

@@ -148,6 +148,7 @@ def _ask_continue_without_upgrading() -> None:
     if click.confirm("Continue without upgrading?", default=False):
         click.secho("Using outdated version.", fg="red")
     else:
+        click.secho("Exiting due to outdated version", fg="red")
         sys.exit(1)
 
 

{indent-0.1.20.dist-info → indent-0.1.28.dist-info}/METADATA

@@ -1,9 +1,10 @@
 Metadata-Version: 2.4
 Name: indent
-Version: 0.1.20
+Version: 0.1.28
 Summary: Indent is an AI Pair Programmer
 Author-email: Sashank Thupukari <sashank@exponent.run>
 Requires-Python: <3.13,>=3.10
+Requires-Dist: aiohttp>=3.13.2
 Requires-Dist: anyio<5,>=4.6.0
 Requires-Dist: async-timeout<5,>=4.0.3; python_version < '3.11'
 Requires-Dist: beautifulsoup4[chardet]<5,>=4.13.4
@@ -22,6 +23,7 @@ Requires-Dist: msgspec>=0.19.0
 Requires-Dist: packaging~=24.1
 Requires-Dist: pip<26,>=25.0.1
 Requires-Dist: prompt-toolkit<4,>=3.0.36
+Requires-Dist: psutil<7,>=5.9.0
 Requires-Dist: pydantic-ai==0.0.30
 Requires-Dist: pydantic-settings<3,>=2.2.1
 Requires-Dist: pydantic[email]<3,>=2.6.4
@@ -30,7 +32,7 @@ Requires-Dist: python-ripgrep==0.0.9
 Requires-Dist: pyyaml>=6.0.2
 Requires-Dist: questionary<3,>=2.0.1
 Requires-Dist: rapidfuzz<4,>=3.9.0
-Requires-Dist: rich<14,>=13.7.1
+Requires-Dist: rich>=13.7.1
 Requires-Dist: sentry-sdk<3,>=2.1.1
 Requires-Dist: toml<0.11,>=0.10.2
 Requires-Dist: websockets~=15.0

{indent-0.1.20.dist-info → indent-0.1.28.dist-info}/RECORD

@@ -1,25 +1,27 @@
-exponent/__init__.py,sha256=P88Jfo9OvOr8LB0vHFqLUwFR7A0eE281KxEewbcCSBc,706
+exponent/__init__.py,sha256=1F4XTGwwdJozvgbsUgvu0kddraJ7P8oKbqLP8wGuYI8,706
 exponent/cli.py,sha256=QnIeDTgWaQJrRs5WESCkQpVEQiJiAO4qWgB0rYlkd78,3344
 exponent/py.typed,sha256=9XZl5avs8yHp89XP_1Fjtbeg_2rjYorCC9I0k_j-h2c,334
-exponent/commands/cloud_commands.py,sha256=TNSbKnc7VBo7VALj44CqV5tdCACJejEGmtYvc5wjza4,19080
-exponent/commands/common.py,sha256=M2KI9yKjB8fecPoDBphMa123c35-iNeaE9q4DxhkaFU,12817
+exponent/commands/cloud_commands.py,sha256=C4BH3pFudWMhtnQJ1IrJu35WyI4i20Twirwzm8qC-kc,16726
+exponent/commands/common.py,sha256=69Qh7EcAsKJeQn9eW41lFHgWbiFLXUy9yG0xah-Ro3Y,13030
 exponent/commands/config_commands.py,sha256=mmQYuyRosODgawoHWsn9xnWnV37GiQaxJjMv-_xreAU,8902
 exponent/commands/run_commands.py,sha256=xn0SJX0PPrmHu8Nh-kG-lJBhGC4qFCc04aLIzdVyFho,6389
 exponent/commands/settings.py,sha256=UwwwoCgCY5hzAFD9slOBbA9Gr1hNfoyJ2blsFDC6V8w,1559
 exponent/commands/types.py,sha256=iDJL3hdwhO1PrhsJTJBioNYSKo0CWV8Nv-ONcDaWIRs,3670
 exponent/commands/upgrade.py,sha256=JZr0sNazziuLByQHdT8GZb-lDbRG1YpHW8VB94q-r8w,803
 exponent/commands/utils.py,sha256=Z3eu3mvYwBh7J_hq17lyt7_MwMG8KcsP7AnsCgOnTNc,4638
-exponent/core/config.py,sha256=TNFLUgLnfSocRMVSav_7E4VcaNHXZ_3Mg5Lp1smP46U,5731
+exponent/core/config.py,sha256=GGEE5sXye5AX88LtmpYfKwPQlRconRILnlaziAsgS9U,5735
+exponent/core/container_build/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+exponent/core/container_build/types.py,sha256=UohT3CuO0fW9G0YMnXIephLbBPrt46j6PFjwZr_KR2Q,628
 exponent/core/graphql/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 exponent/core/graphql/client.py,sha256=SRagD3YPyoYZSO1RtfO-OXD7b5dm1NvgoL6CTbN380o,2009
 exponent/core/graphql/get_chats_query.py,sha256=9-2N1VfapXUZB3IFIKw5U_gKdmfyviJp5JSUntB_Yyk,1177
-exponent/core/graphql/mutations.py,sha256=cAAiSefyamBgy1zJEZ7LNk0rbjwawTPxMyj8eU6yRRI,3502
-exponent/core/graphql/queries.py,sha256=RYsk8bub0esspqgakilhzX07yJf2652Ey9tBZK1l_lY,3297
+exponent/core/graphql/mutations.py,sha256=i1zONxXew5DsQag-tH25IBif6Z8oYXygJmciQjMN9uE,2980
+exponent/core/graphql/queries.py,sha256=9u0PWcTYYRW_2yE7x5p5FzrIwDfeeuPSXrNwMzK5ffc,3246
 exponent/core/graphql/subscriptions.py,sha256=SQngv_nYVNJjiZ_P2k0UcLIu1pzc4vi7q7lhH89NCZM,393
 exponent/core/remote_execution/checkpoints.py,sha256=3QGYMLa8vT7XmxMYTRcGrW8kNGHwRC0AkUfULribJWg,6354
-exponent/core/remote_execution/cli_rpc_types.py,sha256=uBqmxGkhx_SjfMsUeBBKUp1gj273wtsEze0NFmVcwms,8292
-exponent/core/remote_execution/client.py,sha256=9TlYzwh1lBiOeBlhGUhSo73KQGMh6nvbm1xdfxM48OY,31349
-exponent/core/remote_execution/code_execution.py,sha256=jYPB_7dJzS9BTPLX9fKQpsFPatwjbXuaFFSxT9tDTfI,2388
+exponent/core/remote_execution/cli_rpc_types.py,sha256=fnqubTYTUCn2wIeBYkpkpmWPVr-1RqevTKu6NoPNaOA,12883
+exponent/core/remote_execution/client.py,sha256=bUl2w28c_okWndW9jEreVyYKeFNyL8h8OMZNBb4lf3g,40307
+exponent/core/remote_execution/code_execution.py,sha256=QL78v2yHMrIcbNWIczAICWRceziXgG4pzw7gvhUOLxs,3328
 exponent/core/remote_execution/default_env.py,sha256=s44A1Cz9EgYuhF17WO3ESVNSLQw57EoOLyi9k6qliIo,911
 exponent/core/remote_execution/error_info.py,sha256=Rd7OA3ps06qYejPVcOaMBB9AtftP3wqQoOfiILFASnc,1378
 exponent/core/remote_execution/exceptions.py,sha256=eT57lBnBhvh-KJ5lsKWcfgGA5-WisAxhjZx-Z6OupZY,135
@@ -27,15 +29,18 @@ exponent/core/remote_execution/file_write.py,sha256=8Sa70ANIDHGxIAq4_Uy2Qoo55K7-
 exponent/core/remote_execution/files.py,sha256=mIVjhStaEKETW6y3pCVeV8eJKNaPtroWGP_kBK1x8uA,8776
 exponent/core/remote_execution/git.py,sha256=dGjBpeoKJZsYgRwctSq29GmbsNIN9tbSA3VwBnRD0IQ,7810
 exponent/core/remote_execution/http_fetch.py,sha256=aFEyXd0S-MRfisSMuIFiEyc1AEAj9nUZ9Rj_P_YRows,2827
+exponent/core/remote_execution/port_utils.py,sha256=kWje8ikCzBXMeS7qr6NZZOzQOMoMuacgPUDYyloYgwM,2183
 exponent/core/remote_execution/session.py,sha256=jlQIdeUj0f7uOk3BgzlJtBJ_GyTIjCchBp5ApQuF2-I,3847
-exponent/core/remote_execution/system_context.py,sha256=QY1zY8_fWj3sh-fmLYtewvgxh7uTX4ITIJqlUTDkj6c,648
-exponent/core/remote_execution/tool_execution.py,sha256=LSV4vL5eCdjEWxlQF5nL9XwC7PC4c-jPhbfHnBE1pBM,15737
+exponent/core/remote_execution/system_context.py,sha256=I4RNuM60isS-529EuRrrEPPwJssNFC2TZ_7MhBTWEd0,754
+exponent/core/remote_execution/terminal_session.py,sha256=GgPP4BUeNYB6O2sAKeVdlVWewnAXxd0qIQeb3Sebgho,17845
+exponent/core/remote_execution/terminal_types.py,sha256=t4snBiTtidAEJZTvy789x-5HFqjkV9rlonjDO30PfAY,731
+exponent/core/remote_execution/tool_execution.py,sha256=wwKS7GiCsK_I0Na-Vo1xuzUR5wgbZ9bapck2iEuWmQo,20819
 exponent/core/remote_execution/tool_type_utils.py,sha256=7qi6Qd8fvHts019ZSLPbtiy17BUqgqBg3P_gdfvFf7w,1301
-exponent/core/remote_execution/truncation.py,sha256=1L0qwbWgQFRwJZeCXRVKkEhKmzoS1x2usDLPUIp-jlU,9923
-exponent/core/remote_execution/types.py,sha256=nq-7ZppzpEt3pPj3pe18102GG-EYw2I_iLkRrTJjfg8,15078
+exponent/core/remote_execution/truncation.py,sha256=0zFnmqXES2vtQCSpfXIQn5hgg6bZK4Sad_Cfh27xTZU,9849
+exponent/core/remote_execution/types.py,sha256=2tp73g6WLhL3x-5FyP9jhadcRHIswt4wfJJlEvNwlvk,15782
 exponent/core/remote_execution/utils.py,sha256=6PlBqYJ3OQwZ0dgXiIu3br04a-d-glDeDZpD0XGGPAE,14793
 exponent/core/remote_execution/languages/python_execution.py,sha256=nsX_LsXcUcHhiEHpSTjOTVNd7CxM146al0kw_iQX5OU,7724
-exponent/core/remote_execution/languages/shell_streaming.py,sha256=SUlk_dyN0FME3KerwdP-EB7eVe1_dNNn_HRdiEYnTes,7644
+exponent/core/remote_execution/languages/shell_streaming.py,sha256=MpE1XQiu18xWUGp1wD_Hb1nuPCQE-i5-_XO6FnkcNvo,7675
 exponent/core/remote_execution/languages/types.py,sha256=f7FjSRNRSga-ZaE3LddDhxCirUVjlSYMEdoskG6Pta4,314
 exponent/core/types/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 exponent/core/types/command_data.py,sha256=_HqQsnamRZeVoVaTpeO3ecVUzNBdG62WXlFy6Q7rtUM,5294
@@ -45,8 +50,8 @@ exponent/core/types/generated/strategy_info.py,sha256=LN6_ykFMszb21Qc3yw77xEKUtd
 exponent/migration-docs/login.md,sha256=KIeXy3m2nzSUgw-4PW1XzXfHael1D4Zu93CplLMb3hI,4252
 exponent/utils/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 exponent/utils/colors.py,sha256=HBkqe_ZmhJ9YiL2Fpulqek4KvLS5mwBTY4LQSM5N8SM,2762
-exponent/utils/version.py,sha256=GHZ9ET1kMyDubJZU3w2sah5Pw8XpiEakS5IOlt3wUnQ,8888
-indent-0.1.20.dist-info/METADATA,sha256=B9VbGtq-Ukm0jAsk18J05cPOitLswkeV7wVbYfQG2hA,1308
-indent-0.1.20.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
-indent-0.1.20.dist-info/entry_points.txt,sha256=q8q1t1sbl4NULGOR0OV5RmSG4KEjkpEQRU_RUXEGzcs,44
-indent-0.1.20.dist-info/RECORD,,
+exponent/utils/version.py,sha256=_7UyBC-xy7jt2McHIdqMaXPmHmAQSxtzrJJ8q0KZ_8w,8953
+indent-0.1.28.dist-info/METADATA,sha256=SGqskcDWZNEVKGf4q3bdzu-BcfcwhiG91EiOEMrvCmY,1367
+indent-0.1.28.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
+indent-0.1.28.dist-info/entry_points.txt,sha256=q8q1t1sbl4NULGOR0OV5RmSG4KEjkpEQRU_RUXEGzcs,44
+indent-0.1.28.dist-info/RECORD,,