indusagi 0.1.0__py3-none-any.whl

indusagi/__init__.py

@@ -0,0 +1,34 @@
+"""indusagi — terminal-first AI coding agent framework (Python rebuild)."""
+
+from importlib.metadata import PackageNotFoundError, version as _version
+
+try:
+    __version__ = _version("indusagi")
+except PackageNotFoundError:  # running from a source tree without install
+    __version__ = "0.0.0.dev0"
+
+VERSION = __version__
+
+_SUBSYSTEMS = {
+    "gateway": "llmgateway",
+    "runtime": "runtime",
+    "capabilities": "capabilities",
+    "interop": "interop",
+    "saas": "connectors",
+    "swarm": "swarm",
+    "smithy": "smithy",
+    "tracing": "tracing",
+    "shell": "shell_app",
+}
+
+
+def __getattr__(name: str):  # PEP 562 lazy namespace re-exports
+    if name in _SUBSYSTEMS:
+        import importlib
+
+        return importlib.import_module(f"indusagi.{_SUBSYSTEMS[name]}")
+    raise AttributeError(f"module 'indusagi' has no attribute {name!r}")
+
+
+def __dir__() -> list[str]:
+    return sorted([*globals(), *_SUBSYSTEMS])

indusagi/_internal/__init__.py

@@ -0,0 +1,5 @@
+"""Framework-internal shared utilities (not part of the public API)."""
+
+from indusagi._internal.cancel import CancelledByToken, CancelToken
+
+__all__ = ["CancelToken", "CancelledByToken"]

indusagi/_internal/cancel.py

@@ -0,0 +1,96 @@
+"""Cooperative cancellation for the indusagi framework.
+
+Replaces the Web ``AbortSignal``/``AbortController`` pair from the TypeScript
+implementation: queryable, multi-listener, and chainable. Cancellation is
+cooperative — holders poll :attr:`CancelToken.cancelled` or register
+callbacks; nothing is forcibly killed. This is deliberately NOT raw asyncio
+task cancellation: tools must be able to observe cancellation and still
+return typed ``is_error`` results.
+"""
+
+from __future__ import annotations
+
+import threading
+from typing import Callable
+
+Callback = Callable[[], None]
+
+
+class CancelledByToken(Exception):
+    """Raised by :meth:`CancelToken.raise_if_cancelled`.
+
+    Distinct from :class:`asyncio.CancelledError` on purpose — framework code
+    never swallows ``asyncio.CancelledError``.
+    """
+
+    def __init__(self, reason: str | None = None) -> None:
+        super().__init__(reason or "cancelled")
+        self.reason = reason
+
+
+class CancelToken:
+    """One-shot cancellation token shared across all subsystems."""
+
+    __slots__ = ("_cancelled", "_reason", "_callbacks", "_lock")
+
+    def __init__(self) -> None:
+        self._cancelled = False
+        self._reason: str | None = None
+        self._callbacks: list[Callback] = []
+        self._lock = threading.Lock()
+
+    @property
+    def cancelled(self) -> bool:
+        return self._cancelled
+
+    @property
+    def reason(self) -> str | None:
+        return self._reason
+
+    def cancel(self, reason: str | None = None) -> None:
+        with self._lock:
+            if self._cancelled:
+                return
+            self._cancelled = True
+            self._reason = reason
+            callbacks = list(self._callbacks)
+            self._callbacks.clear()
+        for callback in callbacks:
+            try:
+                callback()
+            except Exception:
+                pass  # listener errors never break cancellation fan-out
+
+    def add_callback(self, callback: Callback) -> Callable[[], None]:
+        """Register ``callback`` to fire on cancellation; returns unsubscribe.
+
+        Mirrors ``AbortSignal.addEventListener("abort", ...)``: registering on
+        an already-cancelled token fires the callback immediately.
+        """
+        with self._lock:
+            if not self._cancelled:
+                self._callbacks.append(callback)
+
+                def unsubscribe() -> None:
+                    with self._lock:
+                        try:
+                            self._callbacks.remove(callback)
+                        except ValueError:
+                            pass
+
+                return unsubscribe
+        callback()
+        return lambda: None
+
+    def child(self) -> "CancelToken":
+        """A token cancelled when this one is (but not vice versa)."""
+        token = CancelToken()
+        if self._cancelled:
+            token.cancel(self._reason)
+        else:
+            self.add_callback(lambda: token.cancel(self._reason))
+        return token
+
+    def raise_if_cancelled(self) -> None:
+        if self._cancelled:
+            raise CancelledByToken(self._reason)

indusagi/_internal/env.py

@@ -0,0 +1,41 @@
+"""Single environment-variable registry for the framework.
+
+Every env var the framework reads is declared and resolved here — subsystems
+must not probe ``os.environ`` directly. (The TS codebase had two competing
+resolution tables; the port has one.)
+"""
+
+from __future__ import annotations
+
+import os
+from pathlib import Path
+
+ENV_PREFIX = "INDUSAGI"
+
+
+def env_name(suffix: str) -> str:
+    """Compose a branded env-var name: ``env_name("home") -> "INDUSAGI_HOME"``."""
+    return f"{ENV_PREFIX}_{suffix.upper().replace('-', '_')}"
+
+
+def read_env(suffix: str) -> str | None:
+    return os.environ.get(env_name(suffix))
+
+
+def read_raw(name: str) -> str | None:
+    """Escape hatch for non-branded variables (e.g. ``AWS_REGION``)."""
+    return os.environ.get(name)
+
+
+def indusagi_home() -> Path:
+    """State directory: ``$INDUSAGI_HOME`` or ``~/.pindusagi``.
+
+    Deliberately separate from the TypeScript builds' state directory
+    (owner decision, 2026-06-10): the Python build owns ``~/.pindusagi``
+    and never reads or writes the TS installs' state. The auth.json *format*
+    remains byte-compatible, but the store is no longer shared.
+    """
+    override = read_env("HOME")
+    if override:
+        return Path(override).expanduser()
+    return Path.home() / ".pindusagi"

indusagi/agent/__init__.py

@@ -0,0 +1,249 @@
+"""``indusagi.agent`` — the high-level Agent facade (old vocabulary).
+
+Mirrors the public surface of ``indus-rebuild/src/facade/agent.ts``
+(→ ``facade/bot/index.ts``) as thin shims over the green core:
+
+* :class:`Agent` — prompt / steer / follow_up / wait_for_idle / abort over
+  :func:`indusagi.runtime.create_agent`, with events in the facade
+  vocabulary (``message_start`` / ``message_update`` / ``message_end``,
+  ``tool_execution_*``, ``turn_*``, ``agent_*``).
+* ``create_*_tool`` factories — the facade names over the
+  :mod:`indusagi.capabilities` built-ins.
+* :class:`SessionManager` — append-only JSONL session trees, read-compatible
+  with real TS v3 session files.
+* :func:`register_message_type` — the explicit registry replacing the TS
+  ``CustomAgentMessages`` declaration merging, plus the built-in
+  session-only message kinds.
+
+The message/content models themselves (``UserMessage``, ``AssistantMessage``,
+``TextContent``, ...) live in :mod:`indusagi.ai`; this package consumes them
+without re-exporting (matching the TS module layout, where ``bot/`` imports
+from ``ml/`` but ``facade/agent.ts`` re-exports only the bot surface).
+
+Usage::
+
+    from indusagi.agent import Agent, create_coding_tools
+
+    agent = Agent()
+    agent.set_tools(create_coding_tools("/path/to/project"))
+    await agent.prompt("explain this repository")
+"""
+
+from __future__ import annotations
+
+# --- agent.ts / types.ts -----------------------------------------------------
+from .agent import (
+    AGENT_EVENT_GROUPS,
+    Agent,
+    AgentContext,
+    AgentEndEvent,
+    AgentEvent,
+    AgentOptions,
+    AgentStartEvent,
+    AgentState,
+    AgentTool,
+    AgentToolResult,
+    AgentToolUpdateCallback,
+    MessageEndEvent,
+    MessageStartEvent,
+    MessageUpdateEvent,
+    QueueMode,
+    StreamFn,
+    ThinkingBudgets,
+    ThinkingLevel,
+    ToolExecutionEndEvent,
+    ToolExecutionStartEvent,
+    ToolExecutionUpdateEvent,
+    TurnEndEvent,
+    TurnStartEvent,
+    is_thinking,
+    is_tool_call,
+    is_user_message,
+    validate_message,
+)
+
+# --- messages.ts + the CustomAgentMessages replacement ------------------------
+from .registry import (
+    BRANCH_SUMMARY_PREFIX,
+    BRANCH_SUMMARY_SUFFIX,
+    COMPACTION_SUMMARY_PREFIX,
+    COMPACTION_SUMMARY_SUFFIX,
+    BashExecutionMessage,
+    BranchSummaryMessage,
+    CompactionSummaryMessage,
+    CustomMessage,
+    bash_execution_to_text,
+    convert_to_llm,
+    create_branch_summary_message,
+    create_compaction_summary_message,
+    create_custom_message,
+    get_message_type,
+    register_message_type,
+    registered_message_types,
+    unregister_message_type,
+)
+
+# --- session-manager.ts --------------------------------------------------------
+from .sessions import (
+    CURRENT_SESSION_VERSION,
+    FileEntry,
+    SessionContext,
+    SessionEntry,
+    SessionHeader,
+    SessionInfo,
+    SessionListProgress,
+    SessionManager,
+    SessionTreeNode,
+    build_session_context,
+    find_most_recent_session,
+    get_latest_compaction_entry,
+    load_entries_from_file,
+    migrate_session_entries,
+    parse_session_entries,
+)
+
+# --- actions/index.ts (the tool factories + collections) -------------------------
+from .tools import (
+    TOOL_METADATA,
+    FacadeAgentTool,
+    TodoStore,
+    ToolCategory,
+    ToolMetadata,
+    ToolName,
+    ToolRegistry,
+    all_tools,
+    bash_tool,
+    coding_tools,
+    create_all_tools,
+    create_bash_tool,
+    create_coding_tools,
+    create_edit_tool,
+    create_find_tool,
+    create_grep_tool,
+    create_ls_tool,
+    create_process_tool,
+    create_read_only_tools,
+    create_read_tool,
+    create_todo_read_tool,
+    create_todo_write_tool,
+    create_tool_registry,
+    create_web_fetch_tool,
+    create_web_search_tool,
+    create_write_tool,
+    edit_tool,
+    find_tool,
+    grep_tool,
+    ls_tool,
+    process_tool,
+    read_only_tools,
+    read_tool,
+    todo_read_tool,
+    todo_write_tool,
+    web_fetch_tool,
+    web_search_tool,
+    write_tool,
+)
+
+__all__ = [
+    # agent.ts / types.ts
+    "AGENT_EVENT_GROUPS",
+    "Agent",
+    "AgentContext",
+    "AgentEndEvent",
+    "AgentEvent",
+    "AgentOptions",
+    "AgentStartEvent",
+    "AgentState",
+    "AgentTool",
+    "AgentToolResult",
+    "AgentToolUpdateCallback",
+    "MessageEndEvent",
+    "MessageStartEvent",
+    "MessageUpdateEvent",
+    "QueueMode",
+    "StreamFn",
+    "ThinkingBudgets",
+    "ThinkingLevel",
+    "ToolExecutionEndEvent",
+    "ToolExecutionStartEvent",
+    "ToolExecutionUpdateEvent",
+    "TurnEndEvent",
+    "TurnStartEvent",
+    "is_thinking",
+    "is_tool_call",
+    "is_user_message",
+    "validate_message",
+    # messages.ts / registry
+    "BRANCH_SUMMARY_PREFIX",
+    "BRANCH_SUMMARY_SUFFIX",
+    "COMPACTION_SUMMARY_PREFIX",
+    "COMPACTION_SUMMARY_SUFFIX",
+    "BashExecutionMessage",
+    "BranchSummaryMessage",
+    "CompactionSummaryMessage",
+    "CustomMessage",
+    "bash_execution_to_text",
+    "convert_to_llm",
+    "create_branch_summary_message",
+    "create_compaction_summary_message",
+    "create_custom_message",
+    "get_message_type",
+    "register_message_type",
+    "registered_message_types",
+    "unregister_message_type",
+    # session-manager.ts
+    "CURRENT_SESSION_VERSION",
+    "FileEntry",
+    "SessionContext",
+    "SessionEntry",
+    "SessionHeader",
+    "SessionInfo",
+    "SessionListProgress",
+    "SessionManager",
+    "SessionTreeNode",
+    "build_session_context",
+    "find_most_recent_session",
+    "get_latest_compaction_entry",
+    "load_entries_from_file",
+    "migrate_session_entries",
+    "parse_session_entries",
+    # actions/index.ts
+    "TOOL_METADATA",
+    "FacadeAgentTool",
+    "TodoStore",
+    "ToolCategory",
+    "ToolMetadata",
+    "ToolName",
+    "ToolRegistry",
+    "all_tools",
+    "bash_tool",
+    "coding_tools",
+    "create_all_tools",
+    "create_bash_tool",
+    "create_coding_tools",
+    "create_edit_tool",
+    "create_find_tool",
+    "create_grep_tool",
+    "create_ls_tool",
+    "create_process_tool",
+    "create_read_only_tools",
+    "create_read_tool",
+    "create_todo_read_tool",
+    "create_todo_write_tool",
+    "create_tool_registry",
+    "create_web_fetch_tool",
+    "create_web_search_tool",
+    "create_write_tool",
+    "edit_tool",
+    "find_tool",
+    "grep_tool",
+    "ls_tool",
+    "process_tool",
+    "read_only_tools",
+    "read_tool",
+    "todo_read_tool",
+    "todo_write_tool",
+    "web_fetch_tool",
+    "web_search_tool",
+    "write_tool",
+]