indent 0.1.21__tar.gz → 0.1.22__tar.gz

{indent-0.1.21 → indent-0.1.22}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: indent
-Version: 0.1.21
+Version: 0.1.22
 Summary: Indent is an AI Pair Programmer
 Author-email: Sashank Thupukari <sashank@exponent.run>
 Requires-Python: <3.13,>=3.10
@@ -22,6 +22,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

{indent-0.1.21 → indent-0.1.22}/exponent/__init__.py

@@ -28,7 +28,7 @@ version_tuple: VERSION_TUPLE
 commit_id: COMMIT_ID
 __commit_id__: COMMIT_ID
 
-__version__ = version = '0.1.21'
-__version_tuple__ = version_tuple = (0, 1, 21)
+__version__ = version = '0.1.22'
+__version_tuple__ = version_tuple = (0, 1, 22)
 
 __commit_id__ = commit_id = None

{indent-0.1.21 → indent-0.1.22}/exponent/commands/cloud_commands.py

@@ -111,6 +111,7 @@ async def create_cloud_chat_from_repository(
     base_api_url: str,
     base_ws_url: str,
     repository_id: str,
+    provider: str | None = None,
 ) -> dict[str, Any]:
     graphql_client = GraphQLClient(
         api_key=api_key, base_api_url=base_api_url, base_ws_url=base_ws_url
@@ -118,6 +119,7 @@ async def create_cloud_chat_from_repository(
 
     variables = {
         "repositoryId": repository_id,
+        "provider": provider,
     }
 
     result = await graphql_client.execute(

{indent-0.1.21 → indent-0.1.22}/exponent/core/graphql/mutations.py

@@ -76,8 +76,8 @@ mutation CreateCloudChat($configId: String!) {
 
 
 CREATE_CLOUD_CHAT_FROM_REPOSITORY_MUTATION = """
-mutation CreateCloudChatFromRepository($repositoryId: String!) {
-  createCloudChat(repositoryId: $repositoryId) {
+mutation CreateCloudChatFromRepository($repositoryId: String!, $provider: String) {
+  createCloudChat(repositoryId: $repositoryId, provider: $provider) {
     __typename
     ...on Chat {
       chatUuid

indent-0.1.22/exponent/core/remote_execution/port_utils.py

@@ -0,0 +1,73 @@
+import time
+
+import psutil
+
+from exponent.core.remote_execution.types import PortInfo
+
+
+def get_port_usage() -> list[PortInfo] | None:
+    """
+    Get information about all listening ports on the system.
+
+    Returns:
+        List of PortInfo objects containing process name, port, protocol, pid, and uptime.
+        Returns None if there's a permission error.
+        Returns empty list if no listening ports are found.
+    """
+    try:
+        connections = psutil.net_connections(kind="tcp")
+    except (psutil.AccessDenied, PermissionError):
+        # If we don't have permission to see connections, return None
+        return None
+    except Exception:
+        # For any other unexpected errors, return None
+        return None
+
+    port_info_list: list[PortInfo] = []
+    current_time = time.time()
+
+    for conn in connections:
+        # Only include TCP ports in LISTEN state
+        if conn.status != "LISTEN":
+            continue
+
+        # Skip if no local address (shouldn't happen for LISTEN, but be safe)
+        if not conn.laddr:
+            continue
+
+        port = conn.laddr.port
+        pid = conn.pid
+
+        # Try to get process information
+        process_name = "unknown"
+        uptime_seconds = None
+
+        if pid:
+            try:
+                process = psutil.Process(pid)
+                process_name = process.name()
+
+                # Calculate uptime
+                create_time = process.create_time()
+                uptime_seconds = current_time - create_time
+            except (psutil.NoSuchProcess, psutil.AccessDenied, psutil.ZombieProcess):
+                # Process disappeared or we don't have permission
+                pass
+            except Exception:
+                # Any other unexpected error, just skip process info
+                pass
+
+        port_info = PortInfo(
+            process_name=process_name,
+            port=port,
+            protocol="TCP",
+            pid=pid,
+            uptime_seconds=uptime_seconds,
+        )
+        port_info_list.append(port_info)
+
+        # Limit to 50 ports to avoid bloating the heartbeat payload
+        if len(port_info_list) >= 50:
+            break
+
+    return port_info_list

{indent-0.1.21 → indent-0.1.22}/exponent/core/remote_execution/system_context.py

@@ -4,6 +4,7 @@ import platform
 
 from exponent.core.remote_execution.git import get_git_info
 from exponent.core.remote_execution.languages import python_execution
+from exponent.core.remote_execution.port_utils import get_port_usage
 from exponent.core.remote_execution.types import (
     SystemInfo,
 )
@@ -17,6 +18,7 @@ async def get_system_info(working_directory: str) -> SystemInfo:
         shell=_get_user_shell(),
         git=await get_git_info(working_directory),
         python_env=python_execution.get_python_env_info(),
+        port_usage=get_port_usage(),
     )
 
 

{indent-0.1.21 → indent-0.1.22}/exponent/core/remote_execution/types.py

@@ -120,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
@@ -127,6 +135,7 @@ class SystemInfo(BaseModel):
     shell: str
     git: GitInfo | None
     python_env: PythonEnvInfo | None
+    port_usage: list[PortInfo] | None = None
 
 
 class HeartbeatInfo(BaseModel):

{indent-0.1.21 → indent-0.1.22}/pyproject.toml

@@ -19,6 +19,7 @@ dependencies = [
     "jupyter-client>=8.6.1,<9",
     "packaging~=24.1",
     "prompt-toolkit>=3.0.36,<4",
+    "psutil>=5.9.0,<7",
     "pydantic[email]>=2.6.4,<3",
     "pydantic-settings>=2.2.1,<3",
     "pydantic-ai==0.0.30",