ommlds 0.0.0.dev463__py3-none-any.whl → 0.0.0.dev464__py3-none-any.whl

ommlds/cli/sessions/chat2/inject.py

@@ -1,143 +0,0 @@
-import typing as ta
-
-from omlish import inject as inj
-from omlish import lang
-
-from .... import minichain as mc
-from . import _inject as _inj
-from .configs import DEFAULT_CHAT_MODEL_BACKEND
-from .configs import ChatConfig
-
-
-ItemT = ta.TypeVar('ItemT')
-
-
-##
-
-
-CHAT_OPTIONS = inj.items_binder_helper[mc.ChatChoicesOption](_inj.ChatChoicesServiceOptions)
-BACKEND_CONFIGS = inj.items_binder_helper[mc.Config](_inj.BackendConfigs)
-PHASE_CALLBACKS = inj.items_binder_helper[_inj.ChatPhaseCallback](_inj.ChatPhaseCallbacks)
-
-
-##
-
-
-def bind_chat(cfg: ChatConfig) -> inj.Elements:
-    els: list[inj.Elemental] = []
-
-    #
-
-    els.extend([
-        CHAT_OPTIONS.bind_items_provider(singleton=True),
-        BACKEND_CONFIGS.bind_items_provider(singleton=True),
-        PHASE_CALLBACKS.bind_items_provider(singleton=True),
-    ])
-
-    #
-
-    if cfg.state in ('continue', 'new'):
-        els.extend([
-            inj.bind(_inj.StateStorageChatStateManager, singleton=True),
-            inj.bind(_inj.ChatStateManager, to_key=_inj.StateStorageChatStateManager),
-        ])
-
-        if cfg.state == 'new':
-            els.append(PHASE_CALLBACKS.bind_item(to_fn=lang.typed_lambda(cm=_inj.ChatStateManager)(
-                lambda cm: _inj.ChatPhaseCallback(_inj.ChatPhase.STARTING, cm.clear_state),
-            )))
-
-    elif cfg.state == 'ephemeral':
-        els.extend([
-            inj.bind(_inj.InMemoryChatStateManager, singleton=True),
-            inj.bind(_inj.ChatStateManager, to_key=_inj.InMemoryChatStateManager),
-        ])
-
-    else:
-        raise TypeError(cfg.state)
-
-    #
-
-    if cfg.interactive:
-        if cfg.initial_content is not None:
-            async def add_initial_content(cm: '_inj.ChatStateManager') -> None:
-                await cm.extend_chat([mc.UserMessage(cfg.initial_content)])
-
-            els.append(PHASE_CALLBACKS.bind_item(to_fn=lang.typed_lambda(cm=_inj.ChatStateManager)(
-                lambda cm: _inj.ChatPhaseCallback(_inj.ChatPhase.STARTED, lambda: add_initial_content(cm)),
-            )))
-
-            raise NotImplementedError
-
-        els.extend([
-            inj.bind(_inj.InteractiveUserChatInput, singleton=True),
-            inj.bind(_inj.UserChatInput, to_key=_inj.InteractiveUserChatInput),
-        ])
-
-    else:
-        if cfg.initial_content is None:
-            raise ValueError('Initial content is required for non-interactive chat')
-
-        els.extend([
-            inj.bind(_inj.OneshotUserChatInputInitialChat, to_const=[mc.UserMessage(cfg.initial_content)]),
-            inj.bind(_inj.OneshotUserChatInput, singleton=True),
-            inj.bind(_inj.UserChatInput, to_key=_inj.OneshotUserChatInput),
-        ])
-
-    #
-
-    if cfg.stream:
-        raise NotImplementedError
-
-    else:
-        ai_stack = inj.wrapper_binder_helper(_inj.AiChatGenerator)
-
-        els.append(ai_stack.push_bind(to_ctor=_inj.ChatChoicesServiceAiChatGenerator, singleton=True))
-
-        if not cfg.silent:
-            if cfg.markdown:
-                els.extend([
-                    inj.bind(_inj.MarkdownContentRendering, singleton=True),
-                    inj.bind(_inj.ContentRendering, to_key=_inj.MarkdownContentRendering),
-                ])
-            else:
-                els.extend([
-                    inj.bind(_inj.RawContentRendering, singleton=True),
-                    inj.bind(_inj.ContentRendering, to_key=_inj.RawContentRendering),
-                ])
-
-            els.append(ai_stack.push_bind(to_ctor=_inj.RenderingAiChatGenerator, singleton=True))
-
-        els.append(inj.bind(_inj.AiChatGenerator, to_key=ai_stack.top))
-
-    #
-
-    els.append(inj.bind(_inj.BackendName, to_const=cfg.backend or DEFAULT_CHAT_MODEL_BACKEND))
-
-    els.extend([
-        inj.bind(_inj.CatalogChatChoicesServiceBackendProvider),
-        inj.bind(_inj.ChatChoicesServiceBackendProvider, to_key=_inj.CatalogChatChoicesServiceBackendProvider),
-    ])
-
-    #
-
-    els.extend([
-        inj.bind(_inj.ToolUseExecutorImpl, singleton=True),
-        inj.bind(_inj.ToolUseExecutor, to_key=_inj.ToolUseExecutorImpl),
-    ])
-
-    #
-
-    els.extend([
-        inj.bind(_inj.ChatPhaseManager, singleton=True),
-    ])
-
-    #
-
-    els.extend([
-        inj.bind(_inj.ChatDriver, singleton=True),
-    ])
-
-    #
-
-    return inj.as_elements(*els)

ommlds/cli/tools/config.py

@@ -1,14 +0,0 @@
-from omlish import dataclasses as dc
-
-
-##
-
-
-@dc.dataclass(frozen=True, kw_only=True)
-class ToolsConfig:
-    enable_fs_tools: bool = False
-    enable_todo_tools: bool = False
-
-    enable_unsafe_tools_do_not_use_lol: bool = False
-
-    enable_test_weather_tool: bool = False

ommlds/cli/tools/inject.py

@@ -1,75 +0,0 @@
-import dataclasses as dc
-import typing as ta
-
-from omlish import inject as inj
-from omlish import lang
-
-from ... import minichain as mc
-from ..sessions.chat.inject import bind_chat_options
-from .config import ToolsConfig
-from .weather import WEATHER_TOOL
-
-
-##
-
-
-@dc.dataclass(frozen=True, eq=False)
-class _InjectedTool:
-    tce: mc.ToolCatalogEntry
-
-
-def bind_tool(tce: mc.ToolCatalogEntry) -> inj.Element | inj.Elements:
-    return inj.as_elements(
-        inj.bind_set_entry_const(ta.AbstractSet[_InjectedTool], _InjectedTool(tce)),
-
-        bind_chat_options(mc.Tool(tce.spec)),
-    )
-
-
-##
-
-
-def bind_tools(tools_config: ToolsConfig) -> inj.Elements:
-    els: list[inj.Elemental] = [
-        inj.bind(tools_config),
-
-        inj.bind(mc.ToolCatalog, singleton=True),
-
-        inj.set_binder[_InjectedTool](),
-        inj.bind(
-            lang.typed_lambda(mc.ToolCatalogEntries, s=ta.AbstractSet[_InjectedTool])(
-                lambda s: [it.tce for it in s],
-            ),
-            singleton=True,
-        ),
-    ]
-
-    #
-
-    if tools_config.enable_fs_tools:
-        from ...minichain.lib.fs.tools.ls import ls_tool
-        els.append(bind_tool(ls_tool()))
-
-        from ...minichain.lib.fs.tools.read import read_tool
-        els.append(bind_tool(read_tool()))
-
-    if tools_config.enable_todo_tools:
-        from ...minichain.lib.todo.tools.read import todo_read_tool
-        els.append(bind_tool(todo_read_tool()))
-
-        from ...minichain.lib.todo.tools.write import todo_write_tool
-        els.append(bind_tool(todo_write_tool()))
-
-    if tools_config.enable_unsafe_tools_do_not_use_lol:
-        from ...minichain.lib.bash import bash_tool
-        els.append(bind_tool(bash_tool()))
-
-        from ...minichain.lib.fs.tools.edit import edit_tool
-        els.append(bind_tool(edit_tool()))
-
-    if tools_config.enable_test_weather_tool:
-        els.append(bind_tool(WEATHER_TOOL))
-
-    #
-
-    return inj.as_elements(*els)