codebind 0.1.1__tar.gz → 0.1.2__tar.gz

{codebind-0.1.1 → codebind-0.1.2}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: codebind
-Version: 0.1.1
+Version: 0.1.2
 Summary: A minimal model loop over a persistent IPython session.
 Keywords: ai,ipython,llm,repl,agents
 Author: Giovanni Gravili

{codebind-0.1.1 → codebind-0.1.2}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "codebind"
-version = "0.1.1"
+version = "0.1.2"
 description = "A minimal model loop over a persistent IPython session."
 readme = "README.md"
 requires-python = ">=3.13"

{codebind-0.1.1 → codebind-0.1.2}/pyproject.toml.orig

@@ -1,6 +1,6 @@
 [project]
 name = "codebind"
-version = "0.1.1"
+version = "0.1.2"
 description = "A minimal model loop over a persistent IPython session."
 readme = "README.md"
 authors = [

{codebind-0.1.1 → codebind-0.1.2}/src/codebind/execution.py

@@ -39,14 +39,18 @@ class IPythonExecutor:
             raise ValueError("cell must be a non-empty string")
 
         original_prompts = getattr(self.shell, "prompts", None)
+        displayhook = self.shell.displayhook
+        original_output_prompt = displayhook.write_output_prompt
         if prompts is not None and original_prompts is not None:
             self.shell.prompts = prompts
+            displayhook.write_output_prompt = lambda: None
         try:
             with capture_output() as captured:
                 result = self.shell.run_cell(cell, store_history=False)
         finally:
             if prompts is not None and original_prompts is not None:
                 self.shell.prompts = original_prompts
+                displayhook.write_output_prompt = original_output_prompt
 
         displays: list[str] = []
         for output in captured.outputs:

{codebind-0.1.1 → codebind-0.1.2}/src/codebind/prompts.py

@@ -2,6 +2,8 @@
 
 from __future__ import annotations
 
+import sys
+
 from IPython.terminal.prompts import Prompts
 from IPython.terminal.ptutils import IPythonPTLexer
 from prompt_toolkit.document import Document
@@ -44,6 +46,7 @@ def render_cell(shell: object, cell: str, prompts: CellPrompts) -> None:
     lexer = IPythonPTLexer().lex_document(Document(cell))
     pt_app = getattr(shell, "pt_app", None)
     style = pt_app.app.style if pt_app is not None else None
+    sys.stdout.write(getattr(shell, "separate_in", "\n"))
 
     for index, _line in enumerate(cell.split("\n")):
         prompt_tokens = (
@@ -53,3 +56,10 @@ def render_cell(shell: object, cell: str, prompts: CellPrompts) -> None:
         )
         print_formatted_text(PygmentsTokens(prompt_tokens), style=style, end="")
         print_formatted_text(FormattedText(lexer(index)), style=style)
+
+
+def render_output_prompt(shell: object, prompts: CellPrompts) -> None:
+    """Render an output prompt with IPython's terminal style."""
+    pt_app = getattr(shell, "pt_app", None)
+    style = pt_app.app.style if pt_app is not None else None
+    print_formatted_text(PygmentsTokens(prompts.out_prompt_tokens()), style=style, end="")

{codebind-0.1.1 → codebind-0.1.2}/src/codebind/rendering.py

@@ -10,9 +10,10 @@ from rich.segment import Segments
 from rich.text import Text
 
 from .execution import ExecutionReport
+from .prompts import CellPrompts, render_output_prompt
 
 
-_MAX_OUTPUT_LINES = 10
+_MAXIMUM_OUTPUT_LINES = 10
 
 
 def _visible_output(report: ExecutionReport) -> str:
@@ -32,7 +33,7 @@ class TerminalRenderer:
     def __init__(self) -> None:
         self.console = Console()
 
-    def tool_output(self, report: ExecutionReport) -> None:
+    def tool_output(self, report: ExecutionReport, shell: object, prompts: CellPrompts) -> None:
         """Show a bounded preview while leaving the report itself intact for the model."""
         visible = _visible_output(report)
         if not visible:
@@ -40,7 +41,9 @@ class TerminalRenderer:
         output = Text.from_ansi(visible)
         options = self.console.options
         complete = self.console.render_lines(output, options, pad=False, new_lines=True)
-        preview = complete[:_MAX_OUTPUT_LINES]
+        preview = complete[:_MAXIMUM_OUTPUT_LINES]
+        if report.ok:
+            render_output_prompt(shell, prompts)
         self.console.print(Segments(chain.from_iterable(preview)), end="")
         if len(complete) > len(preview):
             self.console.print(

{codebind-0.1.1 → codebind-0.1.2}/src/codebind/session.py

@@ -77,7 +77,6 @@ class Session:
         self.instructions = instructions.strip() if instructions else None
         self.messages: list[BaseMessage] = []
         self.last_response: AIMessage | None = None
-        self.cell_number = 0
         if self.instructions:
             self.messages.append(SystemMessage(self.instructions))
 
@@ -85,7 +84,6 @@ class Session:
         """Clear conversation history without clearing the shared Python namespace."""
         self.messages.clear()
         self.last_response = None
-        self.cell_number = 0
         if self.instructions:
             self.messages.append(SystemMessage(self.instructions))
 
@@ -136,12 +134,11 @@ class Session:
             return _tool_error("InvalidArguments", "ipython requires a string cell argument")
 
         cell = arguments["cell"]
-        self.cell_number += 1
-        prompts = CellPrompts(self.shell, self.cell_number)
+        prompts = CellPrompts(self.shell, self.shell.execution_count - 1)
         render_cell(self.shell, cell, prompts)
         try:
             report = self.executor.execute(cell, prompts=prompts)
         except Exception as error:  # The failure must be returned to the model, not end the session.
             report = _tool_error(type(error).__name__, str(error))
-        self.renderer.tool_output(report)
+        self.renderer.tool_output(report, self.shell, prompts)
         return report