kiari 0.1.0__py3-none-any.whl

kiari/README.md

@@ -0,0 +1,31 @@
+## watch モード
+
+```sh
+# ファイルの変更を監視
+kiari watch "file?paths=.&include_patterns=*.md"
+
+# PubSub を監視
+kiari ext pubsub create-topic --project-id kiarina --topic-id tmp
+kiari ext pubsub create-subscription --project-id kiarina --topic-id tmp --subscription-id tmp
+kiari watch "pubsub?project_id=kiarina&subscription_id=tmp"
+kiari ext pubsub publish-message --project-id kiarina --topic-id tmp --attribute hoge=fuga "hello"
+
+# Realtime Database を監視
+DATABASE_URL=https://kiarina-python.firebaseio.com/
+kiari watch "rtdb?database_url=$DATABASE_URL&path=/posts/kiarina"
+# - 複数のパスを監視
+kiari watch \
+"rtdb?database_url=$DATABASE_URL&path=/posts/kiarina" \
+"rtdb?database_url=$DATABASE_URL&path=/status/kiarina"
+kiari ext rtdb set --database-url $DATABASE_URL --path /posts/kiarina '{"message":"hello"}'
+kiari ext rtdb get --database-url $DATABASE_URL --path /posts/kiarina
+kiari ext rtdb watch --database-url $DATABASE_URL --path /posts/kiarina
+
+# Slack を監視
+CHANNEL_ID=C077QKNDCUR
+kiari watch "slack?"
+kiari watch --watch-handler slack "slack?"
+kiari ext slack post-message --channel $CHANNEL_ID "hello"
+kiari ext -v slack get-channel-messages --channel $CHANNEL_ID --limit 1
+kiari ext slack watch-channel --channel $CHANNEL_ID
+```

kiari/__init__.py

@@ -0,0 +1,46 @@
+from importlib import import_module
+from typing import TYPE_CHECKING
+
+if TYPE_CHECKING:
+    from kiarina.agi.agent import BaseAgent as Agent
+
+__all__ = [
+    # --------------------------------------------------
+    # kiarina_agi
+    # --------------------------------------------------
+    # kiarina.agi.agent
+    "Agent",
+]
+
+
+def __getattr__(name: str) -> object:
+    if name not in __all__:  # pragma: no cover
+        raise AttributeError(f"module {__name__} has no attribute {name}")
+
+    # Map attribute names to their module paths
+    # Format: "name": "module_path" or "name": ("module_path", "attr_name")
+    # fmg: off
+    module_map: dict[str, str | tuple[str, str]] = {
+        # --------------------------------------------------
+        # kiarina_agi
+        # --------------------------------------------------
+        # kiarina.agi.agent
+        "Agent": ("kiarina.agi.agent", "BaseAgent"),
+    }
+    # fmt: on
+
+    module_info = module_map[name]
+
+    if isinstance(module_info, tuple):
+        # Tuple format: (module_path, attr_name)
+        module_path, attr_name = module_info
+        module = import_module(module_path)
+        attr = getattr(module, attr_name)
+    else:
+        # String format: module_path (attr_name is the same as name)
+        module_path = module_info
+        module = import_module(module_path)
+        attr = getattr(module, name)
+
+    globals()[name] = attr
+    return attr

kiari/__main__.py

@@ -0,0 +1,9 @@
+from .cli.cli import kiari
+
+
+def main() -> None:
+    kiari()
+
+
+if __name__ == "__main__":
+    main()

kiari/cli/__init__.py

@@ -0,0 +1,29 @@
+from ._constants.common_option_groups import COMMON_OPTION_GROUPS
+from ._decorators.common_options import common_options
+from ._helpers.build_cli_args import build_cli_args
+from ._helpers.render_bootstrap_message import render_bootstrap_message
+from ._helpers.run import run
+from ._helpers.setup_profile import setup_profile
+from ._schemas.cli_args import CLIArgs
+from ._types.extra_args import ExtraArgs
+from ._types.save_mode import SaveMode
+from ._utils.graceful_shutdown import graceful_shutdown
+
+__all__ = [
+    # ._constants
+    "COMMON_OPTION_GROUPS",
+    # ._decorators
+    "common_options",
+    # ._helpers
+    "build_cli_args",
+    "render_bootstrap_message",
+    "run",
+    "setup_profile",
+    # ._schemas
+    "CLIArgs",
+    # ._types
+    "ExtraArgs",
+    "SaveMode",
+    # ._utils
+    "graceful_shutdown",
+]

kiari/cli/_constants/common_option_groups.py

@@ -0,0 +1,129 @@
+from rich_click.utils import OptionGroupDict
+
+COMMON_OPTION_GROUPS: list[OptionGroupDict] = [
+    {
+        "name": "History Repository",
+        "options": [
+            "--history-repository",
+            "--stateless",
+            "--no-load",
+            "--no-save",
+            "--allow-active-missing-tools",
+        ],
+    },
+    {
+        "name": "History",
+        "options": [
+            "--event",
+            "--file-info",
+            "--tool-info",
+            "--default-tool-state",
+        ],
+    },
+    {
+        "name": "Agent",
+        "options": [
+            "--agent",
+            "--file-limits",
+            "--max-iterations",
+            "--until-end",
+            "--until-tool-call",
+            "--until-tool-run",
+        ],
+    },
+    {
+        "name": "Tool",
+        "options": [
+            "--tool",
+            "--pre-hook",
+            "--post-hook",
+        ],
+    },
+    {
+        "name": "Workflow",
+        "options": [
+            "--workflow",
+        ],
+    },
+    {
+        "name": "Prompt",
+        "options": [
+            "--prompt",
+            "--prompt-limits",
+            "--system-message",
+        ],
+    },
+    {
+        "name": "Chat",
+        "options": [
+            "--chat-model",
+            "--openai",
+            "--anthropic",
+            "--google",
+            "--tool-choice",
+            "--parallel-tool-calls",
+            "--streaming",
+        ],
+    },
+    {
+        "name": "Cost Recorder",
+        "options": [
+            "--cost-recorder",
+        ],
+    },
+    {
+        "name": "Observability",
+        "options": [
+            "--request-logger",
+            "--cost-logger",
+            "--chat-logger",
+            "--tool-logger",
+        ],
+    },
+    {
+        "name": "Run Context",
+        "options": [
+            "--organization-id",
+            "--user-id",
+            "--agent-id",
+            "--runner-id",
+            "--language",
+            "--time-zone",
+            "--currency",
+        ],
+    },
+    {
+        "name": "GitHub",
+        "options": [
+            "--github-ignore-cache",
+            "--github-trusted-username",
+            "--github-skip-trust-verification",
+        ],
+    },
+    {
+        "name": "Config",
+        "options": [
+            "--i18n-catalog",
+            "--config",
+            "--config-var",
+            "--plugin",
+        ],
+    },
+    {
+        "name": "Profile",
+        "options": [
+            "--profile",
+            "--set-profile",
+            "--reset-profile",
+        ],
+    },
+    {
+        "name": "Logging",
+        "options": [
+            "--logger-name",
+            "--log-level",
+            "--verbose",
+            "--quiet",
+        ],
+    },
+]

kiari/cli/_decorators/common_options.py

@@ -0,0 +1,188 @@
+import functools
+from collections.abc import Callable
+from typing import Any
+
+import rich_click as click
+from kiarina.i18n import get_i18n, get_system_language
+
+from .._i18n import CLII18n
+
+t = get_i18n(CLII18n, get_system_language())
+
+
+def common_options(func: Callable) -> Callable:
+    # fmt: off
+    # --------------------------------------------------
+    # History Repository
+    # --------------------------------------------------
+    @click.option("--history-repository", type=str, help=t.history_repository_help)
+    @click.option("--stateless", is_flag=True, is_eager=True, expose_value=False, callback=_set_stateless, help=t.stateless_help)
+    @click.option("--no-load", is_flag=True, default=None, help=t.no_load_help)
+    @click.option("--no-save", is_flag=True, default=None, help=t.no_save_help)
+    @click.option("--allow-active-missing-tools", is_flag=True, default=None, help=t.allow_active_missing_tools_help)
+    # --------------------------------------------------
+    # History
+    # --------------------------------------------------
+    @click.option("--event", "events", multiple=True, type=str, help=t.events_help)
+    @click.option("-f", "--file-info", "file_infos", multiple=True, type=str, help=t.file_infos_help)
+    @click.option("--tool-info", "tool_infos", multiple=True, type=str, help=t.tool_infos_help)
+    @click.option("--default-tool-state", type=click.Choice(["active", "inactive", "disabled"]), help=t.default_tool_state_help)
+    # --------------------------------------------------
+    # Agent
+    # --------------------------------------------------
+    @click.option("--agent", type=str, help=t.agent_help)
+    @click.option("--file-limits", type=str, help=t.file_limits_help)
+    @click.option("-n", "--max-iterations", type=int, help=t.max_iterations_help)
+    @click.option("--until-end/--no-until-end", is_flag=True, default=None, help=t.until_end_help)
+    @click.option("--until-tool-call", "until_tool_calls", multiple=True, type=str, help=t.until_tool_calls_help)
+    @click.option("--until-tool-run", "until_tool_runs", multiple=True, type=str, help=t.until_tool_runs_help)
+    # --------------------------------------------------
+    # Tool
+    # --------------------------------------------------
+    @click.option("-t", "--tool", "tools", multiple=True, type=str, help=t.tools_help)
+    @click.option("--pre-hook", "pre_hooks", multiple=True, type=str, help=t.pre_hooks_help)
+    @click.option("--post-hook", "post_hooks", multiple=True, type=str, help=t.post_hooks_help)
+    # --------------------------------------------------
+    # Workflow
+    # --------------------------------------------------
+    @click.option("--workflow", type=str, help=t.workflow_help)
+    # --------------------------------------------------
+    # Prompt
+    # --------------------------------------------------
+    @click.option("--prompt", type=str, help=t.prompt_help)
+    @click.option("--prompt-limits", type=str, help=t.prompt_limits_help)
+    @click.option("--system-message", "system_messages", multiple=True, type=str, help=t.system_messages_help)
+    # --------------------------------------------------
+    # Chat
+    # --------------------------------------------------
+    @click.option("--chat-model", type=str, help=t.chat_model_help)
+    @click.option("--openai", is_flag=True, default=None, is_eager=True, expose_value=False, callback=_set_openai, help=t.openai_help)
+    @click.option("--anthropic", is_flag=True, default=None, is_eager=True, expose_value=False, callback=_set_anthropic, help=t.anthropic_help)
+    @click.option("--google", is_flag=True, default=None, is_eager=True, expose_value=False, callback=_set_google, help=t.google_help)
+    @click.option("--tool-choice", type=str, help=t.tool_choice_help)
+    @click.option("--parallel-tool-calls/--no-parallel-tool-calls", default=None, help=t.parallel_tool_calls_help)
+    @click.option("--streaming/--no-streaming", default=None, help=t.streaming_help)
+    # --------------------------------------------------
+    # Cost Recorder
+    # --------------------------------------------------
+    @click.option("--cost-recorder", type=str, help=t.cost_recorder_help)
+    # --------------------------------------------------
+    # Observability
+    # --------------------------------------------------
+    @click.option("--request-logger", type=str, help=t.request_logger_help)
+    @click.option("--cost-logger", type=str, help=t.cost_logger_help)
+    @click.option("--chat-logger", type=str, help=t.chat_logger_help)
+    @click.option("--tool-logger", type=str, help=t.tool_logger_help)
+    # --------------------------------------------------
+    # Run Context
+    # --------------------------------------------------
+    @click.option("--organization-id", type=str, help=t.organization_id_help)
+    @click.option("--user-id", type=str, help=t.user_id_help)
+    @click.option("--agent-id", type=str, help=t.agent_id_help)
+    @click.option("--runner-id", type=str, help=t.node_id_help)
+    @click.option("--language", type=str, help=t.language_help)
+    @click.option("--time-zone", type=str, help=t.time_zone_help)
+    @click.option("--currency", type=str, help=t.currency_help)
+    # --------------------------------------------------
+    # GitHub
+    # --------------------------------------------------
+    @click.option("--github-ignore-cache", is_flag=True, default=None, help=t.github_ignore_cache_help)
+    @click.option("--github-trusted-username", "github_trusted_usernames", multiple=True, type=str, help=t.github_trusted_username_help)
+    @click.option("--github-skip-trust-verification", is_flag=True, default=None, help=t.github_skip_trust_verification_help)
+    @functools.wraps(func)
+    # --------------------------------------------------
+    # Config
+    # --------------------------------------------------
+    @click.option("--i18n-catalog", "i18n_catalogs", multiple=True, type=str, help=t.i18n_catalog_help)
+    @click.option("--config", "configs", multiple=True, type=str, help=t.config_file_help)
+    @click.option("-c", "--config-var", "config_vars", multiple=True, type=str, help=t.config_var_help)
+    @click.option("--plugin", "plugins", multiple=True, type=str, help=t.plugin_help)
+    # --------------------------------------------------
+    # Logging
+    # --------------------------------------------------
+    @click.option("--logger-name", "logger_names", multiple=True, type=str, help=t.logger_name_help)
+    @click.option("--log-level", type=click.Choice(["DEBUG", "INFO", "WARNING", "ERROR", "CRITICAL"]), default=None, help=t.log_level_help)
+    @click.option("-v", "--verbose", is_flag=True, default=None, callback=_set_verbose, help=t.verbose_help)
+    @click.option("-q", "--quiet", is_flag=True, default=None, callback=_set_quiet, help=t.quiet_help)
+    # --------------------------------------------------
+    # Profile
+    # --------------------------------------------------
+    @click.option("-p", "--profile", "profile_name", type=str, default=None, help=t.profile_help)
+    @click.option("--set", "--set-profile", is_flag=True, is_eager=True, expose_value=False, callback=_set_profile, help=t.set_profile_help)
+    @click.option("--reset", "--reset-profile", is_flag=True, is_eager=True, expose_value=False, callback=_reset_profile, help=t.reset_profile_help)
+    # --------------------------------------------------
+    # CLI
+    # --------------------------------------------------
+    @click.option("-x", "--exec", "exec_file", type=str, help=t.exec_file_help)
+    @click.option("--finalizer", "finalizers", multiple=True, type=str, help=t.finalizers_help)
+    @functools.wraps(func)
+    # fmt: on
+    def wrapper(*args, **kwargs):
+        return func(*args, **kwargs)
+
+    return wrapper
+
+
+def _set_profile(ctx: click.Context, _: click.Parameter, value: Any) -> Any:
+    if not value:
+        return value
+
+    ctx.params["save_mode"] = "set"
+    return value
+
+
+def _reset_profile(ctx: click.Context, _: click.Parameter, value: Any) -> Any:
+    if not value:
+        return value
+
+    ctx.params["save_mode"] = "reset"
+    return value
+
+
+def _set_stateless(ctx: click.Context, _: click.Parameter, value: Any) -> Any:
+    if not value:
+        return value
+
+    ctx.params["no_load"] = True
+    ctx.params["no_save"] = True
+    return value
+
+
+def _set_openai(ctx: click.Context, _: click.Parameter, value: Any) -> Any:
+    if not value:
+        return value
+
+    ctx.params["chat_model"] = "openai"
+    return value
+
+
+def _set_anthropic(ctx: click.Context, _: click.Parameter, value: Any) -> Any:
+    if not value:
+        return value
+
+    ctx.params["chat_model"] = "anthropic"
+    return value
+
+
+def _set_google(ctx: click.Context, _: click.Parameter, value: Any) -> Any:
+    if not value:
+        return value
+
+    ctx.params["chat_model"] = "google"
+    return value
+
+
+def _set_verbose(ctx: click.Context, _: click.Parameter, value: Any) -> Any:
+    if not value:
+        return value
+
+    ctx.params["log_level"] = "DEBUG"
+    return value
+
+
+def _set_quiet(ctx: click.Context, _: click.Parameter, value: Any) -> Any:
+    if not value:
+        return value
+
+    ctx.params["log_level"] = "WARNING"
+    return value

kiari/cli/_helpers/build_cli_args.py

@@ -0,0 +1,70 @@
+from typing import Any
+
+from kiari.core.profile import ProfileName
+
+from .._schemas.cli_args import CLIArgs
+from .._types.save_mode import SaveMode
+
+
+def build_cli_args(
+    exec_file: str | None = None,
+    profile_name: ProfileName | None = None,
+    save_mode: SaveMode | None = None,
+    extra_args_keys: list[str] | None = None,
+    markdown_content_key: str | None = None,
+    **kwargs: Any,
+) -> CLIArgs:
+    run_spec = _normalize_kwargs(kwargs)
+
+    extra_args = {}
+
+    if exec_file:
+        file_args: dict[str, Any] | None = None
+
+        if exec_file.endswith(".json"):
+            from kiarina.utils.file import read_json_dict
+
+            file_args = read_json_dict(exec_file)
+
+        elif exec_file.endswith(".yaml") or exec_file.endswith(".yml"):
+            from kiarina.utils.file import read_yaml_dict
+
+            file_args = read_yaml_dict(exec_file)
+
+        elif exec_file.endswith(".md") or exec_file.endswith(".markdown"):
+            from kiarina.utils.file import read_markdown
+
+            if markdown_content := read_markdown(exec_file):
+                file_args = markdown_content.metadata
+
+                if markdown_content_key:
+                    extra_args[markdown_content_key] = markdown_content.content
+
+        else:  # pragma: no cover
+            raise ValueError(f"Unsupported file type: {exec_file}")
+
+        if file_args is None:
+            raise FileNotFoundError(f"File not found: {exec_file}")
+
+        run_spec = {**file_args, **run_spec}
+
+    if extra_args_keys:
+        for key in extra_args_keys:
+            if key in run_spec:
+                extra_args[key] = run_spec.pop(key)
+
+    return CLIArgs(
+        exec_file=exec_file,
+        profile_name=profile_name,
+        save_mode=save_mode,
+        run_spec=run_spec,
+        extra_args=extra_args,
+    )
+
+
+def _normalize_kwargs(kwargs: dict[str, Any]) -> dict[str, Any]:
+    return {
+        k: v if not isinstance(v, tuple) else list(v)
+        for k, v in kwargs.items()
+        if v is not None and (not isinstance(v, tuple) or len(v) > 0)
+    }

kiari/cli/_helpers/render_bootstrap_message.py

@@ -0,0 +1,37 @@
+from kiarina.agi.console_utils import divider, section_header
+from rich.console import Group, RenderableType
+from rich.text import Text
+
+from kiari.core.profile import ProfileName, RunOptions, RunSpec
+
+
+def render_bootstrap_message(
+    exec_file: str | None,
+    profile_name: ProfileName,
+    run_spec: RunSpec,
+    run_options: RunOptions,
+) -> RenderableType | None:
+    if run_options.log_level not in ("DEBUG", "INFO"):
+        return None
+
+    style = "black"
+
+    renderables: list[RenderableType] = [
+        Text("kiari version 1.0.0", style=style),
+    ]
+
+    if exec_file:
+        renderables.append(Text(f"+ exec: {exec_file}", style=style))
+
+    renderables.append(Text(f"+ profile: {profile_name}", style=style))
+    renderables.append(Text(section_header("RUN OPTIONS"), style=style))
+
+    if run_spec:
+        for k, v in run_spec.items():
+            renderables.append(Text(f"{k}: {v}", style=style))
+    else:
+        renderables.append(Text("No run options specified", style=style))
+
+    renderables.append(Text(divider(), style=style))
+
+    return Group(*renderables)

kiari/cli/_helpers/run.py

@@ -0,0 +1,18 @@
+from collections.abc import Callable
+from typing import Any
+
+from kiari.core.finalizer import run_finalizers
+from kiari.core.profile import ProfileName, RunOptions
+
+
+async def run(
+    runner: Callable,
+    profile_name: ProfileName,
+    run_options: RunOptions,
+    *args: Any,
+    **kwargs: Any,
+) -> None:
+    try:
+        await runner(profile_name, run_options, *args, **kwargs)
+    finally:
+        await run_finalizers(run_options.finalizers)

kiari/cli/_helpers/setup_profile.py

@@ -0,0 +1,24 @@
+from kiari.core.profile import ProfileName, RunOptions, RunSpec, profile_store
+
+from .._types.save_mode import SaveMode
+
+
+def setup_profile(
+    profile_name: ProfileName | None,
+    save_mode: SaveMode | None,
+    run_spec: RunSpec,
+) -> tuple[ProfileName, RunSpec, RunOptions]:
+    if not profile_name:
+        profile_name = profile_store.get_current()
+
+    if save_mode != "reset":
+        run_spec = {
+            **profile_store.load_run_spec(profile_name),
+            **run_spec,
+        }
+
+    if save_mode in ("set", "reset"):
+        profile_store.save_run_spec(profile_name, run_spec)
+
+    run_options = RunOptions.model_validate(run_spec)
+    return profile_name, run_spec, run_options