indent 0.1.14__py3-none-any.whl → 0.1.15__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.14'
-__version_tuple__ = version_tuple = (0, 1, 14)
+__version__ = version = '0.1.15'
+__version_tuple__ = version_tuple = (0, 1, 15)
 
 __commit_id__ = commit_id = None

exponent/core/remote_execution/cli_rpc_types.py

@@ -2,6 +2,7 @@ from typing import TYPE_CHECKING, Any
 
 import msgspec
 import yaml
+from pydantic_ai.format_as_xml import format_as_xml
 
 if TYPE_CHECKING:
     from exponent_server.core.tools.edit_tool import (
@@ -28,6 +29,7 @@ class ToolInput(msgspec.Struct, tag_field="tool_name", omit_defaults=True):
         return msgspec.to_builtins(self)  # type: ignore[no-any-return]
 
     def to_text(self) -> str:
+        """This text is purely used for human debugging, it's not fed to the llm"""
         d = msgspec.to_builtins(self)
         del d["tool_name"]
         return yaml.dump(d)
@@ -41,7 +43,7 @@ class ToolResult(msgspec.Struct, tag_field="tool_name", omit_defaults=True):
         This provides a default textual representation of the tool result. Override it as needed for your tool."""
         d = msgspec.to_builtins(self)
         del d["tool_name"]
-        return yaml.dump(d)
+        return format_as_xml(d, include_root_tag=False, item_tag="item")
 
 
 class ErrorToolResult(ToolResult, tag="error"):

exponent/core/remote_execution/client.py

@@ -5,7 +5,7 @@ import json
 import logging
 import time
 import uuid
-from collections.abc import AsyncGenerator, Callable, Generator
+from collections.abc import AsyncGenerator, Callable, Coroutine, Generator
 from contextlib import asynccontextmanager
 from dataclasses import dataclass
 from typing import Any, TypeVar, cast
@@ -30,6 +30,7 @@ from exponent.core.remote_execution.cli_rpc_types import (
     CliRpcRequest,
     CliRpcResponse,
     ErrorResponse,
+    ErrorToolResult,
     GetAllFilesRequest,
     GetAllFilesResponse,
     HttpRequest,
@@ -611,33 +612,36 @@ class RemoteExecutionClient:
                     response=GetAllFilesResponse(files=files),
                 )
             elif isinstance(request.request, BatchToolExecutionRequest):
-                results: list[ToolResultType] = []
+                coros: list[Coroutine[Any, Any, ToolResultType]] = []
                 for tool_input in request.request.tool_inputs:
-                    try:
-                        if isinstance(tool_input, BashToolInput):
-                            raw_result = await execute_bash_tool(
+                    if isinstance(tool_input, BashToolInput):
+                        coros.append(
+                            execute_bash_tool(
                                 tool_input,
                                 self.working_directory,
                                 should_halt=self.get_halt_check(request.request_id),
                             )
-                        else:
-                            raw_result = await execute_tool(  # type: ignore[assignment]
-                                tool_input, self.working_directory
-                            )
-                        tool_result = truncate_result(raw_result)
-                        results.append(tool_result)
-                    except Exception as e:
-                        logger.error(f"Error executing tool {tool_input}: {e}")
-                        from exponent.core.remote_execution.cli_rpc_types import (
-                            ErrorToolResult,
                         )
+                    else:
+                        coros.append(execute_tool(tool_input, self.working_directory))
+
+                results: list[ToolResultType | BaseException] = await asyncio.gather(
+                    *coros, return_exceptions=True
+                )
 
-                        results.append(ErrorToolResult(error_message=str(e)))
+                processed_results: list[ToolResultType] = []
+                for result in results:
+                    if not isinstance(result, BaseException):
+                        processed_results.append(truncate_result(result))
+                    else:
+                        processed_results.append(
+                            ErrorToolResult(error_message=str(result))
+                        )
 
                 return CliRpcResponse(
                     request_id=request.request_id,
                     response=BatchToolExecutionResponse(
-                        tool_results=results,
+                        tool_results=processed_results,
                     ),
                 )
             elif isinstance(request.request, HttpRequest):

exponent/core/remote_execution/tool_type_utils.py

@@ -0,0 +1,39 @@
+from typing import Any
+
+import msgspec
+
+from exponent.core.remote_execution.cli_rpc_types import ToolResult
+
+
+def to_mostly_xml(tool_result: ToolResult) -> str:
+    """
+    This provides a default textual representation of the tool result. Override it as needed for your tool."""
+    d = msgspec.to_builtins(tool_result)
+    del d["tool_name"]
+    return to_mostly_xml_helper(d)
+
+
+def to_mostly_xml_helper(
+    d: Any,
+) -> str:
+    if isinstance(d, dict):
+        # No outer wrapper at top level, each field gets XML tags
+        parts = []
+        for key, value in d.items():
+            if isinstance(value, list):
+                # Handle lists with item tags
+                list_items = "\n".join(
+                    f"<item>\n{to_mostly_xml_helper(item)}\n</item>" for item in value
+                )
+                parts.append(f"<{key}>\n{list_items}\n</{key}>")
+            elif isinstance(value, dict):
+                # Nested dict
+                parts.append(f"<{key}>\n{to_mostly_xml_helper(value)}\n</{key}>")
+            else:
+                # Scalar value
+                parts.append(f"<{key}>\n{value!s}\n</{key}>")
+        return "\n".join(parts)
+    elif isinstance(d, list):
+        raise ValueError("Lists are not allowed at the top level")
+    else:
+        return str(d)

exponent/core/remote_execution/types.py

@@ -517,6 +517,10 @@ class ChatMode(str, Enum):
     DATABASE = "DATABASE"  # chat with database connection
     WORKFLOW = "WORKFLOW"
 
+    @classmethod
+    def requires_cli(cls, mode: "ChatMode") -> bool:
+        return mode not in [cls.DATABASE]
+
 
 class ChatSource(str, Enum):
     CLI_SHELL = "CLI_SHELL"

{indent-0.1.14.dist-info → indent-0.1.15.dist-info}/METADATA

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

{indent-0.1.14.dist-info → indent-0.1.15.dist-info}/RECORD

@@ -1,4 +1,4 @@
-exponent/__init__.py,sha256=1asLxKIxr0ym19WewGl0URtkfnEuN7mK8ZckB9dZw6Q,706
+exponent/__init__.py,sha256=HPqQHR9pVxIxlFt4vovkyoe7k6UO3ag2isBN2lHFL8g,706
 exponent/cli.py,sha256=u3hhZnn5uqVXyNz6wU8j4U5vLJ--pbA-PweHhpvRzY4,3444
 exponent/py.typed,sha256=9XZl5avs8yHp89XP_1Fjtbeg_2rjYorCC9I0k_j-h2c,334
 exponent/commands/cloud_commands.py,sha256=TNSbKnc7VBo7VALj44CqV5tdCACJejEGmtYvc5wjza4,19080
@@ -19,8 +19,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=bmpfZASzzsD_NDIF2Mnym79km2fem8Oc0tSlSC-0EoY,7478
-exponent/core/remote_execution/client.py,sha256=eeUiWXebJ3WuDjgvBqg0OZk0heUHcBSRM96xnIE0wFE,28942
+exponent/core/remote_execution/cli_rpc_types.py,sha256=8xCDy-3kA-PQew8l7CZtoCrCFHz9FsMWI8aXljh5wCg,7659
+exponent/core/remote_execution/client.py,sha256=kQ3WKyGQwA68H76RL_XhL6PEOJkEPFNWAt3L5vB_3tc,29008
 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
@@ -31,8 +31,9 @@ exponent/core/remote_execution/http_fetch.py,sha256=aFEyXd0S-MRfisSMuIFiEyc1AEAj
 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=ZBDC2RN_l8oCFO8Fc8dkKoLY_0rn-5h394PvWeXqTnI,12972
+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/types.py,sha256=8caWaHyiD_QXYFkP6gt8Ieyu3ctLVNDvoM0RfN-Zy_s,15006
+exponent/core/remote_execution/types.py,sha256=2lMSikVHo9WMUNEYFHQN-EqyFbZOJAfG-AUV6zlti7o,15119
 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=eEhngdLhiMiDM0KUfLUtj6BdU3ay3G1zJy6dlR2V6nw,7389
@@ -46,7 +47,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=utp6uc6x_-diQ8scs0GkLHxFXG1gffAmZL77dWXI9cw,8890
-indent-0.1.14.dist-info/METADATA,sha256=ab2iaWn7PdoS835Lj5nTf6DYV8Q5WRh1z5t4ClE3bGc,1273
-indent-0.1.14.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
-indent-0.1.14.dist-info/entry_points.txt,sha256=q8q1t1sbl4NULGOR0OV5RmSG4KEjkpEQRU_RUXEGzcs,44
-indent-0.1.14.dist-info/RECORD,,
+indent-0.1.15.dist-info/METADATA,sha256=OkrSKctJfgz1C5duJtqthUG5Z5_sv-gYXCTPQjUGwrE,1273
+indent-0.1.15.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
+indent-0.1.15.dist-info/entry_points.txt,sha256=q8q1t1sbl4NULGOR0OV5RmSG4KEjkpEQRU_RUXEGzcs,44
+indent-0.1.15.dist-info/RECORD,,