refactor(coding-agent): simplify AgentSession

This commit is contained in:
Mario Zechner 2026-01-24 01:38:58 +01:00
parent 254c00b788
commit f9eb190ef9
5 changed files with 33 additions and 51 deletions

View file

@ -27,6 +27,7 @@ import { isContextOverflow, modelsAreEqual, supportsXhigh } from "@mariozechner/
import { getAuthPath } from "../config.js"; import { getAuthPath } from "../config.js";
import { theme } from "../modes/interactive/theme/theme.js"; import { theme } from "../modes/interactive/theme/theme.js";
import { stripFrontmatter } from "../utils/frontmatter.js"; import { stripFrontmatter } from "../utils/frontmatter.js";
import { sleep } from "../utils/sleep.js";
import { type BashResult, executeBash as executeBashCommand, executeBashWithOperations } from "./bash-executor.js"; import { type BashResult, executeBash as executeBashCommand, executeBashWithOperations } from "./bash-executor.js";
import { import {
type CompactionResult, type CompactionResult,
@ -62,10 +63,9 @@ import {
import type { BashExecutionMessage, CustomMessage } from "./messages.js"; import type { BashExecutionMessage, CustomMessage } from "./messages.js";
import type { ModelRegistry } from "./model-registry.js"; import type { ModelRegistry } from "./model-registry.js";
import { expandPromptTemplate, type PromptTemplate } from "./prompt-templates.js"; import { expandPromptTemplate, type PromptTemplate } from "./prompt-templates.js";
import type { ResourceDiagnostic, ResourceLoader } from "./resource-loader.js"; import type { ResourceLoader } from "./resource-loader.js";
import type { BranchSummaryEntry, CompactionEntry, NewSessionOptions, SessionManager } from "./session-manager.js"; import type { BranchSummaryEntry, CompactionEntry, NewSessionOptions, SessionManager } from "./session-manager.js";
import type { SettingsManager } from "./settings-manager.js"; import type { SettingsManager } from "./settings-manager.js";
import type { Skill } from "./skills.js";
import { buildSystemPrompt } from "./system-prompt.js"; import { buildSystemPrompt } from "./system-prompt.js";
import type { BashOperations } from "./tools/bash.js"; import type { BashOperations } from "./tools/bash.js";
import { createAllTools } from "./tools/index.js"; import { createAllTools } from "./tools/index.js";
@ -817,7 +817,7 @@ export class AgentSession {
const skillName = spaceIndex === -1 ? text.slice(7) : text.slice(7, spaceIndex); const skillName = spaceIndex === -1 ? text.slice(7) : text.slice(7, spaceIndex);
const args = spaceIndex === -1 ? "" : text.slice(spaceIndex + 1).trim(); const args = spaceIndex === -1 ? "" : text.slice(spaceIndex + 1).trim();
const skill = this.skills.find((s) => s.name === skillName); const skill = this.resourceLoader.getSkills().skills.find((s) => s.name === skillName);
if (!skill) return text; // Unknown skill, pass through if (!skill) return text; // Unknown skill, pass through
try { try {
@ -1031,16 +1031,6 @@ export class AgentSession {
return this._followUpMessages; return this._followUpMessages;
} }
/** Skills loaded by resource loader */
get skills(): readonly Skill[] {
return this._resourceLoader.getSkills().skills;
}
/** Skill loading diagnostics (warnings, errors, collisions) */
get skillWarnings(): readonly ResourceDiagnostic[] {
return this._resourceLoader.getSkills().diagnostics;
}
get resourceLoader(): ResourceLoader { get resourceLoader(): ResourceLoader {
return this._resourceLoader; return this._resourceLoader;
} }
@ -1212,13 +1202,6 @@ export class AgentSession {
return { model: nextModel, thinkingLevel: this.thinkingLevel, isScoped: false }; return { model: nextModel, thinkingLevel: this.thinkingLevel, isScoped: false };
} }
/**
* Get all available models with valid API keys.
*/
async getAvailableModels(): Promise<Model<any>[]> {
return this._modelRegistry.getAvailable();
}
// ========================================================================= // =========================================================================
// Thinking Level Management // Thinking Level Management
// ========================================================================= // =========================================================================
@ -1937,7 +1920,7 @@ export class AgentSession {
// Wait with exponential backoff (abortable) // Wait with exponential backoff (abortable)
this._retryAbortController = new AbortController(); this._retryAbortController = new AbortController();
try { try {
await this._sleep(delayMs, this._retryAbortController.signal); await sleep(delayMs, this._retryAbortController.signal);
} catch { } catch {
// Aborted during sleep - emit end event so UI can clean up // Aborted during sleep - emit end event so UI can clean up
const attempt = this._retryAttempt; const attempt = this._retryAttempt;
@ -1964,25 +1947,6 @@ export class AgentSession {
return true; return true;
} }
/**
* Sleep helper that respects abort signal.
*/
private _sleep(ms: number, signal?: AbortSignal): Promise<void> {
return new Promise((resolve, reject) => {
if (signal?.aborted) {
reject(new Error("Aborted"));
return;
}
const timeout = setTimeout(resolve, ms);
signal?.addEventListener("abort", () => {
clearTimeout(timeout);
reject(new Error("Aborted"));
});
});
}
/** /**
* Cancel in-progress retry. * Cancel in-progress retry.
*/ */

View file

@ -347,7 +347,7 @@ export class InteractiveMode {
this.skillCommands.clear(); this.skillCommands.clear();
const skillCommandList: SlashCommand[] = []; const skillCommandList: SlashCommand[] = [];
if (this.settingsManager.getEnableSkillCommands()) { if (this.settingsManager.getEnableSkillCommands()) {
for (const skill of this.session.skills) { for (const skill of this.session.resourceLoader.getSkills().skills) {
const commandName = `skill:${skill.name}`; const commandName = `skill:${skill.name}`;
this.skillCommands.set(commandName, skill.filePath); this.skillCommands.set(commandName, skill.filePath);
skillCommandList.push({ name: commandName, description: skill.description }); skillCommandList.push({ name: commandName, description: skill.description });
@ -839,7 +839,7 @@ export class InteractiveMode {
this.chatContainer.addChild(new Spacer(1)); this.chatContainer.addChild(new Spacer(1));
} }
const skills = this.session.skills; const skills = this.session.resourceLoader.getSkills().skills;
if (skills.length > 0) { if (skills.length > 0) {
const skillPaths = skills.map((s) => s.filePath); const skillPaths = skills.map((s) => s.filePath);
const groups = this.groupPathsBySource(skillPaths, metadata); const groups = this.groupPathsBySource(skillPaths, metadata);
@ -848,7 +848,7 @@ export class InteractiveMode {
this.chatContainer.addChild(new Spacer(1)); this.chatContainer.addChild(new Spacer(1));
} }
const skillDiagnostics = this.session.skillWarnings; const skillDiagnostics = this.session.resourceLoader.getSkills().diagnostics;
if (skillDiagnostics.length > 0) { if (skillDiagnostics.length > 0) {
const warningLines = this.formatDiagnostics(skillDiagnostics, metadata); const warningLines = this.formatDiagnostics(skillDiagnostics, metadata);
this.chatContainer.addChild(new Text(`${theme.fg("warning", "[Skill conflicts]")}\n${warningLines}`, 0, 0)); this.chatContainer.addChild(new Text(`${theme.fg("warning", "[Skill conflicts]")}\n${warningLines}`, 0, 0));

View file

@ -365,7 +365,7 @@ export async function runRpcMode(session: AgentSession): Promise<never> {
// ================================================================= // =================================================================
case "set_model": { case "set_model": {
const models = await session.getAvailableModels(); const models = await session.modelRegistry.getAvailable();
const model = models.find((m) => m.provider === command.provider && m.id === command.modelId); const model = models.find((m) => m.provider === command.provider && m.id === command.modelId);
if (!model) { if (!model) {
return error(id, "set_model", `Model not found: ${command.provider}/${command.modelId}`); return error(id, "set_model", `Model not found: ${command.provider}/${command.modelId}`);
@ -383,7 +383,7 @@ export async function runRpcMode(session: AgentSession): Promise<never> {
} }
case "get_available_models": { case "get_available_models": {
const models = await session.getAvailableModels(); const models = await session.modelRegistry.getAvailable();
return success(id, "get_available_models", { models }); return success(id, "get_available_models", { models });
} }

View file

@ -0,0 +1,18 @@
/**
* Sleep helper that respects abort signal.
*/
export function sleep(ms: number, signal?: AbortSignal): Promise<void> {
return new Promise((resolve, reject) => {
if (signal?.aborted) {
reject(new Error("Aborted"));
return;
}
const timeout = setTimeout(resolve, ms);
signal?.addEventListener("abort", () => {
clearTimeout(timeout);
reject(new Error("Aborted"));
});
});
}

View file

@ -45,8 +45,8 @@ This is a test skill.
}); });
// Skills should be discovered and exposed on the session // Skills should be discovered and exposed on the session
expect(session.skills.length).toBeGreaterThan(0); expect(session.resourceLoader.getSkills().skills.length).toBeGreaterThan(0);
expect(session.skills.some((s) => s.name === "test-skill")).toBe(true); expect(session.resourceLoader.getSkills().skills.some((s) => s.name === "test-skill")).toBe(true);
}); });
it("should have empty skills when resource loader returns none (--no-skills)", async () => { it("should have empty skills when resource loader returns none (--no-skills)", async () => {
@ -69,8 +69,8 @@ This is a test skill.
resourceLoader, resourceLoader,
}); });
expect(session.skills).toEqual([]); expect(session.resourceLoader.getSkills().skills).toEqual([]);
expect(session.skillWarnings).toEqual([]); expect(session.resourceLoader.getSkills().diagnostics).toEqual([]);
}); });
it("should use provided skills when resource loader supplies them", async () => { it("should use provided skills when resource loader supplies them", async () => {
@ -101,7 +101,7 @@ This is a test skill.
resourceLoader, resourceLoader,
}); });
expect(session.skills).toEqual([customSkill]); expect(session.resourceLoader.getSkills().skills).toEqual([customSkill]);
expect(session.skillWarnings).toEqual([]); expect(session.resourceLoader.getSkills().diagnostics).toEqual([]);
}); });
}); });