rx4 0.7.2

The agent harness engine — loop, tools, providers, sessions, permissions, computer-use
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81

"""Harbor BaseAgent that runs host-built tk inside the trial container.

Coding Plan only: glm-5.3-flash via ZAI_API_KEY. Not OpenCode Go.
"""

from __future__ import annotations

import os
import shlex
from pathlib import Path

from harbor.agents.base import BaseAgent
from harbor.environments.base import BaseEnvironment
from harbor.models.agent.context import AgentContext


class TkHarborAgent(BaseAgent):
    @staticmethod
    def name() -> str:
        return "tk"

    def version(self) -> str:
        return "0.2.0"

    def _tk_bin(self) -> Path:
        override = os.environ.get("TK_BIN")
        if override:
            return Path(override)
        root = os.environ.get("TELEKINESIS_ROOT", "/home/user/workspace/repo")
        return Path(root) / "ui/tui/target/release/tk"

    async def setup(self, environment: BaseEnvironment) -> None:
        tk = self._tk_bin()
        if not tk.is_file():
            raise FileNotFoundError(f"tk binary missing: {tk}")
        await environment.upload_file(tk, "/usr/local/bin/tk")
        chmod = await environment.exec("chmod +x /usr/local/bin/tk", user="root")
        if chmod.return_code != 0:
            raise RuntimeError(chmod.stderr or chmod.stdout or "chmod tk failed")

    async def run(
        self,
        instruction: str,
        environment: BaseEnvironment,
        context: AgentContext,
    ) -> None:
        key = os.environ.get("ZAI_API_KEY", "")
        if not key:
            raise RuntimeError("ZAI_API_KEY is not set")
        prompt = (
            instruction.strip()
            + "\n\nWork in the current workspace. Finish the task completely. "
            "Do not stop until any required binaries are installed at the paths "
            "the instruction names (for example /usr/local/bin/pmars)."
        )
        quoted = shlex.quote(prompt)
        cmd = (
            "mkdir -p /tmp/tk && printf '%s\\n' "
            + quoted
            + " > /tmp/tk/prompt.txt && "
            "tk exec --provider zai --model glm-5.3-flash --effort low "
            "--cwd /app - < /tmp/tk/prompt.txt"
        )
        result = await environment.exec(
            cmd,
            cwd="/app",
            env={
                "ZAI_API_KEY": key,
                "TK_MAX_TURNS": os.environ.get("TK_MAX_TURNS", "250"),
            },
            timeout_sec=int(os.environ.get("BENCH_AGENT_TIMEOUT", "1800")),
        )
        log = self.logs_dir / "tk-harbor.log"
        log.write_text(
            f"exit={result.return_code}\nstdout:\n{result.stdout or ''}\n"
            f"stderr:\n{result.stderr or ''}\n"
        )
        if result.return_code != 0:
            raise RuntimeError(
                f"tk exec failed ({result.return_code}): {result.stderr or result.stdout}"
            )