drydock-cli 3.0.117__tar.gz → 3.0.119__tar.gz

{drydock_cli-3.0.117/drydock_cli.egg-info → drydock_cli-3.0.119}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: drydock-cli
-Version: 3.0.117
+Version: 3.0.119
 Summary: Drydock — a local, provider-agnostic terminal coding agent for local LLMs
 Author: Frank Bobe III
 License-Expression: Apache-2.0

{drydock_cli-3.0.117 → drydock_cli-3.0.119}/drydock/recipes.py

@@ -245,6 +245,25 @@ RECIPES: list[tuple[str, list[str], str]] = [
      "Build: pdflatex -interaction=nonstopmode report.tex  (run twice for refs).\n"
      "Escape %, _, &, # in text. pandas: df.to_latex('t.tex', index=False). Confirm\n"
      "report.pdf exists and pdflatex exit code is 0."),
+
+    ("Reinforcement learning — Q-learning / policy on a small environment",
+     ["rl", "reinforcement", "q-learning", "qlearning", "policy", "reward", "agent",
+      "environment", "gridworld", "episode", "bellman", "value", "action", "dqn",
+      "exploration", "epsilon", "gym", "cartpole", "bandit"],
+     "Tabular Q-learning for a discrete env with reset()/step(a)->(s2,r,done):\n"
+     "  Q = np.zeros((nS, nA))\n"
+     "  for ep in range(N):\n"
+     "    s = env.reset()\n"
+     "    for t in range(max_steps):\n"
+     "      a = randint(nA) if rand() < eps else int(Q[s].argmax())   # eps-greedy\n"
+     "      s2, r, done = env.step(a)\n"
+     "      Q[s,a] += alpha * (r + gamma * Q[s2].max() - Q[s,a])       # Bellman update\n"
+     "      s = s2\n"
+     "      if done: break\n"
+     "Greedy policy = Q.argmax(1). Typical alpha=0.5, gamma=0.95, eps=0.1-0.2 (decay it).\n"
+     "Sanity: average episode reward should rise and the greedy policy should reach the\n"
+     "goal. For large/continuous states use a neural Q (DQN: replay buffer + target net)\n"
+     "or a policy network (REINFORCE: maximize sum of log_prob(a)*return)."),
 ]
 
 _WORD = re.compile(r"[a-z0-9.]+")

drydock_cli-3.0.119/drydock/suggest.py

@@ -0,0 +1,30 @@
+"""Recommended next command — a single dimmed hint shown after each turn (à la
+Claude Code's ghost suggestion). Pure logic so it's testable; the TUI renders it.
+"""
+from __future__ import annotations
+
+
+def suggest_next_command(
+    *,
+    ctx_pct: int,
+    wrote_files: bool,
+    ran_bash: bool,
+    had_error: bool,
+    in_git: bool,
+    plan_remaining: bool,
+) -> str | None:
+    """The single most useful next step given what the last turn did, or None to
+    show nothing (better no hint than a noisy one). Order = priority."""
+    if ctx_pct >= 78:
+        return "/compact"          # context nearly full — shrink it before continuing
+    if had_error:
+        return "fix the error above"
+    if plan_remaining:
+        return "continue"          # the model left plan steps unfinished
+    if wrote_files and in_git:
+        return "git diff"          # review what changed
+    if wrote_files:
+        return "run the tests"
+    if ran_bash:
+        return None                # ran a command but nothing else obvious — no clutter
+    return None

{drydock_cli-3.0.117 → drydock_cli-3.0.119}/drydock/tui/app.py

@@ -111,6 +111,8 @@ class DrydockApp(App):
         height: auto; margin: 0 2 1 2; padding: 0 1; color: #d7e6ee;
         border-left: thick #c9a227; background: #14241c;
     }
+    /* Dimmed recommended-next-command hint (empty → 0 lines). */
+    #suggest { height: auto; margin: 0 3; color: #4a6b78; text-style: italic; }
     /* One bottom-docked footer holds, top→bottom: working-line, prompt, status. */
     #footer { dock: bottom; height: auto; background: #0b1f2a; }
     #prompt {
@@ -140,6 +142,7 @@ class DrydockApp(App):
         # does the same. priority so it isn't swallowed by the focused prompt.
         Binding("escape", "stop", "Stop the running turn", priority=True),
         Binding("ctrl+o", "toggle_tools", "Expand/collapse details"),
+        Binding("ctrl+n", "use_suggestion", "Use suggested next command", show=False),
         # Scroll the transcript from the keyboard (focus stays on the prompt,
         # and SSH sessions often don't forward the mouse wheel). priority=True
         # so the prompt's TextArea doesn't swallow PageUp/PageDown first.
@@ -250,6 +253,13 @@ class DrydockApp(App):
         self._skills = load_skills(config.get("cwd") or ".")
         self._current_assistant: AssistantMessage | None = None
         self._last_card: ToolCard | None = None
+        # Recommended-next-command hint state.
+        self._turn_tools: set[str] = set()   # tool names used this turn
+        self._turn_error = False
+        self._plan_remaining = False
+        self._suggestion = ""
+        import os
+        self._in_git = os.path.isdir(os.path.join(config.get("cwd") or ".", ".git"))
         self._busy = False
         # STOP signal: Escape / "/stop" sets it; the agent loop checks it at safe
         # points and ends the turn cleanly (session preserved). Lives in config
@@ -300,6 +310,7 @@ class DrydockApp(App):
         with Vertical(id="footer"):
             yield Static("", id="todo")     # pinned task checklist (empty = hidden)
             yield Static("", id="working")  # in-line activity (empty when idle)
+            yield Static("", id="suggest")  # dimmed recommended-next-command hint
             yield PromptArea(id="prompt")
             yield Static(self._status_text(), id="status")
 
@@ -446,6 +457,9 @@ class DrydockApp(App):
         # turn. If this turn emits its own todo, _render_todo repopulates it.
         self.config.pop("_todo", None)
         self.query_one("#todo", Static).update("")
+        self._turn_tools = set()
+        self._turn_error = False
+        self.query_one("#suggest", Static).update("")  # hide the hint while working
         self._busy = True
         self._work_start = time.monotonic()
         self._last_progress = self._work_start
@@ -1210,6 +1224,7 @@ class DrydockApp(App):
 
     def on_agent_tool_start(self, m: AgentToolStart) -> None:
         self._current_assistant = None  # end the current text block
+        self._turn_tools.add(m.name)
         if m.name == "todo":
             self._render_todo(m.inputs.get("tasks", ""))
             self._last_card = None  # tool_end is a no-op for the checklist
@@ -1229,6 +1244,7 @@ class DrydockApp(App):
 
         items = parse_todo(tasks)
         panel = self.query_one("#todo", Static)
+        self._plan_remaining = any(s != "done" for _, s in items)
         if not items:
             panel.update("")  # collapses to 0 height
             return
@@ -1269,9 +1285,38 @@ class DrydockApp(App):
             return
         self._repeat = None
         self._refresh_status()
+        self._update_suggestion()
         self.query_one("#prompt", PromptArea).focus()
 
+    def _update_suggestion(self) -> None:
+        """Compute + render the dimmed recommended-next-command hint (Claude-Code-
+        style). Empty when nothing useful to suggest."""
+        from drydock.suggest import suggest_next_command
+        limit = self.config.get("context_limit", 65536) or 65536
+        pct = min(100, round(self._ctx_tokens / limit * 100)) if limit else 0
+        wrote = bool(self._turn_tools & {"Write", "Edit"})
+        ran = "Bash" in self._turn_tools
+        self._suggestion = suggest_next_command(
+            ctx_pct=pct, wrote_files=wrote, ran_bash=ran, had_error=self._turn_error,
+            in_git=self._in_git, plan_remaining=self._plan_remaining,
+        ) or ""
+        hint = self.query_one("#suggest", Static)
+        if self._suggestion:
+            from rich.markup import escape
+            hint.update(f"→ next: {escape(self._suggestion)}   [dim](ctrl+n to use)[/]")
+        else:
+            hint.update("")
+
+    def action_use_suggestion(self) -> None:
+        """Accept the recommended next command — drop it into the prompt to edit/send."""
+        if self._suggestion and not self._busy:
+            prompt = self.query_one("#prompt", PromptArea)
+            prompt.text = self._suggestion
+            prompt.move_cursor(prompt.document.end)
+            prompt.focus()
+
     def on_agent_error(self, m: AgentError) -> None:
+        self._turn_error = True
         self._mount(ErrorMessage(m.error))
 
 

{drydock_cli-3.0.117 → drydock_cli-3.0.119/drydock_cli.egg-info}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: drydock-cli
-Version: 3.0.117
+Version: 3.0.119
 Summary: Drydock — a local, provider-agnostic terminal coding agent for local LLMs
 Author: Frank Bobe III
 License-Expression: Apache-2.0

{drydock_cli-3.0.117 → drydock_cli-3.0.119}/drydock_cli.egg-info/SOURCES.txt

@@ -25,6 +25,7 @@ drydock/rmf.py
 drydock/rmf_graph.py
 drydock/skills.py
 drydock/stig.py
+drydock/suggest.py
 drydock/tool_registry.py
 drydock/tuning.py
 drydock/web.py
@@ -108,6 +109,7 @@ tests/test_stig.py
 tests/test_stop.py
 tests/test_streaming_newlines.py
 tests/test_subagent.py
+tests/test_suggest.py
 tests/test_system_prompt_help.py
 tests/test_todo.py
 tests/test_tool_arg_coercion.py

{drydock_cli-3.0.117 → drydock_cli-3.0.119}/pyproject.toml

@@ -7,7 +7,7 @@ build-backend = "setuptools.build_meta"
 # PyPI distribution name is drydock-cli (the established install name, continued
 # from the retired v2 fork); the import package + CLI command stay `drydock`.
 name = "drydock-cli"
-version = "3.0.117"
+version = "3.0.119"
 description = "Drydock — a local, provider-agnostic terminal coding agent for local LLMs"
 readme = "README.md"
 requires-python = ">=3.11"

drydock_cli-3.0.119/tests/test_suggest.py

@@ -0,0 +1,20 @@
+"""Recommended-next-command suggestion logic (priority-ordered, None when noisy)."""
+from drydock.suggest import suggest_next_command as s
+
+
+def test_priority_and_cases():
+    base = dict(ctx_pct=10, wrote_files=False, ran_bash=False, had_error=False,
+                in_git=False, plan_remaining=False)
+    assert s(**{**base, "ctx_pct": 80}) == "/compact"          # ctx wins
+    assert s(**{**base, "ctx_pct": 80, "had_error": True}) == "/compact"
+    assert s(**{**base, "had_error": True}) == "fix the error above"
+    assert s(**{**base, "plan_remaining": True}) == "continue"
+    assert s(**{**base, "wrote_files": True, "in_git": True}) == "git diff"
+    assert s(**{**base, "wrote_files": True}) == "run the tests"
+    assert s(**{**base, "ran_bash": True}) is None             # no clutter
+    assert s(**base) is None
+
+
+def test_error_beats_plan_and_writes():
+    assert s(ctx_pct=10, wrote_files=True, ran_bash=True, had_error=True,
+             in_git=True, plan_remaining=True) == "fix the error above"