Refactor and improve code formatting in client modules

Author: joezhoujinjing

src/nexus_client/__init__.py

@@ -47,16 +47,18 @@
 # LangGraph exports (optional - requires langgraph optional dependencies)
 try:
     from nexus_client.langgraph import get_nexus_tools, list_skills
+
     _HAS_LANGGRAPH = True
 except ImportError:
     # LangGraph dependencies not installed
     _HAS_LANGGRAPH = False
+
     # Don't export these if dependencies aren't available
     def _langgraph_not_available():
         raise ImportError(
-            "LangGraph integration requires optional dependencies. "
-            "Install with: pip install nexus-fs-python[langgraph]"
+            "LangGraph integration requires optional dependencies. Install with: pip install nexus-fs-python[langgraph]"
         )
+
     get_nexus_tools = _langgraph_not_available  # type: ignore
     list_skills = _langgraph_not_available  # type: ignore
 

src/nexus_client/async_client.py

@@ -194,8 +194,7 @@ async def _fetch_auth_info(self) -> None:
                     else:
                         self._agent_id = None
                     logger.info(
-                        f"Authenticated as {subject_type}:{auth_info.get('subject_id')} "
-                        f"(tenant: {self._tenant_id})"
+                        f"Authenticated as {subject_type}:{auth_info.get('subject_id')} (tenant: {self._tenant_id})"
                     )
                 else:
                     logger.debug("Not authenticated (anonymous access)")
@@ -208,9 +207,7 @@ async def _fetch_auth_info(self) -> None:
     @retry(
         stop=stop_after_attempt(3),
         wait=wait_exponential(multiplier=1, min=1, max=10),
-        retry=retry_if_exception_type(
-            (httpx.ConnectError, httpx.TimeoutException, RemoteConnectionError)
-        ),
+        retry=retry_if_exception_type((httpx.ConnectError, httpx.TimeoutException, RemoteConnectionError)),
         reraise=True,
     )
     async def _call_rpc(
@@ -285,9 +282,7 @@ async def _call_rpc(
 
             # Check HTTP status
             if response.status_code != 200:
-                logger.error(
-                    f"API call failed: {method} - HTTP {response.status_code} ({elapsed:.3f}s)"
-                )
+                logger.error(f"API call failed: {method} - HTTP {response.status_code} ({elapsed:.3f}s)")
                 raise RemoteFilesystemError(
                     f"Request failed: {response.text}",
                     status_code=response.status_code,
@@ -307,9 +302,7 @@ async def _call_rpc(
 
             # Check for RPC error
             if rpc_response.error:
-                logger.error(
-                    f"API call RPC error: {method} - {rpc_response.error.get('message')} ({elapsed:.3f}s)"
-                )
+                logger.error(f"API call RPC error: {method} - {rpc_response.error.get('message')} ({elapsed:.3f}s)")
                 self._handle_rpc_error(rpc_response.error)
 
             # Log detailed timing for grep operations
@@ -366,9 +359,7 @@ def _handle_rpc_error(self, error: dict[str, Any]) -> None:
             raise FileExistsError(message)
         elif code == RPCErrorCode.INVALID_PATH.value:
             raise InvalidPathError(message)
-        elif (
-            code == RPCErrorCode.ACCESS_DENIED.value or code == RPCErrorCode.PERMISSION_ERROR.value
-        ):
+        elif code == RPCErrorCode.ACCESS_DENIED.value or code == RPCErrorCode.PERMISSION_ERROR.value:
             raise NexusPermissionError(message)
         elif code == RPCErrorCode.VALIDATION_ERROR.value:
             raise ValidationError(message)
@@ -658,9 +649,7 @@ async def mkdir(
         Returns:
             Directory metadata dict
         """
-        result = await self._call_rpc(
-            "mkdir", {"path": path, "parents": parents, "exist_ok": exist_ok}
-        )
+        result = await self._call_rpc("mkdir", {"path": path, "parents": parents, "exist_ok": exist_ok})
         return result  # type: ignore[no-any-return]
 
     async def glob(
@@ -1071,9 +1060,7 @@ async def diff_versions(
         Returns:
             Diff result (dict for metadata, str for content)
         """
-        result = await self._call_rpc(
-            "diff_versions", {"path": path, "v1": v1, "v2": v2, "mode": mode}
-        )
+        result = await self._call_rpc("diff_versions", {"path": path, "v1": v1, "v2": v2, "mode": mode})
         return result  # type: ignore[no-any-return]
 
     # ============================================================
@@ -2676,9 +2663,7 @@ async def get_trajectory_feedback(
         Returns:
             List of feedback dicts
         """
-        result = await self.remote_fs._call_rpc(
-            "ace_get_trajectory_feedback", {"trajectory_id": trajectory_id}
-        )
+        result = await self.remote_fs._call_rpc("ace_get_trajectory_feedback", {"trajectory_id": trajectory_id})
         return result  # type: ignore[no-any-return]
 
     async def get_effective_score(

src/nexus_client/client.py

@@ -695,9 +695,7 @@ def _fetch_auth_info(self) -> None:
         from the server's /api/auth/whoami endpoint.
         """
         try:
-            response = self.session.get(
-                urljoin(self.server_url, "/api/auth/whoami"), timeout=self.connect_timeout
-            )
+            response = self.session.get(urljoin(self.server_url, "/api/auth/whoami"), timeout=self.connect_timeout)
 
             if response.status_code == 200:
                 auth_info = response.json()
@@ -711,8 +709,7 @@ def _fetch_auth_info(self) -> None:
                     else:
                         self.agent_id = None
                     logger.info(
-                        f"Authenticated as {subject_type}:{auth_info.get('subject_id')} "
-                        f"(tenant: {self.tenant_id})"
+                        f"Authenticated as {subject_type}:{auth_info.get('subject_id')} (tenant: {self.tenant_id})"
                     )
                 else:
                     logger.debug("Not authenticated (anonymous access)")
@@ -725,14 +722,10 @@ def _fetch_auth_info(self) -> None:
     @retry(
         stop=stop_after_attempt(3),
         wait=wait_exponential(multiplier=1, min=1, max=10),
-        retry=retry_if_exception_type(
-            (httpx.ConnectError, httpx.TimeoutException, RemoteConnectionError)
-        ),
+        retry=retry_if_exception_type((httpx.ConnectError, httpx.TimeoutException, RemoteConnectionError)),
         reraise=True,
     )
-    def _call_rpc(
-        self, method: str, params: dict[str, Any] | None = None, read_timeout: float | None = None
-    ) -> Any:
+    def _call_rpc(self, method: str, params: dict[str, Any] | None = None, read_timeout: float | None = None) -> Any:
         """Make RPC call to server with automatic retry logic.
 
         This method automatically retries on transient failures (connection errors,
@@ -807,9 +800,7 @@ def _call_rpc(
 
             # Check HTTP status
             if response.status_code != 200:
-                logger.error(
-                    f"API call failed: {method} - HTTP {response.status_code} ({elapsed:.3f}s)"
-                )
+                logger.error(f"API call failed: {method} - HTTP {response.status_code} ({elapsed:.3f}s)")
                 raise RemoteFilesystemError(
                     f"Request failed: {response.text}",
                     status_code=response.status_code,
@@ -829,9 +820,7 @@ def _call_rpc(
 
             # Check for RPC error
             if rpc_response.error:
-                logger.error(
-                    f"API call RPC error: {method} - {rpc_response.error.get('message')} ({elapsed:.3f}s)"
-                )
+                logger.error(f"API call RPC error: {method} - {rpc_response.error.get('message')} ({elapsed:.3f}s)")
                 self._handle_rpc_error(rpc_response.error)
 
             # Log detailed timing breakdown for grep operations
@@ -896,9 +885,7 @@ def _handle_rpc_error(self, error: dict[str, Any]) -> None:
             raise FileExistsError(message)
         elif code == RPCErrorCode.INVALID_PATH.value:
             raise InvalidPathError(message)
-        elif (
-            code == RPCErrorCode.ACCESS_DENIED.value or code == RPCErrorCode.PERMISSION_ERROR.value
-        ):
+        elif code == RPCErrorCode.ACCESS_DENIED.value or code == RPCErrorCode.PERMISSION_ERROR.value:
             raise NexusPermissionError(message)
         elif code == RPCErrorCode.VALIDATION_ERROR.value:
             raise ValidationError(message)
@@ -1438,9 +1425,7 @@ async def semantic_search(
         )
         return result  # type: ignore[no-any-return]
 
-    async def semantic_search_index(
-        self, path: str = "/", recursive: bool = True
-    ) -> dict[str, int]:
+    async def semantic_search_index(self, path: str = "/", recursive: bool = True) -> dict[str, int]:
         """Index documents for semantic search.
 
         Args:
@@ -1538,9 +1523,7 @@ def rollback(self, path: str, version: int, context: Any = None) -> None:  # noq
         # context is unused in remote client (handled server-side)
         self._call_rpc("rollback", {"path": path, "version": version})
 
-    def diff_versions(
-        self, path: str, v1: int, v2: int, mode: str = "metadata"
-    ) -> dict[str, Any] | str:
+    def diff_versions(self, path: str, v1: int, v2: int, mode: str = "metadata") -> dict[str, Any] | str:
         """Compare two versions of a file."""
         result = self._call_rpc("diff_versions", {"path": path, "v1": v1, "v2": v2, "mode": mode})
         return result  # type: ignore[no-any-return]
@@ -1904,8 +1887,7 @@ def get_acl(self, path: str) -> builtins.list[dict[str, str | bool | None]]:
             - rebac_expand(): Find all subjects with a permission
         """
         raise NotImplementedError(
-            "get_acl() is no longer supported. Use ReBAC instead:\n"
-            f"  nx.rebac_list_tuples(object=('file', '{path}'))"
+            f"get_acl() is no longer supported. Use ReBAC instead:\n  nx.rebac_list_tuples(object=('file', '{path}'))"
         )
 
     # ============================================================
@@ -2638,9 +2620,7 @@ def revoke_consent(self, from_subject: tuple[str, str], to_subject: tuple[str, s
             ...     to_subject=("user", "bob")
             ... )
         """
-        result = self._call_rpc(
-            "revoke_consent", {"from_subject": from_subject, "to_subject": to_subject}
-        )
+        result = self._call_rpc("revoke_consent", {"from_subject": from_subject, "to_subject": to_subject})
         return result  # type: ignore[no-any-return]
 
     def make_public(self, resource: tuple[str, str], tenant_id: str | None = None) -> str:
@@ -3064,9 +3044,7 @@ def list_saved_mounts(
             >>> # List mounts for specific user
             >>> alice_mounts = nx.list_saved_mounts(owner_user_id="google:alice123")
         """
-        result = self._call_rpc(
-            "list_saved_mounts", {"owner_user_id": owner_user_id, "tenant_id": tenant_id}
-        )
+        result = self._call_rpc("list_saved_mounts", {"owner_user_id": owner_user_id, "tenant_id": tenant_id})
         return result  # type: ignore[no-any-return]
 
     def load_mount(self, mount_point: str) -> str:
@@ -3350,8 +3328,7 @@ def register_workspace(
         created_by: str | None = None,
         tags: builtins.list[str] | None = None,
         metadata: dict[str, Any] | None = None,
-        session_id: str
-        | None = None,  # v0.5.0: If provided, workspace is session-scoped (temporary)
+        session_id: str | None = None,  # v0.5.0: If provided, workspace is session-scoped (temporary)
         ttl: timedelta | None = None,  # v0.5.0: Time-to-live for auto-expiry
     ) -> dict[str, Any]:
         """Register a directory as a workspace.
@@ -3755,9 +3732,7 @@ def ace_add_feedback(
         result = self._call_rpc("ace_add_feedback", params)
         return result  # type: ignore[no-any-return]
 
-    def ace_get_trajectory_feedback(
-        self, trajectory_id: str, context: dict | None = None
-    ) -> builtins.list[dict]:
+    def ace_get_trajectory_feedback(self, trajectory_id: str, context: dict | None = None) -> builtins.list[dict]:
         """Get all feedback for a trajectory.
 
         Args:

src/nexus_client/exceptions.py

@@ -119,9 +119,7 @@ def __init__(self, path: str, expected_etag: str, current_etag: str):
 class AuditLogError(NexusError):
     """Raised when audit logging fails and audit_strict_mode is enabled."""
 
-    def __init__(
-        self, message: str, path: str | None = None, original_error: Exception | None = None
-    ):
+    def __init__(self, message: str, path: str | None = None, original_error: Exception | None = None):
         self.original_error = original_error
         super().__init__(message, path)
 

src/nexus_client/langgraph/client.py

@@ -39,8 +39,7 @@ async def _get_nexus_client(config: RunnableConfig, state: dict[str, Any] | None
 
     if not x_auth:
         raise ValueError(
-            "Missing x_auth in metadata. "
-            "Frontend must pass API key via metadata: {'x_auth': 'Bearer <token>'}"
+            "Missing x_auth in metadata. Frontend must pass API key via metadata: {'x_auth': 'Bearer <token>'}"
         )
 
     # Strip "Bearer " prefix if present

src/nexus_client/langgraph/tools.py

@@ -269,9 +269,7 @@ async def read_file(
             # Parse read command
             parts = shlex.split(read_cmd.strip())
             if not parts:
-                return (
-                    "Error: Empty read command. Usage: read_file('[cat|less] path [start] [end]')"
-                )
+                return "Error: Empty read command. Usage: read_file('[cat|less] path [start] [end]')"
 
             # Determine command type, path, and line range
             start_line = None
@@ -352,9 +350,7 @@ async def read_file(
                 if start_line < 1:
                     return f"Error: Start line must be >= 1, got {start_line}"
                 if start_line > total_lines:
-                    return (
-                        f"Error: Start line {start_line} exceeds file length ({total_lines} lines)"
-                    )
+                    return f"Error: Start line {start_line} exceeds file length ({total_lines} lines)"
 
             if end_line is not None:
                 if end_line < 1:
@@ -382,9 +378,7 @@ async def read_file(
                         f"Try a smaller line range."
                     )
 
-                output = (
-                    f"Content of {path} (lines {start_line or 1}-{end_idx} of {total_lines}):\n\n"
-                )
+                output = f"Content of {path} (lines {start_line or 1}-{end_idx} of {total_lines}):\n\n"
                 output += content_str
                 return output
 
@@ -485,9 +479,7 @@ async def python(
                 return "Error: sandbox_id not found in metadata. Please start a sandbox first."
 
             # Execute Python code in sandbox
-            result = await nx.sandbox_run(
-                sandbox_id=sandbox_id, language="python", code=code, timeout=300
-            )
+            result = await nx.sandbox_run(sandbox_id=sandbox_id, language="python", code=code, timeout=300)
 
             # Format output
             output_parts = []
@@ -542,9 +534,7 @@ async def bash(
                 return "Error: sandbox_id not found in metadata. Please start a sandbox first."
 
             # Execute bash command in sandbox
-            result = await nx.sandbox_run(
-                sandbox_id=sandbox_id, language="bash", code=command, timeout=300
-            )
+            result = await nx.sandbox_run(sandbox_id=sandbox_id, language="bash", code=command, timeout=300)
 
             # Format output
             output_parts = []

src/nexus_client/protocol.py

@@ -156,9 +156,7 @@ def default(self, obj: Any) -> Any:
             return {"__type__": "datetime", "data": obj.isoformat()}
         elif hasattr(obj, "__dict__"):
             # Convert objects to dictionaries, filtering out methods
-            return {
-                k: v for k, v in obj.__dict__.items() if not k.startswith("_") and not callable(v)
-            }
+            return {k: v for k, v in obj.__dict__.items() if not k.startswith("_") and not callable(v)}
         return super().default(obj)
 
 
@@ -203,11 +201,7 @@ def _prepare_for_orjson(obj: Any) -> Any:
     elif isinstance(obj, (list, tuple)):
         return [_prepare_for_orjson(item) for item in obj]
     elif hasattr(obj, "__dict__") and not isinstance(obj, type):
-        return {
-            k: _prepare_for_orjson(v)
-            for k, v in obj.__dict__.items()
-            if not k.startswith("_") and not callable(v)
-        }
+        return {k: _prepare_for_orjson(v) for k, v in obj.__dict__.items() if not k.startswith("_") and not callable(v)}
     else:
         return obj