neoctl 0.1.0

package/dist/core/query-engine.js

@@ -0,0 +1,337 @@
+import { ToolRegistry } from "../tools/registry.js";
+import { createSystemInitMessage, createTextMessage } from "../types/messages.js";
+import { query } from "./query.js";
+import { runAgent } from "./run-agent.js";
+import { GENERAL_PURPOSE_AGENT } from "../agents/agent-definition.js";
+import { SessionStore } from "../session/session-store.js";
+const DEFAULT_SESSION_TITLE_DELAY_MS = 5000;
+export class QueryEngine {
+    options;
+    agentId;
+    history = [];
+    titleTimers = new Set();
+    lastTerminalReason;
+    sessionStore;
+    currentModel;
+    currentReasoning;
+    sessionInitialized = false;
+    userTurns = 0;
+    titleSchedulerVersion = 0;
+    titleAgentRun;
+    sessionTitleListeners = new Set();
+    constructor(options) {
+        this.options = options;
+        this.agentId = options.agentId ?? "main";
+        this.currentModel = options.model;
+        this.currentReasoning = cloneReasoningConfig(options.reasoning);
+    }
+    onSessionTitleChange(listener) {
+        this.sessionTitleListeners.add(listener);
+        return () => this.sessionTitleListeners.delete(listener);
+    }
+    async initialize() {
+        if (this.sessionInitialized)
+            return;
+        this.sessionInitialized = true;
+        if (this.options.session?.enabled === false)
+            return;
+        if (!this.options.session)
+            return;
+        await this.openSession({
+            sessionId: this.options.session.sessionId,
+            resume: this.options.session.resume,
+        });
+    }
+    async resumeSession(sessionId) {
+        this.sessionInitialized = true;
+        await this.openSession({ sessionId, resume: true });
+        const snapshot = this.sessionStore?.snapshot();
+        if (!snapshot)
+            throw new Error("session transcripts are disabled");
+        return snapshot;
+    }
+    async listSessions(limit = 10) {
+        if (this.options.session?.enabled === false || !this.options.session)
+            return [];
+        return SessionStore.list({
+            agentId: this.agentId,
+            cwd: process.cwd(),
+            rootDir: this.options.session.rootDir,
+            limit,
+        });
+    }
+    async deleteSession(sessionId) {
+        if (this.options.session?.enabled === false || !this.options.session)
+            throw new Error("session transcripts are disabled");
+        const activeSessionId = this.sessionStore?.sessionId;
+        if (activeSessionId && activeSessionId === sessionId) {
+            throw new Error("cannot delete the active session");
+        }
+        return SessionStore.delete({
+            cwd: process.cwd(),
+            sessionId,
+            rootDir: this.options.session.rootDir,
+        });
+    }
+    async *sendUserText(text, options = {}) {
+        await this.initialize();
+        const userMessage = options.blocks
+            ? { ...createTextMessage("user", options.displayText ?? text), blocks: options.blocks }
+            : createTextMessage("user", text);
+        this.userTurns += 1;
+        this.history.push(userMessage);
+        this.sessionStore?.recordMessage(userMessage);
+        this.scheduleSessionTitleCheck();
+        const initMessage = createSystemInitMessage({
+            agentId: this.agentId,
+            tools: this.options.tools.names(),
+            model: this.currentModel,
+            commands: [...(this.options.commands ?? [])],
+            agents: [...(this.options.agents ?? [])],
+            skills: [...(this.options.skills ?? [])],
+            plugins: [...(this.options.plugins ?? [])],
+        });
+        yield {
+            type: "message",
+            message: initMessage,
+        };
+        this.sessionStore?.recordMessage(initMessage);
+        const queryOptions = {
+            agentId: this.agentId,
+            model: this.currentModel,
+            fallbackModel: this.options.fallbackModel,
+            reasoning: cloneReasoningConfig(this.currentReasoning),
+            queryOrigin: this.options.queryOrigin ?? "repl",
+            maxOutputTokensOverride: this.options.maxOutputTokensOverride,
+            maxTurns: this.options.maxTurns,
+            abortSignal: options.abortSignal,
+        };
+        const stream = query(this.history, {
+            ...this.options,
+            taskNotificationSource: this.options.taskNotificationSource,
+            toolResultMemory: this.sessionStore?.toolResultMemory,
+            recordContentReplacements: (records) => this.sessionStore?.recordContentReplacements(records),
+        }, queryOptions);
+        for await (const event of stream) {
+            if (event.type === "message") {
+                this.history.push(event.message);
+                this.sessionStore?.recordMessage(event.message);
+            }
+            if (event.type === "terminal")
+                this.lastTerminalReason = event.reason;
+            yield event;
+        }
+    }
+    setModel(model, reasoning, updateReasoning = false) {
+        this.currentModel = model?.trim() || undefined;
+        if (updateReasoning)
+            this.currentReasoning = reasoning === null ? null : cloneReasoningConfig(reasoning);
+    }
+    getModelSettings() {
+        return {
+            model: this.currentModel,
+            fallbackModel: this.options.fallbackModel,
+            reasoning: cloneReasoningConfig(this.currentReasoning),
+        };
+    }
+    reset() {
+        this.history.length = 0;
+        this.userTurns = 0;
+        this.lastTerminalReason = undefined;
+        this.cancelPendingTitleWork();
+        this.sessionStore?.reset();
+        this.notifySessionTitleChange(this.sessionStore?.snapshot());
+    }
+    snapshot() {
+        return {
+            agentId: this.agentId,
+            messages: this.history.length,
+            model: this.currentModel,
+            fallbackModel: this.options.fallbackModel,
+            reasoning: cloneReasoningConfig(this.currentReasoning),
+            lastTerminalReason: this.lastTerminalReason,
+            session: this.sessionStore?.snapshot(),
+        };
+    }
+    getHistoryMessages() {
+        return this.history.map(cloneMessage);
+    }
+    get toolResultMemory() {
+        return this.sessionStore?.toolResultMemory;
+    }
+    async openSession(options) {
+        if (this.options.session?.enabled === false || !this.options.session)
+            return;
+        this.cancelPendingTitleWork();
+        this.sessionStore = await SessionStore.open({
+            agentId: this.agentId,
+            cwd: process.cwd(),
+            sessionId: options.sessionId,
+            rootDir: this.options.session.rootDir,
+            resume: options.resume,
+            toolResultThresholdChars: this.options.session.toolResultThresholdChars,
+        });
+        this.history.length = 0;
+        if (options.resume)
+            this.history.push(...this.sessionStore.getInitialMessages());
+        this.userTurns = countUserTurns(this.history);
+        this.notifySessionTitleChange(this.sessionStore.snapshot());
+    }
+    scheduleSessionTitleCheck() {
+        const store = this.sessionStore;
+        if (!store || titleWorkComplete(store))
+            return;
+        const version = this.titleSchedulerVersion;
+        const timer = setTimeout(() => {
+            this.titleTimers.delete(timer);
+            void this.runSessionTitleCheck(version).catch(() => undefined);
+        }, resolveSessionTitleDelayMs());
+        this.titleTimers.add(timer);
+        timer.unref?.();
+    }
+    async runSessionTitleCheck(version) {
+        try {
+            if (version !== this.titleSchedulerVersion)
+                return;
+            const store = this.sessionStore;
+            if (!store || titleWorkComplete(store))
+                return;
+            if (this.titleAgentRun?.version === version)
+                return;
+            const state = store.getTitleState();
+            const kind = state.hasInitialTitle ? "refinement" : "initial";
+            if (kind === "refinement" && !state.title)
+                return;
+            const controller = new AbortController();
+            this.titleAgentRun = { version, controller };
+            const title = await generateSessionTitle({
+                agentId: `${this.agentId}-session-title`,
+                modelGateway: this.options.modelGateway,
+                model: this.currentModel,
+                fallbackModel: this.options.fallbackModel,
+                kind,
+                previousTitle: state.title,
+                messages: this.history,
+                abortSignal: controller.signal,
+            });
+            if (version !== this.titleSchedulerVersion || store !== this.sessionStore || !title)
+                return;
+            const latest = store.getTitleState();
+            if (kind === "initial" && latest.hasInitialTitle)
+                return;
+            if (kind === "refinement" && latest.hasRefinement)
+                return;
+            store.recordTitle(title, kind);
+            this.notifySessionTitleChange(store.snapshot());
+        }
+        finally {
+            if (this.titleAgentRun?.version === version)
+                this.titleAgentRun = undefined;
+        }
+    }
+    notifySessionTitleChange(snapshot) {
+        for (const listener of this.sessionTitleListeners)
+            listener(snapshot);
+    }
+    cancelPendingTitleWork() {
+        this.titleSchedulerVersion += 1;
+        for (const timer of this.titleTimers)
+            clearTimeout(timer);
+        this.titleTimers.clear();
+        this.titleAgentRun?.controller.abort();
+        this.titleAgentRun = undefined;
+    }
+}
+async function generateSessionTitle(input) {
+    try {
+        const prompt = buildSessionTitlePrompt(input.kind, input.messages, input.previousTitle);
+        const stream = runAgent({
+            agentId: input.agentId,
+            agent: {
+                ...GENERAL_PURPOSE_AGENT,
+                agentType: "session-title",
+                tools: [],
+                maxTurns: 1,
+                initialPrompt: "You generate concise conversation titles. Return only the title.",
+            },
+            prompt,
+            dependencies: { modelGateway: input.modelGateway, tools: new ToolRegistry() },
+            model: input.model,
+            fallbackModel: input.fallbackModel,
+            maxTurns: 1,
+            abortSignal: input.abortSignal,
+        });
+        let completed = await stream.next();
+        while (!completed.done)
+            completed = await stream.next();
+        return normalizeGeneratedTitle(completed.value.result.content);
+    }
+    catch {
+        return undefined;
+    }
+}
+function buildSessionTitlePrompt(kind, messages, previousTitle) {
+    const instructions = kind === "initial"
+        ? ["Summarize this session as a short title for a session list."]
+        : [
+            "Refine this existing session title using the updated conversation.",
+            `Previous title: ${previousTitle ?? "<none>"}`,
+        ];
+    return [
+        ...instructions,
+        "Return only the title, without quotes or punctuation decoration.",
+        "Keep it under 8 words and use the user's language when possible.",
+        "",
+        serializeMessagesForTitle(messages),
+    ].join("\n");
+}
+function serializeMessagesForTitle(messages) {
+    return messages
+        .filter((message) => message.role === "user" || message.role === "assistant")
+        .map((message) => {
+        const text = message.blocks
+            .filter((block) => block.type === "text")
+            .map((block) => block.text)
+            .join("\n")
+            .trim();
+        return text ? `${message.role}: ${text}` : undefined;
+    })
+        .filter((line) => Boolean(line))
+        .join("\n")
+        .slice(0, 4000);
+}
+function titleWorkComplete(store) {
+    const state = store.getTitleState();
+    return state.hasInitialTitle && state.hasRefinement;
+}
+function normalizeGeneratedTitle(title) {
+    const normalized = title
+        .replace(/[\r\n]+/g, " ")
+        .replace(/^\s*["'“”‘’`]+|["'“”‘’`]+\s*$/g, "")
+        .replace(/\s+/g, " ")
+        .trim();
+    return normalized ? normalized.slice(0, 120) : undefined;
+}
+function resolveSessionTitleDelayMs() {
+    const raw = process.env.AGENT_SESSION_TITLE_DELAY_MS;
+    if (raw !== undefined && raw !== "") {
+        const parsed = Number(raw);
+        if (Number.isFinite(parsed) && parsed >= 0)
+            return Math.floor(parsed);
+    }
+    return DEFAULT_SESSION_TITLE_DELAY_MS;
+}
+function countUserTurns(messages) {
+    return messages.filter((message) => message.role === "user" && message.blocks.some((block) => block.type === "text")).length;
+}
+function cloneReasoningConfig(reasoning) {
+    if (reasoning === null)
+        return null;
+    if (!reasoning)
+        return undefined;
+    return { ...reasoning };
+}
+function cloneMessage(message) {
+    return JSON.parse(JSON.stringify(message));
+}
+//# sourceMappingURL=query-engine.js.map

package/dist/core/query-engine.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"query-engine.js","sourceRoot":"","sources":["../../src/core/query-engine.ts"],"names":[],"mappings":"AAGA,OAAO,EAAE,YAAY,EAAE,MAAM,sBAAsB,CAAC;AAKpD,OAAO,EAAE,uBAAuB,EAAE,iBAAiB,EAAE,MAAM,sBAAsB,CAAC;AAClF,OAAO,EAAE,KAAK,EAAE,MAAM,YAAY,CAAC;AACnC,OAAO,EAAE,QAAQ,EAAE,MAAM,gBAAgB,CAAC;AAC1C,OAAO,EAAE,qBAAqB,EAAE,MAAM,+BAA+B,CAAC;AAEtE,OAAO,EAAE,YAAY,EAAyE,MAAM,6BAA6B,CAAC;AAElI,MAAM,8BAA8B,GAAG,IAAI,CAAC;AA8B5C,MAAM,OAAO,WAAW;IAcO;IAbZ,OAAO,CAAS;IAChB,OAAO,GAAc,EAAE,CAAC;IACxB,WAAW,GAAG,IAAI,GAAG,EAAiC,CAAC;IAChE,kBAAkB,CAAkB;IACpC,YAAY,CAAgB;IAC5B,YAAY,CAAU;IACtB,gBAAgB,CAA0B;IAC1C,kBAAkB,GAAG,KAAK,CAAC;IAC3B,SAAS,GAAG,CAAC,CAAC;IACd,qBAAqB,GAAG,CAAC,CAAC;IAC1B,aAAa,CAAoD;IACxD,qBAAqB,GAAG,IAAI,GAAG,EAAwD,CAAC;IAEzG,YAA6B,OAA2B;QAA3B,YAAO,GAAP,OAAO,CAAoB;QACtD,IAAI,CAAC,OAAO,GAAG,OAAO,CAAC,OAAO,IAAI,MAAM,CAAC;QACzC,IAAI,CAAC,YAAY,GAAG,OAAO,CAAC,KAAK,CAAC;QAClC,IAAI,CAAC,gBAAgB,GAAG,oBAAoB,CAAC,OAAO,CAAC,SAAS,CAAC,CAAC;IAClE,CAAC;IAED,oBAAoB,CAAC,QAA8D;QACjF,IAAI,CAAC,qBAAqB,CAAC,GAAG,CAAC,QAAQ,CAAC,CAAC;QACzC,OAAO,GAAG,EAAE,CAAC,IAAI,CAAC,qBAAqB,CAAC,MAAM,CAAC,QAAQ,CAAC,CAAC;IAC3D,CAAC;IAED,KAAK,CAAC,UAAU;QACd,IAAI,IAAI,CAAC,kBAAkB;YAAE,OAAO;QACpC,IAAI,CAAC,kBAAkB,GAAG,IAAI,CAAC;QAC/B,IAAI,IAAI,CAAC,OAAO,CAAC,OAAO,EAAE,OAAO,KAAK,KAAK;YAAE,OAAO;QACpD,IAAI,CAAC,IAAI,CAAC,OAAO,CAAC,OAAO;YAAE,OAAO;QAClC,MAAM,IAAI,CAAC,WAAW,CAAC;YACrB,SAAS,EAAE,IAAI,CAAC,OAAO,CAAC,OAAO,CAAC,SAAS;YACzC,MAAM,EAAE,IAAI,CAAC,OAAO,CAAC,OAAO,CAAC,MAAM;SACpC,CAAC,CAAC;IACL,CAAC;IAED,KAAK,CAAC,aAAa,CAAC,SAAkB;QACpC,IAAI,CAAC,kBAAkB,GAAG,IAAI,CAAC;QAC/B,MAAM,IAAI,CAAC,WAAW,CAAC,EAAE,SAAS,EAAE,MAAM,EAAE,IAAI,EAAE,CAAC,CAAC;QACpD,MAAM,QAAQ,GAAG,IAAI,CAAC,YAAY,EAAE,QAAQ,EAAE,CAAC;QAC/C,IAAI,CAAC,QAAQ;YAAE,MAAM,IAAI,KAAK,CAAC,kCAAkC,CAAC,CAAC;QACnE,OAAO,QAAQ,CAAC;IAClB,CAAC;IAED,KAAK,CAAC,YAAY,CAAC,KAAK,GAAG,EAAE;QAC3B,IAAI,IAAI,CAAC,OAAO,CAAC,OAAO,EAAE,OAAO,KAAK,KAAK,IAAI,CAAC,IAAI,CAAC,OAAO,CAAC,OAAO;YAAE,OAAO,EAAE,CAAC;QAChF,OAAO,YAAY,CAAC,IAAI,CAAC;YACvB,OAAO,EAAE,IAAI,CAAC,OAAO;YACrB,GAAG,EAAE,OAAO,CAAC,GAAG,EAAE;YAClB,OAAO,EAAE,IAAI,CAAC,OAAO,CAAC,OAAO,CAAC,OAAO;YACrC,KAAK;SACN,CAAC,CAAC;IACL,CAAC;IAED,KAAK,CAAC,aAAa,CAAC,SAAiB;QACnC,IAAI,IAAI,CAAC,OAAO,CAAC,OAAO,EAAE,OAAO,KAAK,KAAK,IAAI,CAAC,IAAI,CAAC,OAAO,CAAC,OAAO;YAAE,MAAM,IAAI,KAAK,CAAC,kCAAkC,CAAC,CAAC;QAC1H,MAAM,eAAe,GAAG,IAAI,CAAC,YAAY,EAAE,SAAS,CAAC;QACrD,IAAI,eAAe,IAAI,eAAe,KAAK,SAAS,EAAE,CAAC;YACrD,MAAM,IAAI,KAAK,CAAC,kCAAkC,CAAC,CAAC;QACtD,CAAC;QACD,OAAO,YAAY,CAAC,MAAM,CAAC;YACzB,GAAG,EAAE,OAAO,CAAC,GAAG,EAAE;YAClB,SAAS;YACT,OAAO,EAAE,IAAI,CAAC,OAAO,CAAC,OAAO,CAAC,OAAO;SACtC,CAAC,CAAC;IACL,CAAC;IAED,KAAK,CAAC,CAAC,YAAY,CAAC,IAAY,EAAE,UAAwF,EAAE;QAC1H,MAAM,IAAI,CAAC,UAAU,EAAE,CAAC;QACxB,MAAM,WAAW,GAAG,OAAO,CAAC,MAAM;YAChC,CAAC,CAAC,EAAE,GAAG,iBAAiB,CAAC,MAAM,EAAE,OAAO,CAAC,WAAW,IAAI,IAAI,CAAC,EAAE,MAAM,EAAE,OAAO,CAAC,MAAM,EAAE;YACvF,CAAC,CAAC,iBAAiB,CAAC,MAAM,EAAE,IAAI,CAAC,CAAC;QACpC,IAAI,CAAC,SAAS,IAAI,CAAC,CAAC;QACpB,IAAI,CAAC,OAAO,CAAC,IAAI,CAAC,WAAW,CAAC,CAAC;QAC/B,IAAI,CAAC,YAAY,EAAE,aAAa,CAAC,WAAW,CAAC,CAAC;QAC9C,IAAI,CAAC,yBAAyB,EAAE,CAAC;QAEjC,MAAM,WAAW,GAAG,uBAAuB,CAAC;YAC1C,OAAO,EAAE,IAAI,CAAC,OAAO;YACrB,KAAK,EAAE,IAAI,CAAC,OAAO,CAAC,KAAK,CAAC,KAAK,EAAE;YACjC,KAAK,EAAE,IAAI,CAAC,YAAY;YACxB,QAAQ,EAAE,CAAC,GAAG,CAAC,IAAI,CAAC,OAAO,CAAC,QAAQ,IAAI,EAAE,CAAC,CAAC;YAC5C,MAAM,EAAE,CAAC,GAAG,CAAC,IAAI,CAAC,OAAO,CAAC,MAAM,IAAI,EAAE,CAAC,CAAC;YACxC,MAAM,EAAE,CAAC,GAAG,CAAC,IAAI,CAAC,OAAO,CAAC,MAAM,IAAI,EAAE,CAAC,CAAC;YACxC,OAAO,EAAE,CAAC,GAAG,CAAC,IAAI,CAAC,OAAO,CAAC,OAAO,IAAI,EAAE,CAAC,CAAC;SAC3C,CAAC,CAAC;QACH,MAAM;YACJ,IAAI,EAAE,SAAS;YACf,OAAO,EAAE,WAAW;SACrB,CAAC;QACF,IAAI,CAAC,YAAY,EAAE,aAAa,CAAC,WAAW,CAAC,CAAC;QAE9C,MAAM,YAAY,GAAiB;YACjC,OAAO,EAAE,IAAI,CAAC,OAAO;YACrB,KAAK,EAAE,IAAI,CAAC,YAAY;YACxB,aAAa,EAAE,IAAI,CAAC,OAAO,CAAC,aAAa;YACzC,SAAS,EAAE,oBAAoB,CAAC,IAAI,CAAC,gBAAgB,CAAC;YACtD,WAAW,EAAE,IAAI,CAAC,OAAO,CAAC,WAAW,IAAI,MAAM;YAC/C,uBAAuB,EAAE,IAAI,CAAC,OAAO,CAAC,uBAAuB;YAC7D,QAAQ,EAAE,IAAI,CAAC,OAAO,CAAC,QAAQ;YAC/B,WAAW,EAAE,OAAO,CAAC,WAAW;SACjC,CAAC;QAEF,MAAM,MAAM,GAAG,KAAK,CAClB,IAAI,CAAC,OAAO,EACZ;YACE,GAAG,IAAI,CAAC,OAAO;YACf,sBAAsB,EAAE,IAAI,CAAC,OAAO,CAAC,sBAAsB;YAC3D,gBAAgB,EAAE,IAAI,CAAC,YAAY,EAAE,gBAAgB;YACrD,yBAAyB,EAAE,CAAC,OAAO,EAAE,EAAE,CAAC,IAAI,CAAC,YAAY,EAAE,yBAAyB,CAAC,OAAO,CAAC;SAC9F,EACD,YAAY,CACb,CAAC;QACF,IAAI,KAAK,EAAE,MAAM,KAAK,IAAI,MAAM,EAAE,CAAC;YACjC,IAAI,KAAK,CAAC,IAAI,KAAK,SAAS,EAAE,CAAC;gBAC7B,IAAI,CAAC,OAAO,CAAC,IAAI,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC;gBACjC,IAAI,CAAC,YAAY,EAAE,aAAa,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC;YAClD,CAAC;YACD,IAAI,KAAK,CAAC,IAAI,KAAK,UAAU;gBAAE,IAAI,CAAC,kBAAkB,GAAG,KAAK,CAAC,MAAM,CAAC;YACtE,MAAM,KAAK,CAAC;QACd,CAAC;IACH,CAAC;IAED,QAAQ,CAAC,KAAyB,EAAE,SAAkC,EAAE,eAAe,GAAG,KAAK;QAC7F,IAAI,CAAC,YAAY,GAAG,KAAK,EAAE,IAAI,EAAE,IAAI,SAAS,CAAC;QAC/C,IAAI,eAAe;YAAE,IAAI,CAAC,gBAAgB,GAAG,SAAS,KAAK,IAAI,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,oBAAoB,CAAC,SAAS,CAAC,CAAC;IAC3G,CAAC;IAED,gBAAgB;QACd,OAAO;YACL,KAAK,EAAE,IAAI,CAAC,YAAY;YACxB,aAAa,EAAE,IAAI,CAAC,OAAO,CAAC,aAAa;YACzC,SAAS,EAAE,oBAAoB,CAAC,IAAI,CAAC,gBAAgB,CAAC;SACvD,CAAC;IACJ,CAAC;IAED,KAAK;QACH,IAAI,CAAC,OAAO,CAAC,MAAM,GAAG,CAAC,CAAC;QACxB,IAAI,CAAC,SAAS,GAAG,CAAC,CAAC;QACnB,IAAI,CAAC,kBAAkB,GAAG,SAAS,CAAC;QACpC,IAAI,CAAC,sBAAsB,EAAE,CAAC;QAC9B,IAAI,CAAC,YAAY,EAAE,KAAK,EAAE,CAAC;QAC3B,IAAI,CAAC,wBAAwB,CAAC,IAAI,CAAC,YAAY,EAAE,QAAQ,EAAE,CAAC,CAAC;IAC/D,CAAC;IAED,QAAQ;QACN,OAAO;YACL,OAAO,EAAE,IAAI,CAAC,OAAO;YACrB,QAAQ,EAAE,IAAI,CAAC,OAAO,CAAC,MAAM;YAC7B,KAAK,EAAE,IAAI,CAAC,YAAY;YACxB,aAAa,EAAE,IAAI,CAAC,OAAO,CAAC,aAAa;YACzC,SAAS,EAAE,oBAAoB,CAAC,IAAI,CAAC,gBAAgB,CAAC;YACtD,kBAAkB,EAAE,IAAI,CAAC,kBAAkB;YAC3C,OAAO,EAAE,IAAI,CAAC,YAAY,EAAE,QAAQ,EAAE;SACvC,CAAC;IACJ,CAAC;IAED,kBAAkB;QAChB,OAAO,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,YAAY,CAAC,CAAC;IACxC,CAAC;IAED,IAAI,gBAAgB;QAClB,OAAO,IAAI,CAAC,YAAY,EAAE,gBAAgB,CAAC;IAC7C,CAAC;IAEO,KAAK,CAAC,WAAW,CAAC,OAAiD;QACzE,IAAI,IAAI,CAAC,OAAO,CAAC,OAAO,EAAE,OAAO,KAAK,KAAK,IAAI,CAAC,IAAI,CAAC,OAAO,CAAC,OAAO;YAAE,OAAO;QAC7E,IAAI,CAAC,sBAAsB,EAAE,CAAC;QAC9B,IAAI,CAAC,YAAY,GAAG,MAAM,YAAY,CAAC,IAAI,CAAC;YAC1C,OAAO,EAAE,IAAI,CAAC,OAAO;YACrB,GAAG,EAAE,OAAO,CAAC,GAAG,EAAE;YAClB,SAAS,EAAE,OAAO,CAAC,SAAS;YAC5B,OAAO,EAAE,IAAI,CAAC,OAAO,CAAC,OAAO,CAAC,OAAO;YACrC,MAAM,EAAE,OAAO,CAAC,MAAM;YACtB,wBAAwB,EAAE,IAAI,CAAC,OAAO,CAAC,OAAO,CAAC,wBAAwB;SACxE,CAAC,CAAC;QACH,IAAI,CAAC,OAAO,CAAC,MAAM,GAAG,CAAC,CAAC;QACxB,IAAI,OAAO,CAAC,MAAM;YAAE,IAAI,CAAC,OAAO,CAAC,IAAI,CAAC,GAAG,IAAI,CAAC,YAAY,CAAC,kBAAkB,EAAE,CAAC,CAAC;QACjF,IAAI,CAAC,SAAS,GAAG,cAAc,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;QAC9C,IAAI,CAAC,wBAAwB,CAAC,IAAI,CAAC,YAAY,CAAC,QAAQ,EAAE,CAAC,CAAC;IAC9D,CAAC;IAEO,yBAAyB;QAC/B,MAAM,KAAK,GAAG,IAAI,CAAC,YAAY,CAAC;QAChC,IAAI,CAAC,KAAK,IAAI,iBAAiB,CAAC,KAAK,CAAC;YAAE,OAAO;QAC/C,MAAM,OAAO,GAAG,IAAI,CAAC,qBAAqB,CAAC;QAC3C,MAAM,KAAK,GAAG,UAAU,CAAC,GAAG,EAAE;YAC5B,IAAI,CAAC,WAAW,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC;YAC/B,KAAK,IAAI,CAAC,oBAAoB,CAAC,OAAO,CAAC,CAAC,KAAK,CAAC,GAAG,EAAE,CAAC,SAAS,CAAC,CAAC;QACjE,CAAC,EAAE,0BAA0B,EAAE,CAAC,CAAC;QACjC,IAAI,CAAC,WAAW,CAAC,GAAG,CAAC,KAAK,CAAC,CAAC;QAC5B,KAAK,CAAC,KAAK,EAAE,EAAE,CAAC;IAClB,CAAC;IAEO,KAAK,CAAC,oBAAoB,CAAC,OAAe;QAChD,IAAI,CAAC;YACH,IAAI,OAAO,KAAK,IAAI,CAAC,qBAAqB;gBAAE,OAAO;YACnD,MAAM,KAAK,GAAG,IAAI,CAAC,YAAY,CAAC;YAChC,IAAI,CAAC,KAAK,IAAI,iBAAiB,CAAC,KAAK,CAAC;gBAAE,OAAO;YAC/C,IAAI,IAAI,CAAC,aAAa,EAAE,OAAO,KAAK,OAAO;gBAAE,OAAO;YAEpD,MAAM,KAAK,GAAG,KAAK,CAAC,aAAa,EAAE,CAAC;YACpC,MAAM,IAAI,GAAqB,KAAK,CAAC,eAAe,CAAC,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,SAAS,CAAC;YAChF,IAAI,IAAI,KAAK,YAAY,IAAI,CAAC,KAAK,CAAC,KAAK;gBAAE,OAAO;YAElD,MAAM,UAAU,GAAG,IAAI,eAAe,EAAE,CAAC;YACzC,IAAI,CAAC,aAAa,GAAG,EAAE,OAAO,EAAE,UAAU,EAAE,CAAC;YAC7C,MAAM,KAAK,GAAG,MAAM,oBAAoB,CAAC;gBACvC,OAAO,EAAE,GAAG,IAAI,CAAC,OAAO,gBAAgB;gBACxC,YAAY,EAAE,IAAI,CAAC,OAAO,CAAC,YAAY;gBACvC,KAAK,EAAE,IAAI,CAAC,YAAY;gBACxB,aAAa,EAAE,IAAI,CAAC,OAAO,CAAC,aAAa;gBACzC,IAAI;gBACJ,aAAa,EAAE,KAAK,CAAC,KAAK;gBAC1B,QAAQ,EAAE,IAAI,CAAC,OAAO;gBACtB,WAAW,EAAE,UAAU,CAAC,MAAM;aAC/B,CAAC,CAAC;YACH,IAAI,OAAO,KAAK,IAAI,CAAC,qBAAqB,IAAI,KAAK,KAAK,IAAI,CAAC,YAAY,IAAI,CAAC,KAAK;gBAAE,OAAO;YAC5F,MAAM,MAAM,GAAG,KAAK,CAAC,aAAa,EAAE,CAAC;YACrC,IAAI,IAAI,KAAK,SAAS,IAAI,MAAM,CAAC,eAAe;gBAAE,OAAO;YACzD,IAAI,IAAI,KAAK,YAAY,IAAI,MAAM,CAAC,aAAa;gBAAE,OAAO;YAC1D,KAAK,CAAC,WAAW,CAAC,KAAK,EAAE,IAAI,CAAC,CAAC;YAC/B,IAAI,CAAC,wBAAwB,CAAC,KAAK,CAAC,QAAQ,EAAE,CAAC,CAAC;QAClD,CAAC;gBAAS,CAAC;YACT,IAAI,IAAI,CAAC,aAAa,EAAE,OAAO,KAAK,OAAO;gBAAE,IAAI,CAAC,aAAa,GAAG,SAAS,CAAC;QAC9E,CAAC;IACH,CAAC;IAEO,wBAAwB,CAAC,QAA0C;QACzE,KAAK,MAAM,QAAQ,IAAI,IAAI,CAAC,qBAAqB;YAAE,QAAQ,CAAC,QAAQ,CAAC,CAAC;IACxE,CAAC;IAEO,sBAAsB;QAC5B,IAAI,CAAC,qBAAqB,IAAI,CAAC,CAAC;QAChC,KAAK,MAAM,KAAK,IAAI,IAAI,CAAC,WAAW;YAAE,YAAY,CAAC,KAAK,CAAC,CAAC;QAC1D,IAAI,CAAC,WAAW,CAAC,KAAK,EAAE,CAAC;QACzB,IAAI,CAAC,aAAa,EAAE,UAAU,CAAC,KAAK,EAAE,CAAC;QACvC,IAAI,CAAC,aAAa,GAAG,SAAS,CAAC;IACjC,CAAC;CACF;AAED,KAAK,UAAU,oBAAoB,CAAC,KASnC;IACC,IAAI,CAAC;QACH,MAAM,MAAM,GAAG,uBAAuB,CAAC,KAAK,CAAC,IAAI,EAAE,KAAK,CAAC,QAAQ,EAAE,KAAK,CAAC,aAAa,CAAC,CAAC;QACxF,MAAM,MAAM,GAAG,QAAQ,CAAC;YACtB,OAAO,EAAE,KAAK,CAAC,OAAO;YACtB,KAAK,EAAE;gBACL,GAAG,qBAAqB;gBACxB,SAAS,EAAE,eAAe;gBAC1B,KAAK,EAAE,EAAE;gBACT,QAAQ,EAAE,CAAC;gBACX,aAAa,EAAE,kEAAkE;aAClF;YACD,MAAM;YACN,YAAY,EAAE,EAAE,YAAY,EAAE,KAAK,CAAC,YAAY,EAAE,KAAK,EAAE,IAAI,YAAY,EAAE,EAAE;YAC7E,KAAK,EAAE,KAAK,CAAC,KAAK;YAClB,aAAa,EAAE,KAAK,CAAC,aAAa;YAClC,QAAQ,EAAE,CAAC;YACX,WAAW,EAAE,KAAK,CAAC,WAAW;SAC/B,CAAC,CAAC;QAEH,IAAI,SAAS,GAAG,MAAM,MAAM,CAAC,IAAI,EAAE,CAAC;QACpC,OAAO,CAAC,SAAS,CAAC,IAAI;YAAE,SAAS,GAAG,MAAM,MAAM,CAAC,IAAI,EAAE,CAAC;QACxD,OAAO,uBAAuB,CAAC,SAAS,CAAC,KAAK,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC;IACjE,CAAC;IAAC,MAAM,CAAC;QACP,OAAO,SAAS,CAAC;IACnB,CAAC;AACH,CAAC;AAED,SAAS,uBAAuB,CAAC,IAAsB,EAAE,QAA4B,EAAE,aAAsB;IAC3G,MAAM,YAAY,GAAG,IAAI,KAAK,SAAS;QACrC,CAAC,CAAC,CAAC,6DAA6D,CAAC;QACjE,CAAC,CAAC;YACE,oEAAoE;YACpE,mBAAmB,aAAa,IAAI,QAAQ,EAAE;SAC/C,CAAC;IACN,OAAO;QACL,GAAG,YAAY;QACf,kEAAkE;QAClE,kEAAkE;QAClE,EAAE;QACF,yBAAyB,CAAC,QAAQ,CAAC;KACpC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;AACf,CAAC;AAED,SAAS,yBAAyB,CAAC,QAA4B;IAC7D,OAAO,QAAQ;SACZ,MAAM,CAAC,CAAC,OAAO,EAAE,EAAE,CAAC,OAAO,CAAC,IAAI,KAAK,MAAM,IAAI,OAAO,CAAC,IAAI,KAAK,WAAW,CAAC;SAC5E,GAAG,CAAC,CAAC,OAAO,EAAE,EAAE;QACf,MAAM,IAAI,GAAG,OAAO,CAAC,MAAM;aACxB,MAAM,CAAC,CAAC,KAAK,EAA2C,EAAE,CAAC,KAAK,CAAC,IAAI,KAAK,MAAM,CAAC;aACjF,GAAG,CAAC,CAAC,KAAK,EAAE,EAAE,CAAC,KAAK,CAAC,IAAI,CAAC;aAC1B,IAAI,CAAC,IAAI,CAAC;aACV,IAAI,EAAE,CAAC;QACV,OAAO,IAAI,CAAC,CAAC,CAAC,GAAG,OAAO,CAAC,IAAI,KAAK,IAAI,EAAE,CAAC,CAAC,CAAC,SAAS,CAAC;IACvD,CAAC,CAAC;SACD,MAAM,CAAC,CAAC,IAAI,EAAkB,EAAE,CAAC,OAAO,CAAC,IAAI,CAAC,CAAC;SAC/C,IAAI,CAAC,IAAI,CAAC;SACV,KAAK,CAAC,CAAC,EAAE,IAAI,CAAC,CAAC;AACpB,CAAC;AAED,SAAS,iBAAiB,CAAC,KAAmB;IAC5C,MAAM,KAAK,GAAG,KAAK,CAAC,aAAa,EAAE,CAAC;IACpC,OAAO,KAAK,CAAC,eAAe,IAAI,KAAK,CAAC,aAAa,CAAC;AACtD,CAAC;AAED,SAAS,uBAAuB,CAAC,KAAa;IAC5C,MAAM,UAAU,GAAG,KAAK;SACrB,OAAO,CAAC,UAAU,EAAE,GAAG,CAAC;SACxB,OAAO,CAAC,gCAAgC,EAAE,EAAE,CAAC;SAC7C,OAAO,CAAC,MAAM,EAAE,GAAG,CAAC;SACpB,IAAI,EAAE,CAAC;IACV,OAAO,UAAU,CAAC,CAAC,CAAC,UAAU,CAAC,KAAK,CAAC,CAAC,EAAE,GAAG,CAAC,CAAC,CAAC,CAAC,SAAS,CAAC;AAC3D,CAAC;AAED,SAAS,0BAA0B;IACjC,MAAM,GAAG,GAAG,OAAO,CAAC,GAAG,CAAC,4BAA4B,CAAC;IACrD,IAAI,GAAG,KAAK,SAAS,IAAI,GAAG,KAAK,EAAE,EAAE,CAAC;QACpC,MAAM,MAAM,GAAG,MAAM,CAAC,GAAG,CAAC,CAAC;QAC3B,IAAI,MAAM,CAAC,QAAQ,CAAC,MAAM,CAAC,IAAI,MAAM,IAAI,CAAC;YAAE,OAAO,IAAI,CAAC,KAAK,CAAC,MAAM,CAAC,CAAC;IACxE,CAAC;IACD,OAAO,8BAA8B,CAAC;AACxC,CAAC;AAED,SAAS,cAAc,CAAC,QAA4B;IAClD,OAAO,QAAQ,CAAC,MAAM,CAAC,CAAC,OAAO,EAAE,EAAE,CAAC,OAAO,CAAC,IAAI,KAAK,MAAM,IAAI,OAAO,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC,KAAK,EAAE,EAAE,CAAC,KAAK,CAAC,IAAI,KAAK,MAAM,CAAC,CAAC,CAAC,MAAM,CAAC;AAC/H,CAAC;AAED,SAAS,oBAAoB,CAAC,SAA6C;IACzE,IAAI,SAAS,KAAK,IAAI;QAAE,OAAO,IAAI,CAAC;IACpC,IAAI,CAAC,SAAS;QAAE,OAAO,SAAS,CAAC;IACjC,OAAO,EAAE,GAAG,SAAS,EAAE,CAAC;AAC1B,CAAC;AAED,SAAS,YAAY,CAAC,OAAgB;IACpC,OAAO,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,SAAS,CAAC,OAAO,CAAC,CAAY,CAAC;AACxD,CAAC"}

package/dist/core/query.d.ts

@@ -0,0 +1,47 @@
+import type { Compactor, ContextBudgetOptions } from "../context/compaction.js";
+import type { ContextManager } from "../context/context-manager.js";
+import type { ModelGateway, ReasoningConfig } from "../model/model-gateway.js";
+import type { ToolRegistry } from "../tools/registry.js";
+import type { CanUseTool, ToolUseContext } from "../tools/tool.js";
+import type { AgentEvent } from "../types/events.js";
+import { type Message } from "../types/messages.js";
+import { type TerminalReason } from "./state.js";
+export interface QueryOptions {
+    agentId: string;
+    model?: string;
+    fallbackModel?: string;
+    reasoning?: ReasoningConfig | null;
+    queryOrigin?: string;
+    maxOutputTokensOverride?: number;
+    maxTurns?: number;
+    skipCacheWrite?: boolean;
+    taskBudget?: {
+        total: number;
+    };
+    abortSignal?: AbortSignal;
+    /** Resolved workspace root for tools (e.g. subagent `cwd` from parent `agent` tool). */
+    workspaceCwd?: string;
+}
+export interface QueryDependencies {
+    modelGateway: ModelGateway;
+    tools: ToolRegistry;
+    contextManager?: ContextManager;
+    compactor?: Compactor;
+    contextBudget?: ContextBudgetOptions;
+    canUseTool?: CanUseTool;
+    maxToolResultSerializedLength?: number;
+    toolResultMemory?: ToolUseContext["toolResultMemory"];
+    recordContentReplacements?: ToolUseContext["recordContentReplacements"];
+    taskNotificationSource?: TaskNotificationSource;
+}
+export interface TaskNotificationSource {
+    collectUnnotifiedCompletions(): {
+        taskId: string;
+        agentId: string;
+        status: string;
+        type: string;
+        content: string;
+    }[];
+    markNotified(taskId: string): void;
+}
+export declare function query(messages: Message[], dependencies: QueryDependencies, options: QueryOptions): AsyncGenerator<AgentEvent, TerminalReason, void>;