mirror of https://github.com/openclaw/openclaw.git
76 lines
2.3 KiB
TypeScript
76 lines
2.3 KiB
TypeScript
import { SessionManager } from "@mariozechner/pi-coding-agent";
|
|
|
|
type AppendMessageArg = Parameters<SessionManager["appendMessage"]>[0];
|
|
|
|
export type GatewayInjectedAbortMeta = {
|
|
aborted: true;
|
|
origin: "rpc" | "stop-command";
|
|
runId: string;
|
|
};
|
|
|
|
export type GatewayInjectedTranscriptAppendResult = {
|
|
ok: boolean;
|
|
messageId?: string;
|
|
message?: Record<string, unknown>;
|
|
error?: string;
|
|
};
|
|
|
|
export function appendInjectedAssistantMessageToTranscript(params: {
|
|
transcriptPath: string;
|
|
message: string;
|
|
label?: string;
|
|
idempotencyKey?: string;
|
|
abortMeta?: GatewayInjectedAbortMeta;
|
|
now?: number;
|
|
}): GatewayInjectedTranscriptAppendResult {
|
|
const now = params.now ?? Date.now();
|
|
const labelPrefix = params.label ? `[${params.label}]\n\n` : "";
|
|
const usage = {
|
|
input: 0,
|
|
output: 0,
|
|
cacheRead: 0,
|
|
cacheWrite: 0,
|
|
totalTokens: 0,
|
|
cost: {
|
|
input: 0,
|
|
output: 0,
|
|
cacheRead: 0,
|
|
cacheWrite: 0,
|
|
total: 0,
|
|
},
|
|
};
|
|
const messageBody: AppendMessageArg & Record<string, unknown> = {
|
|
role: "assistant",
|
|
content: [{ type: "text", text: `${labelPrefix}${params.message}` }],
|
|
timestamp: now,
|
|
// Pi stopReason is a strict enum; this is not model output, but we still store it as a
|
|
// normal assistant message so it participates in the session parentId chain.
|
|
stopReason: "stop",
|
|
usage,
|
|
// Make these explicit so downstream tooling never treats this as model output.
|
|
api: "openai-responses",
|
|
provider: "openclaw",
|
|
model: "gateway-injected",
|
|
...(params.idempotencyKey ? { idempotencyKey: params.idempotencyKey } : {}),
|
|
...(params.abortMeta
|
|
? {
|
|
openclawAbort: {
|
|
aborted: true,
|
|
origin: params.abortMeta.origin,
|
|
runId: params.abortMeta.runId,
|
|
},
|
|
}
|
|
: {}),
|
|
};
|
|
|
|
try {
|
|
// IMPORTANT: Use SessionManager so the entry is attached to the current leaf via parentId.
|
|
// Raw jsonl appends break the parent chain and can hide compaction summaries from context.
|
|
const sessionManager = SessionManager.open(params.transcriptPath);
|
|
const messageId = sessionManager.appendMessage(messageBody);
|
|
return { ok: true, messageId, message: messageBody };
|
|
} catch (err) {
|
|
return { ok: false, error: err instanceof Error ? err.message : String(err) };
|
|
}
|
|
}
|