mirror of
https://github.com/getpaseo/paseo.git
synced 2026-11-03 07:03:18 +00:00
3447 lines
104 KiB
TypeScript
3447 lines
104 KiB
TypeScript
import { homedir } from "node:os";
|
|
import {
|
|
type AssistantMessage as OpenCodeAssistantMessage,
|
|
type Event as OpenCodeEvent,
|
|
type FilePartInput as OpenCodeFilePartInput,
|
|
type GlobalSession as OpenCodeGlobalSession,
|
|
type Message as OpenCodeMessage,
|
|
type OpencodeClient,
|
|
type Part as OpenCodePart,
|
|
type Session as OpenCodeSession,
|
|
type TextPartInput as OpenCodeTextPartInput,
|
|
} from "@opencode-ai/sdk/v2/client";
|
|
import { findExecutable, isCommandAvailable } from "../../../utils/executable.js";
|
|
import { createPathEquivalenceMatcher } from "../../../utils/path.js";
|
|
import type { Logger } from "pino";
|
|
import { z } from "zod";
|
|
|
|
import {
|
|
getAgentStreamEventTurnId,
|
|
type AgentCapabilityFlags,
|
|
type AgentClient,
|
|
type AgentCreateSessionOptions,
|
|
type AgentFeature,
|
|
type AgentLaunchContext,
|
|
type AgentMode,
|
|
type AgentModelDefinition,
|
|
type AgentPermissionAction,
|
|
type AgentPermissionRequest,
|
|
type AgentPermissionResponse,
|
|
type AgentPersistenceHandle,
|
|
type AgentPromptInput,
|
|
type AgentRunOptions,
|
|
type AgentRunResult,
|
|
type AgentRuntimeInfo,
|
|
type AgentSession,
|
|
type AgentSessionConfig,
|
|
type AgentSlashCommand,
|
|
type AgentStreamEvent,
|
|
type AgentTimelineItem,
|
|
type AgentUsage,
|
|
type ListModelsOptions,
|
|
type ListModesOptions,
|
|
type ListPersistedAgentsOptions,
|
|
type McpServerConfig,
|
|
type PersistedAgentDescriptor,
|
|
type ToolCallDetail,
|
|
type ToolCallTimelineItem,
|
|
} from "../agent-sdk-types.js";
|
|
import { createProviderEnvSpec, type ProviderRuntimeSettings } from "../provider-launch-config.js";
|
|
import { withTimeout } from "../../../utils/promise-timeout.js";
|
|
import { execCommand } from "../../../utils/spawn.js";
|
|
import { buildToolCallDisplayModel } from "../../../shared/tool-call-display.js";
|
|
import { mapOpencodeToolCall } from "./opencode/tool-call-mapper.js";
|
|
import { OpenCodeServerManager } from "./opencode/server-manager.js";
|
|
import {
|
|
formatDiagnosticStatus,
|
|
formatProviderDiagnostic,
|
|
formatProviderDiagnosticError,
|
|
resolveBinaryVersion,
|
|
toDiagnosticErrorMessage,
|
|
} from "./diagnostic-utils.js";
|
|
import { runProviderTurn } from "./provider-runner.js";
|
|
import { renderPromptAttachmentAsText } from "../prompt-attachments.js";
|
|
import { composeSystemPromptParts } from "../system-prompt.js";
|
|
import {
|
|
createSdkOpenCodeClient,
|
|
type OpenCodeRuntime,
|
|
type OpenCodeServerAcquisition,
|
|
} from "./opencode/runtime.js";
|
|
import { normalizeProviderReplayTimestamp } from "../provider-history-timestamps.js";
|
|
import { revertOpenCodeConversationAndFiles } from "./opencode/rewind.js";
|
|
|
|
const OPENCODE_CAPABILITIES: AgentCapabilityFlags = {
|
|
supportsStreaming: true,
|
|
supportsSessionPersistence: true,
|
|
supportsDynamicModes: true,
|
|
supportsMcpServers: true,
|
|
supportsReasoningStream: true,
|
|
supportsToolInvocations: true,
|
|
supportsRewindConversation: false,
|
|
supportsRewindFiles: false,
|
|
supportsRewindBoth: true,
|
|
};
|
|
|
|
const OPENCODE_BUILD_MODE_ID = "build";
|
|
const OPENCODE_FULL_ACCESS_MODE_ID = "full-access";
|
|
const OPENCODE_PERSISTED_SESSION_LIMIT = 200;
|
|
const OPENCODE_PENDING_ABORT_START_TIMEOUT_MS = 10_000;
|
|
const OPENCODE_PERMISSION_ACTION_ALLOW_ONCE = "allow_once";
|
|
const OPENCODE_PERMISSION_ACTION_ALLOW_ALWAYS = "allow_always";
|
|
|
|
const DEFAULT_MODES: AgentMode[] = [
|
|
{
|
|
id: OPENCODE_BUILD_MODE_ID,
|
|
label: "Build",
|
|
description: "Allows edits and tool execution for implementation work",
|
|
},
|
|
{
|
|
id: "plan",
|
|
label: "Plan",
|
|
description: "Read-only planning mode that avoids file edits",
|
|
},
|
|
{
|
|
id: OPENCODE_FULL_ACCESS_MODE_ID,
|
|
label: "Full Access",
|
|
description: "Automatically approves all tool permission prompts for the session",
|
|
},
|
|
];
|
|
|
|
function buildOpenCodePermissionActions(): AgentPermissionAction[] {
|
|
return [
|
|
{
|
|
id: "deny",
|
|
label: "Deny",
|
|
behavior: "deny",
|
|
variant: "danger",
|
|
intent: "dismiss",
|
|
},
|
|
{
|
|
id: OPENCODE_PERMISSION_ACTION_ALLOW_ALWAYS,
|
|
label: "Allow always",
|
|
behavior: "allow",
|
|
variant: "secondary",
|
|
},
|
|
{
|
|
id: OPENCODE_PERMISSION_ACTION_ALLOW_ONCE,
|
|
label: "Allow once",
|
|
behavior: "allow",
|
|
variant: "primary",
|
|
},
|
|
];
|
|
}
|
|
|
|
function resolveOpenCodePermissionReply(
|
|
response: AgentPermissionResponse,
|
|
): "once" | "always" | "reject" {
|
|
if (response.behavior === "deny") {
|
|
return "reject";
|
|
}
|
|
|
|
if (response.selectedActionId === OPENCODE_PERMISSION_ACTION_ALLOW_ALWAYS) {
|
|
return "always";
|
|
}
|
|
|
|
return "once";
|
|
}
|
|
|
|
type OpenCodeAgentConfig = AgentSessionConfig & { provider: "opencode" };
|
|
type OpenCodeMessageRole = "user" | "assistant";
|
|
type OpenCodePersistedSession = OpenCodeSession | OpenCodeGlobalSession;
|
|
|
|
interface OpenCodeSessionMessage {
|
|
info: OpenCodeMessage;
|
|
parts: OpenCodePart[];
|
|
}
|
|
|
|
type OpenCodeMcpConfig =
|
|
| {
|
|
type: "local";
|
|
command: string[];
|
|
environment?: Record<string, string>;
|
|
enabled?: boolean;
|
|
}
|
|
| {
|
|
type: "remote";
|
|
url: string;
|
|
headers?: Record<string, string>;
|
|
enabled?: boolean;
|
|
};
|
|
|
|
const MCP_ALREADY_PRESENT_ERROR_TOKENS = ["already", "exists", "connected"] as const;
|
|
const OPENCODE_PROVIDER_LIST_TIMEOUT_MS = 30_000;
|
|
const OPENCODE_HANDLED_BUILTIN_SLASH_COMMANDS: AgentSlashCommand[] = [
|
|
{ name: "compact", description: "Compact the current session", argumentHint: "" },
|
|
{ name: "summarize", description: "Compact the current session", argumentHint: "" },
|
|
];
|
|
const OPENCODE_HEADERS_TIMEOUT_TOKENS = [
|
|
"headers timeout",
|
|
"headers timeout error",
|
|
"headers_timeout",
|
|
"und_err_headers_timeout",
|
|
] as const;
|
|
|
|
const OpencodeToolStateSchema = z
|
|
.object({
|
|
status: z.string().optional(),
|
|
input: z.unknown().optional(),
|
|
output: z.unknown().optional(),
|
|
error: z.unknown().optional(),
|
|
})
|
|
.passthrough();
|
|
|
|
const OpencodeToolPartBaseSchema = z
|
|
.object({
|
|
tool: z.string().trim().min(1),
|
|
state: OpencodeToolStateSchema.optional(),
|
|
})
|
|
.passthrough();
|
|
|
|
const OpencodeToolPartWithCallIdSchema = OpencodeToolPartBaseSchema.extend({
|
|
callID: z.string().trim().min(1),
|
|
id: z.string().optional(),
|
|
}).transform((part) => ({
|
|
toolName: part.tool,
|
|
callId: part.callID,
|
|
status: part.state?.status,
|
|
input: part.state?.input,
|
|
output: part.state?.output,
|
|
error: part.state?.error,
|
|
}));
|
|
|
|
const OpencodeToolPartWithIdSchema = OpencodeToolPartBaseSchema.extend({
|
|
id: z.string().trim().min(1),
|
|
callID: z.string().optional(),
|
|
}).transform((part) => ({
|
|
toolName: part.tool,
|
|
callId: part.id,
|
|
status: part.state?.status,
|
|
input: part.state?.input,
|
|
output: part.state?.output,
|
|
error: part.state?.error,
|
|
}));
|
|
|
|
const OpencodeToolPartWithoutIdSchema = OpencodeToolPartBaseSchema.extend({
|
|
id: z.string().optional(),
|
|
callID: z.string().optional(),
|
|
}).transform((part) => ({
|
|
toolName: part.tool,
|
|
callId: undefined,
|
|
status: part.state?.status,
|
|
input: part.state?.input,
|
|
output: part.state?.output,
|
|
error: part.state?.error,
|
|
}));
|
|
|
|
const OpencodeToolPartSchema = z.union([
|
|
OpencodeToolPartWithCallIdSchema,
|
|
OpencodeToolPartWithIdSchema,
|
|
OpencodeToolPartWithoutIdSchema,
|
|
]);
|
|
|
|
const OpencodeToolPartTimelineEnvelopeSchema = OpencodeToolPartSchema.transform((part) => ({
|
|
toolName: part.toolName,
|
|
callId: part.callId,
|
|
status: part.status,
|
|
input: part.input,
|
|
output: part.output,
|
|
error: part.error,
|
|
}));
|
|
|
|
const OpencodeToolPartToTimelineItemSchema = OpencodeToolPartTimelineEnvelopeSchema.transform(
|
|
(part) =>
|
|
mapOpencodeToolCall({
|
|
toolName: part.toolName,
|
|
callId: part.callId,
|
|
status: part.status,
|
|
input: part.input,
|
|
output: part.output,
|
|
error: part.error,
|
|
}),
|
|
);
|
|
|
|
function toOpenCodeMcpConfig(config: McpServerConfig): OpenCodeMcpConfig {
|
|
if (config.type === "stdio") {
|
|
return {
|
|
type: "local",
|
|
command: [config.command, ...(config.args ?? [])],
|
|
...(config.env ? { environment: config.env } : {}),
|
|
enabled: true,
|
|
};
|
|
}
|
|
|
|
return {
|
|
type: "remote",
|
|
url: config.url,
|
|
...(config.headers ? { headers: config.headers } : {}),
|
|
enabled: true,
|
|
};
|
|
}
|
|
|
|
type TerminalTurnEvent = Extract<
|
|
AgentStreamEvent,
|
|
{ type: "turn_completed" | "turn_failed" | "turn_canceled" }
|
|
>;
|
|
|
|
function toTerminalTurnEvent(event: AgentStreamEvent): TerminalTurnEvent | null {
|
|
if (event.type === "turn_failed") {
|
|
return {
|
|
type: "turn_failed",
|
|
provider: "opencode",
|
|
error: toDiagnosticErrorMessage(event.error),
|
|
};
|
|
}
|
|
if (event.type === "turn_completed" || event.type === "turn_canceled") {
|
|
return event;
|
|
}
|
|
return null;
|
|
}
|
|
|
|
function isOpenCodeNotFoundError(error: unknown): boolean {
|
|
return (
|
|
typeof error === "object" &&
|
|
error !== null &&
|
|
"name" in error &&
|
|
(error as { name?: unknown }).name === "NotFoundError"
|
|
);
|
|
}
|
|
|
|
async function reconcileOpenCodeSessionClose(params: {
|
|
client: Pick<OpencodeClient, "session">;
|
|
sessionId: string;
|
|
directory: string;
|
|
logger: Logger;
|
|
}): Promise<void> {
|
|
const { client, sessionId, directory, logger } = params;
|
|
|
|
try {
|
|
const response = await client.session.abort({
|
|
sessionID: sessionId,
|
|
directory,
|
|
});
|
|
if (response.error && !isOpenCodeNotFoundError(response.error)) {
|
|
logger.warn(
|
|
{
|
|
sessionId,
|
|
error: toDiagnosticErrorMessage(response.error),
|
|
},
|
|
"Failed to abort OpenCode session during close",
|
|
);
|
|
}
|
|
} catch (error) {
|
|
logger.warn(
|
|
{
|
|
sessionId,
|
|
error: toDiagnosticErrorMessage(error),
|
|
},
|
|
"Failed to abort OpenCode session during close",
|
|
);
|
|
}
|
|
|
|
try {
|
|
const response = await client.session.update({
|
|
sessionID: sessionId,
|
|
directory,
|
|
time: { archived: Date.now() },
|
|
});
|
|
if (response.error && !isOpenCodeNotFoundError(response.error)) {
|
|
logger.warn(
|
|
{
|
|
sessionId,
|
|
error: toDiagnosticErrorMessage(response.error),
|
|
},
|
|
"Failed to archive OpenCode session during close",
|
|
);
|
|
}
|
|
} catch (error) {
|
|
logger.warn(
|
|
{
|
|
sessionId,
|
|
error: toDiagnosticErrorMessage(error),
|
|
},
|
|
"Failed to archive OpenCode session during close",
|
|
);
|
|
}
|
|
}
|
|
|
|
function isOpenCodeHeadersTimeoutFailure(error: unknown): boolean {
|
|
const diagnostics = new Set<string>();
|
|
const queue: unknown[] = [error];
|
|
|
|
while (queue.length > 0) {
|
|
const current = queue.shift();
|
|
if (!current) {
|
|
continue;
|
|
}
|
|
|
|
const normalized = toDiagnosticErrorMessage(current).trim().toLowerCase();
|
|
if (normalized) {
|
|
diagnostics.add(normalized);
|
|
}
|
|
|
|
if (typeof current === "object") {
|
|
const record = current as {
|
|
message?: unknown;
|
|
code?: unknown;
|
|
name?: unknown;
|
|
cause?: unknown;
|
|
};
|
|
|
|
for (const value of [record.message, record.code, record.name]) {
|
|
if (typeof value !== "string") {
|
|
continue;
|
|
}
|
|
const diagnostic = value.trim().toLowerCase();
|
|
if (diagnostic) {
|
|
diagnostics.add(diagnostic);
|
|
}
|
|
}
|
|
|
|
if (record.cause) {
|
|
queue.push(record.cause);
|
|
}
|
|
}
|
|
}
|
|
|
|
return [...diagnostics].some((diagnostic) =>
|
|
OPENCODE_HEADERS_TIMEOUT_TOKENS.some((token) => diagnostic.includes(token)),
|
|
);
|
|
}
|
|
|
|
function isAlreadyPresentMcpError(error: unknown): boolean {
|
|
const normalized = toDiagnosticErrorMessage(error).toLowerCase();
|
|
return MCP_ALREADY_PRESENT_ERROR_TOKENS.some((token) => normalized.includes(token));
|
|
}
|
|
|
|
function resolvePartDedupeKey(
|
|
part: { id: string; messageID: string },
|
|
partType: "text" | "reasoning",
|
|
): string | null {
|
|
if (part.id.trim().length > 0) {
|
|
return `${partType}:${part.id}`;
|
|
}
|
|
if (part.messageID.trim().length > 0) {
|
|
return `${partType}:message:${part.messageID}`;
|
|
}
|
|
return null;
|
|
}
|
|
|
|
function normalizeOpenCodeModeId(modeId: string | null | undefined): string {
|
|
const trimmed = typeof modeId === "string" ? modeId.trim() : "";
|
|
if (!trimmed || trimmed === "default") {
|
|
return OPENCODE_BUILD_MODE_ID;
|
|
}
|
|
return trimmed;
|
|
}
|
|
|
|
function resolveOpenCodeRuntimeAgentId(modeId: string | null | undefined): string {
|
|
const normalizedModeId = normalizeOpenCodeModeId(modeId);
|
|
return normalizedModeId === OPENCODE_FULL_ACCESS_MODE_ID
|
|
? OPENCODE_BUILD_MODE_ID
|
|
: normalizedModeId;
|
|
}
|
|
|
|
function isSelectableOpenCodeAgent(agent: { mode?: string; hidden?: boolean }): boolean {
|
|
return (agent.mode === "primary" || agent.mode === "all") && agent.hidden !== true;
|
|
}
|
|
|
|
function mergeOpenCodeModes(discoveredModes: AgentMode[]): AgentMode[] {
|
|
const modesById = new Map(DEFAULT_MODES.map((mode) => [mode.id, mode]));
|
|
for (const mode of discoveredModes) {
|
|
modesById.set(mode.id, mode);
|
|
}
|
|
return sortOpenCodeModes(Array.from(modesById.values()));
|
|
}
|
|
|
|
function sortOpenCodeModes(modes: AgentMode[]): AgentMode[] {
|
|
const order = new Map(DEFAULT_MODES.map((mode, index) => [mode.id, index]));
|
|
return [...modes].sort((left, right) => {
|
|
const leftOrder = order.get(left.id) ?? Number.MAX_SAFE_INTEGER;
|
|
const rightOrder = order.get(right.id) ?? Number.MAX_SAFE_INTEGER;
|
|
if (leftOrder !== rightOrder) {
|
|
return leftOrder - rightOrder;
|
|
}
|
|
return left.label.localeCompare(right.label);
|
|
});
|
|
}
|
|
|
|
function readPositiveFiniteNumber(value: unknown): number | undefined {
|
|
return typeof value === "number" && Number.isFinite(value) && value > 0 ? value : undefined;
|
|
}
|
|
|
|
function maxFiniteNumber(left: number | undefined, right: number): number {
|
|
return left === undefined ? right : Math.max(left, right);
|
|
}
|
|
|
|
function assignUsageNumber(usage: AgentUsage, key: keyof AgentUsage, value: number | undefined) {
|
|
if (value !== undefined) {
|
|
usage[key] = value;
|
|
}
|
|
}
|
|
|
|
function buildOpenCodeModelLookupKey(providerId: string, modelId: string): string {
|
|
return `${providerId}/${modelId}`;
|
|
}
|
|
|
|
function parseOpenCodeModelLookupKey(modelId: string | null | undefined): string | undefined {
|
|
if (typeof modelId !== "string" || modelId.trim().length === 0) {
|
|
return undefined;
|
|
}
|
|
|
|
const slashIndex = modelId.indexOf("/");
|
|
if (slashIndex <= 0 || slashIndex === modelId.length - 1) {
|
|
return undefined;
|
|
}
|
|
|
|
const providerId = modelId.slice(0, slashIndex).trim();
|
|
const providerModelId = modelId.slice(slashIndex + 1).trim();
|
|
if (!providerId || !providerModelId) {
|
|
return undefined;
|
|
}
|
|
|
|
return buildOpenCodeModelLookupKey(providerId, providerModelId);
|
|
}
|
|
|
|
function extractOpenCodeModelContextWindow(model: unknown): number | undefined {
|
|
if (!model || typeof model !== "object") {
|
|
return undefined;
|
|
}
|
|
const limit = (model as { limit?: { context?: unknown } }).limit;
|
|
return readPositiveFiniteNumber(limit?.context);
|
|
}
|
|
|
|
function buildOpenCodeModelDefinition(
|
|
provider: {
|
|
id: string;
|
|
name: string;
|
|
},
|
|
modelId: string,
|
|
model: {
|
|
name: string;
|
|
family?: string;
|
|
release_date?: string;
|
|
attachment?: boolean;
|
|
reasoning?: boolean;
|
|
tool_call?: boolean;
|
|
cost?: unknown;
|
|
limit?: { context?: number; input?: number; output?: number };
|
|
variants?: Record<string, unknown>;
|
|
},
|
|
): AgentModelDefinition {
|
|
const rawVariants = model.variants ? Object.keys(model.variants) : [];
|
|
const thinkingOptions = rawVariants.map((id, index) => ({
|
|
id,
|
|
label: id,
|
|
isDefault: index === 0,
|
|
}));
|
|
|
|
return {
|
|
provider: "opencode",
|
|
id: `${provider.id}/${modelId}`,
|
|
label: model.name,
|
|
description: `${provider.name} - ${model.family ?? ""}`.trim(),
|
|
thinkingOptions: thinkingOptions.length > 0 ? thinkingOptions : undefined,
|
|
defaultThinkingOptionId: thinkingOptions[0]?.id,
|
|
metadata: {
|
|
providerId: provider.id,
|
|
providerName: provider.name,
|
|
modelId,
|
|
family: model.family,
|
|
releaseDate: model.release_date,
|
|
supportsAttachments: model.attachment,
|
|
supportsReasoning: model.reasoning,
|
|
supportsToolCall: model.tool_call,
|
|
cost: model.cost,
|
|
contextWindowMaxTokens: extractOpenCodeModelContextWindow(model),
|
|
...(model.limit ? { limit: model.limit } : {}),
|
|
},
|
|
};
|
|
}
|
|
|
|
function resolveOpenCodeSelectedModelContextWindow(
|
|
providers:
|
|
| {
|
|
connected?: string[];
|
|
all?: Array<{
|
|
id: string;
|
|
models?: Record<string, unknown>;
|
|
}>;
|
|
}
|
|
| null
|
|
| undefined,
|
|
modelId: string | null | undefined,
|
|
): number | undefined {
|
|
if (!providers) {
|
|
return undefined;
|
|
}
|
|
const modelLookupKey = parseOpenCodeModelLookupKey(modelId);
|
|
if (!modelLookupKey) {
|
|
return undefined;
|
|
}
|
|
const lookup = buildOpenCodeModelContextWindowLookup(providers);
|
|
return lookup.get(modelLookupKey);
|
|
}
|
|
|
|
function buildOpenCodeModelContextWindowLookup(
|
|
providers:
|
|
| {
|
|
connected?: string[];
|
|
all?: Array<{
|
|
id: string;
|
|
source?: string;
|
|
models?: Record<string, unknown>;
|
|
}>;
|
|
}
|
|
| null
|
|
| undefined,
|
|
): Map<string, number> {
|
|
const lookup = new Map<string, number>();
|
|
if (!providers) {
|
|
return lookup;
|
|
}
|
|
|
|
const connectedProviderIds = new Set(providers.connected ?? []);
|
|
for (const provider of providers.all ?? []) {
|
|
// Providers with source "api" are managed by the OpenCode console/subscription and are
|
|
// usable even though they don't appear in `connected` (which only lists env/config providers).
|
|
if (!connectedProviderIds.has(provider.id) && provider.source !== "api") {
|
|
continue;
|
|
}
|
|
for (const [modelId, modelDefinition] of Object.entries(provider.models ?? {})) {
|
|
const contextWindow = extractOpenCodeModelContextWindow(modelDefinition);
|
|
if (contextWindow === undefined) {
|
|
continue;
|
|
}
|
|
lookup.set(buildOpenCodeModelLookupKey(provider.id, modelId), contextWindow);
|
|
}
|
|
}
|
|
|
|
return lookup;
|
|
}
|
|
|
|
function resolveOpenCodeModelLookupKeyFromAssistantMessage(
|
|
info: OpenCodeAssistantMessage,
|
|
): string | undefined {
|
|
const providerId = info.providerID;
|
|
const modelId = info.modelID;
|
|
if (!providerId || !modelId) {
|
|
return undefined;
|
|
}
|
|
|
|
return buildOpenCodeModelLookupKey(providerId, modelId);
|
|
}
|
|
|
|
function mergeOpenCodeStepFinishUsage(
|
|
usage: AgentUsage,
|
|
part: {
|
|
cost?: unknown;
|
|
tokens?: {
|
|
input?: unknown;
|
|
output?: unknown;
|
|
reasoning?: unknown;
|
|
total?: unknown;
|
|
cache?: {
|
|
read?: unknown;
|
|
write?: unknown;
|
|
};
|
|
};
|
|
},
|
|
options: { totalCostUsd?: number } = {},
|
|
): void {
|
|
const inputTokens = readPositiveFiniteNumber(part.tokens?.input);
|
|
const outputTokens = readPositiveFiniteNumber(part.tokens?.output);
|
|
const reasoningTokens = readPositiveFiniteNumber(part.tokens?.reasoning);
|
|
const cacheReadTokens = readPositiveFiniteNumber(part.tokens?.cache?.read);
|
|
const cacheWriteTokens = readPositiveFiniteNumber(part.tokens?.cache?.write);
|
|
const totalTokens =
|
|
(inputTokens ?? 0) +
|
|
(outputTokens ?? 0) +
|
|
(reasoningTokens ?? 0) +
|
|
(cacheReadTokens ?? 0) +
|
|
(cacheWriteTokens ?? 0);
|
|
const cost = readPositiveFiniteNumber(part.cost);
|
|
|
|
assignUsageNumber(usage, "inputTokens", inputTokens);
|
|
assignUsageNumber(usage, "cachedInputTokens", cacheReadTokens);
|
|
assignUsageNumber(usage, "outputTokens", outputTokens);
|
|
if (totalTokens > 0) {
|
|
usage.contextWindowUsedTokens = totalTokens;
|
|
}
|
|
if (cost !== undefined) {
|
|
usage.totalCostUsd = options.totalCostUsd ?? (usage.totalCostUsd ?? 0) + cost;
|
|
}
|
|
}
|
|
|
|
function hasNormalizedOpenCodeUsage(usage: AgentUsage): boolean {
|
|
return [
|
|
usage.inputTokens,
|
|
usage.cachedInputTokens,
|
|
usage.outputTokens,
|
|
usage.totalCostUsd,
|
|
usage.contextWindowMaxTokens,
|
|
usage.contextWindowUsedTokens,
|
|
].some((value) => typeof value === "number" && Number.isFinite(value));
|
|
}
|
|
|
|
function getOpenCodeAttachmentExtension(mimeType: string): string {
|
|
switch (mimeType) {
|
|
case "image/png":
|
|
return "png";
|
|
case "image/jpeg":
|
|
return "jpg";
|
|
case "image/webp":
|
|
return "webp";
|
|
case "image/gif":
|
|
return "gif";
|
|
case "image/svg+xml":
|
|
return "svg";
|
|
default:
|
|
return "bin";
|
|
}
|
|
}
|
|
|
|
function toOpenCodeDataUrl(mimeType: string, data: string): { mimeType: string; url: string } {
|
|
const match = data.match(/^data:([^;,]+);base64,(.+)$/);
|
|
if (match) {
|
|
return {
|
|
mimeType: match[1] ?? mimeType,
|
|
url: data,
|
|
};
|
|
}
|
|
return {
|
|
mimeType,
|
|
url: `data:${mimeType};base64,${data}`,
|
|
};
|
|
}
|
|
|
|
function buildOpenCodePromptParts(
|
|
prompt: AgentPromptInput,
|
|
): Array<OpenCodeTextPartInput | OpenCodeFilePartInput> {
|
|
if (typeof prompt === "string") {
|
|
return [{ type: "text", text: prompt }];
|
|
}
|
|
let attachmentOrdinal = 0;
|
|
const output: Array<OpenCodeTextPartInput | OpenCodeFilePartInput> = [];
|
|
for (const part of prompt) {
|
|
if (part.type === "text") {
|
|
output.push({ type: "text", text: part.text });
|
|
continue;
|
|
}
|
|
if (part.type === "image") {
|
|
attachmentOrdinal += 1;
|
|
const normalized = toOpenCodeDataUrl(part.mimeType, part.data);
|
|
output.push({
|
|
type: "file",
|
|
mime: normalized.mimeType,
|
|
filename: `attachment-${attachmentOrdinal}.${getOpenCodeAttachmentExtension(
|
|
normalized.mimeType,
|
|
)}`,
|
|
url: normalized.url,
|
|
});
|
|
continue;
|
|
}
|
|
output.push({ type: "text", text: renderPromptAttachmentAsText(part) });
|
|
}
|
|
return output;
|
|
}
|
|
|
|
function buildOpenCodeUserTimelineText(prompt: AgentPromptInput): string {
|
|
if (typeof prompt === "string") {
|
|
return prompt;
|
|
}
|
|
return prompt
|
|
.map((part) => {
|
|
if (part.type === "text") {
|
|
return part.text;
|
|
}
|
|
if (part.type === "image") {
|
|
return "[Image]";
|
|
}
|
|
return renderPromptAttachmentAsText(part);
|
|
})
|
|
.filter((text) => text.trim().length > 0)
|
|
.join("\n");
|
|
}
|
|
|
|
async function collectOpenCodePersistedAgentsFromSdk(
|
|
client: Pick<OpencodeClient, "experimental" | "session">,
|
|
options?: ListPersistedAgentsOptions,
|
|
): Promise<PersistedAgentDescriptor[]> {
|
|
const limit = options?.limit ?? OPENCODE_PERSISTED_SESSION_LIMIT;
|
|
const sessionListLimit = options?.cwd ? Math.max(limit, OPENCODE_PERSISTED_SESSION_LIMIT) : limit;
|
|
const response = await client.experimental.session.list({
|
|
archived: true,
|
|
roots: true,
|
|
limit: sessionListLimit,
|
|
});
|
|
|
|
if (response.error) {
|
|
throw new Error(`Failed to list OpenCode sessions: ${JSON.stringify(response.error)}`);
|
|
}
|
|
|
|
const sessions = response.data ?? [];
|
|
const matchesCwd = options?.cwd ? createPathEquivalenceMatcher(options.cwd) : null;
|
|
const candidates = sessions
|
|
.filter((session) => !matchesCwd || matchesCwd(session.directory))
|
|
.sort((left, right) => getOpenCodeSessionTimestamp(right) - getOpenCodeSessionTimestamp(left))
|
|
.slice(0, limit);
|
|
|
|
return await Promise.all(
|
|
candidates.map((session) => buildOpenCodePersistedAgentDescriptor(client, session)),
|
|
);
|
|
}
|
|
|
|
async function buildOpenCodePersistedAgentDescriptor(
|
|
client: Pick<OpencodeClient, "session">,
|
|
session: OpenCodePersistedSession,
|
|
): Promise<PersistedAgentDescriptor> {
|
|
const messages = await readOpenCodeSessionMessagesFromSdk(client, session);
|
|
const timeline = buildOpenCodeSessionTimeline(messages);
|
|
const modeId = resolveOpenCodePersistedSessionModeId(session, messages);
|
|
const model = resolveOpenCodePersistedSessionModel(session, messages);
|
|
return {
|
|
provider: "opencode",
|
|
sessionId: session.id,
|
|
cwd: session.directory,
|
|
title: normalizeOpenCodeSessionTitle(session.title),
|
|
lastActivityAt: new Date(getOpenCodeSessionTimestamp(session)),
|
|
persistence: {
|
|
provider: "opencode",
|
|
sessionId: session.id,
|
|
nativeHandle: session.id,
|
|
metadata: {
|
|
provider: "opencode",
|
|
cwd: session.directory,
|
|
title: normalizeOpenCodeSessionTitle(session.title),
|
|
...(modeId ? { modeId } : {}),
|
|
...(model ? { model } : {}),
|
|
},
|
|
},
|
|
timeline,
|
|
};
|
|
}
|
|
|
|
function normalizeOpenCodeSessionTitle(title: string | null | undefined): string | null {
|
|
const normalized = title?.trim();
|
|
return normalized ? normalized : null;
|
|
}
|
|
|
|
function getOpenCodeSessionTimestamp(session: OpenCodePersistedSession): number {
|
|
return session.time?.updated ?? session.time?.created ?? 0;
|
|
}
|
|
|
|
function resolveOpenCodeReplayTimestamp(params: {
|
|
message: { time?: { created?: number; completed?: number } | undefined };
|
|
part?: unknown;
|
|
}): string | null {
|
|
const timedPart = params.part as
|
|
| { time?: { start?: number; end?: number } | undefined }
|
|
| undefined;
|
|
const partTimestamp =
|
|
timedPart?.time?.start ??
|
|
timedPart?.time?.end ??
|
|
params.message.time?.created ??
|
|
params.message.time?.completed;
|
|
return normalizeProviderReplayTimestamp(partTimestamp);
|
|
}
|
|
|
|
function buildOpenCodeReplayTimelineEvent(params: {
|
|
item: AgentTimelineItem;
|
|
message: { time?: { created?: number; completed?: number } | undefined };
|
|
part?: unknown;
|
|
}): Extract<AgentStreamEvent, { type: "timeline" }> {
|
|
const timestamp = resolveOpenCodeReplayTimestamp({
|
|
message: params.message,
|
|
part: params.part,
|
|
});
|
|
return {
|
|
type: "timeline",
|
|
provider: "opencode",
|
|
item: params.item,
|
|
...(timestamp ? { timestamp } : {}),
|
|
};
|
|
}
|
|
|
|
function buildOpenCodeReplayPartTimelineEvent(params: {
|
|
part: OpenCodePart;
|
|
message: { structured?: unknown; time?: { created?: number; completed?: number } | undefined };
|
|
}): Extract<AgentStreamEvent, { type: "timeline" }> | null {
|
|
const { part, message } = params;
|
|
if (part.type === "text" && part.text) {
|
|
return buildOpenCodeReplayTimelineEvent({
|
|
item: { type: "assistant_message", text: part.text },
|
|
message,
|
|
part,
|
|
});
|
|
}
|
|
if (part.type === "reasoning" && part.text) {
|
|
return buildOpenCodeReplayTimelineEvent({
|
|
item: { type: "reasoning", text: part.text },
|
|
message,
|
|
part,
|
|
});
|
|
}
|
|
if (part.type !== "tool") {
|
|
return null;
|
|
}
|
|
const parsedToolPart = OpencodeToolPartToTimelineItemSchema.safeParse(part);
|
|
if (!parsedToolPart.success || !parsedToolPart.data) {
|
|
return null;
|
|
}
|
|
return buildOpenCodeReplayTimelineEvent({
|
|
item: parsedToolPart.data,
|
|
message,
|
|
part,
|
|
});
|
|
}
|
|
|
|
async function readOpenCodeSessionMessagesFromSdk(
|
|
client: Pick<OpencodeClient, "session">,
|
|
session: OpenCodePersistedSession,
|
|
): Promise<OpenCodeSessionMessage[]> {
|
|
const response = await client.session.messages({
|
|
sessionID: session.id,
|
|
directory: session.directory,
|
|
});
|
|
|
|
if (response.error || !response.data) {
|
|
return [];
|
|
}
|
|
|
|
return filterOpenCodeRevertedMessages(response.data, session.revert);
|
|
}
|
|
|
|
function buildOpenCodeSessionTimeline(
|
|
messages: ReadonlyArray<OpenCodeSessionMessage>,
|
|
): AgentTimelineItem[] {
|
|
return messages.flatMap((message) =>
|
|
buildOpenCodeReplayTimelineEvents(message).map((event) => event.item),
|
|
);
|
|
}
|
|
|
|
function filterOpenCodeRevertedMessages(
|
|
messages: ReadonlyArray<OpenCodeSessionMessage>,
|
|
revert: OpenCodePersistedSession["revert"] | null | undefined,
|
|
): OpenCodeSessionMessage[] {
|
|
if (!revert?.messageID || revert.partID) {
|
|
return [...messages];
|
|
}
|
|
const revertIndex = messages.findIndex((message) => message.info.id === revert.messageID);
|
|
if (revertIndex < 0) {
|
|
return [...messages];
|
|
}
|
|
return messages.slice(0, revertIndex);
|
|
}
|
|
|
|
function resolveOpenCodePersistedSessionModeId(
|
|
session: OpenCodePersistedSession,
|
|
messages: ReadonlyArray<OpenCodeSessionMessage>,
|
|
): string | undefined {
|
|
const agent = session.agent ?? messages.map(readOpenCodeMessageAgent).find(Boolean);
|
|
return agent ? normalizeOpenCodeModeId(agent) : undefined;
|
|
}
|
|
|
|
function readOpenCodeMessageAgent(message: OpenCodeSessionMessage): string | undefined {
|
|
const agent = message.info.agent;
|
|
return typeof agent === "string" && agent.trim() ? agent : undefined;
|
|
}
|
|
|
|
function resolveOpenCodePersistedSessionModel(
|
|
session: OpenCodePersistedSession,
|
|
messages: ReadonlyArray<OpenCodeSessionMessage>,
|
|
): string | undefined {
|
|
if (session.model) {
|
|
return buildOpenCodeModelLookupKey(session.model.providerID, session.model.id);
|
|
}
|
|
|
|
const model = messages.map(readOpenCodeMessageModel).find(Boolean);
|
|
return model ? buildOpenCodeModelLookupKey(model.providerID, model.modelID) : undefined;
|
|
}
|
|
|
|
function readOpenCodeMessageModel(
|
|
message: OpenCodeSessionMessage,
|
|
): { providerID: string; modelID: string } | undefined {
|
|
const { info } = message;
|
|
if (info.role === "user") {
|
|
return info.model;
|
|
}
|
|
return {
|
|
providerID: info.providerID,
|
|
modelID: info.modelID,
|
|
};
|
|
}
|
|
|
|
function buildOpenCodeReplayTimelineEvents(
|
|
message: OpenCodeSessionMessage,
|
|
): Extract<AgentStreamEvent, { type: "timeline" }>[] {
|
|
const { info, parts } = message;
|
|
if (info.role === "user") {
|
|
const text = parts
|
|
.filter((part): part is Extract<OpenCodePart, { type: "text" }> => part.type === "text")
|
|
.map((part) => part.text)
|
|
.join("");
|
|
|
|
return text
|
|
? [
|
|
buildOpenCodeReplayTimelineEvent({
|
|
item: { type: "user_message", text, messageId: info.id },
|
|
message: info,
|
|
}),
|
|
]
|
|
: [];
|
|
}
|
|
|
|
const events: Extract<AgentStreamEvent, { type: "timeline" }>[] = [];
|
|
let emittedAssistantText = false;
|
|
for (const part of parts) {
|
|
if (part.type === "text" && part.text) {
|
|
emittedAssistantText = true;
|
|
}
|
|
const event = buildOpenCodeReplayPartTimelineEvent({ part, message: info });
|
|
if (event) {
|
|
events.push(event);
|
|
}
|
|
}
|
|
|
|
if (!emittedAssistantText) {
|
|
const text = stringifyStructuredAssistantMessage(info.structured);
|
|
if (text) {
|
|
events.push(
|
|
buildOpenCodeReplayTimelineEvent({
|
|
item: { type: "assistant_message", text },
|
|
message: info,
|
|
}),
|
|
);
|
|
}
|
|
}
|
|
|
|
return events;
|
|
}
|
|
|
|
export const __openCodeInternals = {
|
|
buildOpenCodePromptParts,
|
|
buildOpenCodeModelContextWindowLookup,
|
|
buildOpenCodeModelDefinition,
|
|
buildOpenCodeModelLookupKey,
|
|
extractOpenCodeModelContextWindow,
|
|
hasNormalizedOpenCodeUsage,
|
|
mergeOpenCodeStepFinishUsage,
|
|
parseOpenCodeModelLookupKey,
|
|
reconcileOpenCodeSessionClose,
|
|
resolveOpenCodeModelLookupKeyFromAssistantMessage,
|
|
resolveOpenCodeSelectedModelContextWindow,
|
|
isSelectableOpenCodeAgent,
|
|
get OpenCodeAgentSession() {
|
|
return OpenCodeAgentSession;
|
|
},
|
|
};
|
|
|
|
interface OpenCodeAgentClientDeps {
|
|
runtime?: OpenCodeRuntime;
|
|
}
|
|
|
|
class ProductionOpenCodeRuntime implements OpenCodeRuntime {
|
|
constructor(private readonly serverManager: OpenCodeServerManager) {}
|
|
|
|
async acquireServer(options: {
|
|
force: boolean;
|
|
env?: Record<string, string>;
|
|
}): Promise<OpenCodeServerAcquisition> {
|
|
return this.serverManager.acquire(options);
|
|
}
|
|
|
|
async ensureServerRunning(): Promise<{ port: number; url: string }> {
|
|
return this.serverManager.ensureRunning();
|
|
}
|
|
|
|
createClient(options: { baseUrl: string; directory: string }): OpencodeClient {
|
|
return createSdkOpenCodeClient(options);
|
|
}
|
|
|
|
async shutdown(): Promise<void> {
|
|
await this.serverManager.shutdown();
|
|
}
|
|
}
|
|
|
|
export class OpenCodeAgentClient implements AgentClient {
|
|
readonly provider = "opencode" as const;
|
|
readonly capabilities = OPENCODE_CAPABILITIES;
|
|
|
|
private readonly runtime: OpenCodeRuntime;
|
|
private readonly logger: Logger;
|
|
private readonly runtimeSettings?: ProviderRuntimeSettings;
|
|
private readonly modelContextWindows = new Map<string, number>();
|
|
|
|
constructor(
|
|
logger: Logger,
|
|
runtimeSettings?: ProviderRuntimeSettings,
|
|
deps: OpenCodeAgentClientDeps = {},
|
|
) {
|
|
this.logger = logger.child({ module: "agent", provider: "opencode" });
|
|
this.runtimeSettings = runtimeSettings;
|
|
this.runtime =
|
|
deps.runtime ??
|
|
new ProductionOpenCodeRuntime(
|
|
OpenCodeServerManager.getInstance(this.logger, runtimeSettings),
|
|
);
|
|
}
|
|
|
|
async createSession(
|
|
config: AgentSessionConfig,
|
|
launchContext?: AgentLaunchContext,
|
|
options?: AgentCreateSessionOptions,
|
|
): Promise<AgentSession> {
|
|
const openCodeConfig = this.assertConfig(config);
|
|
const acquisition = await this.runtime.acquireServer({
|
|
force: false,
|
|
env: launchContext?.env,
|
|
});
|
|
const { url } = acquisition.server;
|
|
const client = this.runtime.createClient({
|
|
baseUrl: url,
|
|
directory: openCodeConfig.cwd,
|
|
});
|
|
|
|
try {
|
|
const response = await withTimeout(
|
|
client.session.create({ directory: openCodeConfig.cwd }),
|
|
10_000,
|
|
"OpenCode session.create timed out after 10s",
|
|
);
|
|
|
|
if (response.error) {
|
|
throw new Error(`Failed to create OpenCode session: ${JSON.stringify(response.error)}`);
|
|
}
|
|
|
|
const session = response.data;
|
|
if (!session) {
|
|
throw new Error("OpenCode session creation returned no data");
|
|
}
|
|
|
|
await this.populateModelContextWindowCache(client, openCodeConfig.cwd);
|
|
|
|
return new OpenCodeAgentSession(
|
|
openCodeConfig,
|
|
client,
|
|
session.id,
|
|
this.logger,
|
|
new Map(this.modelContextWindows),
|
|
acquisition.release,
|
|
options?.persistSession,
|
|
launchContext?.agentId,
|
|
);
|
|
} catch (error) {
|
|
acquisition.release();
|
|
throw error;
|
|
}
|
|
}
|
|
|
|
async resumeSession(
|
|
handle: AgentPersistenceHandle,
|
|
overrides?: Partial<AgentSessionConfig>,
|
|
launchContext?: AgentLaunchContext,
|
|
): Promise<AgentSession> {
|
|
const metadata = (handle.metadata ?? {}) as Partial<AgentSessionConfig>;
|
|
const cwd = overrides?.cwd ?? metadata.cwd;
|
|
if (!cwd) {
|
|
throw new Error("OpenCode resume requires the original working directory");
|
|
}
|
|
|
|
const config: AgentSessionConfig = {
|
|
...metadata,
|
|
...overrides,
|
|
provider: "opencode",
|
|
cwd,
|
|
};
|
|
const openCodeConfig = this.assertConfig(config);
|
|
const acquisition = await this.runtime.acquireServer({ force: false });
|
|
const { url } = acquisition.server;
|
|
const client = this.runtime.createClient({
|
|
baseUrl: url,
|
|
directory: openCodeConfig.cwd,
|
|
});
|
|
|
|
try {
|
|
await this.populateModelContextWindowCache(client, openCodeConfig.cwd);
|
|
|
|
return new OpenCodeAgentSession(
|
|
openCodeConfig,
|
|
client,
|
|
handle.sessionId,
|
|
this.logger,
|
|
new Map(this.modelContextWindows),
|
|
acquisition.release,
|
|
undefined,
|
|
launchContext?.agentId,
|
|
);
|
|
} catch (error) {
|
|
acquisition.release();
|
|
throw error;
|
|
}
|
|
}
|
|
|
|
async listModels(options: ListModelsOptions): Promise<AgentModelDefinition[]> {
|
|
const acquisition = await this.runtime.acquireServer({ force: options.force });
|
|
const { url } = acquisition.server;
|
|
const client = this.runtime.createClient({
|
|
baseUrl: url,
|
|
directory: options.cwd,
|
|
});
|
|
|
|
try {
|
|
// Background model discovery can be legitimately slow while OpenCode refreshes
|
|
// provider state, so allow longer than turn execution paths.
|
|
const response = await withTimeout(
|
|
client.provider.list({ directory: options.cwd }),
|
|
OPENCODE_PROVIDER_LIST_TIMEOUT_MS,
|
|
`OpenCode provider.list timed out after ${OPENCODE_PROVIDER_LIST_TIMEOUT_MS / 1000}s - server may not be authenticated or connected to any providers`,
|
|
);
|
|
|
|
if (response.error) {
|
|
throw new Error(`Failed to fetch OpenCode providers: ${JSON.stringify(response.error)}`);
|
|
}
|
|
|
|
const providers = response.data;
|
|
if (!providers) {
|
|
return [];
|
|
}
|
|
|
|
const connectedProviderIds = new Set(providers.connected);
|
|
|
|
// Providers with source "api" are managed by the OpenCode console/subscription (e.g. Pi
|
|
// coding agent). They do not appear in `connected` (which only lists env/config providers)
|
|
// but are fully usable — OpenCode authenticates them internally via the console session.
|
|
const isAccessible = (provider: { id: string; source: string }): boolean =>
|
|
connectedProviderIds.has(provider.id) || provider.source === "api";
|
|
|
|
// Fail fast if no providers are accessible at all
|
|
if (!providers.all.some(isAccessible)) {
|
|
throw new Error(
|
|
"OpenCode has no connected providers. Please authenticate with at least one provider " +
|
|
"(e.g., openai, anthropic), set appropriate environment variables (e.g., OPENAI_API_KEY), " +
|
|
"or log in to OpenCode Go via the console.",
|
|
);
|
|
}
|
|
|
|
const models: AgentModelDefinition[] = [];
|
|
this.modelContextWindows.clear();
|
|
for (const provider of providers.all) {
|
|
if (!isAccessible(provider)) {
|
|
continue;
|
|
}
|
|
|
|
for (const [modelId, model] of Object.entries(provider.models)) {
|
|
const definition = buildOpenCodeModelDefinition(provider, modelId, model);
|
|
const contextWindowMaxTokens = extractOpenCodeModelContextWindow(model);
|
|
if (contextWindowMaxTokens !== undefined) {
|
|
this.modelContextWindows.set(
|
|
buildOpenCodeModelLookupKey(provider.id, modelId),
|
|
contextWindowMaxTokens,
|
|
);
|
|
}
|
|
models.push(definition);
|
|
}
|
|
}
|
|
|
|
return models;
|
|
} finally {
|
|
acquisition.release();
|
|
}
|
|
}
|
|
|
|
async listModes(options: ListModesOptions): Promise<AgentMode[]> {
|
|
const acquisition = await this.runtime.acquireServer({ force: options.force });
|
|
const { url } = acquisition.server;
|
|
const directory = options.cwd;
|
|
const client = this.runtime.createClient({ baseUrl: url, directory });
|
|
|
|
try {
|
|
const response = await withTimeout(
|
|
client.app.agents({ directory }),
|
|
10_000,
|
|
"OpenCode app.agents timed out after 10s",
|
|
);
|
|
|
|
if (response.error || !response.data) {
|
|
return DEFAULT_MODES;
|
|
}
|
|
|
|
const discovered = response.data.filter(isSelectableOpenCodeAgent).map((agent) => ({
|
|
id: agent.name,
|
|
label: agent.name.charAt(0).toUpperCase() + agent.name.slice(1),
|
|
description:
|
|
typeof agent.description === "string" && agent.description.trim().length > 0
|
|
? agent.description.trim()
|
|
: DEFAULT_MODES.find((mode) => mode.id === agent.name)?.description,
|
|
}));
|
|
|
|
return mergeOpenCodeModes(discovered);
|
|
} finally {
|
|
acquisition.release();
|
|
}
|
|
}
|
|
|
|
async listCommands(config: AgentSessionConfig): Promise<AgentSlashCommand[]> {
|
|
const openCodeConfig = this.assertConfig(config);
|
|
const acquisition = await this.runtime.acquireServer({ force: false });
|
|
const { url } = acquisition.server;
|
|
const client = this.runtime.createClient({
|
|
baseUrl: url,
|
|
directory: openCodeConfig.cwd,
|
|
});
|
|
|
|
try {
|
|
return await listOpenCodeCommandsFromSdk(client, openCodeConfig.cwd);
|
|
} finally {
|
|
acquisition.release();
|
|
}
|
|
}
|
|
|
|
async listFeatures(_config: AgentSessionConfig): Promise<AgentFeature[]> {
|
|
return [];
|
|
}
|
|
|
|
async listPersistedAgents(
|
|
options?: ListPersistedAgentsOptions,
|
|
): Promise<PersistedAgentDescriptor[]> {
|
|
const acquisition = await this.runtime.acquireServer({ force: false });
|
|
const { url } = acquisition.server;
|
|
const client = this.runtime.createClient({
|
|
baseUrl: url,
|
|
directory: options?.cwd ?? "",
|
|
});
|
|
|
|
try {
|
|
return await collectOpenCodePersistedAgentsFromSdk(client, options);
|
|
} finally {
|
|
acquisition.release();
|
|
}
|
|
}
|
|
|
|
async isAvailable(): Promise<boolean> {
|
|
const command = this.runtimeSettings?.command;
|
|
if (command?.mode === "replace") {
|
|
return await isCommandAvailable(command.argv[0]);
|
|
}
|
|
return await isCommandAvailable("opencode");
|
|
}
|
|
|
|
async getDiagnostic(): Promise<{ diagnostic: string }> {
|
|
try {
|
|
const available = await this.isAvailable();
|
|
const resolvedBinary = await findExecutable("opencode");
|
|
let serverStatus = "Not running";
|
|
let modelsValue = "Not checked";
|
|
let status = formatDiagnosticStatus(available);
|
|
|
|
try {
|
|
const { url } = await this.runtime.ensureServerRunning();
|
|
serverStatus = `Running (${url})`;
|
|
} catch (error) {
|
|
serverStatus = `Unavailable (${toDiagnosticErrorMessage(error)})`;
|
|
}
|
|
|
|
let authValue = "Not checked";
|
|
if (resolvedBinary) {
|
|
try {
|
|
const { stdout, stderr } = await execCommand(resolvedBinary, ["auth", "list"], {
|
|
...createProviderEnvSpec(),
|
|
timeout: 5_000,
|
|
});
|
|
const text = (stdout.trim() || stderr.trim()).trim();
|
|
authValue = text ? `\n ${text.replace(/\n/g, "\n ")}` : "(empty)";
|
|
} catch (error) {
|
|
authValue = `Error - ${toDiagnosticErrorMessage(error)}`;
|
|
}
|
|
}
|
|
|
|
if (available) {
|
|
try {
|
|
const models = await this.listModels({ cwd: homedir(), force: false });
|
|
modelsValue = String(models.length);
|
|
} catch (error) {
|
|
modelsValue = `Error - ${toDiagnosticErrorMessage(error)}`;
|
|
status = formatDiagnosticStatus(available, {
|
|
source: "model fetch",
|
|
cause: error,
|
|
});
|
|
}
|
|
|
|
if (!modelsValue.startsWith("Error -")) {
|
|
try {
|
|
await this.listModes({ cwd: homedir(), force: false });
|
|
} catch (error) {
|
|
status = formatDiagnosticStatus(available, {
|
|
source: "mode fetch",
|
|
cause: error,
|
|
});
|
|
}
|
|
}
|
|
}
|
|
|
|
return {
|
|
diagnostic: formatProviderDiagnostic("OpenCode", [
|
|
{
|
|
label: "Binary",
|
|
value: resolvedBinary ?? "not found",
|
|
},
|
|
{
|
|
label: "Version",
|
|
value: resolvedBinary ? await resolveBinaryVersion(resolvedBinary) : "unknown",
|
|
},
|
|
{ label: "Server", value: serverStatus },
|
|
{ label: "Auth", value: authValue },
|
|
{ label: "Models", value: modelsValue },
|
|
{ label: "Status", value: status },
|
|
]),
|
|
};
|
|
} catch (error) {
|
|
return {
|
|
diagnostic: formatProviderDiagnosticError("OpenCode", error),
|
|
};
|
|
}
|
|
}
|
|
private assertConfig(config: AgentSessionConfig): OpenCodeAgentConfig {
|
|
if (config.provider !== "opencode") {
|
|
throw new Error(`OpenCodeAgentClient received config for provider '${config.provider}'`);
|
|
}
|
|
return { ...config, provider: "opencode" };
|
|
}
|
|
|
|
private async populateModelContextWindowCache(
|
|
client: OpencodeClient,
|
|
cwd: string,
|
|
): Promise<void> {
|
|
const response = await client.provider.list({ directory: cwd });
|
|
if (response.error || !response.data) {
|
|
return;
|
|
}
|
|
|
|
const lookup = buildOpenCodeModelContextWindowLookup(response.data);
|
|
this.modelContextWindows.clear();
|
|
for (const [modelLookupKey, contextWindowMaxTokens] of lookup.entries()) {
|
|
this.modelContextWindows.set(modelLookupKey, contextWindowMaxTokens);
|
|
}
|
|
}
|
|
}
|
|
|
|
export interface OpenCodeEventTranslationState {
|
|
sessionId: string;
|
|
cwd?: string;
|
|
messageRoles: Map<string, OpenCodeMessageRole>;
|
|
pendingUserMessageText?: string | null;
|
|
emittedUserMessageIds?: Set<string>;
|
|
accumulatedUsage: AgentUsage;
|
|
sessionTotalCostUsd?: number;
|
|
streamedPartKeys: Set<string>;
|
|
emittedStructuredMessageIds: Set<string>;
|
|
/** Tracks the type of each part by ID, learned from message.part.updated events. */
|
|
partTypes: Map<string, string>;
|
|
subAgentsByCallId?: Map<string, OpenCodeSubAgentActivityState>;
|
|
subAgentCallIdByChildSessionId?: Map<string, string>;
|
|
pendingChildToolPartsBySessionId?: Map<string, OpenCodeToolPartEventPart[]>;
|
|
modelContextWindowsByModelKey?: ReadonlyMap<string, number>;
|
|
onAssistantModelContextWindowResolved?: (contextWindowMaxTokens: number) => void;
|
|
}
|
|
|
|
interface OpenCodeTraceData {
|
|
turnId?: string;
|
|
[key: string]: unknown;
|
|
}
|
|
|
|
type OpenCodeTraceMessage =
|
|
| "provider.opencode.prompt_async.start"
|
|
| "provider.opencode.prompt_async.response"
|
|
| "provider.opencode.prompt_async.throw"
|
|
| "provider.opencode.subscribe.start"
|
|
| "provider.opencode.subscribe.ready"
|
|
| "provider.opencode.stream.eof"
|
|
| "provider.opencode.turn.fail_eof"
|
|
| "provider.opencode.subscribe.error"
|
|
| "provider.opencode.raw_event"
|
|
| "provider.opencode.event.skip"
|
|
| "provider.opencode.parsed_event"
|
|
| "provider.opencode.parsed_event.skip_active"
|
|
| "provider.opencode.event.terminal"
|
|
| "provider.opencode.finish_foreground_turn"
|
|
| "provider.opencode.event_emit";
|
|
|
|
type OpenCodeToolPartEventPart = Extract<
|
|
Extract<OpenCodeEvent, { type: "message.part.updated" }>["properties"]["part"],
|
|
{ type: "tool" }
|
|
>;
|
|
|
|
interface OpenCodeSubAgentActionEntry {
|
|
index: number;
|
|
key: string;
|
|
toolName: string;
|
|
summary?: string;
|
|
}
|
|
|
|
interface OpenCodeSubAgentActivityState {
|
|
toolCall: ToolCallTimelineItem;
|
|
actions: OpenCodeSubAgentActionEntry[];
|
|
actionIndexByKey: Map<string, number>;
|
|
nextActionIndex: number;
|
|
childSessionId?: string;
|
|
}
|
|
|
|
const MAX_OPENCODE_SUB_AGENT_ACTIONS = 200;
|
|
const MAX_OPENCODE_PENDING_CHILD_TOOL_PARTS = 200;
|
|
|
|
function stringifyStructuredAssistantMessage(value: unknown): string | null {
|
|
if (value === undefined) {
|
|
return null;
|
|
}
|
|
if (typeof value === "string") {
|
|
const trimmed = value.trim();
|
|
return trimmed.length > 0 ? trimmed : null;
|
|
}
|
|
try {
|
|
return JSON.stringify(value);
|
|
} catch {
|
|
return null;
|
|
}
|
|
}
|
|
|
|
async function listOpenCodeCommandsFromSdk(
|
|
client: Pick<OpencodeClient, "command">,
|
|
directory: string,
|
|
): Promise<AgentSlashCommand[]> {
|
|
const result = await client.command.list({ directory });
|
|
const commandsByName = new Map(
|
|
OPENCODE_HANDLED_BUILTIN_SLASH_COMMANDS.map((command) => [command.name, command]),
|
|
);
|
|
if (result.error || !result.data) {
|
|
return Array.from(commandsByName.values());
|
|
}
|
|
|
|
for (const cmd of result.data) {
|
|
commandsByName.set(cmd.name, {
|
|
name: cmd.name,
|
|
description: cmd.description ?? "",
|
|
argumentHint: cmd.hints?.length ? cmd.hints.join(" ") : "",
|
|
});
|
|
}
|
|
|
|
return Array.from(commandsByName.values());
|
|
}
|
|
|
|
function readOpenCodeRecord(value: unknown): Record<string, unknown> | null {
|
|
return typeof value === "object" && value !== null && !Array.isArray(value)
|
|
? (value as Record<string, unknown>)
|
|
: null;
|
|
}
|
|
|
|
function readNonEmptyString(value: unknown): string | null {
|
|
return typeof value === "string" && value.trim().length > 0 ? value.trim() : null;
|
|
}
|
|
|
|
function mapOpenCodeTodosToTimelineItems(
|
|
todos: Array<{ content?: string | null; status?: string | null }>,
|
|
): Extract<AgentTimelineItem, { type: "todo" }> {
|
|
return {
|
|
type: "todo",
|
|
items: todos.flatMap((todo) => {
|
|
const text = readNonEmptyString(todo.content);
|
|
if (!text) {
|
|
return [];
|
|
}
|
|
|
|
return [
|
|
{
|
|
text,
|
|
completed: todo.status === "completed",
|
|
},
|
|
];
|
|
}),
|
|
};
|
|
}
|
|
|
|
function createCompactionTimelineItem(
|
|
status: Extract<AgentTimelineItem, { type: "compaction" }>["status"],
|
|
trigger?: Extract<AgentTimelineItem, { type: "compaction" }>["trigger"],
|
|
): Extract<AgentTimelineItem, { type: "compaction" }> {
|
|
return {
|
|
type: "compaction",
|
|
status,
|
|
...(trigger ? { trigger } : {}),
|
|
};
|
|
}
|
|
|
|
const PERMISSION_COMMAND_KEYS = ["command", "cmd", "shellCommand"] as const;
|
|
const PERMISSION_CWD_KEYS = ["cwd", "directory", "path", "workdir"] as const;
|
|
const PERMISSION_REASON_KEYS = ["reason", "purpose", "description", "message"] as const;
|
|
const PERMISSION_TITLE_BY_NAME: Record<string, string> = {
|
|
external_directory: "Access external directory",
|
|
bash: "Run shell command",
|
|
read: "Read files",
|
|
read_file: "Read files",
|
|
write: "Write files",
|
|
write_file: "Write files",
|
|
create_file: "Write files",
|
|
edit: "Edit files",
|
|
apply_patch: "Edit files",
|
|
apply_diff: "Edit files",
|
|
};
|
|
|
|
function toHumanReadablePermissionTitle(permission: string): string {
|
|
const mapped = PERMISSION_TITLE_BY_NAME[permission];
|
|
if (mapped) {
|
|
return mapped;
|
|
}
|
|
|
|
const normalized = permission
|
|
.split(/[\s_-]+/)
|
|
.map((part) => part.trim())
|
|
.filter((part) => part.length > 0)
|
|
.map((part) => `${part[0]?.toUpperCase() ?? ""}${part.slice(1)}`)
|
|
.join(" ");
|
|
return normalized.length > 0 ? normalized : "Permission request";
|
|
}
|
|
|
|
function readFirstStringFromRecord(
|
|
record: Record<string, unknown> | null,
|
|
keys: readonly string[],
|
|
): string | null {
|
|
if (!record) {
|
|
return null;
|
|
}
|
|
for (const key of keys) {
|
|
const value = readNonEmptyString(record[key]);
|
|
if (value) {
|
|
return value;
|
|
}
|
|
}
|
|
return null;
|
|
}
|
|
|
|
function readPermissionField(
|
|
metadata: Record<string, unknown> | null,
|
|
keys: readonly string[],
|
|
): string | null {
|
|
const direct = readFirstStringFromRecord(metadata, keys);
|
|
if (direct) {
|
|
return direct;
|
|
}
|
|
|
|
const nestedInput = readOpenCodeRecord(metadata?.input);
|
|
return readFirstStringFromRecord(nestedInput, keys);
|
|
}
|
|
|
|
function buildOpenCodePermissionInput(params: {
|
|
patterns: string[];
|
|
metadata: Record<string, unknown> | null;
|
|
tool: Record<string, unknown> | null;
|
|
command: string | null;
|
|
}): Record<string, unknown> {
|
|
return {
|
|
...(params.patterns.length > 0 ? { patterns: params.patterns } : {}),
|
|
...(params.metadata ? { metadata: params.metadata } : {}),
|
|
...(params.tool ? { tool: params.tool } : {}),
|
|
...(params.command ? { command: params.command } : {}),
|
|
};
|
|
}
|
|
|
|
function buildOpenCodePermissionDetail(params: {
|
|
permission: string;
|
|
input: Record<string, unknown>;
|
|
command: string | null;
|
|
cwd: string | null;
|
|
}): ToolCallDetail {
|
|
if (params.command) {
|
|
return {
|
|
type: "shell",
|
|
command: params.command,
|
|
...(params.cwd ? { cwd: params.cwd } : {}),
|
|
};
|
|
}
|
|
|
|
return {
|
|
type: "unknown",
|
|
input: {
|
|
permission: params.permission,
|
|
...params.input,
|
|
},
|
|
output: null,
|
|
};
|
|
}
|
|
|
|
function buildOpenCodePermissionDescription(params: {
|
|
reason: string | null;
|
|
patterns: string[];
|
|
}): string | undefined {
|
|
const parts: string[] = [];
|
|
if (params.reason) {
|
|
parts.push(params.reason);
|
|
}
|
|
if (params.patterns.length > 0) {
|
|
parts.push(`Scope: ${params.patterns.join(", ")}`);
|
|
}
|
|
return parts.length > 0 ? parts.join(" - ") : undefined;
|
|
}
|
|
|
|
export function translateOpenCodeEvent(
|
|
event: OpenCodeEvent,
|
|
state: OpenCodeEventTranslationState,
|
|
): AgentStreamEvent[] {
|
|
const events: AgentStreamEvent[] = [];
|
|
|
|
switch (event.type) {
|
|
case "session.created":
|
|
case "session.updated":
|
|
appendOpenCodeSessionCreatedOrUpdated(event, state, events);
|
|
break;
|
|
case "message.updated":
|
|
appendOpenCodeMessageUpdated(event, state, events);
|
|
break;
|
|
case "message.part.updated":
|
|
appendOpenCodeMessagePartUpdated(event, state, events);
|
|
break;
|
|
case "message.part.delta":
|
|
appendOpenCodeMessagePartDelta(event, state, events);
|
|
break;
|
|
case "permission.asked":
|
|
appendOpenCodePermissionAsked(event, state, events);
|
|
break;
|
|
case "question.asked":
|
|
appendOpenCodeQuestionAsked(event, state, events);
|
|
break;
|
|
case "todo.updated":
|
|
if (event.properties.sessionID === state.sessionId) {
|
|
events.push({
|
|
type: "timeline",
|
|
provider: "opencode",
|
|
item: mapOpenCodeTodosToTimelineItems(event.properties.todos),
|
|
});
|
|
}
|
|
break;
|
|
case "session.compacted":
|
|
if (event.properties.sessionID === state.sessionId) {
|
|
events.push({
|
|
type: "timeline",
|
|
provider: "opencode",
|
|
item: createCompactionTimelineItem("completed"),
|
|
});
|
|
}
|
|
break;
|
|
case "session.idle":
|
|
if (event.properties.sessionID === state.sessionId) {
|
|
resetOpenCodeTurnTrackingState(state);
|
|
events.push({ type: "turn_completed", provider: "opencode", usage: undefined });
|
|
}
|
|
break;
|
|
case "session.error":
|
|
appendOpenCodeSessionError(event, state, events);
|
|
break;
|
|
case "session.status":
|
|
appendOpenCodeSessionStatus(event, state, events);
|
|
break;
|
|
}
|
|
|
|
return events;
|
|
}
|
|
|
|
function resetOpenCodeTurnTrackingState(state: OpenCodeEventTranslationState): void {
|
|
state.streamedPartKeys.clear();
|
|
state.partTypes.clear();
|
|
}
|
|
|
|
function getOpenCodeSubAgentMaps(state: OpenCodeEventTranslationState): {
|
|
byCallId: Map<string, OpenCodeSubAgentActivityState>;
|
|
callIdByChildSessionId: Map<string, string>;
|
|
pendingChildToolPartsBySessionId: Map<string, OpenCodeToolPartEventPart[]>;
|
|
} {
|
|
state.subAgentsByCallId ??= new Map();
|
|
state.subAgentCallIdByChildSessionId ??= new Map();
|
|
state.pendingChildToolPartsBySessionId ??= new Map();
|
|
return {
|
|
byCallId: state.subAgentsByCallId,
|
|
callIdByChildSessionId: state.subAgentCallIdByChildSessionId,
|
|
pendingChildToolPartsBySessionId: state.pendingChildToolPartsBySessionId,
|
|
};
|
|
}
|
|
|
|
function getOpenCodeSubAgentState(
|
|
callId: string,
|
|
state: OpenCodeEventTranslationState,
|
|
toolCall: ToolCallTimelineItem,
|
|
): OpenCodeSubAgentActivityState {
|
|
const maps = getOpenCodeSubAgentMaps(state);
|
|
const existing = maps.byCallId.get(callId);
|
|
if (existing) {
|
|
existing.toolCall = toolCall;
|
|
return existing;
|
|
}
|
|
|
|
const created: OpenCodeSubAgentActivityState = {
|
|
toolCall,
|
|
actions: [],
|
|
actionIndexByKey: new Map(),
|
|
nextActionIndex: 1,
|
|
};
|
|
maps.byCallId.set(callId, created);
|
|
return created;
|
|
}
|
|
|
|
function linkOpenCodeSubAgentChildSession(
|
|
activity: OpenCodeSubAgentActivityState,
|
|
childSessionId: string,
|
|
state: OpenCodeEventTranslationState,
|
|
): void {
|
|
activity.childSessionId = childSessionId;
|
|
const maps = getOpenCodeSubAgentMaps(state);
|
|
maps.callIdByChildSessionId.set(childSessionId, activity.toolCall.callId);
|
|
}
|
|
|
|
function buildOpenCodeSubAgentLog(
|
|
detail: Extract<ToolCallDetail, { type: "sub_agent" }>,
|
|
activity: OpenCodeSubAgentActivityState,
|
|
): string {
|
|
const actionLog = activity.actions
|
|
.map((action) =>
|
|
action.summary ? `[${action.toolName}] ${action.summary}` : `[${action.toolName}]`,
|
|
)
|
|
.join("\n");
|
|
const parts = [actionLog, detail.log].filter((part) => part.trim().length > 0);
|
|
return parts.join("\n\n");
|
|
}
|
|
|
|
function buildOpenCodeSubAgentTimelineItem(
|
|
activity: OpenCodeSubAgentActivityState,
|
|
): ToolCallTimelineItem {
|
|
const toolCall = activity.toolCall;
|
|
if (toolCall.detail.type !== "sub_agent") {
|
|
return toolCall;
|
|
}
|
|
const childSessionId = activity.childSessionId ?? toolCall.detail.childSessionId;
|
|
return {
|
|
...toolCall,
|
|
detail: {
|
|
...toolCall.detail,
|
|
...(childSessionId ? { childSessionId } : {}),
|
|
log: buildOpenCodeSubAgentLog(toolCall.detail, activity),
|
|
},
|
|
};
|
|
}
|
|
|
|
function registerOpenCodeSubAgentToolCall(
|
|
item: ToolCallTimelineItem,
|
|
state: OpenCodeEventTranslationState,
|
|
): ToolCallTimelineItem {
|
|
if (item.detail.type !== "sub_agent") {
|
|
return item;
|
|
}
|
|
const activity = getOpenCodeSubAgentState(item.callId, state, item);
|
|
if (item.detail.childSessionId) {
|
|
linkOpenCodeSubAgentChildSession(activity, item.detail.childSessionId, state);
|
|
}
|
|
return buildOpenCodeSubAgentTimelineItem(activity);
|
|
}
|
|
|
|
function bufferOpenCodeSubAgentChildToolPart(
|
|
part: OpenCodeToolPartEventPart,
|
|
state: OpenCodeEventTranslationState,
|
|
): void {
|
|
const maps = getOpenCodeSubAgentMaps(state);
|
|
if (maps.byCallId.size === 0) {
|
|
return;
|
|
}
|
|
const totalPending = [...maps.pendingChildToolPartsBySessionId.values()].reduce(
|
|
(total, parts) => total + parts.length,
|
|
0,
|
|
);
|
|
if (totalPending >= MAX_OPENCODE_PENDING_CHILD_TOOL_PARTS) {
|
|
return;
|
|
}
|
|
const pending = maps.pendingChildToolPartsBySessionId.get(part.sessionID) ?? [];
|
|
pending.push(part);
|
|
maps.pendingChildToolPartsBySessionId.set(part.sessionID, pending);
|
|
}
|
|
|
|
function flushOpenCodeSubAgentChildToolParts(
|
|
childSessionId: string,
|
|
state: OpenCodeEventTranslationState,
|
|
events: AgentStreamEvent[],
|
|
): void {
|
|
const maps = getOpenCodeSubAgentMaps(state);
|
|
const pending = maps.pendingChildToolPartsBySessionId.get(childSessionId);
|
|
if (!pending || pending.length === 0) {
|
|
return;
|
|
}
|
|
maps.pendingChildToolPartsBySessionId.delete(childSessionId);
|
|
for (const part of pending) {
|
|
appendOpenCodeSubAgentChildToolPart(part, state, events);
|
|
}
|
|
}
|
|
|
|
function findOnlyOpenCodeSubAgentWaitingForChild(
|
|
state: OpenCodeEventTranslationState,
|
|
): OpenCodeSubAgentActivityState | null {
|
|
const maps = getOpenCodeSubAgentMaps(state);
|
|
const candidates = [...maps.byCallId.values()].filter(
|
|
(activity) =>
|
|
activity.toolCall.status === "running" &&
|
|
activity.toolCall.detail.type === "sub_agent" &&
|
|
!activity.childSessionId,
|
|
);
|
|
return candidates.length === 1 ? (candidates[0] ?? null) : null;
|
|
}
|
|
|
|
function summarizeOpenCodeSubAgentAction(
|
|
item: ToolCallTimelineItem,
|
|
cwd: string | undefined,
|
|
): string | undefined {
|
|
const display = buildToolCallDisplayModel({
|
|
name: item.name,
|
|
status: item.status,
|
|
error: item.error,
|
|
metadata: item.metadata,
|
|
detail: item.detail,
|
|
cwd,
|
|
});
|
|
return display.summary ?? display.errorText;
|
|
}
|
|
|
|
function appendOpenCodeSubAgentAction(
|
|
activity: OpenCodeSubAgentActivityState,
|
|
item: ToolCallTimelineItem,
|
|
cwd: string | undefined,
|
|
): boolean {
|
|
const key = item.callId || `${item.name}:${activity.actions.length}`;
|
|
const existingIndex = activity.actionIndexByKey.get(key);
|
|
const summary = summarizeOpenCodeSubAgentAction(item, cwd);
|
|
|
|
if (existingIndex !== undefined) {
|
|
const action = activity.actions[existingIndex];
|
|
if (!action) {
|
|
return false;
|
|
}
|
|
const changed = action.toolName !== item.name || action.summary !== summary;
|
|
action.toolName = item.name;
|
|
if (summary) {
|
|
action.summary = summary;
|
|
} else {
|
|
delete action.summary;
|
|
}
|
|
return changed;
|
|
}
|
|
|
|
if (activity.actions.length >= MAX_OPENCODE_SUB_AGENT_ACTIONS) {
|
|
return false;
|
|
}
|
|
|
|
activity.actionIndexByKey.set(key, activity.actions.length);
|
|
activity.actions.push({
|
|
index: activity.nextActionIndex,
|
|
key,
|
|
toolName: item.name,
|
|
...(summary ? { summary } : {}),
|
|
});
|
|
activity.nextActionIndex += 1;
|
|
return true;
|
|
}
|
|
|
|
function appendOpenCodeToolCallTimelineItem(
|
|
item: ToolCallTimelineItem,
|
|
state: OpenCodeEventTranslationState,
|
|
events: AgentStreamEvent[],
|
|
): void {
|
|
const timelineItem = registerOpenCodeSubAgentToolCall(item, state);
|
|
events.push({
|
|
type: "timeline",
|
|
provider: "opencode",
|
|
item: timelineItem,
|
|
});
|
|
if (timelineItem.detail.type === "sub_agent" && timelineItem.detail.childSessionId) {
|
|
flushOpenCodeSubAgentChildToolParts(timelineItem.detail.childSessionId, state, events);
|
|
}
|
|
}
|
|
|
|
function appendOpenCodeSubAgentChildSessionLinked(
|
|
childSessionId: string,
|
|
state: OpenCodeEventTranslationState,
|
|
events: AgentStreamEvent[],
|
|
): void {
|
|
const activity = findOnlyOpenCodeSubAgentWaitingForChild(state);
|
|
if (!activity) {
|
|
return;
|
|
}
|
|
linkOpenCodeSubAgentChildSession(activity, childSessionId, state);
|
|
events.push({
|
|
type: "timeline",
|
|
provider: "opencode",
|
|
item: buildOpenCodeSubAgentTimelineItem(activity),
|
|
});
|
|
flushOpenCodeSubAgentChildToolParts(childSessionId, state, events);
|
|
}
|
|
|
|
function appendOpenCodeSubAgentChildToolPart(
|
|
part: OpenCodeToolPartEventPart,
|
|
state: OpenCodeEventTranslationState,
|
|
events: AgentStreamEvent[],
|
|
): void {
|
|
const maps = getOpenCodeSubAgentMaps(state);
|
|
const parentCallId = maps.callIdByChildSessionId.get(part.sessionID);
|
|
if (!parentCallId) {
|
|
bufferOpenCodeSubAgentChildToolPart(part, state);
|
|
return;
|
|
}
|
|
const activity = maps.byCallId.get(parentCallId);
|
|
if (!activity) {
|
|
return;
|
|
}
|
|
const parsedToolPart = OpencodeToolPartToTimelineItemSchema.safeParse(part);
|
|
if (!parsedToolPart.success || !parsedToolPart.data) {
|
|
return;
|
|
}
|
|
if (!appendOpenCodeSubAgentAction(activity, parsedToolPart.data, state.cwd)) {
|
|
return;
|
|
}
|
|
events.push({
|
|
type: "timeline",
|
|
provider: "opencode",
|
|
item: buildOpenCodeSubAgentTimelineItem(activity),
|
|
});
|
|
}
|
|
|
|
function appendOpenCodeSessionCreatedOrUpdated(
|
|
event: Extract<OpenCodeEvent, { type: "session.created" | "session.updated" }>,
|
|
state: OpenCodeEventTranslationState,
|
|
events: AgentStreamEvent[],
|
|
): void {
|
|
const info = readOpenCodeRecord(event.properties.info);
|
|
if (event.properties.info.id === state.sessionId) {
|
|
const sessionCost = readPositiveFiniteNumber(info?.cost);
|
|
if (sessionCost !== undefined) {
|
|
state.sessionTotalCostUsd = maxFiniteNumber(state.sessionTotalCostUsd, sessionCost);
|
|
state.accumulatedUsage.totalCostUsd = state.sessionTotalCostUsd;
|
|
}
|
|
events.push({
|
|
type: "thread_started",
|
|
sessionId: state.sessionId,
|
|
provider: "opencode",
|
|
});
|
|
return;
|
|
}
|
|
|
|
const parentSessionId = readNonEmptyString(info?.parentID) ?? readNonEmptyString(info?.parentId);
|
|
if (parentSessionId === state.sessionId) {
|
|
appendOpenCodeSubAgentChildSessionLinked(event.properties.info.id, state, events);
|
|
}
|
|
}
|
|
|
|
function appendOpenCodeMessageUpdated(
|
|
event: Extract<OpenCodeEvent, { type: "message.updated" }>,
|
|
state: OpenCodeEventTranslationState,
|
|
events: AgentStreamEvent[],
|
|
): void {
|
|
const info = event.properties.info;
|
|
if (info.sessionID !== state.sessionId) {
|
|
return;
|
|
}
|
|
state.messageRoles.set(info.id, info.role);
|
|
if (info.role === "user") {
|
|
appendOpenCodeUserMessageUpdated(info, state, events);
|
|
return;
|
|
}
|
|
if (info.role !== "assistant") {
|
|
return;
|
|
}
|
|
const modelLookupKey = resolveOpenCodeModelLookupKeyFromAssistantMessage(info);
|
|
if (modelLookupKey) {
|
|
const contextWindowMaxTokens = state.modelContextWindowsByModelKey?.get(modelLookupKey);
|
|
if (contextWindowMaxTokens !== undefined) {
|
|
state.onAssistantModelContextWindowResolved?.(contextWindowMaxTokens);
|
|
}
|
|
}
|
|
if (state.emittedStructuredMessageIds.has(info.id) || info.time?.completed === undefined) {
|
|
return;
|
|
}
|
|
const text = stringifyStructuredAssistantMessage(info.structured);
|
|
if (!text) {
|
|
return;
|
|
}
|
|
state.emittedStructuredMessageIds.add(info.id);
|
|
events.push({
|
|
type: "timeline",
|
|
provider: "opencode",
|
|
item: { type: "assistant_message", text },
|
|
});
|
|
}
|
|
|
|
function appendOpenCodeUserMessageUpdated(
|
|
info: Extract<OpenCodeMessage, { role: "user" }>,
|
|
state: OpenCodeEventTranslationState,
|
|
events: AgentStreamEvent[],
|
|
): void {
|
|
const text = state.pendingUserMessageText;
|
|
if (!text || text.trim().length === 0 || state.emittedUserMessageIds?.has(info.id)) {
|
|
return;
|
|
}
|
|
state.emittedUserMessageIds?.add(info.id);
|
|
events.push({
|
|
type: "timeline",
|
|
provider: "opencode",
|
|
item: { type: "user_message", text, messageId: info.id },
|
|
});
|
|
}
|
|
|
|
function appendOpenCodeMessagePartUpdated(
|
|
event: Extract<OpenCodeEvent, { type: "message.part.updated" }>,
|
|
state: OpenCodeEventTranslationState,
|
|
events: AgentStreamEvent[],
|
|
): void {
|
|
const part = event.properties.part;
|
|
if (part.sessionID !== state.sessionId) {
|
|
if (part.type === "tool") {
|
|
appendOpenCodeSubAgentChildToolPart(part, state, events);
|
|
}
|
|
return;
|
|
}
|
|
const messageRole = state.messageRoles.get(part.messageID);
|
|
state.partTypes.set(part.id, part.type);
|
|
|
|
if (part.type === "text") {
|
|
appendOpenCodeTextPart(part, messageRole, state, events);
|
|
return;
|
|
}
|
|
if (part.type === "reasoning") {
|
|
appendOpenCodeReasoningPart(part, state, events);
|
|
return;
|
|
}
|
|
if (part.type === "tool") {
|
|
const parsedToolPart = OpencodeToolPartToTimelineItemSchema.safeParse(part);
|
|
if (parsedToolPart.success && parsedToolPart.data) {
|
|
appendOpenCodeToolCallTimelineItem(parsedToolPart.data, state, events);
|
|
}
|
|
return;
|
|
}
|
|
if (part.type === "compaction") {
|
|
events.push({
|
|
type: "timeline",
|
|
provider: "opencode",
|
|
item: createCompactionTimelineItem("loading", part.auto ? "auto" : "manual"),
|
|
});
|
|
return;
|
|
}
|
|
if (part.type === "step-finish") {
|
|
const stepCost = readPositiveFiniteNumber(part.cost);
|
|
if (stepCost !== undefined) {
|
|
state.sessionTotalCostUsd = (state.sessionTotalCostUsd ?? 0) + stepCost;
|
|
}
|
|
mergeOpenCodeStepFinishUsage(state.accumulatedUsage, part, {
|
|
totalCostUsd: state.sessionTotalCostUsd,
|
|
});
|
|
if (hasNormalizedOpenCodeUsage(state.accumulatedUsage)) {
|
|
events.push({
|
|
type: "usage_updated",
|
|
provider: "opencode",
|
|
usage: { ...state.accumulatedUsage },
|
|
});
|
|
}
|
|
}
|
|
}
|
|
|
|
function appendOpenCodeTextPart(
|
|
part: Extract<
|
|
Extract<OpenCodeEvent, { type: "message.part.updated" }>["properties"]["part"],
|
|
{ type: "text" }
|
|
>,
|
|
messageRole: OpenCodeMessageRole | undefined,
|
|
state: OpenCodeEventTranslationState,
|
|
events: AgentStreamEvent[],
|
|
): void {
|
|
if (messageRole === "user") {
|
|
return;
|
|
}
|
|
if (!part.time?.end) {
|
|
return;
|
|
}
|
|
const partKey = resolvePartDedupeKey(part, "text");
|
|
if (partKey && state.streamedPartKeys.delete(partKey)) {
|
|
return;
|
|
}
|
|
if (part.text) {
|
|
events.push({
|
|
type: "timeline",
|
|
provider: "opencode",
|
|
item: { type: "assistant_message", text: part.text },
|
|
});
|
|
}
|
|
}
|
|
|
|
function appendOpenCodeReasoningPart(
|
|
part: Extract<
|
|
Extract<OpenCodeEvent, { type: "message.part.updated" }>["properties"]["part"],
|
|
{ type: "reasoning" }
|
|
>,
|
|
state: OpenCodeEventTranslationState,
|
|
events: AgentStreamEvent[],
|
|
): void {
|
|
if (!part.time.end) {
|
|
return;
|
|
}
|
|
const partKey = resolvePartDedupeKey(part, "reasoning");
|
|
if (partKey && state.streamedPartKeys.delete(partKey)) {
|
|
return;
|
|
}
|
|
if (part.text) {
|
|
events.push({
|
|
type: "timeline",
|
|
provider: "opencode",
|
|
item: { type: "reasoning", text: part.text },
|
|
});
|
|
}
|
|
}
|
|
|
|
function appendOpenCodeMessagePartDelta(
|
|
event: Extract<OpenCodeEvent, { type: "message.part.delta" }>,
|
|
state: OpenCodeEventTranslationState,
|
|
events: AgentStreamEvent[],
|
|
): void {
|
|
const { sessionID, messageID, partID, field, delta } = event.properties;
|
|
if (sessionID !== state.sessionId) {
|
|
return;
|
|
}
|
|
if (!delta || !field) {
|
|
return;
|
|
}
|
|
const messageRole = messageID ? state.messageRoles.get(messageID) : undefined;
|
|
const knownPartType = partID ? state.partTypes.get(partID) : undefined;
|
|
const isReasoning = knownPartType === "reasoning" || field === "reasoning";
|
|
|
|
if (isReasoning) {
|
|
if (partID) {
|
|
state.streamedPartKeys.add(`reasoning:${partID}`);
|
|
}
|
|
events.push({
|
|
type: "timeline",
|
|
provider: "opencode",
|
|
item: { type: "reasoning", text: delta },
|
|
});
|
|
return;
|
|
}
|
|
if (field !== "text") {
|
|
return;
|
|
}
|
|
if (messageRole === "user") {
|
|
return;
|
|
}
|
|
if (partID) {
|
|
state.streamedPartKeys.add(`text:${partID}`);
|
|
}
|
|
events.push({
|
|
type: "timeline",
|
|
provider: "opencode",
|
|
item: { type: "assistant_message", text: delta },
|
|
});
|
|
}
|
|
|
|
function appendOpenCodePermissionAsked(
|
|
event: Extract<OpenCodeEvent, { type: "permission.asked" }>,
|
|
state: OpenCodeEventTranslationState,
|
|
events: AgentStreamEvent[],
|
|
): void {
|
|
if (event.properties.sessionID !== state.sessionId) {
|
|
return;
|
|
}
|
|
const metadata = readOpenCodeRecord(event.properties.metadata);
|
|
const tool = readOpenCodeRecord(event.properties.tool);
|
|
const patterns = Array.isArray(event.properties.patterns)
|
|
? event.properties.patterns.filter((value): value is string => typeof value === "string")
|
|
: [];
|
|
const command = readPermissionField(metadata, PERMISSION_COMMAND_KEYS);
|
|
const cwd = readPermissionField(metadata, PERMISSION_CWD_KEYS);
|
|
const reason = readPermissionField(metadata, PERMISSION_REASON_KEYS);
|
|
const input = buildOpenCodePermissionInput({ patterns, metadata, tool, command });
|
|
const detail = buildOpenCodePermissionDetail({
|
|
permission: event.properties.permission,
|
|
input,
|
|
command,
|
|
cwd,
|
|
});
|
|
const description = buildOpenCodePermissionDescription({ reason, patterns });
|
|
|
|
events.push({
|
|
type: "permission_requested",
|
|
provider: "opencode",
|
|
request: {
|
|
id: event.properties.id,
|
|
provider: "opencode",
|
|
name: event.properties.permission,
|
|
kind: "tool",
|
|
title: toHumanReadablePermissionTitle(event.properties.permission),
|
|
...(description ? { description } : {}),
|
|
input,
|
|
detail,
|
|
actions: buildOpenCodePermissionActions(),
|
|
},
|
|
});
|
|
}
|
|
|
|
function appendOpenCodeQuestionAsked(
|
|
event: Extract<OpenCodeEvent, { type: "question.asked" }>,
|
|
state: OpenCodeEventTranslationState,
|
|
events: AgentStreamEvent[],
|
|
): void {
|
|
if (event.properties.sessionID !== state.sessionId) {
|
|
return;
|
|
}
|
|
const questions = event.properties.questions.flatMap((q) => {
|
|
if (!q.question || !q.header) {
|
|
return [];
|
|
}
|
|
const options =
|
|
q.options?.map((o) => ({
|
|
label: o.label,
|
|
...(o.description ? { description: o.description } : {}),
|
|
})) ?? [];
|
|
return [
|
|
{
|
|
question: q.question,
|
|
header: q.header,
|
|
options,
|
|
...(q.multiple === true ? { multiSelect: true } : {}),
|
|
},
|
|
];
|
|
});
|
|
|
|
if (questions.length === 0) {
|
|
return;
|
|
}
|
|
|
|
events.push({
|
|
type: "permission_requested",
|
|
provider: "opencode",
|
|
request: {
|
|
id: event.properties.id,
|
|
provider: "opencode",
|
|
name: "question",
|
|
kind: "question",
|
|
title: "Question",
|
|
input: { questions },
|
|
metadata: {
|
|
source: "opencode_question",
|
|
...event.properties.tool,
|
|
},
|
|
},
|
|
});
|
|
}
|
|
|
|
function appendOpenCodeSessionError(
|
|
event: Extract<OpenCodeEvent, { type: "session.error" }>,
|
|
state: OpenCodeEventTranslationState,
|
|
events: AgentStreamEvent[],
|
|
): void {
|
|
if (event.properties.sessionID !== state.sessionId) {
|
|
return;
|
|
}
|
|
resetOpenCodeTurnTrackingState(state);
|
|
const error = event.properties.error;
|
|
if (
|
|
typeof error === "object" &&
|
|
error !== null &&
|
|
"name" in error &&
|
|
error.name === "MessageAbortedError"
|
|
) {
|
|
events.push({
|
|
type: "turn_canceled",
|
|
provider: "opencode",
|
|
reason: "interrupted",
|
|
});
|
|
} else {
|
|
events.push({
|
|
type: "turn_failed",
|
|
provider: "opencode",
|
|
error: toDiagnosticErrorMessage(error),
|
|
});
|
|
}
|
|
}
|
|
|
|
function appendOpenCodeSessionStatus(
|
|
event: Extract<OpenCodeEvent, { type: "session.status" }>,
|
|
state: OpenCodeEventTranslationState,
|
|
events: AgentStreamEvent[],
|
|
): void {
|
|
if (event.properties.sessionID !== state.sessionId) {
|
|
return;
|
|
}
|
|
const { status } = event.properties;
|
|
if (status.type === "idle") {
|
|
resetOpenCodeTurnTrackingState(state);
|
|
events.push({ type: "turn_completed", provider: "opencode", usage: undefined });
|
|
return;
|
|
}
|
|
if (status.type === "retry") {
|
|
// Mirror what opencode's TUI shows: retry attempts are visible activity, not
|
|
// terminal. opencode itself never gives up — it backs off and tries again
|
|
// forever. If we silently swallow these the user sees a spinner with no
|
|
// explanation. Forwarding as a timeline error item is a no-op for old
|
|
// clients (the schema already supports it).
|
|
const message = typeof status.message === "string" ? status.message.trim() : "";
|
|
const text = message
|
|
? `Provider retry (attempt ${status.attempt}): ${message}`
|
|
: `Provider retry (attempt ${status.attempt})`;
|
|
events.push({
|
|
type: "timeline",
|
|
provider: "opencode",
|
|
item: { type: "error", message: text },
|
|
});
|
|
return;
|
|
}
|
|
// "busy" is transient — no terminal event, no surfaced activity.
|
|
}
|
|
|
|
interface Deferred<T> {
|
|
promise: Promise<T>;
|
|
resolve: (value: T) => void;
|
|
reject: (error: unknown) => void;
|
|
}
|
|
|
|
function createDeferred<T>(): Deferred<T> {
|
|
let resolve!: (value: T) => void;
|
|
let reject!: (error: unknown) => void;
|
|
const promise = new Promise<T>((res, rej) => {
|
|
resolve = res;
|
|
reject = rej;
|
|
});
|
|
return { promise, resolve, reject };
|
|
}
|
|
|
|
function unwrapOpenCodeGlobalEvent(event: unknown): OpenCodeEvent | null {
|
|
const record = readOpenCodeRecord(event);
|
|
if (!record) {
|
|
return null;
|
|
}
|
|
|
|
const payload = readOpenCodeRecord(record.payload);
|
|
if (typeof payload?.type === "string") {
|
|
return payload as unknown as OpenCodeEvent;
|
|
}
|
|
|
|
if (typeof record.type === "string") {
|
|
return record as unknown as OpenCodeEvent;
|
|
}
|
|
|
|
return null;
|
|
}
|
|
|
|
class OpenCodeAgentSession implements AgentSession {
|
|
readonly provider = "opencode" as const;
|
|
readonly capabilities = OPENCODE_CAPABILITIES;
|
|
|
|
private readonly config: OpenCodeAgentConfig;
|
|
private readonly client: OpencodeClient;
|
|
private readonly sessionId: string;
|
|
private readonly logger: Logger;
|
|
private readonly modelContextWindowsByModelKey: ReadonlyMap<string, number>;
|
|
private currentMode: string = "default";
|
|
private pendingPermissions = new Map<string, AgentPermissionRequest>();
|
|
private abortController: AbortController | null = null;
|
|
private pendingAbortPromise: Promise<void> | null = null;
|
|
private accumulatedUsage: AgentUsage = {};
|
|
private sessionTotalCostUsd: number | undefined;
|
|
private mcpConfigured = false;
|
|
private mcpSetupPromise: Promise<void> | null = null;
|
|
/** Tracks the role of each message by ID to distinguish user from assistant messages */
|
|
private messageRoles = new Map<string, OpenCodeMessageRole>();
|
|
private pendingUserMessageText: string | null = null;
|
|
private emittedUserMessageIds = new Set<string>();
|
|
/** Tracks streamed textual part IDs to suppress final full-text echoes from OpenCode. */
|
|
private streamedPartKeys = new Set<string>();
|
|
/** Tracks assistant messages already emitted from structured payloads. */
|
|
private emittedStructuredMessageIds = new Set<string>();
|
|
/** Tracks the type of each part by ID, learned from message.part.updated events. */
|
|
private partTypes = new Map<string, string>();
|
|
private availableModesCache: AgentMode[] | null = null;
|
|
private readonly subscribers = new Set<(event: AgentStreamEvent) => void>();
|
|
private nextTurnOrdinal = 0;
|
|
private activeForegroundTurnId: string | null = null;
|
|
private readonly runningToolCalls = new Map<string, ToolCallTimelineItem>();
|
|
private subAgentsByCallId = new Map<string, OpenCodeSubAgentActivityState>();
|
|
private subAgentCallIdByChildSessionId = new Map<string, string>();
|
|
private pendingChildToolPartsBySessionId = new Map<string, OpenCodeToolPartEventPart[]>();
|
|
private selectedModelContextWindowMaxTokens: number | undefined;
|
|
private releaseServer: (() => void) | null;
|
|
private eventStreamAbortController: AbortController | null = null;
|
|
private eventStreamReady: Deferred<void> | null = null;
|
|
private closed = false;
|
|
private readonly persistSession: boolean;
|
|
private deletedFromProvider = false;
|
|
constructor(
|
|
config: OpenCodeAgentConfig,
|
|
client: OpencodeClient,
|
|
sessionId: string,
|
|
logger: Logger,
|
|
modelContextWindowsByModelKey: ReadonlyMap<string, number> = new Map(),
|
|
releaseServer?: () => void,
|
|
persistSession = true,
|
|
private readonly agentId?: string,
|
|
) {
|
|
this.config = config;
|
|
this.client = client;
|
|
this.sessionId = sessionId;
|
|
this.logger = logger.child({ agentId: this.agentId });
|
|
this.modelContextWindowsByModelKey = modelContextWindowsByModelKey;
|
|
this.currentMode = normalizeOpenCodeModeId(config.modeId);
|
|
this.releaseServer = releaseServer ?? null;
|
|
this.persistSession = persistSession;
|
|
this.selectedModelContextWindowMaxTokens = this.resolveConfiguredModelContextWindowMaxTokens(
|
|
config.model,
|
|
);
|
|
this.startEventStream();
|
|
}
|
|
|
|
get id(): string | null {
|
|
return this.sessionId;
|
|
}
|
|
|
|
async getRuntimeInfo(): Promise<AgentRuntimeInfo> {
|
|
return {
|
|
provider: "opencode",
|
|
sessionId: this.sessionId,
|
|
model: this.config.model ?? null,
|
|
modeId: this.currentMode,
|
|
};
|
|
}
|
|
|
|
async setModel(modelId: string | null): Promise<void> {
|
|
const normalizedModelId =
|
|
typeof modelId === "string" && modelId.trim().length > 0 ? modelId : null;
|
|
this.config.model = normalizedModelId ?? undefined;
|
|
this.selectedModelContextWindowMaxTokens = this.resolveConfiguredModelContextWindowMaxTokens(
|
|
this.config.model,
|
|
);
|
|
}
|
|
|
|
async setThinkingOption(thinkingOptionId: string | null): Promise<void> {
|
|
const normalizedThinkingOptionId =
|
|
typeof thinkingOptionId === "string" && thinkingOptionId.trim().length > 0
|
|
? thinkingOptionId
|
|
: null;
|
|
this.config.thinkingOptionId = normalizedThinkingOptionId ?? undefined;
|
|
}
|
|
|
|
async run(prompt: AgentPromptInput, options?: AgentRunOptions): Promise<AgentRunResult> {
|
|
return runProviderTurn({
|
|
prompt,
|
|
runOptions: options,
|
|
startTurn: (p, o) => this.startTurn(p, o),
|
|
subscribe: (callback) => this.subscribe(callback),
|
|
getSessionId: () => this.sessionId,
|
|
});
|
|
}
|
|
|
|
async interrupt(): Promise<void> {
|
|
const turnId = this.activeForegroundTurnId;
|
|
const turnAbortController = this.abortController;
|
|
turnAbortController?.abort();
|
|
// COMPAT(opencodeSlowAbort): OpenCode 1.14.42+ blocks session.abort until
|
|
// the running tool actually stops, which can be tens of seconds for
|
|
// long-running tools. Cap the wait so the user-visible cancel lands
|
|
// quickly while still giving OpenCode a chance to confirm the abort
|
|
// cleanly. Drop the timeout once upstream returns abort acknowledgement
|
|
// before tool teardown.
|
|
const abortPromise = this.beginSessionAbort(turnId, "interrupt");
|
|
await withTimeout(abortPromise, 2_000, "OpenCode session.abort").catch((error) => {
|
|
this.logger.warn(
|
|
{ err: error, sessionId: this.sessionId, turnId },
|
|
"OpenCode session.abort exceeded the cancel cap; proceeding with local cancel",
|
|
);
|
|
});
|
|
if (turnId) {
|
|
this.finishForegroundTurn(
|
|
{ type: "turn_canceled", provider: "opencode", reason: "interrupted" },
|
|
turnId,
|
|
);
|
|
}
|
|
}
|
|
|
|
async revertBoth(input: { messageId: string }): Promise<void> {
|
|
await revertOpenCodeConversationAndFiles({
|
|
client: this.client,
|
|
sessionId: this.sessionId,
|
|
cwd: this.config.cwd,
|
|
messageId: input.messageId,
|
|
});
|
|
}
|
|
|
|
private beginSessionAbort(turnId: string | null, reason: string): Promise<void> {
|
|
const abortPromise = this.client.session
|
|
.abort({
|
|
sessionID: this.sessionId,
|
|
directory: this.config.cwd,
|
|
})
|
|
.then(() => undefined)
|
|
.catch((error) => {
|
|
this.logger.warn(
|
|
{ err: error, sessionId: this.sessionId, turnId, reason },
|
|
"OpenCode session.abort rejected",
|
|
);
|
|
});
|
|
const trackedAbortPromise = abortPromise.finally(() => {
|
|
if (this.pendingAbortPromise === trackedAbortPromise) {
|
|
this.pendingAbortPromise = null;
|
|
}
|
|
});
|
|
this.pendingAbortPromise = trackedAbortPromise;
|
|
return trackedAbortPromise;
|
|
}
|
|
|
|
private async awaitPendingAbortBeforeStartingTurn(): Promise<void> {
|
|
const pendingAbortPromise = this.pendingAbortPromise;
|
|
if (!pendingAbortPromise) {
|
|
return;
|
|
}
|
|
|
|
await withTimeout(
|
|
pendingAbortPromise,
|
|
OPENCODE_PENDING_ABORT_START_TIMEOUT_MS,
|
|
"OpenCode pending session.abort",
|
|
).catch((error) => {
|
|
this.logger.warn(
|
|
{ err: error, sessionId: this.sessionId },
|
|
"OpenCode session.abort was still pending before starting the next turn",
|
|
);
|
|
});
|
|
}
|
|
|
|
async startTurn(
|
|
prompt: AgentPromptInput,
|
|
options?: AgentRunOptions,
|
|
): Promise<{ turnId: string }> {
|
|
if (this.activeForegroundTurnId) {
|
|
throw new Error("A foreground turn is already active");
|
|
}
|
|
await this.awaitPendingAbortBeforeStartingTurn();
|
|
|
|
this.runningToolCalls.clear();
|
|
this.subAgentsByCallId.clear();
|
|
this.subAgentCallIdByChildSessionId.clear();
|
|
this.pendingChildToolPartsBySessionId.clear();
|
|
const turnAbortController = new AbortController();
|
|
this.abortController = turnAbortController;
|
|
await this.ensureMcpServersConfigured();
|
|
const contextWindowMaxTokens = this.resolveSelectedModelContextWindowMaxTokens();
|
|
this.accumulatedUsage = contextWindowMaxTokens !== undefined ? { contextWindowMaxTokens } : {};
|
|
|
|
const parts = buildOpenCodePromptParts(prompt);
|
|
this.pendingUserMessageText = buildOpenCodeUserTimelineText(prompt);
|
|
const model = this.parseModel(this.config.model);
|
|
const thinkingOptionId = this.config.thinkingOptionId;
|
|
const effectiveVariant = thinkingOptionId ?? undefined;
|
|
const effectiveMode = resolveOpenCodeRuntimeAgentId(this.currentMode);
|
|
|
|
try {
|
|
await this.ensureEventStreamReady();
|
|
} catch (error) {
|
|
if (this.abortController === turnAbortController) {
|
|
this.abortController = null;
|
|
}
|
|
throw error;
|
|
}
|
|
|
|
const turnId = this.createTurnId();
|
|
this.activeForegroundTurnId = turnId;
|
|
this.notifySubscribers({ type: "turn_started", provider: "opencode" }, turnId);
|
|
|
|
const slashCommand = await this.resolveSlashCommandInvocation(prompt);
|
|
if (slashCommand) {
|
|
if (slashCommand.commandName === "compact" || slashCommand.commandName === "summarize") {
|
|
void this.client.session
|
|
.summarize({
|
|
sessionID: this.sessionId,
|
|
directory: this.config.cwd,
|
|
...(model ? { providerID: model.providerID, modelID: model.modelID } : {}),
|
|
})
|
|
.then((response) => {
|
|
if (response.error) {
|
|
this.finishForegroundTurn(
|
|
{
|
|
type: "turn_failed",
|
|
provider: "opencode",
|
|
error: toDiagnosticErrorMessage(response.error),
|
|
},
|
|
turnId,
|
|
);
|
|
} else {
|
|
this.finishForegroundTurn(
|
|
{ type: "turn_completed", provider: "opencode", usage: undefined },
|
|
turnId,
|
|
);
|
|
}
|
|
return;
|
|
})
|
|
.catch((error) => {
|
|
this.finishForegroundTurn(
|
|
{
|
|
type: "turn_failed",
|
|
provider: "opencode",
|
|
error: toDiagnosticErrorMessage(error),
|
|
},
|
|
turnId,
|
|
);
|
|
});
|
|
return { turnId };
|
|
}
|
|
|
|
// command() is only dispatch acknowledgement. OpenCode session events are
|
|
// the source of truth for when the command turn becomes idle or fails.
|
|
void this.client.session
|
|
.command({
|
|
sessionID: this.sessionId,
|
|
directory: this.config.cwd,
|
|
command: slashCommand.commandName,
|
|
arguments: slashCommand.args ?? "",
|
|
...(this.config.model ? { model: this.config.model } : {}),
|
|
...(effectiveMode ? { agent: effectiveMode } : {}),
|
|
...(effectiveVariant ? { variant: effectiveVariant } : {}),
|
|
})
|
|
.then((response) => {
|
|
if (response.error) {
|
|
if (isOpenCodeHeadersTimeoutFailure(response.error)) {
|
|
this.logger.warn(
|
|
{
|
|
err: response.error,
|
|
commandName: slashCommand.commandName,
|
|
turnId,
|
|
},
|
|
"OpenCode slash command hit a header timeout; waiting for SSE terminal event",
|
|
);
|
|
return;
|
|
}
|
|
const errorMsg = toDiagnosticErrorMessage(response.error);
|
|
this.finishForegroundTurn(
|
|
{ type: "turn_failed", provider: "opencode", error: errorMsg },
|
|
turnId,
|
|
);
|
|
}
|
|
return;
|
|
})
|
|
.catch((err) => {
|
|
if (isOpenCodeHeadersTimeoutFailure(err)) {
|
|
this.logger.warn(
|
|
{
|
|
err,
|
|
commandName: slashCommand.commandName,
|
|
turnId,
|
|
},
|
|
"OpenCode slash command hit a header timeout; waiting for SSE terminal event",
|
|
);
|
|
return;
|
|
}
|
|
this.finishForegroundTurn(
|
|
{ type: "turn_failed", provider: "opencode", error: toDiagnosticErrorMessage(err) },
|
|
turnId,
|
|
);
|
|
});
|
|
} else {
|
|
// Wrap in an async IIFE so a synchronous throw from promptAsync (e.g.
|
|
// SDK input validation) is caught alongside async rejections. A plain
|
|
// `.then().catch()` chain would let a sync throw escape unhandled.
|
|
void (async () => {
|
|
this.traceOpenCode("provider.opencode.prompt_async.start", {
|
|
turnId,
|
|
sessionId: this.sessionId,
|
|
model,
|
|
effectiveMode,
|
|
effectiveVariant,
|
|
partTypes: parts.map((p) => p.type),
|
|
});
|
|
try {
|
|
const systemPrompt = composeSystemPromptParts(
|
|
this.config.systemPrompt,
|
|
this.config.daemonAppendSystemPrompt,
|
|
);
|
|
const promptResponse = await this.client.session.promptAsync({
|
|
sessionID: this.sessionId,
|
|
directory: this.config.cwd,
|
|
parts,
|
|
...(options?.outputSchema
|
|
? {
|
|
format: {
|
|
type: "json_schema" as const,
|
|
schema: options.outputSchema as Record<string, unknown>,
|
|
},
|
|
}
|
|
: {}),
|
|
...(systemPrompt ? { system: systemPrompt } : {}),
|
|
...(model ? { model } : {}),
|
|
...(effectiveMode ? { agent: effectiveMode } : {}),
|
|
...(effectiveVariant ? { variant: effectiveVariant } : {}),
|
|
});
|
|
this.traceOpenCode("provider.opencode.prompt_async.response", {
|
|
turnId,
|
|
hasError: promptResponse.error !== undefined,
|
|
error: promptResponse.error,
|
|
data: promptResponse.data,
|
|
});
|
|
if (promptResponse.error) {
|
|
this.finishForegroundTurn(
|
|
{
|
|
type: "turn_failed",
|
|
provider: "opencode",
|
|
error: toDiagnosticErrorMessage(promptResponse.error),
|
|
},
|
|
turnId,
|
|
);
|
|
}
|
|
} catch (error) {
|
|
this.traceOpenCode("provider.opencode.prompt_async.throw", {
|
|
turnId,
|
|
error:
|
|
error instanceof Error
|
|
? { name: error.name, message: error.message, stack: error.stack }
|
|
: String(error),
|
|
});
|
|
this.finishForegroundTurn(
|
|
{
|
|
type: "turn_failed",
|
|
provider: "opencode",
|
|
error: toDiagnosticErrorMessage(error),
|
|
},
|
|
turnId,
|
|
);
|
|
}
|
|
})();
|
|
}
|
|
|
|
return { turnId };
|
|
}
|
|
subscribe(callback: (event: AgentStreamEvent) => void): () => void {
|
|
this.subscribers.add(callback);
|
|
return () => {
|
|
this.subscribers.delete(callback);
|
|
};
|
|
}
|
|
|
|
private startEventStream(): void {
|
|
void this.ensureEventStreamReady().catch((error) => {
|
|
this.logger.warn({ err: error, sessionId: this.sessionId }, "OpenCode event stream failed");
|
|
});
|
|
}
|
|
|
|
private ensureEventStreamReady(): Promise<void> {
|
|
if (this.eventStreamReady) {
|
|
return this.eventStreamReady.promise;
|
|
}
|
|
|
|
const eventStreamAbortController = new AbortController();
|
|
const eventStreamReady = createDeferred<void>();
|
|
this.eventStreamAbortController = eventStreamAbortController;
|
|
this.eventStreamReady = eventStreamReady;
|
|
void this.consumeEventStream(eventStreamAbortController, eventStreamReady).finally(() => {
|
|
if (this.eventStreamAbortController === eventStreamAbortController) {
|
|
this.eventStreamAbortController = null;
|
|
this.eventStreamReady = null;
|
|
}
|
|
});
|
|
|
|
return eventStreamReady.promise;
|
|
}
|
|
|
|
private async consumeEventStream(
|
|
eventStreamAbortController: AbortController,
|
|
eventStreamReady: Deferred<void>,
|
|
): Promise<void> {
|
|
this.traceOpenCode("provider.opencode.subscribe.start", {
|
|
sessionId: this.sessionId,
|
|
cwd: this.config.cwd,
|
|
});
|
|
let eventStreamReadyResolved = false;
|
|
try {
|
|
const result = await this.client.global.event({
|
|
signal: eventStreamAbortController.signal,
|
|
sseMaxRetryAttempts: 0,
|
|
});
|
|
eventStreamReadyResolved = true;
|
|
this.traceOpenCode("provider.opencode.subscribe.ready", {
|
|
sessionId: this.sessionId,
|
|
});
|
|
eventStreamReady.resolve();
|
|
|
|
let eventCount = 0;
|
|
for await (const rawEvent of result.stream) {
|
|
eventCount += 1;
|
|
await this.consumeOpenCodeStreamEvent({ rawEvent, eventCount });
|
|
}
|
|
|
|
this.traceOpenCode("provider.opencode.stream.eof", {
|
|
eventCount,
|
|
aborted: eventStreamAbortController.signal.aborted,
|
|
activeTurnId: this.activeForegroundTurnId,
|
|
});
|
|
|
|
if (!eventStreamAbortController.signal.aborted) {
|
|
if (!eventStreamReadyResolved) {
|
|
eventStreamReady.reject(new Error("OpenCode event stream ended before it became ready"));
|
|
}
|
|
const activeTurnId = this.activeForegroundTurnId;
|
|
if (activeTurnId) {
|
|
this.traceOpenCode("provider.opencode.turn.fail_eof", {
|
|
turnId: activeTurnId,
|
|
eventCount,
|
|
});
|
|
this.finishForegroundTurn(
|
|
{
|
|
type: "turn_failed",
|
|
provider: "opencode",
|
|
error: "OpenCode event stream ended before the turn reached a terminal state",
|
|
},
|
|
activeTurnId,
|
|
);
|
|
}
|
|
}
|
|
} catch (error) {
|
|
this.traceOpenCode("provider.opencode.subscribe.error", {
|
|
turnId: this.activeForegroundTurnId ?? undefined,
|
|
error:
|
|
error instanceof Error ? { name: error.name, message: error.message } : String(error),
|
|
});
|
|
if (!eventStreamReadyResolved) {
|
|
eventStreamReady.reject(error);
|
|
}
|
|
const activeTurnId = this.activeForegroundTurnId;
|
|
if (!eventStreamAbortController.signal.aborted && activeTurnId) {
|
|
this.finishForegroundTurn(
|
|
{
|
|
type: "turn_failed",
|
|
provider: "opencode",
|
|
error: toDiagnosticErrorMessage(error),
|
|
},
|
|
activeTurnId,
|
|
);
|
|
}
|
|
}
|
|
}
|
|
|
|
private async consumeOpenCodeStreamEvent(params: {
|
|
rawEvent: unknown;
|
|
eventCount: number;
|
|
}): Promise<void> {
|
|
const { rawEvent, eventCount } = params;
|
|
const turnId = this.activeForegroundTurnId;
|
|
const event = unwrapOpenCodeGlobalEvent(rawEvent);
|
|
this.traceOpenCode("provider.opencode.raw_event", {
|
|
turnId: turnId ?? undefined,
|
|
n: eventCount,
|
|
type: event?.type,
|
|
rawType: readOpenCodeRecord(rawEvent)?.type,
|
|
directory: readOpenCodeRecord(rawEvent)?.directory,
|
|
rawEvent,
|
|
properties: event?.properties,
|
|
});
|
|
if (!event) {
|
|
return;
|
|
}
|
|
if (!turnId) {
|
|
this.traceOpenCode("provider.opencode.event.skip", {
|
|
n: eventCount,
|
|
reason: "no_active_turn",
|
|
type: event.type,
|
|
});
|
|
return;
|
|
}
|
|
const translated = await this.translateEvent(event);
|
|
this.traceOpenCode("provider.opencode.parsed_event", {
|
|
turnId,
|
|
n: eventCount,
|
|
count: translated.length,
|
|
types: translated.map((t) => t.type),
|
|
events: translated,
|
|
});
|
|
|
|
for (const e of translated) {
|
|
if (this.activeForegroundTurnId !== turnId) {
|
|
this.traceOpenCode("provider.opencode.parsed_event.skip_active", { turnId, type: e.type });
|
|
return;
|
|
}
|
|
if (e.type === "timeline" && e.item.type === "tool_call") {
|
|
this.trackToolCall(e.item);
|
|
}
|
|
const terminalEvent = toTerminalTurnEvent(e);
|
|
if (terminalEvent) {
|
|
this.traceOpenCode("provider.opencode.event.terminal", {
|
|
turnId,
|
|
type: terminalEvent.type,
|
|
});
|
|
this.finishForegroundTurn(terminalEvent, turnId);
|
|
return;
|
|
}
|
|
this.notifySubscribers(e, turnId);
|
|
}
|
|
}
|
|
|
|
private finishForegroundTurn(
|
|
event: Extract<AgentStreamEvent, { type: "turn_completed" | "turn_failed" | "turn_canceled" }>,
|
|
turnId: string,
|
|
): void {
|
|
this.traceOpenCode("provider.opencode.finish_foreground_turn", {
|
|
turnId,
|
|
activeTurnId: this.activeForegroundTurnId,
|
|
type: event.type,
|
|
error: event.type === "turn_failed" ? event.error : undefined,
|
|
reason: event.type === "turn_canceled" ? event.reason : undefined,
|
|
});
|
|
if (this.activeForegroundTurnId !== turnId) {
|
|
return;
|
|
}
|
|
if (event.type === "turn_canceled" || event.type === "turn_failed") {
|
|
this.synthesizeInterruptedToolCalls(turnId);
|
|
} else {
|
|
this.runningToolCalls.clear();
|
|
}
|
|
this.pendingUserMessageText = null;
|
|
this.activeForegroundTurnId = null;
|
|
this.abortController = null;
|
|
this.notifySubscribers(event, turnId);
|
|
}
|
|
|
|
private trackToolCall(item: ToolCallTimelineItem): void {
|
|
if (item.status === "running") {
|
|
this.runningToolCalls.set(item.callId, item);
|
|
return;
|
|
}
|
|
this.runningToolCalls.delete(item.callId);
|
|
}
|
|
|
|
private synthesizeInterruptedToolCalls(turnId: string): void {
|
|
for (const item of this.runningToolCalls.values()) {
|
|
const error = { message: "Tool execution aborted" };
|
|
this.notifySubscribers(
|
|
{
|
|
type: "timeline",
|
|
provider: "opencode",
|
|
item: {
|
|
...item,
|
|
status: "failed",
|
|
error,
|
|
detail:
|
|
item.detail.type === "sub_agent"
|
|
? {
|
|
...item.detail,
|
|
log: [item.detail.log, error.message]
|
|
.filter((entry) => entry.trim().length > 0)
|
|
.join("\n"),
|
|
}
|
|
: item.detail,
|
|
},
|
|
},
|
|
turnId,
|
|
);
|
|
}
|
|
this.runningToolCalls.clear();
|
|
}
|
|
|
|
private notifySubscribers(event: AgentStreamEvent, turnIdOverride?: string): void {
|
|
if (this.closed) {
|
|
return;
|
|
}
|
|
const turnId = turnIdOverride ?? this.activeForegroundTurnId;
|
|
const tagged = turnId ? { ...event, turnId } : event;
|
|
this.traceOpenCode("provider.opencode.event_emit", {
|
|
turnId: getAgentStreamEventTurnId(tagged),
|
|
event: tagged,
|
|
});
|
|
for (const callback of this.subscribers) {
|
|
try {
|
|
callback(tagged);
|
|
} catch {
|
|
// Subscriber callback error isolation
|
|
}
|
|
}
|
|
}
|
|
|
|
private createTurnId(): string {
|
|
return `opencode-turn-${this.nextTurnOrdinal++}`;
|
|
}
|
|
|
|
private traceOpenCode(msg: OpenCodeTraceMessage, data: OpenCodeTraceData = {}): void {
|
|
this.logger.trace(
|
|
{
|
|
agentId: this.agentId,
|
|
provider: "opencode",
|
|
sessionId: this.sessionId,
|
|
turnId: data.turnId ?? this.activeForegroundTurnId ?? undefined,
|
|
...data,
|
|
},
|
|
msg,
|
|
);
|
|
}
|
|
|
|
async *streamHistory(): AsyncGenerator<AgentStreamEvent> {
|
|
const sessionResponse = await this.client.session.get({
|
|
sessionID: this.sessionId,
|
|
directory: this.config.cwd,
|
|
});
|
|
const response = await this.client.session.messages({
|
|
sessionID: this.sessionId,
|
|
directory: this.config.cwd,
|
|
});
|
|
|
|
if (response.error || !response.data) {
|
|
return;
|
|
}
|
|
|
|
const messages = filterOpenCodeRevertedMessages(
|
|
response.data,
|
|
sessionResponse.error ? null : sessionResponse.data?.revert,
|
|
);
|
|
for (const message of messages) {
|
|
for (const event of buildOpenCodeReplayTimelineEvents(message)) {
|
|
yield event;
|
|
}
|
|
}
|
|
}
|
|
|
|
async getAvailableModes(): Promise<AgentMode[]> {
|
|
if (this.availableModesCache) {
|
|
return this.availableModesCache;
|
|
}
|
|
|
|
const response = await this.client.app.agents({
|
|
directory: this.config.cwd,
|
|
});
|
|
const agents = response.error || !response.data ? [] : response.data;
|
|
|
|
const discoveredModes = agents.filter(isSelectableOpenCodeAgent).map((agent) => ({
|
|
id: agent.name,
|
|
label: agent.name.charAt(0).toUpperCase() + agent.name.slice(1),
|
|
description:
|
|
typeof agent.description === "string" && agent.description.trim().length > 0
|
|
? agent.description.trim()
|
|
: DEFAULT_MODES.find((mode) => mode.id === agent.name)?.description,
|
|
}));
|
|
|
|
this.availableModesCache = mergeOpenCodeModes(discoveredModes);
|
|
return this.availableModesCache;
|
|
}
|
|
|
|
async getCurrentMode(): Promise<string | null> {
|
|
return this.currentMode;
|
|
}
|
|
|
|
async listCommands(): Promise<AgentSlashCommand[]> {
|
|
return await listOpenCodeCommandsFromSdk(this.client, this.config.cwd);
|
|
}
|
|
|
|
async setMode(modeId: string): Promise<void> {
|
|
this.currentMode = normalizeOpenCodeModeId(modeId);
|
|
}
|
|
|
|
getPendingPermissions(): AgentPermissionRequest[] {
|
|
return Array.from(this.pendingPermissions.values());
|
|
}
|
|
|
|
async respondToPermission(requestId: string, response: AgentPermissionResponse): Promise<void> {
|
|
const pending = this.pendingPermissions.get(requestId);
|
|
if (!pending) {
|
|
throw new Error(`No pending permission request with id '${requestId}'`);
|
|
}
|
|
|
|
if (pending.kind === "question") {
|
|
if (response.behavior === "deny") {
|
|
await this.client.question.reject({
|
|
requestID: requestId,
|
|
directory: this.config.cwd,
|
|
});
|
|
} else {
|
|
const answersRecord = readOpenCodeRecord(response.updatedInput?.answers);
|
|
const questions = Array.isArray(pending.input?.questions) ? pending.input.questions : [];
|
|
const answers = questions.map((item) => {
|
|
const header = readNonEmptyString(readOpenCodeRecord(item)?.header);
|
|
const rawAnswer = header ? readNonEmptyString(answersRecord?.[header]) : null;
|
|
if (!rawAnswer) {
|
|
return [];
|
|
}
|
|
return rawAnswer
|
|
.split(",")
|
|
.map((entry) => entry.trim())
|
|
.filter((entry) => entry.length > 0);
|
|
});
|
|
|
|
await this.client.question.reply({
|
|
requestID: requestId,
|
|
directory: this.config.cwd,
|
|
answers,
|
|
});
|
|
}
|
|
|
|
this.pendingPermissions.delete(requestId);
|
|
return;
|
|
}
|
|
|
|
const reply = resolveOpenCodePermissionReply(response);
|
|
await this.client.permission.reply({
|
|
requestID: requestId,
|
|
directory: this.config.cwd,
|
|
reply,
|
|
message: response.behavior === "deny" ? response.message : undefined,
|
|
});
|
|
|
|
this.pendingPermissions.delete(requestId);
|
|
}
|
|
|
|
describePersistence(): AgentPersistenceHandle | null {
|
|
return {
|
|
provider: "opencode",
|
|
sessionId: this.sessionId,
|
|
nativeHandle: this.sessionId,
|
|
metadata: {
|
|
cwd: this.config.cwd,
|
|
...(this.config.modeId ? { modeId: this.config.modeId } : {}),
|
|
...(this.config.model ? { model: this.config.model } : {}),
|
|
},
|
|
};
|
|
}
|
|
|
|
async close(): Promise<void> {
|
|
try {
|
|
// Flip closed before clearing subscribers so any event the SDK delivers
|
|
// after the abort (between here and subscribers.clear) is swallowed by
|
|
// notifySubscribers instead of bubbling through provider-runner as an
|
|
// unhandled rejection in whichever test the daemon hops to next.
|
|
this.closed = true;
|
|
this.abortController?.abort();
|
|
this.eventStreamAbortController?.abort();
|
|
this.eventStreamAbortController = null;
|
|
this.eventStreamReady = null;
|
|
this.subscribers.clear();
|
|
await reconcileOpenCodeSessionClose({
|
|
client: this.client,
|
|
sessionId: this.sessionId,
|
|
directory: this.config.cwd,
|
|
logger: this.logger,
|
|
});
|
|
await this.deleteProviderSessionIfEphemeral();
|
|
this.activeForegroundTurnId = null;
|
|
} finally {
|
|
this.releaseServer?.();
|
|
this.releaseServer = null;
|
|
}
|
|
}
|
|
|
|
private async deleteProviderSessionIfEphemeral(): Promise<void> {
|
|
if (this.persistSession || this.deletedFromProvider) {
|
|
return;
|
|
}
|
|
this.deletedFromProvider = true;
|
|
try {
|
|
const response = await this.client.session.delete({
|
|
sessionID: this.sessionId,
|
|
directory: this.config.cwd,
|
|
});
|
|
if (response.error) {
|
|
throw new Error(`OpenCode session.delete failed: ${JSON.stringify(response.error)}`);
|
|
}
|
|
} catch (error) {
|
|
this.logger.debug(
|
|
{ err: error, sessionId: this.sessionId },
|
|
"Failed to delete non-persistent OpenCode session",
|
|
);
|
|
}
|
|
}
|
|
|
|
private parseSlashCommandInput(text: string): { commandName: string; args?: string } | null {
|
|
const trimmed = text.trim();
|
|
if (!trimmed.startsWith("/") || trimmed.length <= 1) {
|
|
return null;
|
|
}
|
|
const withoutPrefix = trimmed.slice(1);
|
|
const firstWhitespaceIdx = withoutPrefix.search(/\s/);
|
|
const commandName =
|
|
firstWhitespaceIdx === -1 ? withoutPrefix : withoutPrefix.slice(0, firstWhitespaceIdx);
|
|
if (!commandName || commandName.includes("/")) {
|
|
return null;
|
|
}
|
|
const rawArgs =
|
|
firstWhitespaceIdx === -1 ? "" : withoutPrefix.slice(firstWhitespaceIdx + 1).trim();
|
|
return rawArgs.length > 0 ? { commandName, args: rawArgs } : { commandName };
|
|
}
|
|
|
|
private async resolveSlashCommandInvocation(
|
|
prompt: AgentPromptInput,
|
|
): Promise<{ commandName: string; args?: string } | null> {
|
|
if (typeof prompt !== "string") {
|
|
return null;
|
|
}
|
|
const parsed = this.parseSlashCommandInput(prompt);
|
|
if (!parsed) {
|
|
return null;
|
|
}
|
|
try {
|
|
const commands = await this.listCommands();
|
|
return commands.some((command) => command.name === parsed.commandName) ? parsed : null;
|
|
} catch (error) {
|
|
this.logger.warn(
|
|
{ err: error, commandName: parsed.commandName },
|
|
"Failed to resolve slash command; falling back to plain prompt input",
|
|
);
|
|
return null;
|
|
}
|
|
}
|
|
|
|
private parseModel(model?: string): { providerID: string; modelID: string } | undefined {
|
|
if (!model) {
|
|
return undefined;
|
|
}
|
|
const parts = model.split("/");
|
|
if (parts.length >= 2) {
|
|
return { providerID: parts[0], modelID: parts.slice(1).join("/") };
|
|
}
|
|
return { providerID: "opencode", modelID: model };
|
|
}
|
|
|
|
private async ensureMcpServersConfigured(): Promise<void> {
|
|
if (this.mcpConfigured) {
|
|
return;
|
|
}
|
|
|
|
const mcpServers = this.config.mcpServers;
|
|
if (!mcpServers || Object.keys(mcpServers).length === 0) {
|
|
this.mcpConfigured = true;
|
|
return;
|
|
}
|
|
|
|
if (!this.mcpSetupPromise) {
|
|
this.mcpSetupPromise = this.configureMcpServers(mcpServers);
|
|
}
|
|
|
|
try {
|
|
await this.mcpSetupPromise;
|
|
this.mcpConfigured = true;
|
|
} catch (error) {
|
|
this.mcpSetupPromise = null;
|
|
throw error;
|
|
}
|
|
}
|
|
|
|
private async configureMcpServers(mcpServers: Record<string, McpServerConfig>): Promise<void> {
|
|
await Promise.all(
|
|
Object.entries(mcpServers).map(([name, serverConfig]) =>
|
|
this.registerMcpServer(name, toOpenCodeMcpConfig(serverConfig)),
|
|
),
|
|
);
|
|
}
|
|
|
|
private async registerMcpServer(name: string, config: OpenCodeMcpConfig): Promise<void> {
|
|
await this.runMcpOperation("add", name, () =>
|
|
this.client.mcp.add({
|
|
directory: this.config.cwd,
|
|
name,
|
|
config,
|
|
}),
|
|
);
|
|
}
|
|
|
|
private async runMcpOperation(
|
|
operation: "add",
|
|
name: string,
|
|
run: () => Promise<{ error?: unknown }>,
|
|
): Promise<void> {
|
|
const response = await run();
|
|
const error = response.error;
|
|
if (!error) {
|
|
return;
|
|
}
|
|
|
|
if (isAlreadyPresentMcpError(error)) {
|
|
return;
|
|
}
|
|
|
|
throw new Error(
|
|
`Failed to ${operation} OpenCode MCP server '${name}': ${toDiagnosticErrorMessage(error)}`,
|
|
);
|
|
}
|
|
|
|
private async translateEvent(event: OpenCodeEvent): Promise<AgentStreamEvent[]> {
|
|
const translated = translateOpenCodeEvent(event, {
|
|
sessionId: this.sessionId,
|
|
cwd: this.config.cwd,
|
|
messageRoles: this.messageRoles,
|
|
pendingUserMessageText: this.pendingUserMessageText,
|
|
emittedUserMessageIds: this.emittedUserMessageIds,
|
|
accumulatedUsage: this.accumulatedUsage,
|
|
sessionTotalCostUsd: this.sessionTotalCostUsd,
|
|
streamedPartKeys: this.streamedPartKeys,
|
|
emittedStructuredMessageIds: this.emittedStructuredMessageIds,
|
|
partTypes: this.partTypes,
|
|
subAgentsByCallId: this.subAgentsByCallId,
|
|
subAgentCallIdByChildSessionId: this.subAgentCallIdByChildSessionId,
|
|
pendingChildToolPartsBySessionId: this.pendingChildToolPartsBySessionId,
|
|
modelContextWindowsByModelKey: this.modelContextWindowsByModelKey,
|
|
onAssistantModelContextWindowResolved: (contextWindowMaxTokens) => {
|
|
this.accumulatedUsage.contextWindowMaxTokens = contextWindowMaxTokens;
|
|
if (!this.config.model) {
|
|
this.selectedModelContextWindowMaxTokens = contextWindowMaxTokens;
|
|
}
|
|
},
|
|
});
|
|
|
|
const events: AgentStreamEvent[] = [];
|
|
if (typeof this.accumulatedUsage.totalCostUsd === "number") {
|
|
this.sessionTotalCostUsd = maxFiniteNumber(
|
|
this.sessionTotalCostUsd,
|
|
this.accumulatedUsage.totalCostUsd,
|
|
);
|
|
}
|
|
|
|
for (const translatedEvent of translated) {
|
|
if (translatedEvent.type === "permission_requested") {
|
|
const autoApproved = await this.tryAutoApproveToolPermission(translatedEvent.request);
|
|
if (autoApproved) {
|
|
continue;
|
|
}
|
|
this.pendingPermissions.set(translatedEvent.request.id, translatedEvent.request);
|
|
}
|
|
if (translatedEvent.type === "turn_completed") {
|
|
if (hasNormalizedOpenCodeUsage(this.accumulatedUsage)) {
|
|
translatedEvent.usage = this.accumulatedUsage;
|
|
}
|
|
const contextWindowMaxTokens = this.resolveSelectedModelContextWindowMaxTokens();
|
|
this.accumulatedUsage =
|
|
contextWindowMaxTokens !== undefined ? { contextWindowMaxTokens } : {};
|
|
}
|
|
events.push(translatedEvent);
|
|
}
|
|
|
|
return events;
|
|
}
|
|
|
|
private async tryAutoApproveToolPermission(request: AgentPermissionRequest): Promise<boolean> {
|
|
if (this.currentMode !== OPENCODE_FULL_ACCESS_MODE_ID || request.kind !== "tool") {
|
|
return false;
|
|
}
|
|
|
|
try {
|
|
await this.client.permission.reply({
|
|
requestID: request.id,
|
|
directory: this.config.cwd,
|
|
reply: "once",
|
|
});
|
|
return true;
|
|
} catch (error) {
|
|
this.logger.warn(
|
|
{ err: error, requestId: request.id },
|
|
"Failed to auto-approve OpenCode tool permission",
|
|
);
|
|
return false;
|
|
}
|
|
}
|
|
|
|
private resolveSelectedModelContextWindowMaxTokens(): number | undefined {
|
|
return this.selectedModelContextWindowMaxTokens;
|
|
}
|
|
|
|
private resolveConfiguredModelContextWindowMaxTokens(
|
|
modelId: string | undefined,
|
|
): number | undefined {
|
|
const modelLookupKey = parseOpenCodeModelLookupKey(modelId);
|
|
if (!modelLookupKey) {
|
|
return undefined;
|
|
}
|
|
return this.modelContextWindowsByModelKey.get(modelLookupKey);
|
|
}
|
|
}
|