agentshim 0.1.0__py3-none-any.whl

agentshim/usage.py

@@ -0,0 +1,63 @@
+"""Token/turn usage value objects for CLI-based coding agents.
+
+Duplicated in shape (not in import) from ``libs/pydantic_agent/_usage.py``
+so ``agentshim`` does not depend on the pydantic-ai stack. The
+``to_dict()`` output shape is the shared contract between the two copies
+and is pinned by ``tests/unit/test_usage_schema_parity.py``.
+"""
+
+from __future__ import annotations
+
+from dataclasses import dataclass, field
+from typing import Any
+
+
+@dataclass(frozen=True)
+class TokenUsage:
+    """Token counts for a CLI-agent session.
+
+    Invariant: ``cached_input_tokens`` is a subset of ``input_tokens``.
+    """
+
+    input_tokens: int = 0
+    output_tokens: int = 0
+    cached_input_tokens: int = 0
+    turns: int = 0
+
+    def __add__(self, other: TokenUsage) -> TokenUsage:
+        return TokenUsage(
+            input_tokens=self.input_tokens + other.input_tokens,
+            output_tokens=self.output_tokens + other.output_tokens,
+            cached_input_tokens=self.cached_input_tokens + other.cached_input_tokens,
+            turns=self.turns + other.turns,
+        )
+
+    def to_dict(self) -> dict[str, int]:
+        return {
+            "input_tokens": self.input_tokens,
+            "output_tokens": self.output_tokens,
+            "cached_input_tokens": self.cached_input_tokens,
+            "turns": self.turns,
+        }
+
+
+@dataclass(frozen=True)
+class ProviderUsage:
+    """Per-session usage bundle: common token counts + provider extras.
+
+    ``provider`` is the registered provider name ("claude", "codex",
+    "gemini", "opencode"). ``total_cost_usd`` is ``None`` when the
+    provider does not report cost in its stream (codex, gemini, opencode
+    depending on config).
+    """
+
+    tokens: TokenUsage = field(default_factory=TokenUsage)
+    total_cost_usd: float | None = None
+    provider: str = ""
+
+    def to_dict(self) -> dict[str, Any]:
+        return {
+            **self.tokens.to_dict(),
+            "total_cost_usd": self.total_cost_usd,
+            "provider": self.provider,
+        }

agentshim/utils.py

@@ -0,0 +1,120 @@
+import os
+import re
+import subprocess
+from pathlib import Path
+from typing import Any
+
+from loguru import logger
+
+
+def get_interactive_env() -> dict[str, str]:
+    """Capture environment variables from an interactive shell."""
+    try:
+        # Run env in an interactive shell to get the full user environment
+        # Use start_new_session=True (setsid) to detach from TTY and avoid
+        # SIGTTOU/SIGTTIN signals when bash -i tries to set process group
+        result = subprocess.run(
+            ["/bin/bash", "-i", "-c", "env"], capture_output=True, text=True, check=False, start_new_session=True
+        )
+        if result.returncode != 0:
+            return os.environ.copy()
+
+        env: dict[str, str] = {}
+        for line in result.stdout.splitlines():
+            if "=" in line:
+                key, value = line.split("=", 1)
+                env[key] = value
+        return env
+    except Exception:
+        return os.environ.copy()
+
+
+# -- File generation system prompt (shared by RLMCodingAgent & SubagentCodingAgent) --
+
+
+FILE_GEN_SYSTEM_PROMPT = (
+    "You are a deployment assistant. The user will ask you to generate "
+    "one or more files. For EACH file, output a section in this exact format:\n\n"
+    "FILE: .sds/<filename>\n"
+    "```\n"
+    "<file content here>\n"
+    "```\n\n"
+    "Output ONLY these sections. Do not add explanations outside the sections."
+)
+
+
+def generate_and_write_files(
+    raw: str,
+    prompt: str,
+    repo_path: Path,
+    log_label: str,
+) -> list[str]:
+    """Parse ``FILE: .sds/<name>`` sections from *raw* LLM output and write them.
+
+    Returns a list of relative paths that were written.  If no ``FILE:`` sections
+    are found but the prompt references exactly one ``.sds/`` file, the entire
+    *raw* response (with markdown fences stripped) is written as that file
+    (fallback mode).
+
+    Args:
+        raw: Raw LLM response text.
+        prompt: The original prompt (used to detect expected output files).
+        repo_path: Root of the repository.
+        log_label: Label for log messages (e.g. ``"[RLM]"``, ``"[Subagent]"``).
+    """
+    expected_files = re.findall(r"\.sds/[\w._-]+", prompt)
+
+    written: list[str] = []
+    file_sections = re.findall(
+        r"(?:^|\n)\s*(?:#+\s*)?FILE:\s*(\.sds/[\w./-]+)\s*\n```[^\n]*\n(.*?)```",
+        raw,
+        re.DOTALL | re.IGNORECASE,
+    )
+    for rel_path, content in file_sections:
+        out_path = repo_path / rel_path
+        out_path.parent.mkdir(parents=True, exist_ok=True)
+        out_path.write_text(content)
+        logger.info(f"{log_label} Wrote {out_path}")
+        written.append(rel_path)
+
+    if not written and len(expected_files) == 1:
+        out_path = repo_path / expected_files[0]
+        content = re.sub(r"^```[^\n]*\n|```$", "", raw.strip(), flags=re.MULTILINE)
+        out_path.parent.mkdir(parents=True, exist_ok=True)
+        out_path.write_text(content)
+        logger.info(f"{log_label} Wrote {out_path} (fallback)")
+        written.append(expected_files[0])
+
+    return written
+
+
+DEFAULT_TOOL_PARAM_MAX_LEN = 200
+MEMORY_TOOL_PARAM_MAX_LEN = 10000
+_VERBOSE_TOOL_NAMES = frozenset({"recall_incident", "store_incident"})
+
+
+def truncate_params(params: Any, max_len: int = DEFAULT_TOOL_PARAM_MAX_LEN) -> str:
+    """Truncate a parameter representation to *max_len* characters."""
+    s = str(params)
+    if len(s) > max_len:
+        return s[:max_len] + "..."
+    return s
+
+
+def truncate_tool_params(tool_name: str, params: Any) -> str:
+    """Render tool parameters with per-tool truncation rules.
+
+    Most tools keep the historical compact preview so CLI logs stay readable.
+    Incident-memory tools get a much larger budget because their payloads are
+    often the exact debugging context we need to inspect.
+    """
+    max_len = MEMORY_TOOL_PARAM_MAX_LEN if tool_name in _VERBOSE_TOOL_NAMES else DEFAULT_TOOL_PARAM_MAX_LEN
+    return truncate_params(params, max_len=max_len)
+
+
+def truncate_content(content: str, max_lines: int = 10) -> str:
+    """Truncate *content* keeping the first and last *max_lines* lines."""
+    lines = content.splitlines()
+    if len(lines) > max_lines * 2:
+        return "\n".join(lines[:max_lines] + ["... (truncated) ..."] + lines[-max_lines:])
+    return content

agentshim-0.1.0.dist-info/METADATA

@@ -0,0 +1,61 @@
+Metadata-Version: 2.4
+Name: agentshim
+Version: 0.1.0
+Summary: Provider-agnostic coding-agent CLI shims
+Requires-Python: >=3.10
+Requires-Dist: litellm>=1.0.0
+Requires-Dist: loguru>=0.7.2
+Provides-Extra: test
+Requires-Dist: pytest>=8.0.0; extra == 'test'
+Description-Content-Type: text/markdown
+
+# agentshim
+
+`agentshim` is a small Python package that wraps coding-agent CLIs behind a
+stable interface for higher-level runtimes.
+
+It currently includes:
+
+- a `CodingAgent` base class and provider registry
+- CLI adapters for Claude Code, Codex, Gemini, and Opencode
+- MCP server config dataclasses
+- sandbox settings helpers for Claude Code
+- a lightweight litellm client and isolated subagent helper
+- a trajectory protocol with a no-op implementation
+
+## Install
+
+```bash
+pip install agentshim
+```
+
+For development:
+
+```bash
+uv sync --dev
+uv run pytest
+```
+
+## Quick Example
+
+```python
+from agentshim.claude import ClaudeCodeCodingAgent
+
+agent = ClaudeCodeCodingAgent(model="sonnet")
+result = agent.generate("Summarize the repository layout.", cwd=".")
+print(result)
+```
+
+## Publishing
+
+Build locally with:
+
+```bash
+uv build
+```
+
+Publish with:
+
+```bash
+uv publish
+```

agentshim-0.1.0.dist-info/RECORD

@@ -0,0 +1,33 @@
+agentshim/__init__.py,sha256=6jsY9Z3wuEl_OZcyrLJWmSzl6tyAR24oUW99UMXbpZc,692
+agentshim/base.py,sha256=Z23Ca3ufd9qGDBK9ER4OeCGEpgdlNgX8t1aSpWhv1UQ,1425
+agentshim/claude_events.py,sha256=U8A1IH4dsVHxfHrq28ghAUv1m6P2H_1ZHTp6SnJ3irw,375
+agentshim/cli_agent.py,sha256=INdN67V10KOvS9v0miImx3YjRZz6PJscw34WQ3SnaHM,15623
+agentshim/codex_events.py,sha256=GrUlbtvE_xh00tLWpMqBOnoKjEw4R31jubKbSctjwIQ,441
+agentshim/events.py,sha256=meWWo8MX-25ByDtYqmYCzZ93KyYMWgKYpXJcT8dT8aw,622
+agentshim/gemini_events.py,sha256=6mlITPfx_WjX-4mDZbt5FVw4iJCcuSAnjxdeP9Rm4ow,282
+agentshim/llm_client.py,sha256=sOXrvp6qJiFWx6VAy-szNkapli55qRjIK7_yed0fHvQ,2245
+agentshim/mcp_config.py,sha256=q_gJY0k89a0AHPUe_9KNNIHyb1xr5SAHMEke_XMrCmc,710
+agentshim/opencode_events.py,sha256=S54kA-3olnkqm1sHD_CKRt_KnrRzEp6Nhdu2E08ppao,294
+agentshim/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+agentshim/sandbox.py,sha256=pNEZM85cXq3Swh9S_Vkb2V-02QzEF5RH2MKWwWaUk14,6071
+agentshim/subagent.py,sha256=S_mfykQ9envLQ1BshbXEg4e8sTxI3Db-mQgcQ7W3hzg,3264
+agentshim/trajectory.py,sha256=xjLWv-EXc6_U3ubHg8SATvfgTMn4AEnMxiqG4Oc79wY,5064
+agentshim/usage.py,sha256=yf_M9PkDyDZnz9pv8Z8oISeHSRqp0N62NeemFQ1tWS0,2015
+agentshim/utils.py,sha256=qXNOma48LaiTeEKQr8cbNejA8qtErZMXX1JldyuxUb8,4253
+agentshim/claude/__init__.py,sha256=7-enwzQK90mpHy_Z-IKkkoErwCL8aIfalo3SV24l7f0,130
+agentshim/claude/agent.py,sha256=BVbiW7M9cpQpSBEE9mr13BxZArrK6ki8ywNcmHl_gGE,10310
+agentshim/claude/events.py,sha256=90O7QFre0mNKTn7BX8MaSWewg1tpywDLxdiZ7cwHVwI,5075
+agentshim/claude/hooks/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+agentshim/claude/hooks/confine_reads.py,sha256=kWDxAi1MyPtDNnGmuD0ZNrnIkzDdkjctr2oaBwiDIPc,2237
+agentshim/codex/__init__.py,sha256=bgHz_1JcfcEeUwJwCBqWjauiIvhun7HoJkzdhCBTPQE,118
+agentshim/codex/agent.py,sha256=ttZEbiLja5IAjPi1vxp6_X1zzvAEa-m-pFKLjF26aR4,8500
+agentshim/codex/events.py,sha256=I4u5Uug1ku12a44gouBNJgZGVJdCNx2o8xCQFUZi3Rg,7602
+agentshim/gemini/__init__.py,sha256=HYgiOZwekbPGaTn6aQURuLa7ZIoUmienU_Tr4apDdsI,122
+agentshim/gemini/agent.py,sha256=DlJn2YdlLfkRbZhyjyhURJOhzOWE0v1SWlMgqhG434o,8500
+agentshim/gemini/events.py,sha256=Mj_poV6MzCAWRjWLD9-GJ_U3VbZGdzdEPD7jsC5IsnY,2727
+agentshim/opencode/__init__.py,sha256=mQPn-hR4lIIxjKlrdc0byhdqlD9f0ST-UkKK_1vedHk,130
+agentshim/opencode/agent.py,sha256=I9-vqXxgi6eUNOxgFsBsTa7jEW7_r6gfpu4kTVlV35w,7103
+agentshim/opencode/events.py,sha256=svYi3zP9cziaUPnKlxCH3pdj41qUcvOP2FXb0qOhvks,3027
+agentshim-0.1.0.dist-info/METADATA,sha256=FgjWfvVSOLZrUoU3OtDEj-f0vFJzxEp0RIj8JlLJYrE,1167
+agentshim-0.1.0.dist-info/WHEEL,sha256=QccIxa26bgl1E6uMy58deGWi-0aeIkkangHcxk2kWfw,87
+agentshim-0.1.0.dist-info/RECORD,,

agentshim-0.1.0.dist-info/WHEEL

@@ -0,0 +1,4 @@
+Wheel-Version: 1.0
+Generator: hatchling 1.29.0
+Root-Is-Purelib: true
+Tag: py3-none-any