ommlds 0.0.0.dev460__py3-none-any.whl → 0.0.0.dev461__py3-none-any.whl

ommlds/cli/sessions/chat/printing.py

@@ -1,6 +1,7 @@
 import abc
 import typing as ta
 
+from omdev.tui import rich
 from omlish import check
 from omlish import lang
 from omlish.formats import json
@@ -8,11 +9,6 @@ from omlish.formats import json
 from .... import minichain as mc
 
 
-with lang.auto_proxy_import(globals()):
-    from rich import console as rich_console
-    from rich import markdown as rich_markdown
-
-
 ##
 
 
@@ -22,7 +18,7 @@ class ChatSessionPrinter(lang.Abstract):
         raise NotImplementedError
 
 
-##
+#
 
 
 class StringChatSessionPrinter(ChatSessionPrinter, lang.Abstract):
@@ -59,7 +55,7 @@ class StringChatSessionPrinter(ChatSessionPrinter, lang.Abstract):
             raise TypeError(obj)
 
 
-##
+#
 
 
 class SimpleStringChatSessionPrinter(StringChatSessionPrinter):
@@ -82,7 +78,7 @@ class SimpleStringChatSessionPrinter(StringChatSessionPrinter):
         self._str_printer(s)
 
 
-##
+#
 
 
 class MarkdownStringChatSessionPrinter(StringChatSessionPrinter):
@@ -91,4 +87,40 @@ class MarkdownStringChatSessionPrinter(StringChatSessionPrinter):
         if not s:
             return
 
-        rich_console.Console().print(rich_markdown.Markdown(s))
+        rich.Console().print(rich.Markdown(s))
+
+
+##
+
+
+class StreamPrinter(lang.ExitStacked, lang.Abstract):
+    @abc.abstractmethod
+    def feed(self, s: str) -> None:
+        raise NotImplementedError
+
+
+#
+
+
+class SimpleStreamPrinter(StreamPrinter):
+    def feed(self, s: str) -> None:
+        print(s, end='', flush=True)
+
+    def _exit_contexts(self) -> None:
+        super()._exit_contexts()
+        print(flush=True)
+
+
+#
+
+
+class MarkdownStreamPrinter(StreamPrinter):
+    def __init__(self) -> None:
+        super().__init__()
+
+    def _enter_contexts(self) -> None:
+        super()._enter_contexts()
+        self._ir: rich.MarkdownLiveStream = self._enter_context(rich.IncrementalMarkdownLiveStream())
+
+    def feed(self, s: str) -> None:
+        self._ir.feed(s)

ommlds/cli/sessions/chat/prompt.py

@@ -9,6 +9,9 @@ from .base import DEFAULT_CHAT_MODEL_BACKEND
 from .base import ChatOptions
 from .base import ChatSession
 from .printing import ChatSessionPrinter
+from .printing import MarkdownStreamPrinter
+from .printing import SimpleStreamPrinter
+from .printing import StreamPrinter
 from .state import ChatStateManager
 from .tools import ToolUseExecutor
 
@@ -81,13 +84,20 @@ class PromptChatSession(ChatSession['PromptChatSession.Config']):
                     (self._chat_options or []),
             ))).v as st_resp:
                 lst: list[str] = []
-                async for o in st_resp:
-                    if o:
-                        c = check.isinstance(check.single(check.single(o.choices).deltas), mc.ContentAiChoiceDelta).c
-                        if c is not None:
-                            print(check.isinstance(c, str), end='', flush=True)
-                            lst.append(check.isinstance(c, str))
-                print()
+
+                sp_cls: type[StreamPrinter]
+                if self._config.markdown:
+                    sp_cls = MarkdownStreamPrinter
+                else:
+                    sp_cls = SimpleStreamPrinter
+                with sp_cls() as sp:
+                    async for o in st_resp:
+                        if o:
+                            c = check.isinstance(check.single(check.single(o.choices).deltas), mc.ContentAiChoiceDelta).c  # noqa
+                            if c is not None:
+                                s = check.isinstance(c, str)
+                                sp.feed(s)
+                                lst.append(s)
 
             resp_m = mc.AiMessage(''.join(lst))
             new_chat.append(resp_m)

{ommlds-0.0.0.dev460.dist-info → ommlds-0.0.0.dev461.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: ommlds
-Version: 0.0.0.dev460
+Version: 0.0.0.dev461
 Summary: ommlds
 Author: wrmsr
 License-Expression: BSD-3-Clause
@@ -14,8 +14,8 @@ Classifier: Programming Language :: Python :: 3.13
 Requires-Python: >=3.13
 Description-Content-Type: text/markdown
 License-File: LICENSE
-Requires-Dist: omdev==0.0.0.dev460
-Requires-Dist: omlish==0.0.0.dev460
+Requires-Dist: omdev==0.0.0.dev461
+Requires-Dist: omlish==0.0.0.dev461
 Provides-Extra: all
 Requires-Dist: llama-cpp-python~=0.3; extra == "all"
 Requires-Dist: mlx~=0.29; extra == "all"

{ommlds-0.0.0.dev460.dist-info → ommlds-0.0.0.dev461.dist-info}/RECORD

@@ -91,8 +91,8 @@ ommlds/cli/sessions/chat/base.py,sha256=GlDBXZ-KymSokNl1a7uwX1abswmRE67Q0zOIHjhZ
 ommlds/cli/sessions/chat/code.py,sha256=6hnjibVMTxSUciIIsDPsLHLGMlJNNUZtVQJxFkpi8sM,4269
 ommlds/cli/sessions/chat/inject.py,sha256=uF-QJvONu5sntXxaEu5KTYy2MUaSY8c5oJz0HpK3c7E,2621
 ommlds/cli/sessions/chat/interactive.py,sha256=gurYiDCQSY_5llcE75zvEbpVp77AbKSfgY243Z68bLw,2039
-ommlds/cli/sessions/chat/printing.py,sha256=b1HyripqYnRobLyTPgRidcDLvxIpyqPzgUMiXdklBGE,2327
-ommlds/cli/sessions/chat/prompt.py,sha256=T2h04r9LYA9urYKQZZldoT0Jon_jpKgcbQmtS9xquj8,4915
+ommlds/cli/sessions/chat/printing.py,sha256=3f6icLamt7Wd7oiTshGVnIgC5YAsx_CBvZ4QH_7_mFs,2921
+ommlds/cli/sessions/chat/prompt.py,sha256=tB8Wsy1FYXU6Gq-xOUTVpk8PVAu03mh7lcXf7xO7uJ4,5288
 ommlds/cli/sessions/chat/state.py,sha256=CjYkcpZ4fEBE0kPl1g101EDzq0MKy6S6E84dfDhwzcQ,2649
 ommlds/cli/sessions/chat/tools.py,sha256=dj6ADr6RIJnoIKH5wxRLGkgKAfZ_TtnqNLQ6uO-1Gfg,2325
 ommlds/cli/sessions/completion/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
@@ -325,9 +325,9 @@ ommlds/wiki/utils/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU
 ommlds/wiki/utils/io.py,sha256=UKgDJGtmpnWvIqVd2mJc2QNPOqlToEY1GEveNp6_pMo,7088
 ommlds/wiki/utils/progress.py,sha256=EhvKcMFYtsarCQhIahlO6f0SboyAKP3UwUyrnVnP-Vk,3222
 ommlds/wiki/utils/xml.py,sha256=vVV8Ctn13aaRM9eYfs9Wd6rHn5WOCEUzQ44fIhOvJdg,3754
-ommlds-0.0.0.dev460.dist-info/licenses/LICENSE,sha256=B_hVtavaA8zCYDW99DYdcpDLKz1n3BBRjZrcbv8uG8c,1451
-ommlds-0.0.0.dev460.dist-info/METADATA,sha256=5845ATkqi5W-P5CrEj6n1JPkkp-cQ8CypPhcTMRzBfw,3224
-ommlds-0.0.0.dev460.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-ommlds-0.0.0.dev460.dist-info/entry_points.txt,sha256=Z5YWtX7ClfiCKdW-dd_CSVvM0h4yQpJPi-2G3q6gNFo,35
-ommlds-0.0.0.dev460.dist-info/top_level.txt,sha256=Rbnk5d5wi58vnAXx13WFZqdQ4VX8hBCS2hEL3WeXOhY,7
-ommlds-0.0.0.dev460.dist-info/RECORD,,
+ommlds-0.0.0.dev461.dist-info/licenses/LICENSE,sha256=B_hVtavaA8zCYDW99DYdcpDLKz1n3BBRjZrcbv8uG8c,1451
+ommlds-0.0.0.dev461.dist-info/METADATA,sha256=ictJNO5iWFoU9TN6EDbf_r6HMeWvbLYsr-n0zf7-cSg,3224
+ommlds-0.0.0.dev461.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+ommlds-0.0.0.dev461.dist-info/entry_points.txt,sha256=Z5YWtX7ClfiCKdW-dd_CSVvM0h4yQpJPi-2G3q6gNFo,35
+ommlds-0.0.0.dev461.dist-info/top_level.txt,sha256=Rbnk5d5wi58vnAXx13WFZqdQ4VX8hBCS2hEL3WeXOhY,7
+ommlds-0.0.0.dev461.dist-info/RECORD,,