Fix the Ruff CI Errors

Author: Shashikant86

README.md

@@ -1,5 +1,11 @@
 # RLM Code
 
+<p align="center">
+  <a href="https://github.com/SuperagenticAI/rlm-code">
+    <img src="https://raw.githubusercontent.com/SuperagenticAI/rlm-code/main/assets/rlm-code-logo.png" alt="RLM Code logo" width="300">
+  </a>
+</p>
+
 **Run LLM-powered agents in a REPL loop, benchmark them, and compare results.**
 
 RLM Code implements the [Recursive Language Models](https://arxiv.org/abs/2502.07503) (RLM) approach from the 2025 paper release. Instead of stuffing your entire document into the LLM's context window, RLM stores it as a Python variable and lets the LLM write code to analyze it — chunk by chunk, iteration by iteration. This is dramatically more token-efficient for large inputs.

rlm_code/models/llm_connector.py

@@ -814,13 +814,9 @@ def _generate_ollama(
             if detail:
                 lowered = detail.lower()
                 if "not found" in lowered or "model" in lowered and "pull" in lowered:
-                    guidance = (
-                        f". Verify model name '{self.current_model}' and run `ollama pull {self.current_model}`"
-                    )
+                    guidance = f". Verify model name '{self.current_model}' and run `ollama pull {self.current_model}`"
                 elif "insufficient" in lowered or "memory" in lowered:
-                    guidance = (
-                        ". Model may be too large for local memory; try a smaller model or increase resources"
-                    )
+                    guidance = ". Model may be too large for local memory; try a smaller model or increase resources"
             raise ModelError(f"Ollama generation failed: {detail or exc}{guidance}") from exc
 
         except Exception as e:

rlm_code/rlm/frameworks/google_adk_adapter.py

@@ -5,6 +5,7 @@
 from __future__ import annotations
 
 import asyncio
+import sys
 from dataclasses import dataclass
 from typing import Any
 
@@ -21,6 +22,14 @@ class GoogleADKFrameworkAdapter:
     reference_impl: str = "google.adk (installed package)"
 
     def doctor(self) -> tuple[bool, str]:
+        cached = sys.modules.get("google.adk")
+        if cached is not None:
+            return (True, "google-adk available")
+        if "google.adk" in sys.modules and cached is None:
+            return (
+                False,
+                "google-adk not installed. Install with: pip install 'rlm-code[adk]'",
+            )
         try:
             import google.adk  # noqa: F401
         except Exception:

rlm_code/ui/__init__.py

@@ -6,9 +6,21 @@
 
 from .animations import ThinkingAnimation, get_random_thinking_message
 from .prompts import get_user_input
-from .tui_app import run_textual_tui
 from .welcome import show_welcome_screen
 
+
+def run_textual_tui(*args, **kwargs):
+    """
+    Lazily import and launch the Textual TUI.
+
+    Keeps non-TUI imports (e.g., utility modules/tests) working even when the
+    optional ``textual`` extra is not installed.
+    """
+    from .tui_app import run_textual_tui as _run_textual_tui
+
+    return _run_textual_tui(*args, **kwargs)
+
+
 __all__ = [
     "ThinkingAnimation",
     "get_random_thinking_message",

rlm_code/ui/pty_terminal.py

@@ -29,6 +29,8 @@
 # ---- ANSI escape code stripper (for plain-text fallback) ----
 
 _ANSI_RE = re.compile(r"\x1b\[[0-9;]*[A-Za-z]|\x1b\].*?\x07|\x1b\[.*?[a-zA-Z]")
+
+
 def strip_ansi(text: str) -> str:
     """Remove ANSI escape sequences from text."""
     return _ANSI_RE.sub("", text)
@@ -706,7 +708,16 @@ def on_key(self, event: events.Key) -> None:
             elif event.key == "delete":
                 # Keep behavior simple and conservative for status preview.
                 self._update_status()
-            elif event.key in {"up", "down", "left", "right", "home", "end", "page_up", "page_down"}:
+            elif event.key in {
+                "up",
+                "down",
+                "left",
+                "right",
+                "home",
+                "end",
+                "page_up",
+                "page_down",
+            }:
                 # Cursor/history edits are hard to mirror exactly; reset preview to avoid stale text.
                 self._typed_buffer = ""
                 self._update_status()

rlm_code/ui/tui_app.py

@@ -1316,7 +1316,9 @@ def _resolve_connection_mode(self) -> str:
 
         def _render_status_snapshot(self, *, title: str = "Status Snapshot") -> None:
             connected = bool(self.connector.current_model)
-            model = self.connector.current_model_id or self.connector.current_model or "disconnected"
+            model = (
+                self.connector.current_model_id or self.connector.current_model or "disconnected"
+            )
             provider = self.connector.model_type or "-"
             mode = self._resolve_connection_mode()
             route = str(self._last_response_route or "direct-llm").strip().lower() or "direct-llm"
@@ -1357,7 +1359,7 @@ def _render_status_snapshot(self, *, title: str = "Status Snapshot") -> None:
             if connected:
                 tips.append("/models", style=f"bold {PALETTE.info}")
                 tips.append("  ", style=PALETTE.text_dim)
-                tips.append("/rlm run \"task\" steps=4", style=f"bold {PALETTE.success}")
+                tips.append('/rlm run "task" steps=4', style=f"bold {PALETTE.success}")
                 tips.append("  ", style=PALETTE.text_dim)
                 tips.append("/connect", style=f"bold {PALETTE.warning}")
             else:
@@ -1386,7 +1388,7 @@ def _render_connection_success(self, provider: str, model: str) -> None:
             steps.append("Ready: ", style=f"bold {PALETTE.success}")
             steps.append("/status", style=f"bold {PALETTE.info}")
             steps.append("  ", style=PALETTE.text_dim)
-            steps.append("/rlm run \"your task\" steps=6", style=f"bold {PALETTE.success}")
+            steps.append('/rlm run "your task" steps=6', style=f"bold {PALETTE.success}")
             steps.append("  ", style=PALETTE.text_dim)
             steps.append("/models", style=f"bold {PALETTE.info}")
 
@@ -1444,7 +1446,9 @@ def _render_rlm_run_started(self, command: str) -> None:
                 )
             )
             try:
-                summary = self._cached_research_summary or self.query_one("#research_summary", Static)
+                summary = self._cached_research_summary or self.query_one(
+                    "#research_summary", Static
+                )
                 summary.update(
                     "[yellow]Run started...[/yellow] waiting for runtime events. "
                     "Open [cyan]Research Lab -> Events[/cyan] for live logs."
@@ -1499,7 +1503,9 @@ def _update_research_live_from_event(self, name: str, payload: dict[str, Any]) -
                 except Exception:
                     pass
                 try:
-                    state["reward"] = float(payload.get("total_reward", state.get("reward", 0.0)) or 0.0)
+                    state["reward"] = float(
+                        payload.get("total_reward", state.get("reward", 0.0)) or 0.0
+                    )
                 except Exception:
                     pass
             else:
@@ -1535,7 +1541,9 @@ def _update_research_live_from_event(self, name: str, payload: dict[str, Any]) -
                 pass
 
             try:
-                summary = self._cached_research_summary or self.query_one("#research_summary", Static)
+                summary = self._cached_research_summary or self.query_one(
+                    "#research_summary", Static
+                )
                 summary.update(
                     f"{status_text} | Reward: [bold]{reward:.3f}[/bold] | "
                     f"Steps: {steps} | Run: [dim]{run_id}[/dim]{task_line}"
@@ -1581,8 +1589,12 @@ def _render_rlm_run_summary_from_context(self) -> None:
             body.add_column()
             body.add_row(lines)
             if final_response:
-                preview = final_response if len(final_response) <= 320 else f"{final_response[:317]}..."
-                body.add_row(Panel(preview, title="Final Response (preview)", border_style="#3b82f6"))
+                preview = (
+                    final_response if len(final_response) <= 320 else f"{final_response[:317]}..."
+                )
+                body.add_row(
+                    Panel(preview, title="Final Response (preview)", border_style="#3b82f6")
+                )
             body.add_row(tail)
 
             self._chat_log().write(
@@ -2197,9 +2209,7 @@ def _set_command_running(self, command: str) -> None:
             status.append(f"Running {short}", style=PALETTE.text_secondary)
             self._thinking_status().update(status)
 
-        def _render_slash_footer(
-            self, command: str, *, handled: bool, error: str | None
-        ) -> None:
+        def _render_slash_footer(self, command: str, *, handled: bool, error: str | None) -> None:
             started = self._active_slash_started_at
             elapsed = None
             if started is not None:
@@ -3390,7 +3400,9 @@ def _handle_rlm_abort_fast(self, args: list[str]) -> bool:
                     f"[yellow]Requested cancellation for run '{run_id}'.[/yellow]"
                 )
             else:
-                self._chat_log().write("[yellow]Requested cancellation for all active runs.[/yellow]")
+                self._chat_log().write(
+                    "[yellow]Requested cancellation for all active runs.[/yellow]"
+                )
 
             if active_runs:
                 joined = ", ".join(str(item) for item in active_runs)
@@ -3400,9 +3412,7 @@ def _handle_rlm_abort_fast(self, args: list[str]) -> bool:
 
             if pending:
                 joined = ", ".join(str(item) for item in pending)
-                self._chat_log().write(
-                    f"[dim]Pending run-specific cancellations:[/dim] {joined}"
-                )
+                self._chat_log().write(f"[dim]Pending run-specific cancellations:[/dim] {joined}")
 
             return True
 

tests/test_rlm_runner.py

@@ -265,8 +265,7 @@ def test_rlm_status_and_events(tmp_path):
 def test_rlm_run_task_can_be_cancelled(tmp_path):
     connector = _FakeConnector(
         responses=[
-            '{"action":"run_python","code":"print(\\"tick\\")","done":false}'
-            for _ in range(200)
+            '{"action":"run_python","code":"print(\\"tick\\")","done":false}' for _ in range(200)
         ]
     )
     engine = _SlowExecutionEngine(delay_seconds=0.03)
@@ -297,9 +296,7 @@ def test_rlm_run_task_can_be_cancelled(tmp_path):
 
 
 def test_rlm_request_cancel_all_without_active_runs_does_not_latch(tmp_path):
-    connector = _FakeConnector(
-        responses=['{"action":"final","done":true,"final_response":"done"}']
-    )
+    connector = _FakeConnector(responses=['{"action":"final","done":true,"final_response":"done"}'])
     engine = _FakeExecutionEngine()
     runner = RLMRunner(llm_connector=connector, execution_engine=engine, run_dir=tmp_path)