indent 0.1.3__py3-none-any.whl → 0.1.4__py3-none-any.whl

exponent/__init__.py

@@ -1 +1 @@
-__version__ = "0.1.3"  # Keep in sync with pyproject.toml
+__version__ = "0.1.4"  # Keep in sync with pyproject.toml

exponent/cli.py

@@ -10,6 +10,7 @@ from exponent.commands.run_commands import run_cli
 from exponent.commands.shell_commands import shell_cli
 from exponent.commands.types import ExponentGroup, exponent_cli_group
 from exponent.commands.upgrade import upgrade_cli
+from exponent.commands.workflow_commands import workflow_cli
 from exponent.utils.version import (
     get_installed_version,
 )
@@ -35,6 +36,7 @@ sources: list[ExponentGroup] = [
     cloud_cli,  # Cloud commands
     listen_cli,  # Listen to chat events
     github_app_cli,  # Setup github app installation
+    workflow_cli,  # Workflow commands
 ]
 
 for source in sources:

exponent/commands/workflow_commands.py

@@ -0,0 +1,111 @@
+import asyncio
+import os
+import sys
+from typing import cast
+
+import click
+
+from exponent.commands.run_commands import run_chat
+from exponent.commands.settings import use_settings
+from exponent.commands.types import exponent_cli_group
+from exponent.core.config import Settings
+from exponent.core.remote_execution.client import RemoteExecutionClient, WSDisconnected
+from exponent.core.remote_execution.types import (
+    PrReviewWorkflowInput,
+    WorkflowTriggerResponse,
+)
+
+
+@exponent_cli_group(name="workflow")
+def workflow_cli() -> None:
+    """Workflow commands."""
+    pass
+
+
+@workflow_cli.group()
+def workflow() -> None:
+    """Workflow management commands."""
+    pass
+
+
+@workflow.command()
+@use_settings
+@click.argument("workflow_type", type=click.STRING)
+def trigger(settings: Settings, workflow_type: str) -> None:
+    """Trigger a workflow."""
+
+    if not settings.api_key:
+        raise click.ClickException(
+            "No API key found. Use `indent login` to set your API key."
+        )
+
+    if workflow_type != "pr_review":
+        raise click.UsageError("Invalid workflow name. Only 'pr_review' is supported.")
+
+    loop = asyncio.get_event_loop()
+    response = loop.run_until_complete(trigger_pr_review_workflow(settings))
+
+    while True:
+        result = run_chat(
+            loop, settings.api_key, response.chat_uuid, settings, None, None
+        )
+        if result is None or isinstance(result, WSDisconnected):
+            # NOTE: None here means that handle_connection_changes exited
+            # first. We should likely have a different message for this.
+            if result and result.error_message:
+                click.secho(f"Error: {result.error_message}", fg="red")
+                sys.exit(10)
+            else:
+                print("Disconnected upon user request, shutting down...")
+                break
+        else:
+            raise click.ClickException("Workflow run exited unexpectedly")
+
+
+async def _subprocess_check_output(command: str) -> str:
+    process = await asyncio.create_subprocess_shell(
+        command,
+        stdout=asyncio.subprocess.PIPE,
+        stderr=asyncio.subprocess.STDOUT,
+    )
+    stdout, _ = await process.communicate()
+
+    if process.returncode != 0:
+        output = stdout.decode().strip()
+        raise click.ClickException(
+            f"Command '{command}' failed with exit code {process.returncode}:\n{output}"
+        )
+
+    return stdout.decode().strip()
+
+
+async def trigger_pr_review_workflow(settings: Settings) -> WorkflowTriggerResponse:
+    origin_url = await _subprocess_check_output("git ls-remote --get-url origin")
+    url = origin_url.strip().removesuffix(".git")
+    remote = url.split(":")[-1]
+    owner, repo = remote.split("/")[-2:]
+
+    pr_number_str = os.environ.get("PR_NUMBER")
+    if not pr_number_str:
+        raise click.ClickException("PR_NUMBER environment variable is not set")
+    try:
+        pr_number = int(pr_number_str)
+    except ValueError:
+        raise click.ClickException(
+            "PR_NUMBER environment variable is not a valid integer"
+        )
+
+    async with RemoteExecutionClient.session(
+        api_key=cast(str, settings.api_key),
+        base_url=settings.get_base_api_url(),
+        base_ws_url=settings.get_base_ws_url(),
+        working_directory=os.getcwd(),
+    ) as client:
+        return await client.trigger_workflow(
+            workflow_name="pr_review",
+            workflow_input=PrReviewWorkflowInput(
+                repo_owner=owner,
+                repo_name=repo,
+                pr_number=pr_number,
+            ),
+        )

exponent/core/remote_execution/client.py

@@ -54,9 +54,12 @@ from exponent.core.remote_execution.types import (
     CreateChatResponse,
     GitInfo,
     HeartbeatInfo,
+    PrReviewWorkflowInput,
     RemoteExecutionResponseType,
     RunWorkflowRequest,
     StreamingCodeExecutionRequest,
+    WorkflowTriggerRequest,
+    WorkflowTriggerResponse,
 )
 from exponent.core.remote_execution.utils import (
     deserialize_api_response,
@@ -378,6 +381,7 @@ class RemoteExecutionClient:
         )
         return cast(dict[str, Any], response.json())
 
+    # deprecated
     async def run_workflow(self, chat_uuid: str, workflow_id: str) -> dict[str, Any]:
         response = await self.api_client.post(
             "/api/remote_execution/run_workflow",
@@ -393,6 +397,18 @@ class RemoteExecutionClient:
             )
         return cast(dict[str, Any], response.json())
 
+    async def trigger_workflow(
+        self, workflow_name: str, workflow_input: PrReviewWorkflowInput
+    ) -> WorkflowTriggerResponse:
+        response = await self.api_client.post(
+            "/api/remote_execution/trigger_workflow",
+            json=WorkflowTriggerRequest(
+                workflow_name=workflow_name,
+                workflow_input=workflow_input,
+            ).model_dump(),
+        )
+        return await deserialize_api_response(response, WorkflowTriggerResponse)
+
     async def get_heartbeat_info(self) -> HeartbeatInfo:
         return HeartbeatInfo(
             system_info=await system_context.get_system_info(self.working_directory),

exponent/core/remote_execution/types.py

@@ -42,6 +42,23 @@ class RunWorkflowRequest(BaseModel):
     workflow_id: str
 
 
+# note: before adding fields here, probably update
+# get_workflow_run_by_trigger db query
+class PrReviewWorkflowInput(BaseModel):
+    repo_owner: str
+    repo_name: str
+    pr_number: int
+
+
+class WorkflowTriggerRequest(BaseModel):
+    workflow_name: str
+    workflow_input: PrReviewWorkflowInput
+
+
+class WorkflowTriggerResponse(BaseModel):
+    chat_uuid: str
+
+
 class ExecutionEndResponse(BaseModel):
     execution_ended: bool
 

{indent-0.1.3.dist-info → indent-0.1.4.dist-info}/METADATA

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

{indent-0.1.3.dist-info → indent-0.1.4.dist-info}/RECORD

@@ -1,5 +1,5 @@
-exponent/__init__.py,sha256=lpsrWC4wsfQ4-KiEMYUTzSNJlmYYtn6vNyEFjs6ddoc,58
-exponent/cli.py,sha256=9FPB5bqqGn7qVA_q2XZL6B2kRngVQoPdyvUvC5BH1_0,3496
+exponent/__init__.py,sha256=EbzjWm450G-BHYY2B1fRN9gjMU0txzZ-JYSHtaTQGMs,58
+exponent/cli.py,sha256=GA-Bw1nys9JNrlob0t3LcS9OZ4U5JcNYjR_r3kDD_qs,3596
 exponent/py.typed,sha256=9XZl5avs8yHp89XP_1Fjtbeg_2rjYorCC9I0k_j-h2c,334
 exponent/commands/cloud_commands.py,sha256=4DgS7PjCtCFB5uNN-szzAzOj16UU1D9b9_qS7DskoLE,2026
 exponent/commands/common.py,sha256=HUzc9yl5L6GZEF5DDsF3IeArwPKrghxmK43vxtkxgGs,13679
@@ -13,6 +13,7 @@ exponent/commands/theme.py,sha256=XoX9lYDf8YyoU1HoRdctDHjK5MrMRUcBALBLcHDVPc8,81
 exponent/commands/types.py,sha256=iDJL3hdwhO1PrhsJTJBioNYSKo0CWV8Nv-ONcDaWIRs,3670
 exponent/commands/upgrade.py,sha256=SpCdgEsBesPmXB5XgRJsIqonj73dfuuGVblo_0C2pWE,809
 exponent/commands/utils.py,sha256=gmWUPHZtAfuSb3zdrAEkmHVhQYmHghqhvJUFBunn2-4,7572
+exponent/commands/workflow_commands.py,sha256=ytetNMzojzJDe9NAsGdkDBGJs1eZIfY8xLNqesnc3AE,3580
 exponent/core/config.py,sha256=TNFLUgLnfSocRMVSav_7E4VcaNHXZ_3Mg5Lp1smP46U,5731
 exponent/core/graphql/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 exponent/core/graphql/client.py,sha256=KRveVt4lsrSdF9PnhhlQeqx74FQ9-iN28WQp0WKnQ30,1921
@@ -24,7 +25,7 @@ exponent/core/graphql/queries.py,sha256=TXXHLGb7QpeICaofowVYsjyHDfqjCoQ3omLbesuw
 exponent/core/graphql/subscriptions.py,sha256=gg42wG5HqEuNMJU7OUHruNCAGtM6FKPLRD7KfjcKjC4,9995
 exponent/core/remote_execution/checkpoints.py,sha256=3QGYMLa8vT7XmxMYTRcGrW8kNGHwRC0AkUfULribJWg,6354
 exponent/core/remote_execution/cli_rpc_types.py,sha256=3Yj6kIkqPAf7FNNni5V4XNe8SLPoTgbcL6swesW7cw4,4593
-exponent/core/remote_execution/client.py,sha256=3QbSAsukuVHCwmGfkoZBaBwgy4DuRZQs7LGAJtmT068,21010
+exponent/core/remote_execution/client.py,sha256=u5JH1yhbHcWpdee340x9JnYCsFJ-rJuu0-YPgSbojRw,21603
 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
@@ -35,7 +36,7 @@ exponent/core/remote_execution/session.py,sha256=cSJcCG1o74mBE6lZS_9VFmhyZdW6BeI
 exponent/core/remote_execution/system_context.py,sha256=0FkbsSxEVjdjTF0tQpOkYK_VaVM126C3_K8QP0YXxOs,1510
 exponent/core/remote_execution/tool_execution.py,sha256=G7wB_DA0jANqywhm_a6lbWtwjCcpQsuQHlExGohsgeY,9233
 exponent/core/remote_execution/truncation.py,sha256=rFQDfT7qf_u6lvhEADWSpoRe_GTPegXXqknk7OZp0uI,10093
-exponent/core/remote_execution/types.py,sha256=G-5EU-NtIj_zZbV305YMDZEcSacGK7QfzEdCYSHAGmA,17206
+exponent/core/remote_execution/types.py,sha256=3x73SJ3jQtaIu01musXgS-VF_MYwYucG3hyA2nWHbII,17567
 exponent/core/remote_execution/utils.py,sha256=Hw2zGMq0NKVwkVzNOVekAvpM33rfIz2P36mEcJr43Ag,20688
 exponent/core/remote_execution/languages/python_execution.py,sha256=GcQU5PiAOUD9tUZDrSObgM3L7FjXMdzr4FQg6ts2XjY,7780
 exponent/core/remote_execution/languages/shell_streaming.py,sha256=eSeHcK_unGpVRRdfgfwhml_tSlCo0VCeUEQY4zuMbDw,7372
@@ -49,7 +50,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=Q4txP7Rg_KO0u0tUpx8O0DoOt32wrX7ctNeDXVKaOfA,8835
-indent-0.1.3.dist-info/METADATA,sha256=IMTWgeClAXkVN9EFndRFeNFkx77eQQ1PI7_nhVCgjXU,1283
-indent-0.1.3.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
-indent-0.1.3.dist-info/entry_points.txt,sha256=q8q1t1sbl4NULGOR0OV5RmSG4KEjkpEQRU_RUXEGzcs,44
-indent-0.1.3.dist-info/RECORD,,
+indent-0.1.4.dist-info/METADATA,sha256=OTkZrr6729d5hBErMwl_RqvSr1ZrLRfCsx-MiuWA2b8,1283
+indent-0.1.4.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
+indent-0.1.4.dist-info/entry_points.txt,sha256=q8q1t1sbl4NULGOR0OV5RmSG4KEjkpEQRU_RUXEGzcs,44
+indent-0.1.4.dist-info/RECORD,,