@opengeni/worker-bundle 1.1.0 → 1.2.1-canary.37222177070001

package/dist/activities/agent-turn/admission.d.ts

@@ -83,6 +83,7 @@ export declare function ensureRunAllowedBetweenModelCalls(input: {
     chargesOpenGeniCredits: boolean;
     countsTowardTokenCap: boolean;
     initiatingHumanSubjectId: string | null;
+    modelId?: string;
     serializedRunState?: () => string | null;
-}): Promise<void>;
-export declare function ensureRunAllowed(settings: Settings, db: ActivityServices["db"], accountId: string, workspaceId: string, isExternallyBilledTurn: boolean, entitlements?: ActivityServices["entitlements"], chargesOpenGeniCredits?: boolean, countsTowardTokenCap?: boolean, initiatingHumanSubjectId?: string | null): Promise<void>;
+}): Promise<number | undefined>;
+export declare function ensureRunAllowed(settings: Settings, db: ActivityServices["db"], accountId: string, workspaceId: string, isExternallyBilledTurn: boolean, entitlements?: ActivityServices["entitlements"], chargesOpenGeniCredits?: boolean, countsTowardTokenCap?: boolean, initiatingHumanSubjectId?: string | null, modelId?: string): Promise<number | undefined>;

package/dist/activities/agent-turn/compaction-prep.d.ts

@@ -15,6 +15,7 @@ export type CompactionPrepDeps = {
     input: RunAgentTurnInput;
     settings: Settings;
     db: ActivityServices["db"];
+    entitlements?: ActivityServices["entitlements"];
     bus: ActivityServices["bus"];
     observability: ActivityServices["observability"];
     cancellationSignal: AbortSignal | undefined;

package/dist/activities/agent-turn/finalization-monitor.d.ts

@@ -1,4 +1,4 @@
-import type { Observability } from "@opengeni/observability";
+import { type Observability } from "@opengeni/observability";
 import type { TurnHeartbeatDetails } from "../../op-journal.js";
 export declare const TURN_FINALIZATION_STAGES: readonly ["tool_writers", "credential_renewals", "credential_cleanup", "quiescence_receipt", "event_flush", "provider_leases", "workspace_capture", "tool_close", "sandbox_provisioning", "sandbox_rotation", "workspace_snapshot", "sandbox_release"];
 export type TurnFinalizationStage = (typeof TURN_FINALIZATION_STAGES)[number];
@@ -9,6 +9,11 @@ export declare function startTurnFinalizationMonitor(input: {
     details: TurnHeartbeatDetails;
     heartbeat: (details: TurnHeartbeatDetails) => void;
     requestWorkerDrain: () => void;
+    execution?: {
+        workspaceId: string;
+        sessionId: string;
+        attemptId: string;
+    };
     timeoutMs?: number;
     slowAfterMs?: number;
 }): {

package/dist/activities/agent-turn/model-usage.d.ts

@@ -9,6 +9,13 @@ export declare function modelUsageSourceKey(input: {
     dispatchId: string | null;
     positionalKey: string;
 }): string;
+/** Legacy aggregate usage may only debit a policy shared by its completed calls. */
+export declare function aggregateCreditPolicyRevision(input: {
+    responseRevisions: ReadonlySet<number | undefined>;
+    lastAdmittedRevision: number | undefined;
+    chargesOpenGeniCredits: boolean;
+    totalTokens: number | null;
+}): number | undefined;
 export declare function providerContextTokens(usage: {
     inputTokens?: number;
     outputTokens?: number;
@@ -93,6 +100,7 @@ export declare function processModelResponseTerminalEvent(input: {
     externallyBilled: boolean;
     chargesOpenGeniCredits?: boolean;
     countsTowardTokenCap?: boolean;
+    creditPolicyRevision?: number | undefined;
     servingCredentialId: string | null;
     priorSessionCredentialId: string | null;
     emittedSourceKeys: Set<string>;
@@ -139,6 +147,7 @@ export declare function processCompactionModelUsageEvent(input: {
     externallyBilled: boolean;
     chargesOpenGeniCredits?: boolean;
     countsTowardTokenCap?: boolean;
+    creditPolicyRevision?: number | undefined;
     servingCredentialId: string | null;
     priorSessionCredentialId: string | null;
     emittedSourceKeys: Set<string>;
@@ -211,6 +220,7 @@ export declare function recordModelUsageAndDebitCredits(settings: Settings, db:
     externallyBilled: boolean;
     chargesOpenGeniCredits?: boolean;
     countsTowardTokenCap?: boolean;
+    creditPolicyRevision?: number | undefined;
     gatewayBilling?: ModelResponseUsage["gatewayBilling"];
     usage?: ModelUsageInput | ModelCallUsageInput | null;
     normalizedUsage?: ModelCallUsageNormalization;

package/dist/{activities-control-XZAIMFQJ.js → activities-control-QSDPXLLD.js}

@@ -34,7 +34,7 @@ import {
   warnDrainSnapshotFailure,
   withFirstPartyTools,
   workflowIdForSession
-} from "./chunk-OT64XARC.js";
+} from "./chunk-JKMG5ZHH.js";
 import {
   inspectOpenSandboxKubernetesInventory,
   recordActiveUserGauges,
@@ -62,7 +62,7 @@ import {
   recordVerifiedSignupTrialSwitchGauge,
   recordWorkerDeathRecoveryMetrics,
   runtimeMetricsHooksForObservability
-} from "./chunk-3ZTRWDMN.js";
+} from "./chunk-XONQEVAD.js";
 import "./chunk-PZ5AY32C.js";
 
 // src/activities/knowledge-indexing.ts
@@ -78,7 +78,7 @@ import {
   creditDebitAttributionMetadata,
   deferKnowledgeIndexJob,
   freezeKnowledgeIndexBillingMode,
-  getBillingBalance,
+  getSpendableCreditBalance,
   guardPaidKnowledgeIndexPublication,
   knowledgeIndexBillingActivationTime,
   readKnowledgeIndexSource,
@@ -229,7 +229,7 @@ function createKnowledgeIndexingActivities(services, resolveDocumentServices) {
                 return;
               }
               if (paid && current.nextIndex === 0) {
-                const balance = await getBillingBalance(lockedDb, claim.accountId);
+                const balance = await getSpendableCreditBalance(lockedDb, claim.accountId);
                 if (balance.balanceMicros <= 0) {
                   await waitKnowledgeIndexForFunding(lockedDb, claim);
                   result.deferred++;
@@ -8813,4 +8813,4 @@ export {
   createControlActivities,
   createControlActivitiesFromServices
 };
-//# sourceMappingURL=activities-control-XZAIMFQJ.js.map
+//# sourceMappingURL=activities-control-QSDPXLLD.js.map