@gajae-code/coding-agent 0.8.1 → 0.9.0

package/src/sdk.ts

@@ -4,7 +4,7 @@ import {
 	type AgentMessage,
 	type AgentTelemetryConfig,
 	type AgentTool,
-	AppendOnlyContextManager,
+	type AppendOnlyContextManager,
 	INTENT_FIELD,
 	type ThinkingLevel,
 } from "@gajae-code/agent-core";
@@ -31,7 +31,11 @@ import {
 	prompt,
 	Snowflake,
 } from "@gajae-code/utils";
-
+import {
+	createAppendOnlyContextManager,
+	providerSupportsAppendOnlyAuto,
+	resolveAppendOnlyMode,
+} from "./append-only-mode";
 import { type AsyncJob, AsyncJobManager, isBackgroundJobSupportEnabled, jobElapsedMs } from "./async";
 import { loadCapability } from "./capability";
 import { type Rule, ruleCapability, setActiveRules } from "./capability/rule";
@@ -615,23 +619,12 @@ function registerJsVmCleanup(): void {
 	postmortem.register("js-vm-cleanup", disposeAllVmContexts);
 }
 
-/**
- * Resolve whether to enable append-only context mode based on the setting and provider.
- *
- * - `"on"` → always enable
- * - `"off"` → never enable
- * - `"auto"` → enable for DeepSeek (prefix-caching provider)
+/*
+ * Append-only context-mode resolution + manager construction live in
+ * ./append-only-mode so the initial build, the runtime model/setting-change
+ * path, and the status UI share one implementation. Re-exported for importers/tests.
  */
-function resolveAppendOnlyMode(setting: "auto" | "on" | "off" | undefined, provider: string): boolean {
-	switch (setting ?? "auto") {
-		case "on":
-			return true;
-		case "off":
-			return false;
-		default:
-			return provider === "deepseek";
-	}
-}
+export { createAppendOnlyContextManager, providerSupportsAppendOnlyAuto, resolveAppendOnlyMode };
 
 function customToolToDefinition(tool: CustomTool): ToolDefinition {
 	const definition: ToolDefinition & { [TOOL_DEFINITION_MARKER]: true } = {
@@ -866,6 +859,10 @@ function withEmbeddedDefaultGjcSkills(skills: Skill[]): Skill[] {
 	return [...byName.values()];
 }
 
+export function resolveIntentTracingEnabled(intentTracingSetting: boolean | undefined, hasUI: boolean): boolean {
+	return (!!intentTracingSetting || $flag("PI_INTENT_TRACING")) && hasUI;
+}
+
 export async function createAgentSession(options: CreateAgentSessionOptions = {}): Promise<CreateAgentSessionResult> {
 	const cwd = options.cwd ?? getProjectDir();
 	const agentDir = options.agentDir ?? getDefaultAgentDir();
@@ -1125,8 +1122,15 @@ export async function createAgentSession(options: CreateAgentSessionOptions = {}
 				rulebookRules.push(rule);
 			}
 		}
-		if (existingSession.injectedTtsrRules.length > 0) {
-			ttsrManager.restoreInjected(existingSession.injectedTtsrRules);
+		if (ttsrManager.getSettings().enabled !== false) {
+			if ((existingSession.ttsrMessageCount ?? 0) > 0) {
+				ttsrManager.restoreMessageCount(existingSession.ttsrMessageCount ?? 0);
+			}
+			if (existingSession.injectedTtsrRuleRecords && existingSession.injectedTtsrRuleRecords.length > 0) {
+				ttsrManager.restoreInjected(existingSession.injectedTtsrRuleRecords);
+			} else if (existingSession.injectedTtsrRules.length > 0) {
+				ttsrManager.restoreInjected(existingSession.injectedTtsrRules);
+			}
 		}
 		return { ttsrManager, rulebookRules, alwaysApplyRules };
 	});
@@ -1214,8 +1218,8 @@ export async function createAgentSession(options: CreateAgentSessionOptions = {}
 
 	const agentRegistry = options.agentRegistry ?? AgentRegistry.global();
 	const resolvedAgentId = options.agentId ?? options.parentTaskPrefix ?? MAIN_AGENT_ID;
-	const resolvedAgentDisplayName =
-		options.agentDisplayName ?? ((options.taskDepth ?? 0) > 0 || options.parentTaskPrefix ? "sub" : "main");
+	const isSubSession = taskDepth > 0 || Boolean(options.parentTaskPrefix) || Boolean(options.currentAgentType);
+	const resolvedAgentDisplayName = options.agentDisplayName ?? (isSubSession ? "sub" : "main");
 	const evalKernelOwnerId = `agent-session:${Snowflake.next()}`;
 
 	try {
@@ -1520,12 +1524,20 @@ export async function createAgentSession(options: CreateAgentSessionOptions = {}
 		}
 		let notificationCfg: NotificationConfig | undefined;
 		try {
-			notificationCfg = getNotificationConfig(Settings.instance);
+			notificationCfg = getNotificationConfig(settings);
 		} catch {
 			notificationCfg = undefined;
 		}
-		if (shouldRegisterNotificationsExtension({ env: process.env, cfg: notificationCfg })) {
-			inlineExtensions.push(createNotificationsExtension);
+		if (
+			shouldRegisterNotificationsExtension({
+				env: process.env,
+				cfg: notificationCfg,
+				taskDepth,
+				parentTaskPrefix: options.parentTaskPrefix,
+				currentAgentType: options.currentAgentType,
+			})
+		) {
+			inlineExtensions.push(api => createNotificationsExtension(api, { settings }));
 		}
 
 		// Extension/module discovery is quarantined; retain only the private
@@ -1632,6 +1644,12 @@ export async function createAgentSession(options: CreateAgentSessionOptions = {}
 				cwd,
 				sessionManager,
 				modelRegistry,
+				{
+					kind: isSubSession ? "sub" : "main",
+					taskDepth,
+					...(options.parentTaskPrefix ? { parentTaskPrefix: options.parentTaskPrefix } : {}),
+					...(options.currentAgentType ? { currentAgentType: options.currentAgentType } : {}),
+				},
 			);
 		}
 
@@ -1751,7 +1769,11 @@ export async function createAgentSession(options: CreateAgentSessionOptions = {}
 
 		const repeatToolDescriptions = settings.get("repeatToolDescriptions");
 		const eagerTasks = settings.get("task.eager");
-		const intentField = settings.get("tools.intentTracing") || $flag("PI_INTENT_TRACING") ? INTENT_FIELD : undefined;
+		const intentTracingEnabled = resolveIntentTracingEnabled(
+			settings.get("tools.intentTracing"),
+			options.hasUI ?? false,
+		);
+		const intentField = intentTracingEnabled ? INTENT_FIELD : undefined;
 		const rebuildSystemPrompt = async (
 			toolNames: string[],
 			tools: Map<string, AgentTool>,
@@ -1823,6 +1845,7 @@ export async function createAgentSession(options: CreateAgentSessionOptions = {}
 				eagerTasks,
 				secretsEnabled,
 				workspaceTree: workspaceTreePromise,
+				subagent: options.parentTaskPrefix !== undefined,
 			});
 
 			if (options.systemPrompt === undefined) {
@@ -1924,7 +1947,7 @@ export async function createAgentSession(options: CreateAgentSessionOptions = {}
 		agentRegistry.register({
 			id: resolvedAgentId,
 			displayName: resolvedAgentDisplayName,
-			kind: (options.taskDepth ?? 0) > 0 || options.parentTaskPrefix ? "sub" : "main",
+			kind: isSubSession ? "sub" : "main",
 			parentId: options.parentTaskPrefix,
 			session: null,
 			sessionFile: sessionManager.getSessionFile() ?? null,
@@ -2031,7 +2054,7 @@ export async function createAgentSession(options: CreateAgentSessionOptions = {}
 
 		const appendOnlyContext =
 			model && resolveAppendOnlyMode(settings.get("provider.appendOnlyContext"), model.provider)
-				? new AppendOnlyContextManager()
+				? createAppendOnlyContextManager(model.provider)
 				: undefined;
 		if (appendOnlyContext && options.forkContextSeed && !hasExistingSession) {
 			if (options.forkContextSeed.appendOnlyPrefixSnapshot) {
@@ -2202,6 +2225,7 @@ export async function createAgentSession(options: CreateAgentSessionOptions = {}
 			onResponse,
 			convertToLlm: convertToLlmFinal,
 			rebuildSystemPrompt,
+			workspaceTree: resolvedWorkspaceTree,
 			reloadSshTool,
 			requestedToolNames: requestedToolNameSet,
 			discoverableToolAllowedNames: options.discoverableToolAllowedNames,