mirror of
https://github.com/NousResearch/hermes-agent.git
synced 2026-04-29 01:31:41 +00:00
feat: small refactors
This commit is contained in:
parent
e2b3b1c5e4
commit
afd670a36f
12 changed files with 2780 additions and 68 deletions
|
|
@ -3,6 +3,9 @@ import { EventEmitter } from 'node:events'
|
|||
import { resolve } from 'node:path'
|
||||
import { createInterface } from 'node:readline'
|
||||
|
||||
const MAX_GATEWAY_LOG_LINES = 200
|
||||
const MAX_LOG_PREVIEW = 240
|
||||
|
||||
export interface GatewayEvent {
|
||||
type: string
|
||||
session_id?: string
|
||||
|
|
@ -17,6 +20,7 @@ interface Pending {
|
|||
export class GatewayClient extends EventEmitter {
|
||||
private proc: ChildProcess | null = null
|
||||
private reqId = 0
|
||||
private logs: string[] = []
|
||||
private pending = new Map<string, Pending>()
|
||||
|
||||
start() {
|
||||
|
|
@ -24,18 +28,40 @@ export class GatewayClient extends EventEmitter {
|
|||
|
||||
this.proc = spawn(process.env.HERMES_PYTHON ?? resolve(root, 'venv/bin/python'), ['-m', 'tui_gateway.entry'], {
|
||||
cwd: root,
|
||||
stdio: ['pipe', 'pipe', 'inherit']
|
||||
stdio: ['pipe', 'pipe', 'pipe']
|
||||
})
|
||||
|
||||
createInterface({ input: this.proc.stdout! }).on('line', raw => {
|
||||
try {
|
||||
this.dispatch(JSON.parse(raw))
|
||||
} catch {
|
||||
/* malformed line */
|
||||
const preview = raw.trim().slice(0, MAX_LOG_PREVIEW) || '(empty line)'
|
||||
this.pushLog(`[protocol] malformed stdout: ${preview}`)
|
||||
this.emit('event', { type: 'gateway.protocol_error', payload: { preview } } satisfies GatewayEvent)
|
||||
}
|
||||
})
|
||||
|
||||
this.proc.on('exit', code => this.emit('exit', code))
|
||||
createInterface({ input: this.proc.stderr! }).on('line', raw => {
|
||||
const line = raw.trim()
|
||||
|
||||
if (!line) {
|
||||
return
|
||||
}
|
||||
|
||||
this.pushLog(line)
|
||||
this.emit('event', { type: 'gateway.stderr', payload: { line } } satisfies GatewayEvent)
|
||||
})
|
||||
|
||||
this.proc.on('error', err => {
|
||||
this.pushLog(`[spawn] ${err.message}`)
|
||||
this.rejectPending(new Error(`gateway error: ${err.message}`))
|
||||
this.emit('event', { type: 'gateway.stderr', payload: { line: `[spawn] ${err.message}` } } satisfies GatewayEvent)
|
||||
})
|
||||
|
||||
this.proc.on('exit', code => {
|
||||
this.rejectPending(new Error(`gateway exited${code === null ? '' : ` (${code})`}`))
|
||||
this.emit('exit', code)
|
||||
})
|
||||
}
|
||||
|
||||
private dispatch(msg: Record<string, unknown>) {
|
||||
|
|
@ -54,19 +80,57 @@ export class GatewayClient extends EventEmitter {
|
|||
}
|
||||
}
|
||||
|
||||
private pushLog(line: string) {
|
||||
this.logs.push(line)
|
||||
|
||||
if (this.logs.length > MAX_GATEWAY_LOG_LINES) {
|
||||
this.logs.splice(0, this.logs.length - MAX_GATEWAY_LOG_LINES)
|
||||
}
|
||||
}
|
||||
|
||||
private rejectPending(err: Error) {
|
||||
for (const [id, pending] of this.pending) {
|
||||
this.pending.delete(id)
|
||||
pending.reject(err)
|
||||
}
|
||||
}
|
||||
|
||||
getLogTail(limit = 20): string {
|
||||
return this.logs.slice(-Math.max(1, limit)).join('\n')
|
||||
}
|
||||
|
||||
request(method: string, params: Record<string, unknown> = {}): Promise<unknown> {
|
||||
if (!this.proc?.stdin) {
|
||||
return Promise.reject(new Error('gateway not running'))
|
||||
}
|
||||
|
||||
const id = `r${++this.reqId}`
|
||||
|
||||
this.proc!.stdin!.write(JSON.stringify({ jsonrpc: '2.0', id, method, params }) + '\n')
|
||||
|
||||
return new Promise((resolve, reject) => {
|
||||
this.pending.set(id, { resolve, reject })
|
||||
|
||||
setTimeout(() => {
|
||||
const timeout = setTimeout(() => {
|
||||
if (this.pending.delete(id)) {
|
||||
reject(new Error(`timeout: ${method}`))
|
||||
}
|
||||
}, 30_000)
|
||||
|
||||
this.pending.set(id, {
|
||||
reject: e => {
|
||||
clearTimeout(timeout)
|
||||
reject(e)
|
||||
},
|
||||
resolve: v => {
|
||||
clearTimeout(timeout)
|
||||
resolve(v)
|
||||
}
|
||||
})
|
||||
|
||||
try {
|
||||
this.proc!.stdin!.write(JSON.stringify({ jsonrpc: '2.0', id, method, params }) + '\n')
|
||||
} catch (e) {
|
||||
clearTimeout(timeout)
|
||||
this.pending.delete(id)
|
||||
reject(e instanceof Error ? e : new Error(String(e)))
|
||||
}
|
||||
})
|
||||
}
|
||||
|
||||
|
|
|
|||
Loading…
Add table
Add a link
Reference in a new issue