@ponythewhite/base-context 1.0.13 → 1.0.14

package/dist/core/agent-session.js

@@ -12,7 +12,7 @@ import { sleep } from "../utils/sleep.js";
 import { AGENT_MESSAGE_CUSTOM_TYPE, AGENT_MESSAGE_RECEIVED_PREVIEW_LABEL, AGENT_MESSAGE_SKILL_NAME, assertAgentMessageQueueCapacity, assertAgentSessionNameAvailable, assertDirectAgentMessageTarget, createAgentMessageHostHandlers, DEFAULT_AGENT_MESSAGE_MAX_PENDING_PER_SESSION, formatAgentSessionNameUnavailable, isAgentSessionMessage, isAgentSessionMessagePrompt, normalizeAgentSessionMessage, parseAgentSessionMessagePromptId, startsAgentRun, } from "./agent-messages.js";
 import { AGENT_OBSERVE_SKILL_NAME, createAgentObserveHostHandlers, normalizeObserveLimit, normalizeObserveMaxChars, ORCHESTRATION_HEARTBEAT_SKILL_NAME, } from "./agent-observe.js";
 import { addLoginGuidanceToAuthError, formatAuthenticationFailedMessage, formatNoApiKeyFoundMessage, formatNoModelSelectedMessage, isLikelyAuthenticationError, } from "./auth-guidance.js";
-import { addAutonomousContinuation, addAutonomousUsage, autonomousStatus, createAutonomousRuntimeState, nextAutonomousContinuation, refreshAutonomousQualityGates, setAutonomousEnabled, } from "./autonomous.js";
+import { addAutonomousContinuation, addAutonomousUsage, autonomousRunLimitReason, autonomousStatus, createAutonomousRuntimeState, nextAutonomousContinuation, refreshAutonomousQualityGates, setAutonomousEnabled, } from "./autonomous.js";
 import { executeBashWithOperations } from "./bash-executor.js";
 import { stringifyBoundedJson } from "./bounded-json.js";
 import { CanonicalContextCompiler, canonicalRecoveryBoundary, getCanonicalEpochContext, getCanonicalMessageSource, getCanonicalViewSelectionSource, getCanonicalViewUnits, MissingRecoveryReplayContractError, prepareCanonicalEpoch, prepareContextModeEpoch, prepareRecoveryCompaction, readCanonicalContextMode, } from "./canonical-context.js";
@@ -2345,6 +2345,9 @@ export class AgentSession {
             "proceed");
     }
     async _getTurnOutcome(context, signal) {
+        // Usage is recorded by message_end. Join it before admitting another native tool turn.
+        if (this._autonomousState.enabled)
+            await this._agentEventQueue;
         const sourceOwner = signal ? this._invocationCompactionOwner : this._captureCompactionOwner();
         if (!sourceOwner ||
             signal?.aborted ||
@@ -2371,6 +2374,8 @@ export class AgentSession {
                 return { kind: "finish" };
             if (!current())
                 return { kind: "cancelled" };
+            if (this._autonomousState.enabled && autonomousRunLimitReason(this._autonomousState))
+                return { kind: "finish" };
             try {
                 if (await this._accountGoalUsageForAssistantMessage(context.message, goalOwner)) {
                     this._assertGoalContinuationOwner(goalOwner);