indent 0.1.3__tar.gz → 0.1.4__tar.gz

{indent-0.1.3 → indent-0.1.4}/PKG-INFO

@@ -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.4/exponent/__init__.py

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

{indent-0.1.3 → indent-0.1.4}/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:

indent-0.1.4/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,
+            ),
+        )

{indent-0.1.3 → indent-0.1.4}/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),

{indent-0.1.3 → indent-0.1.4}/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 → indent-0.1.4}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "hatchling.build"
 
 [project]
 name = "indent"
-version = "0.1.3"
+version = "0.1.4"
 description = "Indent is an AI Pair Programmer"
 authors = [{ name = "Sashank Thupukari", email = "sashank@exponent.run" }]
 requires-python = ">=3.10,<3.13"

indent-0.1.3/exponent/__init__.py

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