ommlds 0.0.0.dev489__py3-none-any.whl → 0.0.0.dev491__py3-none-any.whl

ommlds/cli/sessions/chat/chat/user/types.py

@@ -1,15 +0,0 @@
-import abc
-import typing as ta
-
-from omlish import lang
-
-from ...... import minichain as mc
-
-
-##
-
-
-class UserChatInput(lang.Abstract):
-    @abc.abstractmethod
-    def get_next_user_messages(self) -> ta.Awaitable['mc.UserChat']:
-        raise NotImplementedError

ommlds/cli/sessions/chat/driver.py

@@ -1,43 +0,0 @@
-from .chat.ai.types import AiChatGenerator
-from .chat.state.types import ChatStateManager
-from .chat.user.types import UserChatInput
-from .phases.manager import ChatPhaseManager
-from .phases.types import ChatPhase
-
-
-##
-
-
-class ChatDriver:
-    def __init__(
-            self,
-            *,
-            phases: ChatPhaseManager,
-            ai_chat_generator: AiChatGenerator,
-            user_chat_input: UserChatInput,
-            chat_state_manager: ChatStateManager,
-    ):
-        super().__init__()
-
-        self._phases = phases
-        self._ai_chat_generator = ai_chat_generator
-        self._user_chat_input = user_chat_input
-        self._chat_state_manager = chat_state_manager
-
-    async def run(self) -> None:
-        await self._phases.set_phase(ChatPhase.STARTING)
-        await self._phases.set_phase(ChatPhase.STARTED)
-
-        while True:
-            next_user_chat = await self._user_chat_input.get_next_user_messages()
-            if not next_user_chat:
-                break
-
-            prev_user_chat = (await self._chat_state_manager.get_state()).chat
-
-            next_ai_chat = await self._ai_chat_generator.get_next_ai_messages([*prev_user_chat, *next_user_chat])
-
-            await self._chat_state_manager.extend_chat([*next_user_chat, *next_ai_chat])
-
-        await self._phases.set_phase(ChatPhase.STOPPING)
-        await self._phases.set_phase(ChatPhase.STOPPED)

ommlds/cli/sessions/chat/interface/bare/inject.py

@@ -1,32 +0,0 @@
-from omlish import inject as inj
-from omlish import lang
-
-from ..base import ChatInterface
-from ..configs import InterfaceConfig
-from .interface import BareChatInterface
-
-
-with lang.auto_proxy_import(globals()):
-    from .....inputs import asyncs as _inputs_asyncs
-    from .....inputs import sync as _inputs_sync
-    from ...chat.user import types as _user_types
-    from . import user as _user
-
-
-##
-
-
-def bind_bare(cfg: InterfaceConfig = InterfaceConfig()) -> inj.Elements:
-    els: list[inj.Elemental] = [
-        inj.bind(ChatInterface, to_ctor=BareChatInterface, singleton=True),
-    ]
-
-    if cfg.interactive:
-        els.append(inj.bind(_user_types.UserChatInput, to_ctor=_user.InteractiveUserChatInput, singleton=True))
-
-        els.extend([
-            inj.bind(_inputs_sync.SyncStringInput, to_const=_inputs_sync.InputSyncStringInput(use_readline=cfg.use_readline)),  # noqa
-            inj.bind(_inputs_asyncs.AsyncStringInput, to_ctor=_inputs_asyncs.ThreadAsyncStringInput, singleton=True),
-        ])
-
-    return inj.as_elements(*els)

ommlds/cli/sessions/chat/interface/textual/app.py

@@ -1,191 +0,0 @@
-import asyncio
-import dataclasses as dc
-import typing as ta
-
-from omdev.tui import textual as tx
-from omlish import check
-
-from ...... import minichain as mc
-from ...driver import ChatDriver
-from .user import QueueUserChatInput
-
-
-##
-
-
-class UserMessage(tx.Static):
-    pass
-
-
-class AiMessage(tx.Static):
-    pass
-
-
-##
-
-
-class InputTextArea(tx.TextArea):
-    @dc.dataclass()
-    class Submitted(tx.Message):
-        text: str
-
-    def __init__(self, **kwargs: ta.Any) -> None:
-        super().__init__(**kwargs)
-
-    async def _on_key(self, event: tx.Key) -> None:
-        if event.key == 'enter':
-            event.prevent_default()
-            event.stop()
-
-            if text := self.text.strip():
-                self.post_message(self.Submitted(text))
-
-        else:
-            await super()._on_key(event)
-
-
-##
-
-
-class ChatApp(tx.App):
-    def __init__(
-            self,
-            *,
-            chat_driver: ChatDriver,
-            queue_user_chat_input: QueueUserChatInput,
-    ) -> None:
-        super().__init__()
-
-        self._chat_driver = chat_driver
-        self._queue_user_chat_input = queue_user_chat_input
-
-    CSS: ta.ClassVar[str] = """
-        #messages-scroll {
-            width: 100%;
-            height: 1fr;
-
-            padding: 0 2 0 2;
-        }
-
-        #messages-container {
-            height: auto;
-            width: 100%;
-
-            margin-top: 1;
-            margin-bottom: 0;
-
-            layout: stream;
-            text-align: left;
-        }
-
-        #input-outer {
-            width: 100%;
-            height: auto;
-        }
-
-        #input-vertical {
-            width: 100%;
-            height: auto;
-
-            margin: 0 2 1 2;
-
-            padding: 0;
-        }
-
-        #input-vertical2 {
-            width: 100%;
-            height: auto;
-
-            border: round $foreground-muted;
-
-            padding: 0 1;
-        }
-
-        #input-horizontal {
-            width: 100%;
-            height: auto;
-        }
-
-        #input-glyph {
-            width: auto;
-
-            padding: 0 1 0 0;
-
-            background: transparent;
-            color: $primary;
-
-            text-style: bold;
-        }
-
-        #input {
-            width: 1fr;
-            height: auto;
-            max-height: 16;
-
-            border: none;
-
-            padding: 0;
-
-            background: transparent;
-            color: $text;
-        }
-    """
-
-    ENABLE_COMMAND_PALETTE: ta.ClassVar[bool] = False
-
-    #
-
-    def compose(self) -> tx.ComposeResult:
-        with tx.VerticalScroll(id='messages-scroll'):
-            yield tx.Static(id='messages-container')
-
-        with tx.Static(id='input-outer'):
-            with tx.Vertical(id='input-vertical'):
-                with tx.Vertical(id='input-vertical2'):
-                    with tx.Horizontal(id='input-horizontal'):
-                        yield tx.Static('>', id='input-glyph')
-                        yield InputTextArea(placeholder='...', id='input')
-
-    #
-
-    def _get_input_text_area(self) -> InputTextArea:
-        return self.query_one('#input', InputTextArea)
-
-    def _get_messages_container(self) -> tx.Static:
-        return self.query_one('#messages-container', tx.Static)
-
-    #
-
-    async def _mount_message(self, *messages: tx.Widget) -> None:
-        msg_ctr = self._get_messages_container()
-
-        for msg in messages:
-            await msg_ctr.mount(msg)
-
-        self.call_after_refresh(lambda: msg_ctr.scroll_end(animate=False))
-
-    #
-
-    _chat_driver_task: asyncio.Task | None = None
-
-    async def on_mount(self) -> None:
-        check.none(self._chat_driver_task)
-        self._chat_driver_task = asyncio.create_task(self._chat_driver.run())
-
-        self._get_input_text_area().focus()
-
-        await self._mount_message(UserMessage('Hello!'))
-
-    async def on_unmount(self) -> None:
-        await self._queue_user_chat_input.push_next_user_messages([])
-        await check.not_none(self._chat_driver_task)
-
-    async def on_input_text_area_submitted(self, event: InputTextArea.Submitted) -> None:
-        self._get_input_text_area().clear()
-
-        await self._mount_message(
-            UserMessage(event.text),
-            # AiMessage(f'You said: {event.text}!'),
-        )
-
-        await self._queue_user_chat_input.push_next_user_messages([mc.UserMessage(event.text)])

ommlds/cli/sessions/chat/interface/textual/inject.py

@@ -1,27 +0,0 @@
-from omlish import inject as inj
-
-from ...chat.user.types import UserChatInput
-from ..base import ChatInterface
-from .app import ChatApp
-from .interface import TextualChatInterface
-from .user import QueueUserChatInput
-
-
-##
-
-
-def bind_textual() -> inj.Elements:
-    els: list[inj.Elemental] = [
-        inj.bind(ChatInterface, to_ctor=TextualChatInterface, singleton=True),
-    ]
-
-    els.extend([
-        inj.bind(ChatApp, singleton=True),
-    ])
-
-    els.extend([
-        inj.bind(QueueUserChatInput, singleton=True),
-        inj.bind(UserChatInput, to_key=QueueUserChatInput),
-    ])
-
-    return inj.as_elements(*els)

ommlds/cli/sessions/chat/interface/textual/user.py

@@ -1,20 +0,0 @@
-import asyncio
-
-from ...... import minichain as mc
-from ...chat.user.types import UserChatInput
-
-
-##
-
-
-class QueueUserChatInput(UserChatInput):
-    def __init__(self) -> None:
-        super().__init__()
-
-        self._queue: asyncio.Queue[mc.UserChat] = asyncio.Queue()
-
-    async def push_next_user_messages(self, chat: 'mc.UserChat') -> None:
-        await self._queue.put(chat)
-
-    async def get_next_user_messages(self) -> 'mc.UserChat':
-        return await self._queue.get()