@zhihand/mcp 0.22.1 → 0.23.0
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 +1 -1
- package/dist/daemon/dispatcher.js +2 -2
- package/dist/daemon/heartbeat.js +48 -18
- package/dist/daemon/prompt-listener.d.ts +2 -0
- package/dist/daemon/prompt-listener.js +46 -28
- package/dist/index.d.ts +1 -1
- package/dist/index.js +1 -1
- package/package.json +1 -1
package/README.md
CHANGED
|
@@ -5,7 +5,7 @@ import os from "node:os";
|
|
|
5
5
|
import { fileURLToPath } from "node:url";
|
|
6
6
|
import { DEFAULT_MODELS } from "../core/config.js";
|
|
7
7
|
import { resolveGemini, resolveClaude, resolveCodex } from "../core/resolve-path.js";
|
|
8
|
-
const CLI_TIMEOUT =
|
|
8
|
+
const CLI_TIMEOUT = 300_000; // 300s (5min) per prompt — MCP tool chains need multiple turns
|
|
9
9
|
const SIGKILL_DELAY = 2_000; // 2s after SIGTERM
|
|
10
10
|
const MAX_OUTPUT_BYTES = 100 * 1024; // 100KB (for one-shot backends)
|
|
11
11
|
const MAX_HISTORY_TURNS = 20; // keep last N exchanges in conversation history
|
|
@@ -208,7 +208,7 @@ function pollGeminiSession(child, startTime, promptText, log, knownSessionFile,
|
|
|
208
208
|
}
|
|
209
209
|
closeChild(child);
|
|
210
210
|
settle({
|
|
211
|
-
text: "Gemini timed out after
|
|
211
|
+
text: "Gemini timed out after 5 minutes.",
|
|
212
212
|
success: false,
|
|
213
213
|
durationMs: elapsed,
|
|
214
214
|
});
|
package/dist/daemon/heartbeat.js
CHANGED
|
@@ -2,6 +2,7 @@ const HEARTBEAT_INTERVAL = 30_000; // 30s
|
|
|
2
2
|
const HEARTBEAT_RETRY_INTERVAL = 5_000; // 5s on failure
|
|
3
3
|
let heartbeatTimer;
|
|
4
4
|
let retryTimer;
|
|
5
|
+
let stopped = true;
|
|
5
6
|
let currentMeta = {};
|
|
6
7
|
/** Update the backend/model metadata that will be sent with the next heartbeat. */
|
|
7
8
|
export function setBrainMeta(meta) {
|
|
@@ -40,36 +41,65 @@ export async function sendBrainOffline(config) {
|
|
|
40
41
|
}
|
|
41
42
|
export function startHeartbeatLoop(config, log) {
|
|
42
43
|
let retrying = false;
|
|
44
|
+
stopped = false;
|
|
43
45
|
async function beat() {
|
|
46
|
+
// Skip main-timer beats while retry loop is active (avoids overlap & flapping)
|
|
47
|
+
if (retrying || stopped)
|
|
48
|
+
return;
|
|
44
49
|
const ok = await sendBrainOnline(config);
|
|
45
|
-
if (
|
|
46
|
-
|
|
47
|
-
|
|
48
|
-
|
|
49
|
-
|
|
50
|
-
const recovered = await sendBrainOnline(config);
|
|
51
|
-
if (recovered) {
|
|
52
|
-
retrying = false;
|
|
53
|
-
if (retryTimer) {
|
|
54
|
-
clearInterval(retryTimer);
|
|
55
|
-
retryTimer = undefined;
|
|
56
|
-
}
|
|
57
|
-
log("[heartbeat] Recovered.");
|
|
58
|
-
}
|
|
59
|
-
}, HEARTBEAT_RETRY_INTERVAL);
|
|
50
|
+
if (stopped)
|
|
51
|
+
return; // check after await — stopHeartbeatLoop() may have been called
|
|
52
|
+
if (ok) {
|
|
53
|
+
scheduleNextBeat();
|
|
54
|
+
return;
|
|
60
55
|
}
|
|
56
|
+
// Enter retry mode
|
|
57
|
+
retrying = true;
|
|
58
|
+
log("[heartbeat] Failed, retrying every 5s...");
|
|
59
|
+
scheduleRetry();
|
|
60
|
+
}
|
|
61
|
+
/** Recursive setTimeout for retry — waits for fetch to settle before scheduling next. */
|
|
62
|
+
function scheduleRetry() {
|
|
63
|
+
if (stopped)
|
|
64
|
+
return;
|
|
65
|
+
retryTimer = setTimeout(async () => {
|
|
66
|
+
if (!retrying || stopped)
|
|
67
|
+
return;
|
|
68
|
+
const recovered = await sendBrainOnline(config);
|
|
69
|
+
if (stopped)
|
|
70
|
+
return; // check after await
|
|
71
|
+
if (recovered) {
|
|
72
|
+
retrying = false;
|
|
73
|
+
retryTimer = undefined;
|
|
74
|
+
log("[heartbeat] Recovered.");
|
|
75
|
+
// Resume normal beat cycle
|
|
76
|
+
scheduleNextBeat();
|
|
77
|
+
return;
|
|
78
|
+
}
|
|
79
|
+
// Still failing — schedule another retry
|
|
80
|
+
if (retrying && !stopped)
|
|
81
|
+
scheduleRetry();
|
|
82
|
+
}, HEARTBEAT_RETRY_INTERVAL);
|
|
83
|
+
}
|
|
84
|
+
/** Schedule next normal heartbeat using setTimeout (not setInterval, to avoid overlap). */
|
|
85
|
+
function scheduleNextBeat() {
|
|
86
|
+
if (stopped)
|
|
87
|
+
return;
|
|
88
|
+
if (heartbeatTimer)
|
|
89
|
+
clearTimeout(heartbeatTimer);
|
|
90
|
+
heartbeatTimer = setTimeout(beat, HEARTBEAT_INTERVAL);
|
|
61
91
|
}
|
|
62
92
|
// Immediate first heartbeat
|
|
63
93
|
beat();
|
|
64
|
-
heartbeatTimer = setInterval(beat, HEARTBEAT_INTERVAL);
|
|
65
94
|
}
|
|
66
95
|
export function stopHeartbeatLoop() {
|
|
96
|
+
stopped = true;
|
|
67
97
|
if (heartbeatTimer) {
|
|
68
|
-
|
|
98
|
+
clearTimeout(heartbeatTimer);
|
|
69
99
|
heartbeatTimer = undefined;
|
|
70
100
|
}
|
|
71
101
|
if (retryTimer) {
|
|
72
|
-
|
|
102
|
+
clearTimeout(retryTimer);
|
|
73
103
|
retryTimer = undefined;
|
|
74
104
|
}
|
|
75
105
|
}
|
|
@@ -27,6 +27,8 @@ export declare class PromptListener {
|
|
|
27
27
|
private resetWatchdog;
|
|
28
28
|
private handleSSEEvent;
|
|
29
29
|
private startPolling;
|
|
30
|
+
/** Recursive setTimeout: waits for fetch to complete before scheduling next poll. */
|
|
31
|
+
private schedulePoll;
|
|
30
32
|
private stopPolling;
|
|
31
33
|
private poll;
|
|
32
34
|
}
|
|
@@ -1,4 +1,4 @@
|
|
|
1
|
-
const SSE_WATCHDOG_TIMEOUT =
|
|
1
|
+
const SSE_WATCHDOG_TIMEOUT = 120_000; // 120s no data → reconnect (servers may not send keepalive frequently)
|
|
2
2
|
const SSE_RECONNECT_DELAY = 3_000;
|
|
3
3
|
const POLL_INTERVAL = 2_000;
|
|
4
4
|
export class PromptListener {
|
|
@@ -24,7 +24,7 @@ export class PromptListener {
|
|
|
24
24
|
this.sseAbort?.abort();
|
|
25
25
|
this.sseAbort = null;
|
|
26
26
|
if (this.pollTimer) {
|
|
27
|
-
|
|
27
|
+
clearTimeout(this.pollTimer);
|
|
28
28
|
this.pollTimer = null;
|
|
29
29
|
}
|
|
30
30
|
}
|
|
@@ -63,34 +63,39 @@ export class PromptListener {
|
|
|
63
63
|
const decoder = new TextDecoder();
|
|
64
64
|
let buffer = "";
|
|
65
65
|
let watchdog = this.resetWatchdog();
|
|
66
|
-
|
|
67
|
-
|
|
68
|
-
|
|
69
|
-
|
|
70
|
-
|
|
71
|
-
|
|
72
|
-
|
|
73
|
-
|
|
74
|
-
|
|
75
|
-
|
|
76
|
-
|
|
77
|
-
|
|
78
|
-
|
|
79
|
-
|
|
80
|
-
|
|
81
|
-
else if (line === "" && eventData) {
|
|
82
|
-
try {
|
|
83
|
-
const event = JSON.parse(eventData);
|
|
84
|
-
this.handleSSEEvent(event);
|
|
66
|
+
try {
|
|
67
|
+
while (!this.stopped) {
|
|
68
|
+
const { done, value } = await reader.read();
|
|
69
|
+
if (done)
|
|
70
|
+
break;
|
|
71
|
+
// Reset watchdog on any data (including keepalive comments)
|
|
72
|
+
clearTimeout(watchdog);
|
|
73
|
+
watchdog = this.resetWatchdog();
|
|
74
|
+
buffer += decoder.decode(value, { stream: true });
|
|
75
|
+
const lines = buffer.split("\n");
|
|
76
|
+
buffer = lines.pop() ?? "";
|
|
77
|
+
let eventData = "";
|
|
78
|
+
for (const line of lines) {
|
|
79
|
+
if (line.startsWith("data: ")) {
|
|
80
|
+
eventData += (eventData ? "\n" : "") + line.slice(6);
|
|
85
81
|
}
|
|
86
|
-
|
|
87
|
-
|
|
82
|
+
else if (line === "" && eventData) {
|
|
83
|
+
try {
|
|
84
|
+
const event = JSON.parse(eventData);
|
|
85
|
+
this.handleSSEEvent(event);
|
|
86
|
+
}
|
|
87
|
+
catch {
|
|
88
|
+
// Malformed event
|
|
89
|
+
}
|
|
90
|
+
eventData = "";
|
|
88
91
|
}
|
|
89
|
-
eventData = "";
|
|
90
92
|
}
|
|
91
93
|
}
|
|
92
94
|
}
|
|
93
|
-
|
|
95
|
+
finally {
|
|
96
|
+
// Always clear watchdog — prevents leaked timer from aborting next connection
|
|
97
|
+
clearTimeout(watchdog);
|
|
98
|
+
}
|
|
94
99
|
}
|
|
95
100
|
catch (err) {
|
|
96
101
|
if (this.stopped)
|
|
@@ -104,7 +109,7 @@ export class PromptListener {
|
|
|
104
109
|
}
|
|
105
110
|
resetWatchdog() {
|
|
106
111
|
return setTimeout(() => {
|
|
107
|
-
this.log("[sse] Watchdog timeout (
|
|
112
|
+
this.log("[sse] Watchdog timeout (120s no data). Reconnecting...");
|
|
108
113
|
this.sseAbort?.abort();
|
|
109
114
|
}, SSE_WATCHDOG_TIMEOUT);
|
|
110
115
|
}
|
|
@@ -123,11 +128,24 @@ export class PromptListener {
|
|
|
123
128
|
startPolling() {
|
|
124
129
|
if (this.pollTimer)
|
|
125
130
|
return;
|
|
126
|
-
this.
|
|
131
|
+
this.schedulePoll();
|
|
132
|
+
}
|
|
133
|
+
/** Recursive setTimeout: waits for fetch to complete before scheduling next poll. */
|
|
134
|
+
schedulePoll() {
|
|
135
|
+
if (this.pollTimer)
|
|
136
|
+
return;
|
|
137
|
+
this.pollTimer = setTimeout(async () => {
|
|
138
|
+
this.pollTimer = null;
|
|
139
|
+
await this.poll();
|
|
140
|
+
// Schedule next poll only if SSE is still disconnected
|
|
141
|
+
if (!this.sseConnected && !this.stopped) {
|
|
142
|
+
this.schedulePoll();
|
|
143
|
+
}
|
|
144
|
+
}, POLL_INTERVAL);
|
|
127
145
|
}
|
|
128
146
|
stopPolling() {
|
|
129
147
|
if (this.pollTimer) {
|
|
130
|
-
|
|
148
|
+
clearTimeout(this.pollTimer);
|
|
131
149
|
this.pollTimer = null;
|
|
132
150
|
}
|
|
133
151
|
}
|
package/dist/index.d.ts
CHANGED
|
@@ -1,4 +1,4 @@
|
|
|
1
1
|
import { McpServer } from "@modelcontextprotocol/sdk/server/mcp.js";
|
|
2
|
-
export declare const PACKAGE_VERSION = "0.
|
|
2
|
+
export declare const PACKAGE_VERSION = "0.23.0";
|
|
3
3
|
export declare function createServer(deviceName?: string): McpServer;
|
|
4
4
|
export declare function startStdioServer(deviceName?: string): Promise<void>;
|
package/dist/index.js
CHANGED
|
@@ -5,7 +5,7 @@ import { controlSchema, screenshotSchema, pairSchema } from "./tools/schemas.js"
|
|
|
5
5
|
import { executeControl } from "./tools/control.js";
|
|
6
6
|
import { handleScreenshot } from "./tools/screenshot.js";
|
|
7
7
|
import { handlePair } from "./tools/pair.js";
|
|
8
|
-
export const PACKAGE_VERSION = "0.
|
|
8
|
+
export const PACKAGE_VERSION = "0.23.0";
|
|
9
9
|
export function createServer(deviceName) {
|
|
10
10
|
const server = new McpServer({
|
|
11
11
|
name: "zhihand",
|