ommlds 0.0.0.dev475__py3-none-any.whl → 0.0.0.dev476__py3-none-any.whl

ommlds/cli/main2.py

@@ -1,220 +0,0 @@
-"""
-TODO:
- - bootstrap lol
-"""
-import abc
-import functools
-import typing as ta
-
-import anyio
-
-from omlish import check
-from omlish import dataclasses as dc
-from omlish import inject as inj
-from omlish import lang
-from omlish.argparse import all as ap
-from omlish.logs import all as logs
-
-from .inject import bind_main
-from .sessions.base import Session
-from .sessions.chat.configs import ChatConfig
-
-
-##
-
-
-MAIN_EXTRA_ARGS: ta.Sequence[ap.Arg] = [
-    ap.arg('-v', '--verbose', action='store_true'),
-]
-
-
-def _process_main_extra_args(args: ap.Namespace) -> None:
-    if args.verbose:
-        logs.configure_standard_logging('DEBUG')
-    else:
-        logs.configure_standard_logging('INFO')
-        logs.silence_noisy_loggers()
-
-
-##
-
-
-class Profile(lang.Abstract):
-    @abc.abstractmethod
-    def run(self, argv: ta.Sequence[str]) -> ta.Awaitable[None]:
-        raise NotImplementedError
-
-
-##
-
-
-# class ChatAspect(lang.Abstract):
-#     def get_parser_args(self) -> ta.Sequence[ap.Arg]: ...
-#     def set_args(self, args: ap.Namespace) -> None: ...
-#     def configure(self, cfg: ChatConfig) -> ChatConfig: ...
-
-
-class ChatProfile(Profile):
-    _args: ap.Namespace
-
-    #
-
-    BACKEND_ARGS: ta.ClassVar[ta.Sequence[ap.Arg]] = [
-        ap.arg('-b', '--backend', group='backend'),
-    ]
-
-    def configure_backend(self, cfg: ChatConfig) -> ChatConfig:
-        return dc.replace(
-            cfg,
-            backend=self._args.backend,
-        )
-
-    #
-
-    INPUT_ARGS: ta.ClassVar[ta.Sequence[ap.Arg]] = [
-        ap.arg('message', nargs='*', group='input'),
-        ap.arg('-i', '--interactive', action='store_true', group='input'),
-        ap.arg('-e', '--editor', action='store_true', group='input'),
-    ]
-
-    def configure_input(self, cfg: ChatConfig) -> ChatConfig:
-        if self._args.editor:
-            check.arg(not self._args.interactive)
-            check.arg(not self._args.message)
-            raise NotImplementedError
-
-        elif self._args.interactive:
-            check.arg(not self._args.message)
-            return dc.replace(
-                cfg,
-                interactive=True,
-            )
-
-        elif self._args.message:
-            # TODO: '-' -> stdin
-            return dc.replace(
-                cfg,
-                initial_user_content=' '.join(self._args.message),
-            )
-
-        else:
-            raise ValueError('Must specify input')
-
-    #
-
-    STATE_ARGS: ta.ClassVar[ta.Sequence[ap.Arg]] = [
-        ap.arg('-n', '--new', action='store_true', group='state'),
-        ap.arg('--ephemeral', action='store_true', group='state'),
-    ]
-
-    def configure_state(self, cfg: ChatConfig) -> ChatConfig:
-        return dc.replace(
-            cfg,
-            state='ephemeral' if self._args.ephemeral else 'new' if self._args.new else 'continue',
-        )
-
-    #
-
-    OUTPUT_ARGS: ta.ClassVar[ta.Sequence[ap.Arg]] = [
-        ap.arg('-s', '--stream', action='store_true', group='output'),
-        ap.arg('-M', '--markdown', action='store_true', group='output'),
-    ]
-
-    def configure_output(self, cfg: ChatConfig) -> ChatConfig:
-        return dc.replace(
-            cfg,
-            stream=bool(self._args.stream),
-            markdown=bool(self._args.markdown),
-        )
-
-    #
-
-    async def run(self, argv: ta.Sequence[str]) -> None:
-        parser = ap.ArgumentParser()
-
-        for grp_name, grp_args in [
-            ('backend', self.BACKEND_ARGS),
-            ('input', self.INPUT_ARGS),
-            ('state', self.STATE_ARGS),
-            ('output', self.OUTPUT_ARGS),
-        ]:
-            grp = parser.add_argument_group(grp_name)
-            for a in grp_args:
-                grp.add_argument(*a.args, **a.kwargs)
-
-        self._args = parser.parse_args(argv)
-
-        cfg = ChatConfig()
-        cfg = self.configure_backend(cfg)
-        cfg = self.configure_input(cfg)
-        cfg = self.configure_state(cfg)
-        cfg = self.configure_output(cfg)
-
-        # session_cfg = ChatConfig(
-        #     initial_system_content=system_content,
-        #     enable_tools=(
-        #             args.enable_fs_tools or
-        #             args.enable_todo_tools or
-        #             args.enable_unsafe_tools_do_not_use_lol or
-        #             args.enable_test_weather_tool or
-        #             args.code
-        #     ),
-        #     enabled_tools={  # noqa
-        #         *(['fs'] if args.enable_fs_tools else []),
-        #         *(['todo'] if args.enable_todo_tools else []),
-        #         *(['weather'] if args.enable_test_weather_tool else []),
-        #         # FIXME: enable_unsafe_tools_do_not_use_lol
-        #     },
-        #     dangerous_no_tool_confirmation=bool(args.dangerous_no_tool_confirmation),
-        # )
-
-        with inj.create_managed_injector(bind_main(
-                session_cfg=cfg,
-                enable_backend_strings=True,
-        )) as injector:
-            await injector[Session].run()
-
-
-##
-
-
-PROFILE_TYPES: ta.Mapping[str, type[Profile]] = {
-    'chat': ChatProfile,
-}
-
-
-##
-
-
-MAIN_PROFILE_ARGS: ta.Sequence[ap.Arg] = [
-    ap.arg('profile', nargs='?', default='chat'),
-    ap.arg('args', nargs=ap.REMAINDER),
-]
-
-
-async def _a_main(argv: ta.Any = None) -> None:
-    parser = ap.ArgumentParser()
-
-    for a in [*MAIN_EXTRA_ARGS, *MAIN_PROFILE_ARGS]:
-        parser.add_argument(*a.args, **a.kwargs)
-
-    args, unk_args = parser.parse_known_args(argv)
-
-    _process_main_extra_args(args)
-
-    profile_cls = PROFILE_TYPES[args.profile]
-    profile = profile_cls()
-    await profile.run([*unk_args, *args.args])
-
-
-def _main(args: ta.Any = None) -> None:
-    anyio.run(
-        functools.partial(
-            _a_main,
-            args,
-        ),
-    )  # noqa
-
-
-if __name__ == '__main__':
-    _main()

ommlds/cli/sessions/chat/backends/inject.py

@@ -1,53 +0,0 @@
-import typing as ta
-
-from omlish import inject as inj
-from omlish import lang
-from omlish import typedvalues as tv
-
-from ..... import minichain as mc
-from .injection import backend_configs
-
-
-with lang.auto_proxy_import(globals()):
-    from . import catalog as _catalog
-    from . import types as _types
-
-
-##
-
-
-def bind_backends(
-        *,
-        backend: str | None = None,
-) -> inj.Elements:
-    els: list[inj.Elemental] = []
-
-    #
-
-    els.append(backend_configs().bind_items_provider(singleton=True))
-
-    #
-
-    if backend is not None:
-        els.append(inj.bind(_types.BackendName, to_const=backend))
-
-    els.extend([
-        inj.bind(_types.ChatChoicesServiceBackendProvider, to_ctor=_catalog.CatalogChatChoicesServiceBackendProvider, singleton=True),  # noqa
-        inj.bind(_types.ChatChoicesStreamServiceBackendProvider, to_ctor=_catalog.CatalogChatChoicesStreamServiceBackendProvider, singleton=True),  # noqa
-    ])
-
-    #
-
-    async def catalog_backend_instantiator_provider(injector: inj.AsyncInjector) -> _catalog.CatalogBackendProvider.Instantiator:  # noqa
-        async def inner(be: 'mc.BackendCatalog.Backend', cfgs: _types.BackendConfigs | None) -> ta.Any:
-            kwt = inj.build_kwargs_target(be.factory, non_strict=True)
-            kw = await injector.provide_kwargs(kwt)
-            return be.factory(*tv.collect(*(be.configs or []), *(cfgs or []), override=True), **kw)
-
-        return _catalog.CatalogBackendProvider.Instantiator(inner)
-
-    els.append(inj.bind(_catalog.CatalogBackendProvider.Instantiator, to_async_fn=catalog_backend_instantiator_provider))  # noqa
-
-    #
-
-    return inj.as_elements(*els)