codebind 0.6.8__tar.gz → 0.6.9__tar.gz

{codebind-0.6.8 → codebind-0.6.9}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: codebind
-Version: 0.6.8
+Version: 0.6.9
 Summary: A frontend-neutral model loop for persistent IPython sessions.
 Keywords: ai,ipython,llm,repl,agents
 Author: Giovanni Gravili
@@ -75,7 +75,7 @@ Inspect this project and tell me what to implement first.
 ```
 
 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. To keep terminal scrollback usable, Codebind shows at most 10 gray lines of each model-executed cell's output, then displays a notice such as `… 990 more output lines. Run %output 8 to view all.` The complete output remains available to the model and in IPython history; `%output 8` opens its full text in IPython's pager. User-entered Python cells keep IPython's normal display behavior. 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.
+Model-executed cells appear with IPython's native input counter, syntax highlighting, output prompts, streams, and tracebacks. Assistant Markdown appears before the following tool cell, and tool output appears as it is written. To keep terminal scrollback usable, Codebind shows at most 10 gray lines of each model-executed cell's output, then displays a notice such as `… 990 more output lines. Run %output 8 to view all.` The complete output remains available to the model and in IPython history; `%output 8` opens its full text in IPython's pager. User-entered Python cells keep IPython's normal display behavior. 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.8 → codebind-0.6.9}/README.md

@@ -47,7 +47,7 @@ Inspect this project and tell me what to implement first.
 ```
 
 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. To keep terminal scrollback usable, Codebind shows at most 10 gray lines of each model-executed cell's output, then displays a notice such as `… 990 more output lines. Run %output 8 to view all.` The complete output remains available to the model and in IPython history; `%output 8` opens its full text in IPython's pager. User-entered Python cells keep IPython's normal display behavior. 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.
+Model-executed cells appear with IPython's native input counter, syntax highlighting, output prompts, streams, and tracebacks. Assistant Markdown appears before the following tool cell, and tool output appears as it is written. To keep terminal scrollback usable, Codebind shows at most 10 gray lines of each model-executed cell's output, then displays a notice such as `… 990 more output lines. Run %output 8 to view all.` The complete output remains available to the model and in IPython history; `%output 8` opens its full text in IPython's pager. User-entered Python cells keep IPython's normal display behavior. 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.8 → codebind-0.6.9}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "codebind"
-version = "0.6.8"
+version = "0.6.9"
 description = "A frontend-neutral model loop for persistent IPython sessions."
 readme = "README.md"
 requires-python = ">=3.13"

{codebind-0.6.8 → codebind-0.6.9}/pyproject.toml.orig

@@ -1,6 +1,6 @@
 [project]
 name = "codebind"
-version = "0.6.8"
+version = "0.6.9"
 description = "A frontend-neutral model loop for persistent IPython sessions."
 readme = "README.md"
 authors = [

{codebind-0.6.8 → codebind-0.6.9}/src/codebind/session.py

@@ -274,8 +274,24 @@ class Session:
         for identifier in self._pending_calls(turn_id):
             self._append_message_sync(self._interrupted_tool_message(identifier), turn_id)
 
-    def _update_assistant_stream(self, cell_id: str | None, text: str) -> str | None:
-        if self.bridge is None or not text:
+    def _update_assistant_stream(
+        self,
+        cell_id: str | None,
+        text: str,
+        shown: str,
+        terminal_displayed: list[str] | None = None,
+    ) -> str | None:
+        if not text:
+            return cell_id
+        if self.bridge is None:
+            previous = terminal_displayed[0] if terminal_displayed is not None else shown
+            if previous.startswith(text):
+                return cell_id
+            addition = text[len(previous) :] if text.startswith(previous) else text
+            if addition:
+                self._display_assistant(addition)
+            if terminal_displayed is not None:
+                terminal_displayed[0] = text
             return cell_id
         if cell_id is None:
             return self.bridge.start_markdown_cell(text)
@@ -284,7 +300,9 @@ class Session:
 
     def _finish_assistant_stream(self, cell_id: str | None, response: AIMessage) -> None:
         answer = _message_text(response)
-        if cell_id is not None and self.bridge is not None:
+        if self.bridge is None:
+            return
+        if cell_id is not None:
             self.bridge.finish_markdown_cell(cell_id, answer)
         elif answer:
             self._display_assistant(answer)
@@ -302,9 +320,10 @@ class Session:
         self,
         model: BaseChatModel,
     ) -> tuple[AIMessage, str | None]:
+        terminal_displayed = [""]
         for retry in range(len(_MODEL_RETRY_DELAYS) + 1):
             try:
-                return self._stream_model(self._bind(model), model)
+                return self._stream_model(self._bind(model), model, terminal_displayed)
             except TransientProviderError as error:
                 if retry == len(_MODEL_RETRY_DELAYS):
                     raise
@@ -316,9 +335,10 @@ class Session:
         self,
         model: BaseChatModel,
     ) -> tuple[AIMessage, str | None]:
+        terminal_displayed = [""]
         for retry in range(len(_MODEL_RETRY_DELAYS) + 1):
             try:
-                return await self._astream_model(self._bind(model), model)
+                return await self._astream_model(self._bind(model), model, terminal_displayed)
             except TransientProviderError as error:
                 if retry == len(_MODEL_RETRY_DELAYS):
                     raise
@@ -344,6 +364,7 @@ class Session:
         self,
         bound_model: Runnable[Any, BaseMessage],
         model: BaseChatModel,
+        terminal_displayed: list[str] | None = None,
     ) -> tuple[AIMessage, str | None]:
         aggregate: AIMessageChunk | None = None
         response: AIMessage | None = None
@@ -366,12 +387,14 @@ class Session:
                     raise TypeError("model stream must return AIMessage chunks")
                 text = _message_text(current)
                 if tool_call_started and text != shown:
-                    cell_id = self._update_assistant_stream(cell_id, text)
+                    cell_id = self._update_assistant_stream(
+                        cell_id, text, shown, terminal_displayed
+                    )
                     shown = text
             completed = self._completed_stream_message(aggregate, response)
             answer = _message_text(completed)
             if answer != shown:
-                cell_id = self._update_assistant_stream(cell_id, answer)
+                cell_id = self._update_assistant_stream(cell_id, answer, shown, terminal_displayed)
             return completed, cell_id
         except BaseException:
             self._cancel_assistant_stream(cell_id)
@@ -381,6 +404,7 @@ class Session:
         self,
         bound_model: Runnable[Any, BaseMessage],
         model: BaseChatModel,
+        terminal_displayed: list[str] | None = None,
     ) -> tuple[AIMessage, str | None]:
         aggregate: AIMessageChunk | None = None
         response: AIMessage | None = None
@@ -403,12 +427,14 @@ class Session:
                     raise TypeError("model stream must return AIMessage chunks")
                 text = _message_text(current)
                 if tool_call_started and text != shown:
-                    cell_id = self._update_assistant_stream(cell_id, text)
+                    cell_id = self._update_assistant_stream(
+                        cell_id, text, shown, terminal_displayed
+                    )
                     shown = text
             completed = self._completed_stream_message(aggregate, response)
             answer = _message_text(completed)
             if answer != shown:
-                cell_id = self._update_assistant_stream(cell_id, answer)
+                cell_id = self._update_assistant_stream(cell_id, answer, shown, terminal_displayed)
             return completed, cell_id
         except BaseException:
             self._cancel_assistant_stream(cell_id)

{codebind-0.6.8 → codebind-0.6.9}/src/codebind/terminal.py

@@ -30,6 +30,20 @@ def _plain_ansi(value: str) -> str:
     return "".join(text for _, text in to_formatted_text(ANSI(value)))
 
 
+def _split_incomplete_ansi(value: str) -> tuple[str, str]:
+    last_escape = value.rfind("\x1b")
+    if last_escape < 0:
+        return value, ""
+    tail = value[last_escape:]
+    if tail == "\x1b":
+        return value[:last_escape], tail
+    if tail.startswith("\x1b[") and not any("@" <= char <= "~" for char in tail[2:]):
+        return value[:last_escape], tail
+    if tail.startswith("\x1b]") and "\x07" not in tail and "\x1b\\" not in tail:
+        return value[:last_escape], tail
+    return value, ""
+
+
 def _magic_input(transformed: str) -> tuple[str, str] | None:
     """Recognize a complete IPython magic call in transformed history."""
     try:
@@ -220,12 +234,13 @@ class _OutputPreview:
         return len(data)
 
     def show_notice(self, number: int, error: dict[str, str] | None = None) -> None:
+        if self.visible_characters and not self.visible_ends_line:
+            sys.stdout.write("\n")
+            self.visible_ends_line = True
         if not self.hidden_characters:
             return
         if sys.stdout.isatty():
             sys.stdout.write("\x1b[0m")
-        if not self.visible_ends_line:
-            sys.stdout.write("\n")
         hidden_lines = self.hidden_line_breaks + int(self.hidden_has_tail)
         label = "line" if hidden_lines == 1 else "lines"
         sys.stdout.write(
@@ -246,20 +261,12 @@ class _PreviewStream:
         self.pending = ""
 
     def write(self, data: str) -> int:
-        self.pending += data
-        lines = self.pending.splitlines(keepends=True)
-        if lines and not lines[-1].endswith(("\n", "\r")):
-            self.pending = lines.pop()
-        else:
-            self.pending = ""
-        for line in lines:
-            self.preview.write(self.stream, _plain_ansi(line))
+        complete, self.pending = _split_incomplete_ansi(self.pending + data)
+        if complete:
+            self.preview.write(self.stream, _plain_ansi(complete))
         return len(data)
 
     def flush(self) -> None:
-        if self.pending:
-            self.preview.write(self.stream, _plain_ansi(self.pending))
-            self.pending = ""
         self.stream.flush()
 
     def __getattr__(self, name: str):
@@ -388,6 +395,7 @@ def show_input(shell: TerminalInteractiveShell, cell: str) -> None:
             style=style,
             color_depth=shell.color_depth,
         )
+    sys.stdout.write("\n")
 
 
 def install_markdown_renderer(shell: TerminalInteractiveShell) -> Callable[[], None]:
@@ -399,6 +407,7 @@ def install_markdown_renderer(shell: TerminalInteractiveShell) -> Callable[[], N
     previous_renderer = shell.mime_renderers.get(mime, _MISSING)
 
     def render(text: str, _metadata: object) -> None:
+        sys.stdout.write("\n")
         Console(file=sys.stdout, soft_wrap=True, no_color=shell.colors.lower() == "nocolor").print(
             Markdown(text)
         )