@liwu1995/afu-context 0.1.0 → 0.1.1
This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
- package/README.md +64 -70
- package/README.zh-CN.md +71 -46
- package/bin/afu.mjs +13 -3
- package/bin/host-autoload.py +328 -0
- package/package.json +1 -1
package/README.md
CHANGED
|
@@ -1,110 +1,104 @@
|
|
|
1
|
-
# AFU Context
|
|
1
|
+
# AFU Context · Local asynchronous context compression
|
|
2
2
|
|
|
3
|
-
**English** · [简体中文](README.zh-CN.md)
|
|
3
|
+
**English** · [简体中文](README.zh-CN.md) · [npm](https://www.npmjs.com/package/@liwu1995/afu-context) · [Install](#install) · [Results](#results) · [Dashboard](#dashboard)
|
|
4
4
|
|
|
5
|
-
|
|
5
|
+
In a long session, the same tool output can travel with every model request. AFU uses the host’s selected model to summarize completed history in the background. Once ready, the summary can replace that history on a later request while preserving newer tool calls and messages.
|
|
6
6
|
|
|
7
|
-
##
|
|
7
|
+
## How AFU compacts a session
|
|
8
8
|
|
|
9
|
-
|
|
9
|
+
| When | What AFU does |
|
|
10
|
+
| --- | --- |
|
|
11
|
+
| A session reaches its threshold | Prepares a summary in the background while the session continues. |
|
|
12
|
+
| The host next needs the history | Checks that the covered prefix still matches before replacing it. |
|
|
13
|
+
| After replacement | Records a successful upstream response or a host commit so the dashboard can distinguish generation from application. |
|
|
10
14
|
|
|
11
|
-
|
|
15
|
+
<a id="install"></a>
|
|
12
16
|
|
|
13
|
-
|
|
14
|
-
npx @liwu1995/afu-context install # detects installed hosts; asks when multiple are found
|
|
15
|
-
# Non-interactive: npx @liwu1995/afu-context install claude
|
|
16
|
-
```
|
|
17
|
+
## Quick start
|
|
17
18
|
|
|
18
|
-
|
|
19
|
+
Install and sign in to your host first. You need npm and Node.js **22.19+ (22.x) or 24+**. Pick one row, run the install command in your terminal, then use the launch command.
|
|
19
20
|
|
|
20
|
-
|
|
21
|
-
|
|
22
|
-
| Host | Integration | Launch | Validated compatibility |
|
|
21
|
+
| Host | System | Install | Launch after installation |
|
|
23
22
|
| --- | --- | --- | --- |
|
|
24
|
-
| Claude Code |
|
|
25
|
-
|
|
|
26
|
-
|
|
|
27
|
-
|
|
|
28
|
-
|
|
29
|
-
Native services use user launchd on macOS, systemd user services on Linux, and the existing Windows scheduled-task integration for Claude Code. Linux native automatic installation requires a working `systemctl --user`. Codex may require approving its hooks. Host credentials remain owned by the host. Official Claude subscription/OAuth and Windows real-machine acceptance remain pending.
|
|
23
|
+
| <a id="afu-claude"></a>Claude Code | macOS / Linux | `npx --yes @liwu1995/afu-context@next install claude` | `claude` |
|
|
24
|
+
| Claude Code | Windows 10/11¹ | `npx.cmd --yes "@liwu1995/afu-context@next" install claude` | `claude` |
|
|
25
|
+
| <a id="afu-codex"></a>Codex CLI | macOS / Linux | `npx --yes @liwu1995/afu-context@next install codex` | `codex` |
|
|
26
|
+
| <a id="afu-pi"></a>Pi | macOS / Linux | `npx --yes @liwu1995/afu-context@next install pi` | `pi` |
|
|
27
|
+
| <a id="afu-dsh"></a>DSH | macOS / Linux | `npx --yes @liwu1995/afu-context@next install dsh --profile YOUR_PROFILE` | `dsh --profile YOUR_PROFILE` |
|
|
30
28
|
|
|
31
|
-
|
|
32
|
-
|
|
33
|
-
```sh
|
|
34
|
-
afu status
|
|
35
|
-
afu status claude --json
|
|
36
|
-
afu ui --open
|
|
37
|
-
```
|
|
29
|
+
The macOS packages target Apple Silicon (ARM64), and the Linux packages target x64. Both platforms have been tested and include Python. Start a new session for Claude Code or Codex. Pi loads the extension in new sessions. Replace `YOUR_PROFILE` with the DSH profile to integrate, such as `headless` or `web`. For web, you can also launch with `dsh web`. To remove just one profile, run `npx --yes @liwu1995/afu-context@next remove dsh --profile YOUR_PROFILE`.
|
|
38
30
|
|
|
39
|
-
|
|
31
|
+
¹ On Windows, use PowerShell, Python **3.11+** and a Claude Code version with PowerShell hook support. Native Claude Code is implemented and awaits Windows testing. Native Codex is unsupported, and Pi and DSH are unverified on Windows.
|
|
40
32
|
|
|
41
|
-
|
|
33
|
+
<details>
|
|
34
|
+
<summary>Linux manual startup and Pi / DSH launch options</summary>
|
|
42
35
|
|
|
43
|
-
|
|
36
|
+
On Linux without user systemd, append `--no-autostart` to the Claude Code or Codex install command. Run it again after a reboot to start the gateway.
|
|
44
37
|
|
|
45
|
-
|
|
46
|
-
afu remove claude # substitute codex, pi or dsh
|
|
47
|
-
```
|
|
38
|
+
For Pi **0.84.4**, pass `--provider` and `--model` on the Pi command line. DSH was tested with **0.1.2-rc.1** and Cordis **4.0.2**. If a web profile reports an incompatible HMR API, run `npx --yes @liwu1995/afu-context@next dsh --web-compat -- --profile web`.
|
|
48
39
|
|
|
49
|
-
|
|
40
|
+
</details>
|
|
50
41
|
|
|
51
|
-
|
|
42
|
+
<a id="results"></a>
|
|
52
43
|
|
|
53
|
-
|
|
54
|
-
- `AFU_RUNTIME_DIR`: isolated Python runtime cache.
|
|
55
|
-
- `AFU_STATE_DIR`: integration receipts and metadata logs.
|
|
56
|
-
- `afu install claude --context-window 200000`: explicit CC window; substitute Codex where appropriate. A changed build or explicit configuration creates a healthy new generation before switching future sessions; existing sessions keep their original endpoint.
|
|
57
|
-
- Pi/DSH follow the selected model's declared window. Default trigger ratio: 40% at 1M or above, 50% otherwise. Host metadata unavailable means AFU yields to native behavior.
|
|
58
|
-
- `--config-dir` and `--no-autostart` support disposable native sandbox tests; do not substitute hooks-only tests for production service-manager acceptance.
|
|
44
|
+
## Benchmark results
|
|
59
45
|
|
|
60
|
-
|
|
46
|
+
In a LongBench V2 replay of 39 history questions (20 agent histories and 19 dialogue histories), AFU’s asynchronous strategy answered **23 correctly** with an average context of **15.9k tokens**.
|
|
61
47
|
|
|
62
|
-
|
|
48
|
+
| Replay strategy | First-answer accuracy | Mean context tokens |
|
|
49
|
+
| --- | ---: | ---: |
|
|
50
|
+
| No compaction | 19/39 (48.72%) | 72.6k |
|
|
51
|
+
| Codex Compact Sync | 20/39 (51.28%) | 21.7k |
|
|
52
|
+
| DSH Compact Sync | 18/39 (46.15%) | 26.1k |
|
|
53
|
+
| Pi Compact Sync | 13/39 (33.33%) | 21.6k |
|
|
54
|
+
| **AFU Compact Async** | **23/39 (58.97%)** | **15.9k** |
|
|
63
55
|
|
|
64
|
-
|
|
56
|
+
**4.56×** is the ratio of mean context tokens without compaction to those with AFU. The replay used DeepSeek-V3.2 for both answers and summaries, a 131,072-token logical window, and a 20k AFU trigger with a one-turn delay. Synchronous strategies replayed their respective prompts and policies, with at least one compaction for the context-size comparison. This metric excludes answer output and does not measure total API cost or production savings for the npm release.
|
|
65
57
|
|
|
66
|
-
|
|
58
|
+
<a id="dashboard"></a>
|
|
67
59
|
|
|
68
|
-
|
|
60
|
+
## Web dashboard
|
|
69
61
|
|
|
70
|
-
|
|
62
|
+
For Codex, check the integration status, then open the dashboard and select your session. For another host, replace `codex` with `claude`, `pi` or `dsh`.
|
|
71
63
|
|
|
72
|
-
|
|
64
|
+
```sh
|
|
65
|
+
npx --yes @liwu1995/afu-context@next status codex
|
|
66
|
+
npx --yes @liwu1995/afu-context@next ui
|
|
67
|
+
```
|
|
73
68
|
|
|
74
|
-
|
|
69
|
+
The dashboard distinguishes **request observed → summary ready → summary applied**. For Claude Code and Codex, application requires a complete successful response to a request carrying the summary. For Pi and DSH it means the host committed the summary to session history. The dashboard also shows summary duration and provider-reported cache usage.
|
|
75
70
|
|
|
76
|
-
|
|
71
|
+
Start in **Overview** for active sessions, summary duration and application counts. Open a **session** to follow a summary through its timeline to the application receipt and inspect reported cache usage. Open **Diagnostics** to investigate missing evidence, errors or runtime versions.
|
|
77
72
|
|
|
78
|
-
|
|
73
|
+
## Models and thresholds
|
|
79
74
|
|
|
80
|
-
|
|
75
|
+
All four integrations reuse the conversation’s model and authentication. The default trigger is **40%** for windows of **1,000,000 tokens or more** and **50%** for smaller windows. Pi and DSH follow the host’s selected model and window. Claude Code and Codex query provider metadata after a complete response. When the window is unknown, status reports `manual_window_required` and chat continues.
|
|
81
76
|
|
|
82
|
-
|
|
77
|
+
<details>
|
|
78
|
+
<summary>Set a window or ratio, or restore automatic settings</summary>
|
|
83
79
|
|
|
84
|
-
|
|
80
|
+
For a Codex model with a known 200,000-token window, the first command sets a 30% trigger. The second restores automatic settings.
|
|
85
81
|
|
|
86
|
-
|
|
82
|
+
```sh
|
|
83
|
+
npx --yes @liwu1995/afu-context@next install codex --context-window 200000 --ratio 0.3
|
|
84
|
+
npx --yes @liwu1995/afu-context@next install codex --context-window auto --ratio auto
|
|
85
|
+
```
|
|
87
86
|
|
|
88
|
-
|
|
87
|
+
</details>
|
|
89
88
|
|
|
90
|
-
|
|
89
|
+
For frequent use, run `npm install -g @liwu1995/afu-context@next` and use `afu` in place of the full `npx` invocation. For example, `afu ui` opens the dashboard, and `afu ui --prices ./afu-prices.json` uses your price file to estimate costs.
|
|
91
90
|
|
|
92
|
-
|
|
91
|
+
## Upgrade and remove
|
|
93
92
|
|
|
94
|
-
|
|
95
|
-
changed. Enable your systemd user service, or explicitly choose manual startup:
|
|
93
|
+
After upgrading a Claude Code or Codex integration, start a new session to use the new gateway. To uninstall, remove the host integration first, then restart the host. Use the appropriate command below.
|
|
96
94
|
|
|
97
95
|
```sh
|
|
98
|
-
afu
|
|
99
|
-
|
|
96
|
+
npx --yes @liwu1995/afu-context@next upgrade codex
|
|
97
|
+
npx --yes @liwu1995/afu-context@next remove codex
|
|
100
98
|
```
|
|
101
99
|
|
|
102
|
-
|
|
103
|
-
|
|
104
|
-
|
|
105
|
-
command with `--no-autostart`. Do not expect the gateway to restart automatically.
|
|
100
|
+
Use `remove pi` or `remove dsh` for those integrations. In Windows PowerShell, use `npx.cmd` and double-quote the package name.
|
|
101
|
+
|
|
102
|
+
The dashboard listens on loopback. Logs record usage and receipts, and Pi/DSH also persist summaries in local session state. Generating summaries uses model tokens. Savings depend on subsequent reuse and actual cache hits.
|
|
106
103
|
|
|
107
|
-
|
|
108
|
-
service file and registration receipt produce `registration_recorded`; this is
|
|
109
|
-
not a live systemd enablement check or proof that a reboot has been tested. Missing
|
|
110
|
-
or edited service files are reported as `registration_file_changed_or_missing`.
|
|
104
|
+
[Apache-2.0](LICENSE)
|
package/README.zh-CN.md
CHANGED
|
@@ -1,79 +1,104 @@
|
|
|
1
|
-
# AFU Context
|
|
1
|
+
# AFU Context · 本地异步上下文压缩
|
|
2
2
|
|
|
3
|
-
[English](README.md) · **简体中文**
|
|
3
|
+
[English](README.md) · **简体中文** · [npm](https://www.npmjs.com/package/@liwu1995/afu-context) · [一键安装](#install) · [评测效果](#results) · [Web 面板](#dashboard)
|
|
4
4
|
|
|
5
|
-
|
|
5
|
+
长对话里，同一段工具输出可能随每次请求反复发送。AFU 用宿主当前的模型在后台整理已经结束的对话。摘要就绪后，下一次合适的请求可以用摘要替换旧历史，同时保留之后发生的工具调用和新消息。
|
|
6
6
|
|
|
7
|
-
##
|
|
7
|
+
## AFU 如何压缩对话
|
|
8
8
|
|
|
9
|
-
|
|
9
|
+
| 时机 | AFU 的行为 |
|
|
10
|
+
| --- | --- |
|
|
11
|
+
| 达到压缩阈值 | 后台生成摘要，当前会话可以继续。 |
|
|
12
|
+
| 下一次使用历史 | 检查摘要覆盖的对话前缀是否仍然匹配，匹配才替换。 |
|
|
13
|
+
| 使用之后 | 记录模型请求成功回执或宿主提交结果，面板区分已生成与已应用。 |
|
|
10
14
|
|
|
11
|
-
|
|
12
|
-
npx @liwu1995/afu-context install
|
|
13
|
-
# 非交互安装：npx @liwu1995/afu-context install claude
|
|
14
|
-
```
|
|
15
|
+
<a id="install"></a>
|
|
15
16
|
|
|
16
|
-
|
|
17
|
+
## 快速开始
|
|
17
18
|
|
|
18
|
-
|
|
19
|
+
先安装并登录要使用的宿主，准备 npm 和 Node.js **22.19+（22.x）或 24+**。选一行，在终端执行安装命令，然后用最后一列的命令启动。
|
|
20
|
+
|
|
21
|
+
| 宿主 | 系统 | 一键安装 | 安装后启动 |
|
|
19
22
|
| --- | --- | --- | --- |
|
|
20
|
-
| Claude Code |
|
|
21
|
-
|
|
|
22
|
-
|
|
|
23
|
-
|
|
|
23
|
+
| <a id="afu-claude"></a>Claude Code | macOS / Linux | `npx --yes @liwu1995/afu-context@next install claude` | `claude` |
|
|
24
|
+
| Claude Code | Windows 10/11¹ | `npx.cmd --yes "@liwu1995/afu-context@next" install claude` | `claude` |
|
|
25
|
+
| <a id="afu-codex"></a>Codex CLI | macOS / Linux | `npx --yes @liwu1995/afu-context@next install codex` | `codex` |
|
|
26
|
+
| <a id="afu-pi"></a>Pi | macOS / Linux | `npx --yes @liwu1995/afu-context@next install pi` | `pi` |
|
|
27
|
+
| <a id="afu-dsh"></a>DSH | macOS / Linux | `npx --yes @liwu1995/afu-context@next install dsh --profile YOUR_PROFILE` | `dsh --profile YOUR_PROFILE` |
|
|
24
28
|
|
|
25
|
-
|
|
29
|
+
macOS 指 Apple Silicon（ARM64），Linux 指 x64。这两种平台均已验证，自带 Python 运行时。Claude Code 和 Codex 安装后新开会话。Pi 扩展在新开的会话自动加载。将 DSH 的 `YOUR_PROFILE` 换成要接入的 profile，例如 `headless` 或 `web`。选择 web 后也可以用 `dsh web` 启动。仅移除某个 profile 时运行 `npx --yes @liwu1995/afu-context@next remove dsh --profile YOUR_PROFILE`。
|
|
26
30
|
|
|
27
|
-
|
|
31
|
+
¹ Windows 使用 PowerShell，需本机 Python **3.11+** 和支持 PowerShell Hook 的 Claude Code。Windows 原生 Claude Code 已实现，待实机验收。Codex 未支持，Pi 和 DSH 未验收。
|
|
28
32
|
|
|
29
|
-
|
|
33
|
+
<details>
|
|
34
|
+
<summary>Linux 手动启动与 Pi / DSH 启动参数</summary>
|
|
30
35
|
|
|
31
|
-
|
|
32
|
-
|
|
33
|
-
afu
|
|
34
|
-
|
|
35
|
-
|
|
36
|
+
Linux 没有用户级 systemd 时，在 Claude Code 或 Codex 安装命令末尾加 `--no-autostart`，系统重启后重新执行同一命令启动网关。
|
|
37
|
+
|
|
38
|
+
Pi **0.84.4** 可直接传 `--provider` 和 `--model`。DSH 已适配 **0.1.2-rc.1** 和 Cordis **4.0.2**。若 Web profile 提示 HMR API 不兼容，可使用 `npx --yes @liwu1995/afu-context@next dsh --web-compat -- --profile web`。
|
|
39
|
+
|
|
40
|
+
</details>
|
|
41
|
+
|
|
42
|
+
<a id="results"></a>
|
|
36
43
|
|
|
37
|
-
|
|
44
|
+
## 评测效果
|
|
38
45
|
|
|
39
|
-
|
|
46
|
+
在 LongBench V2 的 39 道历史问答回放中（20 道 Agent 历史、19 道对话历史），AFU 异步压缩答对 **23 道**，平均上下文为 **15.9k tokens**。
|
|
40
47
|
|
|
41
|
-
|
|
48
|
+
| 回放策略 | 首次回答正确率 | 平均上下文 tokens |
|
|
49
|
+
| --- | ---: | ---: |
|
|
50
|
+
| 无压缩 | 19/39 (48.72%) | 72.6k |
|
|
51
|
+
| Codex Compact Sync | 20/39 (51.28%) | 21.7k |
|
|
52
|
+
| DSH Compact Sync | 18/39 (46.15%) | 26.1k |
|
|
53
|
+
| Pi Compact Sync | 13/39 (33.33%) | 21.6k |
|
|
54
|
+
| **AFU Compact Async** | **23/39 (58.97%)** | **15.9k** |
|
|
55
|
+
|
|
56
|
+
**4.56×** 是无压缩与 AFU 的平均上下文 token 之比。回放使用 DeepSeek-V3.2 回答和压缩，逻辑窗口为 131,072 tokens，AFU 在 20k 触发，摘要延迟一轮使用。同步策略按各自提示词和策略回放，比较上下文规模时至少压缩一次。此指标不含回答输出，也不能直接换算成 API 总费用或当前 npm 版本的线上收益。
|
|
57
|
+
|
|
58
|
+
<a id="dashboard"></a>
|
|
59
|
+
|
|
60
|
+
## Web 面板：查看压缩是否生效
|
|
61
|
+
|
|
62
|
+
以 Codex 为例，先看接入状态，再打开面板查看刚才的会话。其它宿主将 `codex` 换成 `claude`、`pi` 或 `dsh`。
|
|
42
63
|
|
|
43
64
|
```sh
|
|
44
|
-
|
|
45
|
-
|
|
65
|
+
npx --yes @liwu1995/afu-context@next status codex
|
|
66
|
+
npx --yes @liwu1995/afu-context@next ui
|
|
46
67
|
```
|
|
47
68
|
|
|
48
|
-
|
|
69
|
+
面板分别记录**请求已接入 → 摘要已生成 → 摘要已应用**。Claude Code 和 Codex 的应用回执表示替换历史后的请求已收到完整成功响应。Pi 和 DSH 的回执表示宿主已把摘要写入会话历史。面板也展示摘要耗时和模型返回的缓存用量。
|
|
49
70
|
|
|
50
|
-
|
|
71
|
+
在**总览**查看活跃会话、摘要生成耗时和应用数量。点进一个**会话**，沿时间线找到摘要及后续的应用回执，再看对应请求的缓存用量。遇到失败或证据缺失时，打开**高级诊断**查看错误原因与运行版本。
|
|
51
72
|
|
|
52
|
-
##
|
|
73
|
+
## 模型与压缩阈值
|
|
53
74
|
|
|
54
|
-
|
|
55
|
-
- CC/Codex 在主请求结束后异步查询当前模型的供应商元数据；默认窗口 ≥1M 时 40%，其它窗口 50%。未知窗口暂停新压缩，聊天继续。
|
|
56
|
-
- `afu install claude --context-window 200000 --ratio 0.2` 可显式覆盖；`--context-window auto --ratio auto` 恢复自动配置。省略选项保留已有设置，非法值不修改配置。
|
|
57
|
-
- Pi/DSH 跟随宿主当前模型、认证和声明窗口。窗口缺失时保留原生行为。支持 `afu install pi --context-window 200000 --ratio 0.2`，也可换为 dsh；优先级为启动选项、环境变量、保存的配置、自动默认值。阈值使用可压缩工作集的 token 估算，不等同状态栏总用量。
|
|
58
|
-
- `--config-dir`、`--state`、`--no-autostart` 用于独立沙箱或手动管理。不要将 Hook 测试当作系统服务恢复验收。
|
|
75
|
+
四个平台都复用主对话的模型和认证。模型窗口为 **1,000,000 tokens 或更大** 时，默认在 **40%** 触发，更小的窗口按 **50%** 触发。Pi 和 DSH 跟随宿主选择的模型和窗口。Claude Code 和 Codex 在完整响应后查询提供方的窗口信息，查不到时显示 `manual_window_required`，聊天可以继续。
|
|
59
76
|
|
|
60
|
-
|
|
77
|
+
<details>
|
|
78
|
+
<summary>手动指定窗口、调整比例或恢复自动设置</summary>
|
|
61
79
|
|
|
62
|
-
|
|
80
|
+
下面以已知窗口为 200,000 tokens 的 Codex 模型为例。第一条将触发比例设为 30%，第二条恢复自动设置。
|
|
63
81
|
|
|
64
|
-
|
|
82
|
+
```sh
|
|
83
|
+
npx --yes @liwu1995/afu-context@next install codex --context-window 200000 --ratio 0.3
|
|
84
|
+
npx --yes @liwu1995/afu-context@next install codex --context-window auto --ratio auto
|
|
85
|
+
```
|
|
65
86
|
|
|
66
|
-
|
|
87
|
+
</details>
|
|
67
88
|
|
|
68
|
-
|
|
89
|
+
如果经常使用 AFU，可运行 `npm install -g @liwu1995/afu-context@next`，以后用 `afu` 代替完整的 `npx` 命令。例如，`afu ui` 打开面板，`afu ui --prices ./afu-prices.json` 使用自己的单价文件估算费用。
|
|
69
90
|
|
|
70
|
-
|
|
91
|
+
## 升级与移除
|
|
92
|
+
|
|
93
|
+
升级 Claude Code 或 Codex 接入后，新开会话即可使用新版网关。需要移除时，先撤销对应宿主的接入，再重启宿主。以下两条命令分别用于升级和移除。
|
|
71
94
|
|
|
72
95
|
```sh
|
|
73
|
-
afu
|
|
74
|
-
|
|
96
|
+
npx --yes @liwu1995/afu-context@next upgrade codex
|
|
97
|
+
npx --yes @liwu1995/afu-context@next remove codex
|
|
75
98
|
```
|
|
76
99
|
|
|
77
|
-
|
|
100
|
+
Pi 和 DSH 可分别用 `remove pi` 或 `remove dsh` 撤销。Windows PowerShell 请使用 `npx.cmd` 并将包名放在双引号中。
|
|
101
|
+
|
|
102
|
+
本地面板只监听回环地址。日志记录用量和回执，Pi/DSH 的本地会话状态还会保存摘要。生成摘要本身会调用模型，节省多少取决于后续复用和实际缓存命中。
|
|
78
103
|
|
|
79
|
-
|
|
104
|
+
[Apache-2.0](LICENSE)
|
package/bin/afu.mjs
CHANGED
|
@@ -19,7 +19,7 @@ if(['install','upgrade','status','remove','gc','native'].includes(args[0])){
|
|
|
19
19
|
let release;
|
|
20
20
|
try{
|
|
21
21
|
if(!args.length || (args.includes('--help') && (args.indexOf('--')<0 || args.indexOf('--help')<args.indexOf('--')))){
|
|
22
|
-
console.log('AFU Context: install [claude|codex|pi|dsh], status, remove HOST, gc --dry-run, ui.\nRun npx @liwu1995/afu-context install to detect your installed host. Host login remains managed by the host.');
|
|
22
|
+
console.log('AFU Context: install [claude|codex|pi|dsh], status, remove HOST, gc --dry-run, ui.\nRun npx @liwu1995/afu-context@next install to detect your installed host. Host login remains managed by the host.');
|
|
23
23
|
process.exit(0);
|
|
24
24
|
}
|
|
25
25
|
args=await guide(args,env);
|
|
@@ -40,14 +40,24 @@ try{
|
|
|
40
40
|
child.on('error',()=>{console.error('AFU: HOST_EXECUTABLE_UNAVAILABLE');process.exitCode=1;});
|
|
41
41
|
child.on('exit',(code,signal)=>{process.exitCode=code??(signal?130:1);});
|
|
42
42
|
}else{
|
|
43
|
+
const hostAction = ['pi','dsh'].includes(args[1]) && ['install','upgrade','remove','status'].includes(args[0]);
|
|
44
|
+
const aggregateStatus = args[0]==='status' && args.slice(1).every(arg=>arg==='--json');
|
|
45
|
+
const directHost = process.platform!=='win32' && (hostAction || aggregateStatus);
|
|
46
|
+
if (directHost && ['install','upgrade'].includes(args[0])) {
|
|
47
|
+
// Autoload paths must outlive npx eviction and the ordinary launcher cache GC.
|
|
48
|
+
const state = env.AFU_STATE_DIR || join(homedir(), '.local/share/afu-context');
|
|
49
|
+
env.AFU_RUNTIME_DIR = join(state, 'autoload', 'runtimes');
|
|
50
|
+
}
|
|
51
|
+
if (args[0] === 'pi') env.AFU_EXPLICIT_EXTENSION = 'pi';
|
|
43
52
|
release=await holdCache(manifest.build,env);
|
|
44
53
|
const location=await prepareRuntime(root,manifest,env);
|
|
45
54
|
const python=join(location,process.platform==='win32'?'Scripts/python.exe':'bin/python');
|
|
46
55
|
env.AFU_INSTALL_ROOT=await preparePlatform(root,manifest,args,env);
|
|
47
|
-
const
|
|
56
|
+
const entry = directHost ? [join(root,'bin/host-autoload.py')] : ['-m','afu_context.local_cli'];
|
|
57
|
+
const child=spawn(python,[...entry,...args],{stdio:'inherit',env});
|
|
48
58
|
// Relay signals only to our own foreground child; never enumerate other sessions.
|
|
49
59
|
for(const signal of ['SIGINT','SIGTERM'])process.on(signal,()=>child.kill(signal));
|
|
50
60
|
child.on('error',()=>{release?.();console.error('AFU failed to start');process.exitCode=1;});
|
|
51
|
-
child.on('exit',(code,signal)=>{release?.();if(code===0&&['install','upgrade'].includes(args[0]))console.error(`AFU: installed ${args[1]}; ${['claude','codex'].includes(args[1])?'service checked; open a new '+args[1]+' session':'launch with afu '+args[1]}. Waiting for the first observed request; compression is not yet verified.`);process.exitCode=code ?? (signal?130:1);});
|
|
61
|
+
child.on('exit',(code,signal)=>{release?.();if(code===0&&['install','upgrade'].includes(args[0]))console.error(`AFU: installed ${args[1]}; ${['claude','codex'].includes(args[1])?'service checked; open a new '+args[1]+' session':directHost?'open a new host session':'launch with afu '+args[1]}. Waiting for the first observed request; compression is not yet verified.`);process.exitCode=code ?? (signal?130:1);});
|
|
52
62
|
}
|
|
53
63
|
}catch(error){release?.();console.error('AFU: '+error.message);process.exitCode=1;}
|
|
@@ -0,0 +1,328 @@
|
|
|
1
|
+
"""Install npm-hosted AFU extensions in Pi and DSH's normal discovery paths."""
|
|
2
|
+
|
|
3
|
+
import argparse
|
|
4
|
+
import hashlib
|
|
5
|
+
import json
|
|
6
|
+
import os
|
|
7
|
+
import re
|
|
8
|
+
import shutil
|
|
9
|
+
import subprocess
|
|
10
|
+
import sys
|
|
11
|
+
from pathlib import Path
|
|
12
|
+
|
|
13
|
+
from afu_context import native_install as fs
|
|
14
|
+
from afu_context.diagnostics import state_root
|
|
15
|
+
from afu_context.management import report, validate_policy
|
|
16
|
+
|
|
17
|
+
|
|
18
|
+
def run(*args, stdin=None):
|
|
19
|
+
result = subprocess.run(
|
|
20
|
+
args, input=stdin, capture_output=True, text=True, timeout=30
|
|
21
|
+
)
|
|
22
|
+
if result.returncode:
|
|
23
|
+
raise ValueError("Host configuration could not be loaded")
|
|
24
|
+
return result.stdout
|
|
25
|
+
|
|
26
|
+
|
|
27
|
+
def write_owned(file, content, marker):
|
|
28
|
+
if file.is_symlink():
|
|
29
|
+
raise ValueError("Refusing to replace an extension symlink")
|
|
30
|
+
if file.exists() and file.read_text() == content:
|
|
31
|
+
return
|
|
32
|
+
if file.exists() and not file.read_text().startswith(marker):
|
|
33
|
+
raise ValueError("Existing extension belongs to another installation")
|
|
34
|
+
fs.atomic(file, content)
|
|
35
|
+
|
|
36
|
+
|
|
37
|
+
def install_pi(root, adapter, settings, previous):
|
|
38
|
+
agent = (
|
|
39
|
+
Path(os.environ.get("PI_CODING_AGENT_DIR", "~/.pi/agent"))
|
|
40
|
+
.expanduser()
|
|
41
|
+
.resolve()
|
|
42
|
+
)
|
|
43
|
+
entry = agent / "extensions" / "afu-context.ts"
|
|
44
|
+
if previous and str(entry) != previous["entry"]:
|
|
45
|
+
raise ValueError(
|
|
46
|
+
"Remove the previous Pi agent registration before switching directories"
|
|
47
|
+
)
|
|
48
|
+
if (
|
|
49
|
+
entry.exists()
|
|
50
|
+
and previous
|
|
51
|
+
and hashlib.sha256(entry.read_bytes()).hexdigest() != previous["digest"]
|
|
52
|
+
):
|
|
53
|
+
raise ValueError("Pi extension changed since installation; user file preserved")
|
|
54
|
+
marker = "// Managed by AFU " + root.as_uri() + "\n"
|
|
55
|
+
# The explicit launcher supplies its own extension; avoid installing hooks twice.
|
|
56
|
+
body = marker + "import register from " + json.dumps(str(adapter)) + ";\n"
|
|
57
|
+
body += "export default function(pi) {\n if (process.env.AFU_EXPLICIT_EXTENSION === 'pi' || (process.env.AFU_CONTEXT_COMPACT_WORKER_COMMAND && process.argv.some(a => a.endsWith('/afu_host_integrations/pi/dist/index.js')))) return;\n"
|
|
58
|
+
body += (
|
|
59
|
+
" Object.assign(process.env, "
|
|
60
|
+
+ json.dumps(settings)
|
|
61
|
+
+ ");\n register(pi);\n}\n"
|
|
62
|
+
)
|
|
63
|
+
write_owned(entry, body, marker)
|
|
64
|
+
return {"entry": str(entry), "digest": hashlib.sha256(body.encode()).hexdigest()}
|
|
65
|
+
|
|
66
|
+
|
|
67
|
+
def profile_manifest(binary, platform, profile):
|
|
68
|
+
args = [binary, "--profile", profile, "--dump-config"]
|
|
69
|
+
tree = run(*args)
|
|
70
|
+
info = json.loads(
|
|
71
|
+
run("node", platform / "scripts/dsh-web-probe.mjs", binary, profile)
|
|
72
|
+
)
|
|
73
|
+
return Path(info["profileDir"]) / "package.json", tree
|
|
74
|
+
|
|
75
|
+
|
|
76
|
+
def install_dsh(root, platform, adapter, settings, profile, previous):
|
|
77
|
+
binary = shutil.which("dsh")
|
|
78
|
+
manifest, tree = profile_manifest(binary, platform, profile)
|
|
79
|
+
if previous and previous["manifest"] != str(manifest):
|
|
80
|
+
raise ValueError(
|
|
81
|
+
"DSH profile directory changed; existing registration preserved"
|
|
82
|
+
)
|
|
83
|
+
profile_dir = manifest.parent
|
|
84
|
+
key = hashlib.sha256(str(profile_dir).encode()).hexdigest()[:20]
|
|
85
|
+
bundle_name = "afu-context-profile-" + key
|
|
86
|
+
bundle = root / "autoload" / key
|
|
87
|
+
entry = bundle / "provider.mjs"
|
|
88
|
+
link = profile_dir / "node_modules" / bundle_name
|
|
89
|
+
marker = "// Managed by AFU " + root.as_uri() + "\n"
|
|
90
|
+
original = manifest.read_text()
|
|
91
|
+
config = json.loads(original)
|
|
92
|
+
bundles = config["dsh"]["profile"]["bundles"]
|
|
93
|
+
deps = config.setdefault("dependencies", {})
|
|
94
|
+
owned_dep = "file:" + str(bundle)
|
|
95
|
+
if link.parent.is_symlink():
|
|
96
|
+
raise ValueError("Cannot register a bundle in a shared node_modules symlink")
|
|
97
|
+
if (link.is_symlink() and link.resolve() != bundle) or (
|
|
98
|
+
link.exists() and not link.is_symlink()
|
|
99
|
+
):
|
|
100
|
+
raise ValueError("Existing DSH bundle directory conflicts with AFU")
|
|
101
|
+
if bundle_name in deps and deps[bundle_name] != owned_dep:
|
|
102
|
+
raise ValueError("DSH bundle dependency has been edited")
|
|
103
|
+
if bundle_name in bundles and deps.get(bundle_name) != owned_dep:
|
|
104
|
+
raise ValueError("DSH bundle is not owned by this installation")
|
|
105
|
+
bundle.mkdir(parents=True, exist_ok=True, mode=0o700)
|
|
106
|
+
# Bundle membership is read on host startup. Never edit watched cordis.patch.yml.
|
|
107
|
+
if bundle_name not in bundles:
|
|
108
|
+
patches = run(
|
|
109
|
+
"node",
|
|
110
|
+
platform / "scripts/dsh-overlay.mjs",
|
|
111
|
+
entry,
|
|
112
|
+
bundle,
|
|
113
|
+
binary,
|
|
114
|
+
stdin=tree,
|
|
115
|
+
)
|
|
116
|
+
preset = bundle / "presets.mjs"
|
|
117
|
+
if preset.exists():
|
|
118
|
+
old_arg = ", " + json.dumps(str(bundle)) + ");"
|
|
119
|
+
source = preset.read_text()
|
|
120
|
+
if old_arg not in source:
|
|
121
|
+
raise ValueError("Unsupported preset loader format")
|
|
122
|
+
cleanup = (
|
|
123
|
+
"import {mkdtempSync,rmSync} from 'node:fs';\n"
|
|
124
|
+
"import {tmpdir} from 'node:os';\nimport {join} from 'node:path';\n"
|
|
125
|
+
"const scratch=mkdtempSync(join(tmpdir(),'afu-dsh-presets-'));\n"
|
|
126
|
+
"process.once('exit',()=>rmSync(scratch,{recursive:true,force:true}));\n"
|
|
127
|
+
)
|
|
128
|
+
fs.atomic(preset, cleanup + source.replace(old_arg, ", scratch);"))
|
|
129
|
+
fs.atomic(bundle / "cordis.patch.yml", patches)
|
|
130
|
+
fs.write(
|
|
131
|
+
bundle / "package.json",
|
|
132
|
+
{
|
|
133
|
+
"name": bundle_name,
|
|
134
|
+
"version": "1.0.0",
|
|
135
|
+
"dsh": {"bundle": {"patch": "./cordis.patch.yml"}},
|
|
136
|
+
},
|
|
137
|
+
)
|
|
138
|
+
content = marker + "Object.assign(process.env," + json.dumps(settings) + ");\n"
|
|
139
|
+
content += (
|
|
140
|
+
"export default (await import(" + json.dumps(adapter.as_uri()) + ")).default;\n"
|
|
141
|
+
)
|
|
142
|
+
write_owned(entry, content, marker)
|
|
143
|
+
if not link.exists():
|
|
144
|
+
link.parent.mkdir(parents=True, exist_ok=True)
|
|
145
|
+
link.symlink_to(bundle, target_is_directory=True)
|
|
146
|
+
if bundle_name not in bundles:
|
|
147
|
+
bundles.append(bundle_name)
|
|
148
|
+
deps[bundle_name] = owned_dep
|
|
149
|
+
if manifest.read_text() != original:
|
|
150
|
+
raise ValueError("Concurrent DSH configuration change")
|
|
151
|
+
fs.write(manifest, config)
|
|
152
|
+
try:
|
|
153
|
+
composed = run(binary, "--profile", profile, "--dump-config")
|
|
154
|
+
if (
|
|
155
|
+
str(entry) not in composed
|
|
156
|
+
and str(bundle / "presets.mjs") not in composed
|
|
157
|
+
):
|
|
158
|
+
raise ValueError("DSH did not resolve the AFU provider")
|
|
159
|
+
except Exception:
|
|
160
|
+
if fs.read(manifest) == config:
|
|
161
|
+
fs.atomic(manifest, original)
|
|
162
|
+
raise
|
|
163
|
+
return {"manifest": str(manifest), "bundle": str(bundle), "name": bundle_name}
|
|
164
|
+
|
|
165
|
+
|
|
166
|
+
def remove_record(host, record):
|
|
167
|
+
if host == "pi":
|
|
168
|
+
entry = Path(record["entry"])
|
|
169
|
+
if entry.exists():
|
|
170
|
+
if (
|
|
171
|
+
entry.is_symlink()
|
|
172
|
+
or hashlib.sha256(entry.read_bytes()).hexdigest() != record["digest"]
|
|
173
|
+
):
|
|
174
|
+
raise ValueError(
|
|
175
|
+
"Pi extension changed since installation; user file preserved"
|
|
176
|
+
)
|
|
177
|
+
entry.unlink()
|
|
178
|
+
return
|
|
179
|
+
manifest, name = Path(record["manifest"]), record["name"]
|
|
180
|
+
config = fs.read(manifest)
|
|
181
|
+
if not config:
|
|
182
|
+
return
|
|
183
|
+
bundles = config["dsh"]["profile"]["bundles"]
|
|
184
|
+
deps = config.get("dependencies", {})
|
|
185
|
+
if name in deps and deps[name] != "file:" + record["bundle"]:
|
|
186
|
+
raise ValueError(
|
|
187
|
+
"DSH dependency changed since installation; user file preserved"
|
|
188
|
+
)
|
|
189
|
+
if name in bundles and name not in deps:
|
|
190
|
+
raise ValueError("DSH bundle ownership changed; user configuration preserved")
|
|
191
|
+
if name in bundles:
|
|
192
|
+
bundles.remove(name)
|
|
193
|
+
deps.pop(name, None)
|
|
194
|
+
fs.write(manifest, config)
|
|
195
|
+
# Keep bundle files for running sessions (some presets resolve lazily).
|
|
196
|
+
link = manifest.parent / "node_modules" / name
|
|
197
|
+
if link.is_symlink() and link.resolve() == Path(record["bundle"]):
|
|
198
|
+
link.unlink()
|
|
199
|
+
|
|
200
|
+
|
|
201
|
+
def main():
|
|
202
|
+
p = argparse.ArgumentParser()
|
|
203
|
+
p.add_argument("action", choices=["install", "upgrade", "remove", "status"])
|
|
204
|
+
p.add_argument("host", choices=["pi", "dsh"], nargs="?")
|
|
205
|
+
p.add_argument("--profile")
|
|
206
|
+
p.add_argument("--state", type=Path)
|
|
207
|
+
p.add_argument("--ratio")
|
|
208
|
+
p.add_argument("--context-window")
|
|
209
|
+
p.add_argument("--json", action="store_true")
|
|
210
|
+
args = p.parse_args()
|
|
211
|
+
root = (args.state or state_root()).expanduser().resolve()
|
|
212
|
+
host = args.host
|
|
213
|
+
if args.profile and (
|
|
214
|
+
host != "dsh"
|
|
215
|
+
or not re.fullmatch(r"[\w-]+", args.profile)
|
|
216
|
+
or args.profile == "node_modules"
|
|
217
|
+
):
|
|
218
|
+
p.error("invalid DSH profile")
|
|
219
|
+
key = args.profile or ("web" if host == "dsh" else "user")
|
|
220
|
+
if args.action == "status":
|
|
221
|
+
data = report(host, root=root)
|
|
222
|
+
for install in data["installations"]:
|
|
223
|
+
name = install["host"]
|
|
224
|
+
receipt = fs.read(root / "integrations" / (name + ".json"))
|
|
225
|
+
if receipt.get("autoload"):
|
|
226
|
+
install.update(
|
|
227
|
+
mode="host",
|
|
228
|
+
launch=name if name == "pi" else "dsh --profile PROFILE",
|
|
229
|
+
registrations=sorted(receipt["autoload"]),
|
|
230
|
+
)
|
|
231
|
+
print(json.dumps(data))
|
|
232
|
+
return
|
|
233
|
+
path = root / "integrations" / (host + ".json")
|
|
234
|
+
with fs.lock(path.parent):
|
|
235
|
+
old = fs.read(path)
|
|
236
|
+
registrations = dict(old.get("autoload", {}))
|
|
237
|
+
previous = registrations.get(key, {})
|
|
238
|
+
if args.action == "remove":
|
|
239
|
+
for item in [key] if args.profile or host == "pi" else list(registrations):
|
|
240
|
+
if item in registrations:
|
|
241
|
+
remove_record(host, registrations.pop(item))
|
|
242
|
+
fs.write(
|
|
243
|
+
path,
|
|
244
|
+
{
|
|
245
|
+
**old,
|
|
246
|
+
"host": host,
|
|
247
|
+
"autoload": registrations,
|
|
248
|
+
"removed": not registrations,
|
|
249
|
+
},
|
|
250
|
+
)
|
|
251
|
+
else:
|
|
252
|
+
platform = Path(os.environ["AFU_INSTALL_ROOT"]).resolve()
|
|
253
|
+
adapter = platform / "src/afu_host_integrations" / host / "dist/index.js"
|
|
254
|
+
if not adapter.is_file():
|
|
255
|
+
raise ValueError("Adapter not installed")
|
|
256
|
+
|
|
257
|
+
def policy(raw, field, cast):
|
|
258
|
+
if raw is None:
|
|
259
|
+
return previous.get(
|
|
260
|
+
field, old.get(field) if not registrations else None
|
|
261
|
+
)
|
|
262
|
+
return None if raw == "auto" else cast(raw)
|
|
263
|
+
|
|
264
|
+
ratio = policy(args.ratio, "ratio", float)
|
|
265
|
+
window = policy(args.context_window, "window", int)
|
|
266
|
+
validate_policy(window, ratio)
|
|
267
|
+
if window and ratio and int(window * ratio) < 1:
|
|
268
|
+
raise ValueError("The configured threshold must be at least one token")
|
|
269
|
+
settings = {
|
|
270
|
+
"AFU_CONTEXT_COMPACT_WORKER_COMMAND": sys.executable,
|
|
271
|
+
"AFU_CONTEXT_COMPACT_WORKER_ARGS": json.dumps(
|
|
272
|
+
[
|
|
273
|
+
"-m",
|
|
274
|
+
"afu_context",
|
|
275
|
+
"compact",
|
|
276
|
+
"worker",
|
|
277
|
+
"--factory",
|
|
278
|
+
"afu_context.local_worker:create_plugin",
|
|
279
|
+
]
|
|
280
|
+
),
|
|
281
|
+
"AFU_CONTEXT_COMPACT_WORKER_CWD": str(platform),
|
|
282
|
+
"AFU_INSTALL_ROOT": str(platform),
|
|
283
|
+
"AFU_STATE_DIR": str(root),
|
|
284
|
+
"AFU_HOST_CONTEXT_WINDOW_OVERRIDE": str(window or "auto"),
|
|
285
|
+
}
|
|
286
|
+
if ratio is not None:
|
|
287
|
+
settings["AFU_TRIGGER_RATIO"] = str(ratio)
|
|
288
|
+
if host == "pi":
|
|
289
|
+
record = install_pi(root, adapter, settings, previous)
|
|
290
|
+
else:
|
|
291
|
+
record = install_dsh(root, platform, adapter, settings, key, previous)
|
|
292
|
+
registrations[key] = {**record, "window": window, "ratio": ratio}
|
|
293
|
+
fs.write(
|
|
294
|
+
path,
|
|
295
|
+
{
|
|
296
|
+
**old,
|
|
297
|
+
"host": host,
|
|
298
|
+
"mode": "host",
|
|
299
|
+
"removed": False,
|
|
300
|
+
"window": window,
|
|
301
|
+
"ratio": ratio,
|
|
302
|
+
"build": fs.read(platform / "release.json")["build"],
|
|
303
|
+
"autoload": registrations,
|
|
304
|
+
},
|
|
305
|
+
)
|
|
306
|
+
print(
|
|
307
|
+
json.dumps(
|
|
308
|
+
{
|
|
309
|
+
"host": host,
|
|
310
|
+
"installed": bool(registrations),
|
|
311
|
+
"upgraded": args.action == "upgrade",
|
|
312
|
+
"mode": "host",
|
|
313
|
+
"registrations": sorted(registrations),
|
|
314
|
+
"launch": "pi" if host == "pi" else "dsh --profile " + key,
|
|
315
|
+
}
|
|
316
|
+
)
|
|
317
|
+
)
|
|
318
|
+
|
|
319
|
+
|
|
320
|
+
if __name__ == "__main__":
|
|
321
|
+
try:
|
|
322
|
+
main()
|
|
323
|
+
except (OSError, ValueError, KeyError, subprocess.TimeoutExpired) as error:
|
|
324
|
+
sys.exit(
|
|
325
|
+
"AFU registration failed: " + str(error)
|
|
326
|
+
if type(error) is ValueError
|
|
327
|
+
else "AFU registration failed: " + type(error).__name__
|
|
328
|
+
)
|