indent 0.1.17__py3-none-any.whl → 0.1.18__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.17'
-__version_tuple__ = version_tuple = (0, 1, 17)
+__version__ = version = '0.1.18'
+__version_tuple__ = version_tuple = (0, 1, 18)
 
 __commit_id__ = commit_id = None

exponent/core/remote_execution/cli_rpc_types.py

@@ -51,6 +51,7 @@ class ErrorToolResult(ToolResult, tag="error"):
 
 
 READ_TOOL_NAME = "read"
+READ_TOOL_ARTIFACT_NAME = "read_tool_artifact"
 
 
 class ReadToolInput(ToolInput, tag=READ_TOOL_NAME):
@@ -80,6 +81,15 @@ 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"
 
 
@@ -217,6 +227,7 @@ PartialToolResultType = PartialBashToolResult
 
 ToolResultType = (
     ReadToolResult
+    | ReadToolArtifactResult
     | WriteToolResult
     | ListToolResult
     | GlobToolResult
@@ -273,6 +284,16 @@ class KeepAliveCliChatResponse(msgspec.Struct, tag="keep_alive_cli_chat"):
     pass
 
 
+class GenerateUploadUrlRequest(msgspec.Struct, tag="generate_upload_url"):
+    s3_key: str
+    content_type: str
+
+
+class GenerateUploadUrlResponse(msgspec.Struct, tag="generate_upload_url"):
+    upload_url: str
+    s3_uri: str
+
+
 class CliRpcRequest(msgspec.Struct):
     request_id: str
     request: (
@@ -283,6 +304,7 @@ class CliRpcRequest(msgspec.Struct):
         | BatchToolExecutionRequest
         | SwitchCLIChatRequest
         | KeepAliveCliChatRequest
+        | GenerateUploadUrlRequest
     )
 
 
@@ -305,4 +327,5 @@ class CliRpcResponse(msgspec.Struct):
         | HttpResponse
         | SwitchCLIChatResponse
         | KeepAliveCliChatResponse
+        | GenerateUploadUrlResponse
     )

exponent/core/remote_execution/client.py

@@ -31,6 +31,8 @@ from exponent.core.remote_execution.cli_rpc_types import (
     CliRpcResponse,
     ErrorResponse,
     ErrorToolResult,
+    GenerateUploadUrlRequest,
+    GenerateUploadUrlResponse,
     GetAllFilesRequest,
     GetAllFilesResponse,
     HttpRequest,
@@ -115,6 +117,13 @@ class RemoteExecutionClient:
         # Track last request time for timeout functionality
         self._last_request_time: float | None = None
 
+        # Track pending upload URL requests
+        self._pending_upload_requests: dict[
+            str, asyncio.Future[GenerateUploadUrlResponse]
+        ] = {}
+        self._upload_request_lock = asyncio.Lock()
+        self._websocket: ClientConnection | None = None
+
     @property
     def working_directory(self) -> str:
         return self.current_session.working_directory
@@ -186,7 +195,21 @@ class RemoteExecutionClient:
         self._last_request_time = time.time()
 
         msg_data = json.loads(msg)
-        if msg_data["type"] != "request":
+        if msg_data["type"] == "result":
+            data = json.dumps(msg_data["data"])
+            try:
+                response = msgspec.json.decode(data, type=CliRpcResponse)
+                if isinstance(response.response, GenerateUploadUrlResponse):
+                    async with self._upload_request_lock:
+                        if response.request_id in self._pending_upload_requests:
+                            future = self._pending_upload_requests.pop(
+                                response.request_id
+                            )
+                            future.set_result(response.response)
+            except Exception as e:
+                logger.error(f"Error handling upload URL response: {e}")
+            return None
+        elif msg_data["type"] != "request":
             return None
 
         data = json.dumps(msg_data["data"])
@@ -445,6 +468,8 @@ class RemoteExecutionClient:
         if connection_tracker is not None:
             await connection_tracker.set_connected(True)
 
+        self._websocket = websocket
+
         beats: asyncio.Queue[HeartbeatInfo] = asyncio.Queue()
         requests: asyncio.Queue[CliRpcRequest] = asyncio.Queue()
         results: asyncio.Queue[CliRpcResponse] = asyncio.Queue()
@@ -579,6 +604,38 @@ class RemoteExecutionClient:
         logger.info(f"Heartbeat response: {connected_state}")
         return connected_state
 
+    async def request_upload_url(
+        self, s3_key: str, content_type: str
+    ) -> GenerateUploadUrlResponse:
+        if self._websocket is None:
+            raise RuntimeError("No active websocket connection")
+
+        request_id = str(uuid.uuid4())
+        request = CliRpcRequest(
+            request_id=request_id,
+            request=GenerateUploadUrlRequest(s3_key=s3_key, content_type=content_type),
+        )
+
+        future: asyncio.Future[GenerateUploadUrlResponse] = asyncio.Future()
+        async with self._upload_request_lock:
+            self._pending_upload_requests[request_id] = future
+
+        try:
+            await self._websocket.send(
+                json.dumps({"type": "request", "data": msgspec.to_builtins(request)})
+            )
+
+            response = await asyncio.wait_for(future, timeout=30)
+            return response
+        except TimeoutError:
+            async with self._upload_request_lock:
+                self._pending_upload_requests.pop(request_id, None)
+            raise RuntimeError("Timeout waiting for upload URL response")
+        except Exception as e:
+            async with self._upload_request_lock:
+                self._pending_upload_requests.pop(request_id, None)
+            raise e
+
     async def handle_request(self, request: CliRpcRequest) -> CliRpcResponse:
         # Update last request time for timeout functionality
         self._last_request_time = time.time()
@@ -593,7 +650,7 @@ class RemoteExecutionClient:
                     )
                 else:
                     raw_result = await execute_tool(  # type: ignore[assignment]
-                        request.request.tool_input, self.working_directory
+                        request.request.tool_input, self.working_directory, self
                     )
                 tool_result = truncate_result(raw_result)
                 return CliRpcResponse(
@@ -620,7 +677,9 @@ class RemoteExecutionClient:
                             )
                         )
                     else:
-                        coros.append(execute_tool(tool_input, self.working_directory))
+                        coros.append(
+                            execute_tool(tool_input, self.working_directory, self)
+                        )
 
                 results: list[ToolResultType | BaseException] = await asyncio.gather(
                     *coros, return_exceptions=True

exponent/core/remote_execution/tool_execution.py

@@ -3,6 +3,7 @@ import uuid
 from collections.abc import Callable
 from pathlib import Path
 from time import time
+from typing import TYPE_CHECKING
 
 from anyio import Path as AsyncPath
 
@@ -19,6 +20,7 @@ from exponent.core.remote_execution.cli_rpc_types import (
     GrepToolResult,
     ListToolInput,
     ListToolResult,
+    ReadToolArtifactResult,
     ReadToolInput,
     ReadToolResult,
     ToolInputType,
@@ -26,6 +28,9 @@ from exponent.core.remote_execution.cli_rpc_types import (
     WriteToolInput,
     WriteToolResult,
 )
+
+if TYPE_CHECKING:
+    from exponent.core.remote_execution.client import RemoteExecutionClient
 from exponent.core.remote_execution.code_execution import (
     execute_code_streaming,
 )
@@ -45,10 +50,12 @@ logger = logging.getLogger(__name__)
 
 
 async def execute_tool(
-    tool_input: ToolInputType, working_directory: str
+    tool_input: ToolInputType,
+    working_directory: str,
+    upload_client: "RemoteExecutionClient | None" = None,
 ) -> ToolResultType:
     if isinstance(tool_input, ReadToolInput):
-        return await execute_read_file(tool_input, working_directory)
+        return await execute_read_file(tool_input, working_directory, upload_client)
     elif isinstance(tool_input, WriteToolInput):
         return await execute_write_file(tool_input, working_directory)
     elif isinstance(tool_input, ListToolInput):
@@ -69,9 +76,20 @@ def truncate_result[T: ToolResultType](tool_result: T) -> T:
     return truncate_tool_result(tool_result)
 
 
+def is_image_file(file_path: str) -> tuple[bool, str | None]:
+    ext = Path(file_path).suffix.lower()
+    if ext == ".png":
+        return (True, "image/png")
+    elif ext in [".jpg", ".jpeg"]:
+        return (True, "image/jpeg")
+    return (False, None)
+
+
 async def execute_read_file(  # noqa: PLR0911, PLR0915
-    tool_input: ReadToolInput, working_directory: str
-) -> ReadToolResult | ErrorToolResult:
+    tool_input: ReadToolInput,
+    working_directory: str,
+    upload_client: "RemoteExecutionClient | None" = None,
+) -> ReadToolResult | ReadToolArtifactResult | ErrorToolResult:
     # Validate absolute path requirement
     if not tool_input.file_path.startswith("/"):
         return ErrorToolResult(
@@ -87,6 +105,39 @@ async def execute_read_file(  # noqa: PLR0911, PLR0915
 
     file = AsyncPath(working_directory, tool_input.file_path)
 
+    # Check if this is an image file and we have an upload client
+    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 aiohttp
+
+            file_name = Path(tool_input.file_path).name
+            s3_key = f"images/{uuid.uuid4()}/{file_name}"
+
+            upload_response = await upload_client.request_upload_url(s3_key, media_type)
+
+            async with aiohttp.ClientSession() as session:
+                f = await file.open("rb")
+                async with f:
+                    file_data = await f.read()
+                    async with session.put(
+                        upload_response.upload_url,
+                        data=file_data,
+                        headers={"Content-Type": media_type},
+                    ) as resp:
+                        if resp.status != 200:
+                            raise RuntimeError(
+                                f"Upload failed with status {resp.status}"
+                            )
+
+            return 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}")
+
     try:
         exists = await file.exists()
     except (OSError, PermissionError) as e:

exponent/core/remote_execution/truncation.py

@@ -11,6 +11,7 @@ from exponent.core.remote_execution.cli_rpc_types import (
     GlobToolResult,
     GrepToolResult,
     ListToolResult,
+    ReadToolArtifactResult,
     ReadToolResult,
     ToolResult,
     WriteToolResult,
@@ -173,6 +174,14 @@ class TailTruncation(TruncationStrategy):
         return replace(result, **updates)
 
 
+class NoOpTruncation(TruncationStrategy):
+    def should_truncate(self, result: ToolResult) -> bool:
+        return False
+
+    def truncate(self, result: ToolResult) -> ToolResult:
+        return result
+
+
 class StringListTruncation(TruncationStrategy):
     """Truncation for lists of strings that limits both number of items and individual string length."""
 
@@ -264,6 +273,7 @@ class StringListTruncation(TruncationStrategy):
 
 TRUNCATION_REGISTRY: dict[type[ToolResult], TruncationStrategy] = {
     ReadToolResult: StringFieldTruncation("content"),
+    ReadToolArtifactResult: NoOpTruncation(),
     WriteToolResult: StringFieldTruncation("message"),
     BashToolResult: TailTruncation("shell_output"),
     GrepToolResult: StringListTruncation("matches"),

{indent-0.1.17.dist-info → indent-0.1.18.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: indent
-Version: 0.1.17
+Version: 0.1.18
 Summary: Indent is an AI Pair Programmer
 Author-email: Sashank Thupukari <sashank@exponent.run>
 Requires-Python: <3.13,>=3.10

{indent-0.1.17.dist-info → indent-0.1.18.dist-info}/RECORD

@@ -1,4 +1,4 @@
-exponent/__init__.py,sha256=TtVjjQ5FSnY_MX0ZAPLaNmAfTJWa0sEMBdMs65ngXMM,706
+exponent/__init__.py,sha256=Nx3lULyklTDQB2p2ofjQ59zAxYunJHGjMIsvHePGZsI,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
@@ -18,8 +18,8 @@ exponent/core/graphql/mutations.py,sha256=cAAiSefyamBgy1zJEZ7LNk0rbjwawTPxMyj8eU
 exponent/core/graphql/queries.py,sha256=RYsk8bub0esspqgakilhzX07yJf2652Ey9tBZK1l_lY,3297
 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=8xCDy-3kA-PQew8l7CZtoCrCFHz9FsMWI8aXljh5wCg,7659
-exponent/core/remote_execution/client.py,sha256=l0x0wUlhsEBDAmC_llS4ihP0f-3gdrCvf4wwgDpWKNc,28916
+exponent/core/remote_execution/cli_rpc_types.py,sha256=39MQRwi2WRBBOfIwUpbBDs_ffLgURTaT6kBby3TVTZQ,8255
+exponent/core/remote_execution/client.py,sha256=9TlYzwh1lBiOeBlhGUhSo73KQGMh6nvbm1xdfxM48OY,31349
 exponent/core/remote_execution/code_execution.py,sha256=jYPB_7dJzS9BTPLX9fKQpsFPatwjbXuaFFSxT9tDTfI,2388
 exponent/core/remote_execution/error_info.py,sha256=Rd7OA3ps06qYejPVcOaMBB9AtftP3wqQoOfiILFASnc,1378
 exponent/core/remote_execution/exceptions.py,sha256=eT57lBnBhvh-KJ5lsKWcfgGA5-WisAxhjZx-Z6OupZY,135
@@ -29,9 +29,9 @@ exponent/core/remote_execution/git.py,sha256=dGjBpeoKJZsYgRwctSq29GmbsNIN9tbSA3V
 exponent/core/remote_execution/http_fetch.py,sha256=aFEyXd0S-MRfisSMuIFiEyc1AEAj9nUZ9Rj_P_YRows,2827
 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=g5C0CAlOJhxsNauVQVhXvYGFpA8rDnO05VsWO2ag05k,13654
+exponent/core/remote_execution/tool_execution.py,sha256=jhsZCV_XDyXC9moN01xUTLvl-q84jMFtpauPqPGz2cQ,15591
 exponent/core/remote_execution/tool_type_utils.py,sha256=7qi6Qd8fvHts019ZSLPbtiy17BUqgqBg3P_gdfvFf7w,1301
-exponent/core/remote_execution/truncation.py,sha256=noB6c4eaebqq5ghTlYJkXbe2XY8Bz_GBeh9DazJUrrU,9644
+exponent/core/remote_execution/truncation.py,sha256=1L0qwbWgQFRwJZeCXRVKkEhKmzoS1x2usDLPUIp-jlU,9923
 exponent/core/remote_execution/types.py,sha256=uo8h_ftcafmDp8YTxAI-H9qSaDygHYRFnnrmrBxukm4,14934
 exponent/core/remote_execution/utils.py,sha256=6PlBqYJ3OQwZ0dgXiIu3br04a-d-glDeDZpD0XGGPAE,14793
 exponent/core/remote_execution/languages/python_execution.py,sha256=nsX_LsXcUcHhiEHpSTjOTVNd7CxM146al0kw_iQX5OU,7724
@@ -46,7 +46,7 @@ exponent/migration-docs/login.md,sha256=KIeXy3m2nzSUgw-4PW1XzXfHael1D4Zu93CplLMb
 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.17.dist-info/METADATA,sha256=w5ahImXeo_nq_h3eZd92XsjsYzNmPU8nR-D9xSRIRRI,1308
-indent-0.1.17.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
-indent-0.1.17.dist-info/entry_points.txt,sha256=q8q1t1sbl4NULGOR0OV5RmSG4KEjkpEQRU_RUXEGzcs,44
-indent-0.1.17.dist-info/RECORD,,
+indent-0.1.18.dist-info/METADATA,sha256=fhSeEdMOl6WPMNlWcAWROiZT13E1-b2DooMkOwemXm4,1308
+indent-0.1.18.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
+indent-0.1.18.dist-info/entry_points.txt,sha256=q8q1t1sbl4NULGOR0OV5RmSG4KEjkpEQRU_RUXEGzcs,44
+indent-0.1.18.dist-info/RECORD,,