codebind 0.6.6__py3-none-any.whl → 0.6.7__py3-none-any.whl

codebind/extension.py

@@ -13,13 +13,19 @@ from .configuration import load_configuration, load_models
 from .conversation import MemoryConversationStore, NotebookConversationStore
 from .jupyter import JupyterLabBridge, TARGET_NAME
 from .session import Session
-from .terminal import install_markdown_renderer
+from .terminal import TerminalCellHistory, install_markdown_renderer, install_question_mode
 
 
 _STATE_ATTRIBUTE = "_codebind_extension_state"
 
 
 def _close_state(state: dict[str, Any]) -> None:
+    terminal_history = state.get("terminal_history")
+    if isinstance(terminal_history, TerminalCellHistory):
+        terminal_history.close()
+    restore_terminal_question_mode = state.get("restore_terminal_question_mode")
+    if callable(restore_terminal_question_mode):
+        restore_terminal_question_mode()
     restore_terminal_display = state.get("restore_terminal_display")
     if callable(restore_terminal_display):
         restore_terminal_display()
@@ -44,6 +50,9 @@ def _close_state(state: dict[str, Any]) -> None:
 def load_ipython_extension(ipython: InteractiveShell) -> None:
     """Load Codebind into the active IPython session."""
     previous = getattr(ipython, _STATE_ATTRIBUTE, None)
+    previous_terminal_history = (
+        previous.get("terminal_history") if isinstance(previous, dict) else None
+    )
     if isinstance(previous, dict):
         _close_state(previous)
 
@@ -78,12 +87,23 @@ def load_ipython_extension(ipython: InteractiveShell) -> None:
         "comm_manager": manager,
         "comm_target": None,
         "restore_terminal_display": None,
+        "restore_terminal_question_mode": None,
+        "terminal_history": None,
     }
 
     if manager is None:
         state["session"] = Session(shell=ipython, store=MemoryConversationStore())
         if isinstance(ipython, TerminalInteractiveShell):
+            state["terminal_history"] = TerminalCellHistory(
+                ipython,
+                previous=(
+                    previous_terminal_history
+                    if isinstance(previous_terminal_history, TerminalCellHistory)
+                    else None
+                ),
+            )
             state["restore_terminal_display"] = install_markdown_renderer(ipython)
+            state["restore_terminal_question_mode"] = install_question_mode(ipython)
     else:
 
         def accept_comm(comm: Any, message: dict[str, Any]) -> None:
@@ -142,8 +162,14 @@ def load_ipython_extension(ipython: InteractiveShell) -> None:
             session = Session(shell=ipython, store=MemoryConversationStore())
             state["session"] = session
         selected = get_model()
+        terminal_history = state.get("terminal_history")
+        history = (
+            terminal_history.snapshot()
+            if isinstance(terminal_history, TerminalCellHistory)
+            else None
+        )
         try:
-            session.send(cell if cell is not None else line, selected)
+            session.send(cell if cell is not None else line, selected, notebook=history)
         except BaseException:
             asyncio.run(discard_model())
             raise

codebind/instructions.md

@@ -11,7 +11,7 @@ You are a facts-rooted coding agent, operating through the user’s live IPython
 - Ground factual claims and reported results in direct observations or valid, identifiable sources that actually support them; give the user enough detail to verify the source. Never invent or simulate factual data, sources, citations, quotes, tool output, actions, or observations, or present guesses as facts. If evidence is missing, say so rather than fabricate it. Create synthetic or fictional material only when explicitly requested, label it clearly, and never present it as real; treat fabricated evidence presented as real as fraud.
 - Keep the user oriented while working. Before meaningful tool work, emit one brief Markdown update saying what you are about to do. Between tool calls, emit another update only for a new phase, an important finding, or a change of plan. Do not perform several meaningful steps silently, but do not narrate trivial operations.
 - Keep responses brief and direct, free of jargon and human-written with a clear intent.
-- Treat notebook-context messages as the authoritative notebook document state; later deltas override earlier cell versions.
-- Notebook cells and supported images displayed in them, including images from your IPython tool, enter your context automatically. Inspect those images directly before claiming you cannot see them; mention any reported image omission.
+- Treat notebook-context messages as the authoritative cell history for the active notebook or terminal; later deltas override earlier cell versions. Terminal context includes completed IPython inputs and outputs.
+- Notebook and terminal cells, plus supported images displayed in them, including images from your IPython tool, enter your context automatically. Inspect those images directly before claiming you cannot see them; mention any reported image omission.
 - Before importing or using a non-standard library, check whether it is available in the active IPython interpreter. If it is missing, determine a working installer for that environment (such as `uv` or `pip`), install it into that interpreter, and verify the import before proceeding; do not assume installation succeeded.
 - Be very proactive, instead of giving up. You can ask questions to the user to clarify.

codebind/terminal.py

@@ -2,15 +2,20 @@
 
 from __future__ import annotations
 
+import ast
 import sys
 from collections.abc import Callable, Iterator
 from contextlib import contextmanager
+from copy import deepcopy
 from unittest.mock import patch
 
 from IPython.terminal.interactiveshell import IPythonPTLexer, TerminalInteractiveShell
+from prompt_toolkit.enums import DEFAULT_BUFFER
+from prompt_toolkit.filters import Condition, has_focus
 from prompt_toolkit.document import Document
 from prompt_toolkit.formatted_text import FormattedText, PygmentsTokens
 from prompt_toolkit.shortcuts import print_formatted_text
+from prompt_toolkit.styles import DynamicStyle, Style, merge_styles
 from rich.console import Console
 from rich.markdown import Markdown
 
@@ -19,6 +24,206 @@ _MISSING = object()
 _IMAGE_FORMATS = ("image/png", "image/jpeg", "image/svg+xml")
 
 
+def _question_text(transformed: str) -> str | None:
+    """Recover a Codebind question from IPython's own transformed history."""
+    try:
+        statements = ast.parse(transformed).body
+    except SyntaxError:
+        return None
+    if len(statements) != 1 or not isinstance(statements[0], ast.Expr):
+        return None
+    call = statements[0].value
+    if not (
+        isinstance(call, ast.Call)
+        and isinstance(call.func, ast.Attribute)
+        and call.func.attr in {"run_cell_magic", "run_line_magic"}
+        and isinstance(call.func.value, ast.Call)
+        and isinstance(call.func.value.func, ast.Name)
+        and call.func.value.func.id == "get_ipython"
+        and bool(call.args)
+        and isinstance(call.args[0], ast.Constant)
+        and call.args[0].value == "question"
+    ):
+        return None
+    text_index = 2 if call.func.attr == "run_cell_magic" else 1
+    if len(call.args) <= text_index or not isinstance(call.args[text_index], ast.Constant):
+        return None
+    text = call.args[text_index].value
+    return text.strip() if isinstance(text, str) else None
+
+
+def _history_cell(
+    shell: TerminalInteractiveShell,
+    number: int,
+    source: str,
+    transformed: str,
+) -> dict[str, object] | None:
+    """Capture one finished IPython cell before later displays can alter its history."""
+    if not source.strip():
+        return None
+    history = shell.history_manager
+    identifier = f"terminal-{number}"
+    question = _question_text(transformed)
+    if question is not None:
+        return {"id": identifier, "type": "markdown", "source": question}
+
+    outputs: list[dict[str, object]] = []
+    for record in history.outputs.get(number, ()):
+        if record.output_type in {"out_stream", "err_stream"}:
+            stream = record.bundle.get("stream", [])
+            outputs.append(
+                {
+                    "output_type": "stream",
+                    "name": "stdout" if record.output_type == "out_stream" else "stderr",
+                    "text": "".join(stream),
+                }
+            )
+        elif record.output_type in {"display_data", "execute_result"}:
+            output: dict[str, object] = {
+                "output_type": record.output_type,
+                "data": deepcopy(record.bundle),
+                "metadata": {},
+            }
+            if record.output_type == "execute_result":
+                output["execution_count"] = number
+            outputs.append(output)
+    error = history.exceptions.get(number)
+    if error is not None:
+        outputs.append({"output_type": "error", **deepcopy(error)})
+    return {
+        "id": identifier,
+        "type": "code",
+        "source": source,
+        "execution_count": number,
+        "outputs": outputs,
+    }
+
+
+class TerminalCellHistory:
+    """Track completed IPython cells, without reinterpreting mutable output history."""
+
+    def __init__(
+        self,
+        shell: TerminalInteractiveShell,
+        previous: TerminalCellHistory | None = None,
+    ) -> None:
+        self.shell = shell
+        self._cells = deepcopy(previous._cells) if previous is not None else {}
+        history = shell.history_manager
+        for number in range(1, shell.execution_count):
+            if number in self._cells or number >= len(history.input_hist_raw):
+                continue
+            cell = _history_cell(
+                shell,
+                number,
+                history.input_hist_raw[number],
+                history.input_hist_parsed[number],
+            )
+            if cell is not None:
+                self._cells[number] = cell
+        self._callback = self._record
+        shell.events.register("post_run_cell", self._callback)
+
+    def _record(self, result) -> None:
+        if result is None or result.execution_count is None or not result.info.store_history:
+            return
+        cell = _history_cell(
+            self.shell,
+            result.execution_count,
+            result.info.raw_cell,
+            result.info.transformed_cell,
+        )
+        if cell is not None:
+            self._cells[result.execution_count] = cell
+
+    def snapshot(self) -> list[dict[str, object]]:
+        return [deepcopy(self._cells[number]) for number in sorted(self._cells)]
+
+    def close(self) -> None:
+        self.shell.events.unregister("post_run_cell", self._callback)
+
+
+def install_question_mode(shell: TerminalInteractiveShell) -> Callable[[], None]:
+    """Let Shift+Tab switch the terminal prompt between Python and questions."""
+    prompt_session = shell.pt_app
+    if prompt_session is None:
+        return lambda: None
+
+    original_style = prompt_session.style
+    original_prompt_for_code = shell.prompt_for_code
+    had_prompt_override = "prompt_for_code" in vars(shell)
+    bindings = prompt_session.key_bindings
+    transforms = shell.input_transformer_manager.cleanup_transforms
+    question_mode = False
+    active_prompt = False
+    pending_question = False
+
+    def prompt_for_code() -> str:
+        nonlocal active_prompt, pending_question
+        active_prompt = True
+        try:
+            text = original_prompt_for_code()
+        finally:
+            active_prompt = False
+        pending_question = question_mode and bool(text.strip())
+        return text
+
+    def transform_question(lines: list[str]) -> list[str]:
+        nonlocal pending_question
+        if not pending_question:
+            return lines
+        pending_question = False
+        return ["%%question\n", *lines]
+
+    def toggle_question(event) -> None:
+        nonlocal question_mode
+        question_mode = not question_mode
+        event.app.invalidate()
+
+    def submit_question(event) -> None:
+        event.current_buffer.validate_and_handle()
+
+    question_active = Condition(lambda: active_prompt and question_mode)
+    bindings.add("s-tab", filter=has_focus(DEFAULT_BUFFER))(toggle_question)
+    bindings.add("enter", filter=has_focus(DEFAULT_BUFFER) & question_active)(submit_question)
+    shell.prompt_for_code = prompt_for_code
+    question_style = merge_styles(
+        [
+            original_style,
+            Style.from_dict(
+                {
+                    key: "#a855f7"
+                    for key in (
+                        "pygments.prompt",
+                        "pygments.promptnum",
+                        "pygments.prompt.mode",
+                        "pygments.prompt.linenumber",
+                        "pygments.prompt.continuation",
+                        "pygments.prompt.padding",
+                        "pygments.prompt.wrap",
+                    )
+                }
+            ),
+        ]
+    )
+    prompt_session.style = DynamicStyle(
+        lambda: question_style if active_prompt and question_mode else original_style
+    )
+    transforms.insert(0, transform_question)
+
+    def restore() -> None:
+        bindings.remove(toggle_question)
+        bindings.remove(submit_question)
+        transforms.remove(transform_question)
+        if had_prompt_override:
+            shell.prompt_for_code = original_prompt_for_code
+        else:
+            del shell.prompt_for_code
+        prompt_session.style = original_style
+
+    return restore
+
+
 def show_input(shell: TerminalInteractiveShell, cell: str) -> None:
     """Show a model cell with the shell's own input and continuation prompts."""
     sys.stdout.write(shell.separate_in)
@@ -96,4 +301,10 @@ def record_tool_output(shell: TerminalInteractiveShell) -> Iterator[None]:
             shell.mime_renderers["image/png"] = previous_png_renderer
 
 
-__all__ = ["install_markdown_renderer", "record_tool_output", "show_input"]
+__all__ = [
+    "TerminalCellHistory",
+    "install_markdown_renderer",
+    "install_question_mode",
+    "record_tool_output",
+    "show_input",
+]

{codebind-0.6.6.dist-info → codebind-0.6.7.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: codebind
-Version: 0.6.6
+Version: 0.6.7
 Summary: A frontend-neutral model loop for persistent IPython sessions.
 Keywords: ai,ipython,llm,repl,agents
 Author: Giovanni Gravili
@@ -65,14 +65,16 @@ Start a new conversation:
 uvx codebind
 ```
 
-Ask through ordinary IPython magic syntax:
+Press `Shift+Tab` to switch the prompt into Question mode; the prompt turns purple. Type a question and press `Enter` to send it. Press `Shift+Tab` again to return to Python mode. Ordinary `Tab` completion is unchanged.
+
+The `%question` and `%%question` magics remain available:
 
 ```python
 %%question
 Inspect this project and tell me what to implement first.
 ```
 
-Terminal conversations remain live for the current IPython process. Durable automatic resumption belongs to notebooks, where the notebook file provides an unambiguous conversation identity.
+Terminal conversations remain live for the current IPython process. Previous terminal inputs and outputs, including Python cells you ran yourself, enter the model's context when you ask a question. Durable automatic resumption belongs to notebooks, where the notebook file provides an unambiguous conversation identity.
 Model-executed cells appear with IPython's native input counter, syntax highlighting, output prompts, streams, and tracebacks. Assistant Markdown is rendered through IPython's terminal MIME renderer. A terminal is a stream rather than a notebook document, so these displays are not editable notebook cells.
 
 ## JupyterLab

{codebind-0.6.6.dist-info → codebind-0.6.7.dist-info}/RECORD

@@ -4,21 +4,21 @@ codebind/configuration.py,sha256=b9ux-MiepzAFfRhorZp4w2pdLLs44necescIGg_IKfo,182
 codebind/conversation.py,sha256=nFMe2QgFkFOA2GYInN-cO_jhA9010LpM6_Ij2g4Q2BA,17500
 codebind/display.py,sha256=IpxcElyNcr5olRIatEduN_d-sRwpsMlaq1UtrOXocQA,494
 codebind/execution.py,sha256=gqpZP2BFlSnGkdjLXCsouU6Zim9oETz7dmAiHlIybj8,16261
-codebind/extension.py,sha256=kLlpPZxW5OrQSAA0Oh5tXTHqTsSat7B_HjG8qsG5F_o,6050
+codebind/extension.py,sha256=y6fFRzPJIEoTNoP7NAd33tGLq-x2kzKm1Ss2eeHimA4,7233
 codebind/images.py,sha256=9bJHjXc75_YRBFTAqAWiknRTCjs8ApyGVVSxC_S5lio,4342
-codebind/instructions.md,sha256=WQ42UB5mmGcRqj_hQBzYLpcyUh1EuKDFU5dZ3GpB6JM,3664
+codebind/instructions.md,sha256=_8CVGEuKgPRLmxfPnJMP0sNGg_Nh_o6z5iuKHKj57RM,3768
 codebind/jupyter.py,sha256=07H_S9G3iyyUtWRzlvFr5-5X6CuLBXVficu5Fz8jpVw,8652
 codebind/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 codebind/session.py,sha256=b8fEFOFdZn4h1wIr9RE2K5SPWmoVyro5pRhGUr4rI6c,23722
-codebind/terminal.py,sha256=zSnVSHSxVqKg6wyjZm6KtxUTmuyCM46PQoXwVIVxmNw,3819
-codebind-0.6.6.dist-info/licenses/LICENSE,sha256=ujX46e0LFBIBTnCDY0u3Q7LV363B7CiI7QJZxIkEXlk,1073
-codebind-0.6.6.data/data/share/jupyter/labextensions/codebind-jupyterlab/install.json,sha256=78rEThkfrk1pXtekX5WrG9_Y0lJYi5XBEm8n9athpFE,164
-codebind-0.6.6.data/data/share/jupyter/labextensions/codebind-jupyterlab/package.json,sha256=hawl8ZytgxN-WbUjgbzdIVemPPczzmSFoc2HwDxtgIc,1192
-codebind-0.6.6.data/data/share/jupyter/labextensions/codebind-jupyterlab/static/590.dae4561c5427a855.js,sha256=lMbXilytbQEc_kzyVkgzhKZxU_zgdNLTz4uLJfC7APY,16141
-codebind-0.6.6.data/data/share/jupyter/labextensions/codebind-jupyterlab/static/remoteEntry.c10098fb6e87897c.js,sha256=U0JBoOwkRkzPct8MzcUO1pc2XeVGvxgztXBMtJMg2b4,11794
-codebind-0.6.6.data/data/share/jupyter/labextensions/codebind-jupyterlab/static/style.js,sha256=Opy15wyuR_ABqsrk-cOcOj2yZtvtq0r36aDkErnloD0,113
-codebind-0.6.6.data/data/share/jupyter/labextensions/codebind-jupyterlab/static/third-party-licenses.json,sha256=MNToQfru1YnHAie-2Virp23yWsc5co3qG0IzrO6SEfw,20
-codebind-0.6.6.dist-info/WHEEL,sha256=7hzKWg-J8I3Buqyw5tBii5z_MmAVsDYXXijd_QAtNZ8,81
-codebind-0.6.6.dist-info/entry_points.txt,sha256=GiPcU63sULDlywJSKigAS4M3vaziYIrubXOb3fAg2_c,48
-codebind-0.6.6.dist-info/METADATA,sha256=2M2ZoO3gCWYhx_BSEFhhECMyXh4prs8mo2V3nF4wMlY,6959
-codebind-0.6.6.dist-info/RECORD,,
+codebind/terminal.py,sha256=5ehefVtlrx3j6pURz9kA3nCmwE3wnnnJParOD9VqgfI,11263
+codebind-0.6.7.dist-info/licenses/LICENSE,sha256=ujX46e0LFBIBTnCDY0u3Q7LV363B7CiI7QJZxIkEXlk,1073
+codebind-0.6.7.data/data/share/jupyter/labextensions/codebind-jupyterlab/install.json,sha256=78rEThkfrk1pXtekX5WrG9_Y0lJYi5XBEm8n9athpFE,164
+codebind-0.6.7.data/data/share/jupyter/labextensions/codebind-jupyterlab/package.json,sha256=hawl8ZytgxN-WbUjgbzdIVemPPczzmSFoc2HwDxtgIc,1192
+codebind-0.6.7.data/data/share/jupyter/labextensions/codebind-jupyterlab/static/590.dae4561c5427a855.js,sha256=lMbXilytbQEc_kzyVkgzhKZxU_zgdNLTz4uLJfC7APY,16141
+codebind-0.6.7.data/data/share/jupyter/labextensions/codebind-jupyterlab/static/remoteEntry.c10098fb6e87897c.js,sha256=U0JBoOwkRkzPct8MzcUO1pc2XeVGvxgztXBMtJMg2b4,11794
+codebind-0.6.7.data/data/share/jupyter/labextensions/codebind-jupyterlab/static/style.js,sha256=Opy15wyuR_ABqsrk-cOcOj2yZtvtq0r36aDkErnloD0,113
+codebind-0.6.7.data/data/share/jupyter/labextensions/codebind-jupyterlab/static/third-party-licenses.json,sha256=MNToQfru1YnHAie-2Virp23yWsc5co3qG0IzrO6SEfw,20
+codebind-0.6.7.dist-info/WHEEL,sha256=7hzKWg-J8I3Buqyw5tBii5z_MmAVsDYXXijd_QAtNZ8,81
+codebind-0.6.7.dist-info/entry_points.txt,sha256=GiPcU63sULDlywJSKigAS4M3vaziYIrubXOb3fAg2_c,48
+codebind-0.6.7.dist-info/METADATA,sha256=M_JwPsbx6ZC9FZb8qjs34iZ5DONwxtN0eszXB3Oy8L8,7325
+codebind-0.6.7.dist-info/RECORD,,