openab 0.3.0__tar.gz

openab-0.3.0/LICENSE

@@ -0,0 +1,21 @@
+MIT License
+
+Copyright (c) 2025
+
+Permission is hereby granted, free of charge, to any person obtaining a copy
+of this software and associated documentation files (the "Software"), to deal
+in the Software without restriction, including without limitation the rights
+to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
+copies of the Software, and to permit persons to whom the Software is
+furnished to do so, subject to the following conditions:
+
+The above copyright notice and this permission notice shall be included in all
+copies or substantial portions of the Software.
+
+THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
+IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
+FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
+AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
+LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
+OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
+SOFTWARE.

openab-0.3.0/PKG-INFO

@@ -0,0 +1,70 @@
+Metadata-Version: 2.4
+Name: openab
+Version: 0.3.0
+Summary: Open Agent Bridge — connect agent backends (CLIs, APIs) to chat platforms (Telegram, …).
+License-Expression: MIT
+Requires-Python: >=3.10
+Description-Content-Type: text/markdown
+License-File: LICENSE
+Requires-Dist: python-telegram-bot>=21.0
+Requires-Dist: discord.py>=2.3.0
+Requires-Dist: typer>=0.12.0
+Requires-Dist: python-dotenv>=1.0.0
+Requires-Dist: pyyaml>=6.0
+Provides-Extra: dev
+Requires-Dist: build; extra == "dev"
+Dynamic: license-file
+
+<p align="center">
+  <img src="assets/icon.png" width="256" alt="OpenAB" />
+</p>
+
+# OpenAB
+
+**Open Agent Bridge** — Connect AI agents (Cursor, Codex, Gemini, Claude, OpenClaw) to chat platforms. One config, one bridge.
+
+[中文](docs/README.zh-CN.md) · [Configuration & usage](docs/guide.md)
+
+---
+
+## What it does
+
+OpenAB forwards messages from **chat platforms** to an agent backend you choose and sends the reply back. No per-agent bots — configure once and chat.
+
+| Agents | Chats |
+|--------|-------|
+| Cursor, Codex, Gemini, Claude, OpenClaw | Telegram, Discord, _more planned_ |
+
+---
+
+## Quick start
+
+**1. Install** (Python 3.10+)
+
+```bash
+pip install openab
+# or: uv tool install openab
+```
+
+**2. Config** — Put a YAML/JSON config in `~/.config/openab/` (see [config.example.yaml](config.example.yaml)). You need at least a bot token and an allowlist of user IDs. Full options: [docs/guide.md](docs/guide.md).
+
+**3. Run**
+
+```bash
+openab              # e.g. Telegram
+openab run-discord  # e.g. Discord
+```
+
+Then open the bot in your chat app and send a message.
+
+---
+
+## Docs
+
+- **[Configuration & usage](docs/guide.md)** — Config keys, agent setup, commands, security, i18n.
+
+---
+
+## License
+
+MIT — [LICENSE](LICENSE)

openab-0.3.0/README.md

@@ -0,0 +1,53 @@
+<p align="center">
+  <img src="assets/icon.png" width="256" alt="OpenAB" />
+</p>
+
+# OpenAB
+
+**Open Agent Bridge** — Connect AI agents (Cursor, Codex, Gemini, Claude, OpenClaw) to chat platforms. One config, one bridge.
+
+[中文](docs/README.zh-CN.md) · [Configuration & usage](docs/guide.md)
+
+---
+
+## What it does
+
+OpenAB forwards messages from **chat platforms** to an agent backend you choose and sends the reply back. No per-agent bots — configure once and chat.
+
+| Agents | Chats |
+|--------|-------|
+| Cursor, Codex, Gemini, Claude, OpenClaw | Telegram, Discord, _more planned_ |
+
+---
+
+## Quick start
+
+**1. Install** (Python 3.10+)
+
+```bash
+pip install openab
+# or: uv tool install openab
+```
+
+**2. Config** — Put a YAML/JSON config in `~/.config/openab/` (see [config.example.yaml](config.example.yaml)). You need at least a bot token and an allowlist of user IDs. Full options: [docs/guide.md](docs/guide.md).
+
+**3. Run**
+
+```bash
+openab              # e.g. Telegram
+openab run-discord  # e.g. Discord
+```
+
+Then open the bot in your chat app and send a message.
+
+---
+
+## Docs
+
+- **[Configuration & usage](docs/guide.md)** — Config keys, agent setup, commands, security, i18n.
+
+---
+
+## License
+
+MIT — [LICENSE](LICENSE)

openab-0.3.0/openab/__init__.py

@@ -0,0 +1,2 @@
+# OpenAB — Open Agent Bridge
+__version__ = "0.2.0"

openab-0.3.0/openab/__main__.py

@@ -0,0 +1,5 @@
+"""Entry point: python -m openab"""
+from openab.cli.main import app
+
+if __name__ == "__main__":
+    app()

openab-0.3.0/openab/agents/__init__.py

@@ -0,0 +1,71 @@
+"""Agent backends: Cursor, Codex, Gemini, Claude, OpenClaw. 由 agent_config 或环境变量指定后端与选项。"""
+from __future__ import annotations
+
+import asyncio
+import os
+from pathlib import Path
+from typing import Any, Optional
+
+from openab.core.i18n import t
+
+from . import claude, codex, cursor, gemini, openclaw
+
+
+def get_backend(agent_config: dict[str, Any] | None = None) -> str:
+    if agent_config:
+        agent = agent_config.get("agent") or {}
+        b = agent.get("backend") or "cursor"
+        return str(b).strip().lower()
+    return (os.environ.get("OPENAB_AGENT") or "cursor").strip().lower()
+
+
+def run_agent(
+    prompt: str,
+    *,
+    workspace: Optional[Path] = None,
+    timeout: Optional[int] = None,
+    agent_config: Optional[dict[str, Any]] = None,
+) -> str:
+    """同步执行当前配置的 agent，返回完整 stdout 文本。"""
+    return asyncio.get_event_loop().run_until_complete(
+        run_agent_async(
+            prompt,
+            workspace=workspace,
+            timeout=timeout,
+            agent_config=agent_config,
+        )
+    )
+
+
+async def run_agent_async(
+    prompt: str,
+    *,
+    workspace: Optional[Path] = None,
+    timeout: int = 300,
+    lang: str = "en",
+    agent_config: Optional[dict[str, Any]] = None,
+) -> str:
+    """异步执行 agent；backend 与各后端选项来自 agent_config，缺省时回退到环境变量。"""
+    backend = get_backend(agent_config)
+    if backend == "codex":
+        return await codex.run_async(
+            prompt, workspace=workspace, timeout=timeout, lang=lang, agent_config=agent_config
+        )
+    if backend == "gemini":
+        return await gemini.run_async(
+            prompt, workspace=workspace, timeout=timeout, lang=lang, agent_config=agent_config
+        )
+    if backend == "claude":
+        return await claude.run_async(
+            prompt, workspace=workspace, timeout=timeout, lang=lang, agent_config=agent_config
+        )
+    if backend == "openclaw":
+        return await openclaw.run_async(
+            prompt, workspace=workspace, timeout=timeout, lang=lang, agent_config=agent_config
+        )
+    return await cursor.run_async(
+        prompt, workspace=workspace, timeout=timeout, lang=lang, agent_config=agent_config
+    )
+
+
+__all__ = ["run_agent", "run_agent_async", "get_backend"]

openab-0.3.0/openab/agents/claude.py

@@ -0,0 +1,100 @@
+"""Claude Code CLI backend (https://code.claude.com/docs/en/cli-reference).
+
+Print mode: claude -p "query" — response to stdout, then exit.
+Flags used: --output-format text, --no-session-persistence; optional: --model, --max-turns, --add-dir.
+"""
+from __future__ import annotations
+
+import asyncio
+import os
+import shutil
+from pathlib import Path
+from typing import Any, Optional
+
+from openab.core.i18n import t
+
+
+def _find_cmd(agent_config: dict[str, Any] | None = None) -> str:
+    cmd = "claude"
+    if agent_config:
+        c = (agent_config.get("claude") or {}).get("cmd")
+        if c:
+            cmd = str(c)
+    if not cmd:
+        cmd = os.environ.get("CLAUDE_CLI_CMD", "claude")
+    if os.path.isabs(cmd):
+        return cmd
+    exe = shutil.which(cmd)
+    return exe or cmd
+
+
+def _build_args(
+    prompt: str,
+    workspace: Optional[Path],
+    agent_config: dict[str, Any] | None = None,
+) -> list[str]:
+    cmd = _find_cmd(agent_config)
+    args = [
+        cmd,
+        "--print",
+        "--output-format", "text",
+        "--no-session-persistence",
+    ]
+    model = ""
+    max_turns = ""
+    add_dirs: list[str] = []
+    if agent_config:
+        c = agent_config.get("claude") or {}
+        if c.get("model"):
+            model = str(c["model"]).strip()
+        if c.get("max_turns") is not None:
+            max_turns = str(c["max_turns"]).strip()
+        ad = c.get("add_dir")
+        if isinstance(ad, str) and ad:
+            add_dirs = [d.strip() for d in ad.split(os.pathsep) if d.strip()]
+        elif isinstance(ad, list):
+            add_dirs = [str(x).strip() for x in ad if str(x).strip()]
+    if not model:
+        model = os.environ.get("CLAUDE_CLI_MODEL", "").strip()
+    if not max_turns:
+        max_turns = os.environ.get("CLAUDE_CLI_MAX_TURNS", "").strip()
+    if not add_dirs:
+        raw = os.environ.get("CLAUDE_CLI_ADD_DIR", "").strip()
+        if raw:
+            add_dirs = [d.strip() for d in raw.split(os.pathsep) if d.strip()]
+    if model:
+        args.extend(["--model", model])
+    if max_turns and max_turns.isdigit():
+        args.extend(["--max-turns", max_turns])
+    for d in add_dirs:
+        args.extend(["--add-dir", d])
+    args.append(prompt)
+    return args
+
+
+async def run_async(
+    prompt: str,
+    *,
+    workspace: Optional[Path] = None,
+    timeout: int = 300,
+    lang: str = "en",
+    agent_config: Optional[dict[str, Any]] = None,
+) -> str:
+    """Run Claude Code CLI in print mode; return stdout as reply."""
+    args = _build_args(prompt, workspace, agent_config)
+    cwd = str(workspace) if workspace else None
+    proc = await asyncio.create_subprocess_exec(
+        *args,
+        stdout=asyncio.subprocess.PIPE,
+        stderr=asyncio.subprocess.PIPE,
+        env=os.environ.copy(),
+        cwd=cwd,
+    )
+    try:
+        stdout, _ = await asyncio.wait_for(proc.communicate(), timeout=timeout)
+    except asyncio.TimeoutError:
+        proc.kill()
+        await proc.wait()
+        return t(lang, "agent_timeout")
+    text = (stdout or b"").decode("utf-8", errors="replace").strip()
+    return text or t(lang, "agent_no_output")

openab-0.3.0/openab/agents/codex.py

@@ -0,0 +1,64 @@
+"""OpenAI Codex CLI backend."""
+from __future__ import annotations
+
+import asyncio
+import os
+import shutil
+from pathlib import Path
+from typing import Any, Optional
+
+from openab.core.i18n import t
+
+
+def _find_cmd(agent_config: dict[str, Any] | None = None) -> str:
+    cmd = "codex"
+    if agent_config:
+        c = (agent_config.get("codex") or {}).get("cmd")
+        if c:
+            cmd = str(c)
+    if not cmd:
+        cmd = os.environ.get("CODEX_CMD", "codex")
+    if os.path.isabs(cmd):
+        return cmd
+    exe = shutil.which(cmd)
+    return exe or cmd
+
+
+def _skip_git_check(agent_config: dict[str, Any] | None = None) -> bool:
+    if agent_config:
+        v = (agent_config.get("codex") or {}).get("skip_git_check")
+        if v in (True, "true", "1", 1):
+            return True
+    return os.environ.get("CODEX_SKIP_GIT_CHECK", "").strip().lower() in ("1", "true", "yes")
+
+
+async def run_async(
+    prompt: str,
+    *,
+    workspace: Optional[Path] = None,
+    timeout: int = 300,
+    lang: str = "en",
+    agent_config: Optional[dict[str, Any]] = None,
+) -> str:
+    """Codex CLI: codex exec --ephemeral, final message on stdout."""
+    cmd = _find_cmd(agent_config)
+    args = [cmd, "exec", "--ephemeral"]
+    if _skip_git_check(agent_config):
+        args.append("--skip-git-repo-check")
+    args.append(prompt)
+    cwd = str(workspace) if workspace else None
+    proc = await asyncio.create_subprocess_exec(
+        *args,
+        stdout=asyncio.subprocess.PIPE,
+        stderr=asyncio.subprocess.DEVNULL,
+        env=os.environ.copy(),
+        cwd=cwd,
+    )
+    try:
+        stdout, _ = await asyncio.wait_for(proc.communicate(), timeout=timeout)
+    except asyncio.TimeoutError:
+        proc.kill()
+        await proc.wait()
+        return t(lang, "agent_timeout")
+    text = (stdout or b"").decode("utf-8", errors="replace").strip()
+    return text or t(lang, "agent_no_output")

openab-0.3.0/openab/agents/cursor.py

@@ -0,0 +1,62 @@
+"""Cursor Agent CLI backend."""
+from __future__ import annotations
+
+import asyncio
+import os
+import shutil
+from pathlib import Path
+from typing import Any, Optional
+
+from openab.core.i18n import t
+
+
+def _find_cmd(agent_config: dict[str, Any] | None = None) -> str:
+    cmd = "agent"
+    if agent_config:
+        c = (agent_config.get("cursor") or {}).get("cmd")
+        if c:
+            cmd = str(c)
+    if not cmd:
+        cmd = os.environ.get("CURSOR_AGENT_CMD", "agent")
+    if os.path.isabs(cmd):
+        return cmd
+    exe = shutil.which(cmd)
+    return exe or cmd
+
+
+async def run_async(
+    prompt: str,
+    *,
+    workspace: Optional[Path] = None,
+    timeout: int = 300,
+    lang: str = "en",
+    agent_config: Optional[dict[str, Any]] = None,
+) -> str:
+    """Cursor Agent CLI: agent --print --trust."""
+    cmd = _find_cmd(agent_config)
+    base_args = [
+        cmd,
+        "agent",
+        "--print",
+        "--output-format", "text",
+        "--trust",
+    ]
+    if workspace is not None:
+        base_args.extend(["--workspace", str(workspace)])
+    base_args.extend(["--", prompt])
+    env = os.environ.copy()
+    proc = await asyncio.create_subprocess_exec(
+        *base_args,
+        stdout=asyncio.subprocess.PIPE,
+        stderr=asyncio.subprocess.STDOUT,
+        env=env,
+        cwd=str(workspace) if workspace else None,
+    )
+    try:
+        stdout, _ = await asyncio.wait_for(proc.communicate(), timeout=timeout)
+    except asyncio.TimeoutError:
+        proc.kill()
+        await proc.wait()
+        return t(lang, "agent_timeout")
+    text = (stdout or b"").decode("utf-8", errors="replace").strip()
+    return text or t(lang, "agent_no_output")

openab-0.3.0/openab/agents/gemini.py

@@ -0,0 +1,53 @@
+"""Gemini CLI backend (google-gemini/gemini-cli)."""
+from __future__ import annotations
+
+import asyncio
+import os
+import shutil
+from pathlib import Path
+from typing import Any, Optional
+
+from openab.core.i18n import t
+
+
+def _find_cmd(agent_config: dict[str, Any] | None = None) -> str:
+    cmd = "gemini"
+    if agent_config:
+        c = (agent_config.get("gemini") or {}).get("cmd")
+        if c:
+            cmd = str(c)
+    if not cmd:
+        cmd = os.environ.get("GEMINI_CLI_CMD", "gemini")
+    if os.path.isabs(cmd):
+        return cmd
+    exe = shutil.which(cmd)
+    return exe or cmd
+
+
+async def run_async(
+    prompt: str,
+    *,
+    workspace: Optional[Path] = None,
+    timeout: int = 300,
+    lang: str = "en",
+    agent_config: Optional[dict[str, Any]] = None,
+) -> str:
+    """Gemini CLI: gemini -p \"prompt\" → stdout."""
+    cmd = _find_cmd(agent_config)
+    args = [cmd, "-p", prompt]
+    cwd = str(workspace) if workspace else None
+    proc = await asyncio.create_subprocess_exec(
+        *args,
+        stdout=asyncio.subprocess.PIPE,
+        stderr=asyncio.subprocess.PIPE,
+        env=os.environ.copy(),
+        cwd=cwd,
+    )
+    try:
+        stdout, _ = await asyncio.wait_for(proc.communicate(), timeout=timeout)
+    except asyncio.TimeoutError:
+        proc.kill()
+        await proc.wait()
+        return t(lang, "agent_timeout")
+    text = (stdout or b"").decode("utf-8", errors="replace").strip()
+    return text or t(lang, "agent_no_output")

openab-0.3.0/openab/agents/openclaw.py

@@ -0,0 +1,77 @@
+"""OpenClaw 后端：通过 CLI `openclaw agent --message "..."` 调用，需先运行 OpenClaw Gateway（或使用本地回退）。
+
+参考：https://docs.openclaw.ai/tools/agent-send
+安装：npm install -g openclaw@latest，并运行 openclaw onboard / openclaw gateway。
+"""
+from __future__ import annotations
+
+import asyncio
+import os
+import shutil
+from pathlib import Path
+from typing import Any, Optional
+
+from openab.core.i18n import t
+
+
+def _find_cmd(agent_config: dict[str, Any] | None = None) -> str:
+    cmd = "openclaw"
+    if agent_config:
+        c = (agent_config.get("openclaw") or {}).get("cmd")
+        if c:
+            cmd = str(c)
+    if not cmd:
+        cmd = os.environ.get("OPENCLAW_CMD", "openclaw")
+    if os.path.isabs(cmd):
+        return cmd
+    exe = shutil.which(cmd)
+    return exe or cmd
+
+
+def _strip_media_lines(text: str) -> str:
+    """去掉 OpenClaw 输出中的 MEDIA: 行，只保留回复正文。"""
+    lines = []
+    for line in text.splitlines():
+        if line.strip().startswith("MEDIA:"):
+            continue
+        lines.append(line)
+    return "\n".join(lines).strip()
+
+
+async def run_async(
+    prompt: str,
+    *,
+    workspace: Optional[Path] = None,
+    timeout: int = 300,
+    lang: str = "en",
+    agent_config: Optional[dict[str, Any]] = None,
+) -> str:
+    """调用 openclaw agent --message \"<prompt>\"，从 stdout 取回复；默认会过滤 MEDIA: 行。"""
+    cmd = _find_cmd(agent_config)
+    args = [cmd, "agent", "--message", prompt]
+    oc_cfg = (agent_config or {}).get("openclaw") or {}
+    to = oc_cfg.get("timeout") or timeout
+    if to and int(to) > 0:
+        args.extend(["--timeout", str(int(to))])
+    thinking = (oc_cfg.get("thinking") or "").strip().lower()
+    if thinking in ("off", "minimal", "low", "medium", "high", "xhigh"):
+        args.extend(["--thinking", thinking])
+    cwd = str(workspace) if workspace else None
+    proc = await asyncio.create_subprocess_exec(
+        *args,
+        stdout=asyncio.subprocess.PIPE,
+        stderr=asyncio.subprocess.PIPE,
+        env=os.environ.copy(),
+        cwd=cwd,
+    )
+    try:
+        stdout, stderr = await asyncio.wait_for(proc.communicate(), timeout=timeout)
+    except asyncio.TimeoutError:
+        proc.kill()
+        await proc.wait()
+        return t(lang, "agent_timeout")
+    text = (stdout or b"").decode("utf-8", errors="replace")
+    text = _strip_media_lines(text)
+    if not text.strip():
+        return t(lang, "agent_no_output")
+    return text

openab-0.3.0/openab/chats/__init__.py

@@ -0,0 +1 @@
+# Chat frontends: Telegram, (future: Discord, Slack, …)

openab-0.3.0/openab/chats/discord/__init__.py

@@ -0,0 +1,3 @@
+from .bot import run_bot
+
+__all__ = ["run_bot"]