pytk-ai 0.1.0__tar.gz

pytk_ai-0.1.0/PKG-INFO

@@ -0,0 +1,64 @@
+Metadata-Version: 2.4
+Name: pytk-ai
+Version: 0.1.0
+Summary: Python command planning, execution, and output filtering for PYTK-AI
+Author: RTK AI
+License: MIT
+Keywords: llm,cli,token,hooks,productivity
+Classifier: Development Status :: 3 - Alpha
+Classifier: License :: OSI Approved :: MIT License
+Classifier: Programming Language :: Python :: 3
+Classifier: Programming Language :: Python :: 3.10
+Classifier: Programming Language :: Python :: 3.11
+Classifier: Programming Language :: Python :: 3.12
+Classifier: Programming Language :: Python :: 3.13
+Classifier: Topic :: Software Development :: Build Tools
+Classifier: Topic :: Utilities
+Requires-Python: >=3.10
+Description-Content-Type: text/markdown
+
+# PYTK-AI (Python)
+
+This repository is a Python port of the PYTK-AI command rewrite and output-filtering engine.
+
+## Migration direction
+
+- target public CLI: `pytk-ai run <command>`
+- target public library API: `from pytk_ai.runner import run_command`
+- command rewrite/planning is internal implementation detail
+- `rtk/` is read-only reference material, not a compatibility contract
+- the old rewrite-first scaffold has been replaced by the execution-oriented `pytk-ai run` flow
+
+## What it does
+
+- plans shell commands internally for PYTK-AI-managed handling when useful
+- executes raw shell commands through a small Python runner
+- filters stdout/stderr into a compact token-efficient result
+- preserves compound commands such as `&&`, `||`, `;`, `&`, and simple pipes
+- exposes a small CLI plus hook-oriented JSON adapters
+
+## Install locally
+
+```bash
+pip install -e .
+```
+
+## CLI usage
+
+```bash
+pytk-ai run git status
+pytk-ai run "cargo test && git push"
+```
+
+## Library usage
+
+```python
+from pytk_ai.runner import run_command
+
+result = run_command("git status")
+print(result.filtered_output)
+print(result.exit_code)
+```
+
+The package is intentionally small and dependency-free so it can be embedded in
+agent toolchains and hook scripts.

pytk_ai-0.1.0/README.md

@@ -0,0 +1,45 @@
+# PYTK-AI (Python)
+
+This repository is a Python port of the PYTK-AI command rewrite and output-filtering engine.
+
+## Migration direction
+
+- target public CLI: `pytk-ai run <command>`
+- target public library API: `from pytk_ai.runner import run_command`
+- command rewrite/planning is internal implementation detail
+- `rtk/` is read-only reference material, not a compatibility contract
+- the old rewrite-first scaffold has been replaced by the execution-oriented `pytk-ai run` flow
+
+## What it does
+
+- plans shell commands internally for PYTK-AI-managed handling when useful
+- executes raw shell commands through a small Python runner
+- filters stdout/stderr into a compact token-efficient result
+- preserves compound commands such as `&&`, `||`, `;`, `&`, and simple pipes
+- exposes a small CLI plus hook-oriented JSON adapters
+
+## Install locally
+
+```bash
+pip install -e .
+```
+
+## CLI usage
+
+```bash
+pytk-ai run git status
+pytk-ai run "cargo test && git push"
+```
+
+## Library usage
+
+```python
+from pytk_ai.runner import run_command
+
+result = run_command("git status")
+print(result.filtered_output)
+print(result.exit_code)
+```
+
+The package is intentionally small and dependency-free so it can be embedded in
+agent toolchains and hook scripts.

pytk_ai-0.1.0/pyproject.toml

@@ -0,0 +1,39 @@
+[build-system]
+requires = ["setuptools>=69", "wheel"]
+build-backend = "setuptools.build_meta"
+
+[project]
+name = "pytk-ai"
+version = "0.1.0"
+description = "Python command planning, execution, and output filtering for PYTK-AI"
+readme = "README.md"
+requires-python = ">=3.10"
+license = { text = "MIT" }
+authors = [{ name = "RTK AI" }]
+keywords = ["llm", "cli", "token", "hooks", "productivity"]
+classifiers = [
+  "Development Status :: 3 - Alpha",
+  "License :: OSI Approved :: MIT License",
+  "Programming Language :: Python :: 3",
+  "Programming Language :: Python :: 3.10",
+  "Programming Language :: Python :: 3.11",
+  "Programming Language :: Python :: 3.12",
+  "Programming Language :: Python :: 3.13",
+  "Topic :: Software Development :: Build Tools",
+  "Topic :: Utilities",
+]
+
+[project.scripts]
+pytk-ai = "pytk_ai.cli:main"
+
+[tool.setuptools]
+package-dir = {"" = "src"}
+
+[tool.setuptools.packages.find]
+where = ["src"]
+
+[tool.setuptools.package-data]
+pytk_ai = ["data/*.json"]
+
+[tool.pytest.ini_options]
+testpaths = ["tests"]

pytk_ai-0.1.0/setup.cfg

@@ -0,0 +1,4 @@
+[egg_info]
+tag_build = 
+tag_date = 0
+

pytk_ai-0.1.0/src/pytk_ai/__init__.py

@@ -0,0 +1,13 @@
+from .config import RunOptions
+from .models import CommandResult, ExecutionResult, FilterResult
+from .runner import run_command
+
+__all__ = [
+    "CommandResult",
+    "ExecutionResult",
+    "FilterResult",
+    "RunOptions",
+    "run_command",
+]
+
+__version__ = "0.1.0"

pytk_ai-0.1.0/src/pytk_ai/__main__.py

@@ -0,0 +1,4 @@
+from .cli import main
+
+if __name__ == "__main__":
+    raise SystemExit(main())

pytk_ai-0.1.0/src/pytk_ai/cli.py

@@ -0,0 +1,146 @@
+from __future__ import annotations
+
+import argparse
+import json
+import shlex
+import sys
+from typing import Any
+
+from .rewrite import rewrite_command
+from .runner import run_command
+
+
+def _join_command(parts: list[str]) -> str:
+    return shlex.join(parts).strip()
+
+
+def _read_stdin_json() -> dict[str, Any] | None:
+    raw = sys.stdin.read().strip()
+    if not raw:
+        return None
+    try:
+        payload = json.loads(raw)
+    except json.JSONDecodeError:
+        return None
+    if isinstance(payload, dict):
+        return payload
+    return None
+
+
+def _hook_command_from_payload(payload: dict[str, Any]) -> str | None:
+    tool_input = payload.get("tool_input")
+    if isinstance(tool_input, dict):
+        command = tool_input.get("command")
+        if isinstance(command, str):
+            return command
+    tool_args = payload.get("toolArgs")
+    if isinstance(tool_args, str):
+        try:
+            parsed = json.loads(tool_args)
+        except json.JSONDecodeError:
+            return None
+        if isinstance(parsed, dict):
+            command = parsed.get("command")
+            if isinstance(command, str):
+                return command
+    return None
+
+
+def _emit_claude_hook(rewritten: str) -> None:
+    payload = {
+        "hookSpecificOutput": {
+            "hookEventName": "PreToolUse",
+            "permissionDecision": "allow",
+            "permissionDecisionReason": "PYTK-AI auto-rewrite",
+            "updatedInput": {"command": rewritten},
+        }
+    }
+    sys.stdout.write(json.dumps(payload))
+
+
+def _emit_cursor_hook(rewritten: str) -> None:
+    payload = {"permission": "allow", "updated_input": {"command": rewritten}}
+    sys.stdout.write(json.dumps(payload))
+
+
+def _run_command(args: argparse.Namespace) -> int:
+    command = _join_command(args.args)
+    result = run_command(
+        command,
+        excluded=tuple(args.exclude or ()),
+        max_output_lines=args.max_output_lines,
+    )
+    if result.filtered_output:
+        sys.stdout.write(result.filtered_output)
+        if not result.filtered_output.endswith("\n"):
+            sys.stdout.write("\n")
+    return result.exit_code
+
+
+def _run_hook(args: argparse.Namespace) -> int:
+    payload = _read_stdin_json()
+    if not payload:
+        if args.agent == "cursor":
+            sys.stdout.write("{}")
+        return 0
+
+    command = _hook_command_from_payload(payload)
+    if not command:
+        if args.agent == "cursor":
+            sys.stdout.write("{}")
+        return 0
+
+    result = rewrite_command(command)
+    if result is None or not result.changed:
+        if args.agent == "cursor":
+            sys.stdout.write("{}")
+        return 0
+
+    if args.agent == "cursor":
+        _emit_cursor_hook(result.output)
+    else:
+        _emit_claude_hook(result.output)
+    return 0
+
+
+def build_parser() -> argparse.ArgumentParser:
+    parser = argparse.ArgumentParser(prog="pytk-ai", description="PYTK-AI command runner")
+    sub = parser.add_subparsers(dest="command", required=True)
+
+    run = sub.add_parser("run", help="execute a raw shell command and filter output")
+    run.add_argument("args", nargs=argparse.REMAINDER, help="raw command tokens")
+    run.add_argument(
+        "--exclude",
+        action="append",
+        default=[],
+        help="base commands to never plan as PYTK-AI-managed",
+    )
+    run.add_argument(
+        "--max-output-lines",
+        type=int,
+        default=200,
+        help="maximum number of filtered output lines",
+    )
+    run.set_defaults(func=_run_command)
+
+    hook = sub.add_parser("hook", help="process hook JSON from stdin")
+    hook_sub = hook.add_subparsers(dest="agent", required=True)
+    for agent in ("claude", "cursor"):
+        agent_parser = hook_sub.add_parser(agent)
+        agent_parser.set_defaults(func=_run_hook)
+
+    return parser
+
+
+def main(argv: list[str] | None = None) -> int:
+    parser = build_parser()
+    args = parser.parse_args(argv)
+    func = getattr(args, "func", None)
+    if func is None:
+        parser.print_help()
+        return 2
+    return func(args)
+
+
+if __name__ == "__main__":
+    raise SystemExit(main())

pytk_ai-0.1.0/src/pytk_ai/config.py

@@ -0,0 +1,13 @@
+from __future__ import annotations
+
+from dataclasses import dataclass, field
+
+
+@dataclass(frozen=True)
+class RunOptions:
+    cwd: str | None = None
+    env: dict[str, str] = field(default_factory=dict)
+    timeout: float | None = None
+    plan: bool = True
+    excluded_commands: tuple[str, ...] = ()
+    max_output_lines: int = 200

pytk_ai-0.1.0/src/pytk_ai/data/__init__.py

@@ -0,0 +1 @@
+"""Packaged command registry data for PYTK-AI."""