@opengeni/worker-bundle 2.2.0-canary.36855269964001 → 2.2.0

package/dist/{chunk-ZCCLJK3V.js → chunk-RZAAGRFQ.js}

@@ -1242,7 +1242,7 @@ function createDocumentActivities(services, resolveDocumentServices) {
 // src/activities/goals.ts
 import {
   allowedFirstPartyMcpToolsForSession,
-  configuredStaticUsageLimits as configuredStaticUsageLimits2,
+  configuredStaticUsageLimits,
   policyProviderIdForModel,
   resolveModelProvider,
   resolveTurnExecutionPolicyV1,
@@ -1258,92 +1258,19 @@ import {
 import { isCodexBilledModel } from "@opengeni/codex";
 import {
   enqueueSessionWorkflowWakeIfRunnable,
+  getBillingBalance,
   getWorkspaceModelPolicy,
   getSessionGoal,
+  isCodexBilledTurn,
   materializeGoalContinuation,
-  requireSession
+  requireSession,
+  sumUsageQuantity
 } from "@opengeni/db";
 import {
-  modelFundingForAdmission as modelFundingForAdmission2,
+  modelFundingForAdmission,
   resolveCatalogSettings as resolveCatalogSettings2,
   resolveWorkspaceCatalogSettings
 } from "@opengeni/core";
-
-// src/activities/agent-run-admission.ts
-import { configuredStaticUsageLimits } from "@opengeni/config";
-import { modelFundingForAdmission } from "@opengeni/core";
-import {
-  checkWorkspaceAllowance,
-  getBillingBalance,
-  isCodexBilledTurn,
-  sumUsageQuantity
-} from "@opengeni/db";
-async function agentRunAdmissionDenial(services, input) {
-  const codexBilled = await isCodexBilledTurn({
-    db: services.db,
-    settings: services.settings,
-    workspaceId: input.workspaceId,
-    model: input.model
-  });
-  const externallyBilled = modelFundingForAdmission(
-    services.settings,
-    input.model,
-    codexBilled
-  ).fundedWithoutCredits;
-  if (!externallyBilled && (services.settings.billingMode === "stripe" || services.settings.usageLimitsMode === "managed")) {
-    if (services.entitlements) {
-      const decision = await services.entitlements.admitRun({
-        accountId: input.accountId,
-        workspaceId: input.workspaceId,
-        action: "agent_run:create",
-        quantity: input.requestedAgentRuns
-      });
-      if (!decision.allowed) return "insufficient_credits";
-    } else {
-      const balance = await getBillingBalance(services.db, input.accountId);
-      if (balance.balanceMicros <= 0) return "insufficient_credits";
-    }
-  }
-  if (!externallyBilled) {
-    const refusal = await checkWorkspaceAllowance(services.db, {
-      accountId: input.accountId,
-      workspaceId: input.workspaceId,
-      subjectId: input.initiatingHumanSubjectId ?? null
-    });
-    if (refusal) return refusal.code;
-  }
-  if (services.settings.usageLimitsMode !== "static" && services.settings.usageLimitsMode !== "managed") {
-    return null;
-  }
-  const limits = configuredStaticUsageLimits(services.settings);
-  if (!externallyBilled && limits.maxMonthlyCostMicrosPerAccount) {
-    const used = await sumUsageQuantity(services.db, {
-      accountId: input.accountId,
-      eventType: "model.cost",
-      since: startOfUtcMonth()
-    });
-    if (used >= limits.maxMonthlyCostMicrosPerAccount) {
-      return "monthly_model_cost_limit";
-    }
-  }
-  if (limits.maxMonthlyAgentRunsPerWorkspace) {
-    const used = await sumUsageQuantity(services.db, {
-      workspaceId: input.workspaceId,
-      eventType: "agent_run.created",
-      since: startOfUtcMonth()
-    });
-    if (used + input.requestedAgentRuns > limits.maxMonthlyAgentRunsPerWorkspace) {
-      return "monthly_agent_run_limit";
-    }
-  }
-  return null;
-}
-function startOfUtcMonth() {
-  const now = /* @__PURE__ */ new Date();
-  return new Date(Date.UTC(now.getUTCFullYear(), now.getUTCMonth(), 1));
-}
-
-// src/activities/goals.ts
 function createGoalActivities(services) {
   async function enqueueGoalRetryWake(input) {
     const { db } = await services();
@@ -1390,6 +1317,24 @@ function createGoalActivities(services) {
     if (!modelPolicyBlocked && session.codexCompactionMode === "remote_v2" && !isCodexBilledModel(continuationModel)) {
       modelPolicyBlocked = `session is locked to Codex remote compaction v2; model "${continuationModel}" is not a Codex subscription model`;
     }
+    const isCodexRun = await isCodexBilledTurn({
+      db,
+      settings,
+      workspaceId: input.workspaceId,
+      model: continuationModel
+    });
+    const fundedWithoutCredits = goalContinuationFundedWithoutCredits(
+      settings,
+      continuationModel,
+      isCodexRun
+    );
+    const budgetBlocked = await goalRunBudgetBlocked(
+      settings,
+      db,
+      input.accountId,
+      input.workspaceId,
+      fundedWithoutCredits
+    );
     const turnExecutionPolicy = resolveTurnExecutionPolicyV1(settings, {
       modelId: continuationModel,
       requestedModelId: null,
@@ -1414,21 +1359,7 @@ function createGoalActivities(services) {
       // A model-policy block takes precedence: it is deterministic (a budget
       // pause can clear on its own; a policy pause needs a model/policy change)
       // and rides the same visible-pause channel.
-      admission: async (tx, causalTurn) => {
-        const budgetBlocked = modelPolicyBlocked ? null : await goalRunBudgetBlocked(
-          { ...service, settings, db: tx },
-          {
-            accountId: input.accountId,
-            workspaceId: input.workspaceId,
-            model: continuationModel,
-            initiatingHumanSubjectId: causalTurn?.initiatingHumanSubjectId ?? null
-          }
-        );
-        return {
-          budgetBlocked: modelPolicyBlocked ?? budgetBlocked?.message ?? null,
-          budgetPausedReason: modelPolicyBlocked ? "limits" : budgetBlocked?.pausedReason ?? "limits"
-        };
-      },
+      budgetBlocked: modelPolicyBlocked ?? budgetBlocked,
       policy: {
         model: continuationModel,
         reasoningEffort: continuationReasoningEffort,
@@ -1486,6 +1417,9 @@ function goalContinuationModelDecision(input) {
     blocked: `workspace model policy blocks model "${input.inheritedModel}"; pick an allowed model or change the workspace model policy`
   };
 }
+function goalContinuationFundedWithoutCredits(settings, model, codexBilled) {
+  return modelFundingForAdmission(settings, model, codexBilled).fundedWithoutCredits;
+}
 function goalContinuationPrompt(_goal, _autoContinuation, _cap, options = {}) {
   const waitingGuidance = options.inputWaitAvailable ? [
     "Waiting on child sessions, background commands, or external events:",
@@ -1541,8 +1475,6 @@ function goalContinuationPrompt(_goal, _autoContinuation, _cap, options = {}) {
     "- For document report deliverables (a document the user asked for, or a large report meant to be kept or shared), follow the Documents Skill: create the durable native document first, inspect its relevant final head after the last edit, and provide the returned artifact reference. Declare report requirements through the available goal tools before authoring and satisfy every persisted report requirement with verified artifact delivery evidence before completion. Sandbox paths and raw file IDs do not prove report delivery. If artifact tooling or access is unavailable, keep that deliverable incomplete and state the blocker; never invent proof or silently substitute a local report. Ordinary chat answers, short progress updates, source-code links, and explicitly requested local-file work remain outside this report contract.",
     "",
     "Do not rely on intent, partial progress, memory of earlier work, or a plausible final answer as proof of completion. Call opengeni__goal_complete with concrete evidence only when the full objective is actually achieved and no required work remains.",
-    "Goal evidence is a short proof for the ledger, not the deliverable. After goal_complete succeeds, finish this same turn with the requested user-facing answer, or a concise summary and retained artifact link. Goal completion stops future automatic continuations; it does not send the answer or end this turn. Never compress a report into evidence or omit the final reply.",
-    "Goal progress notes are short human-readable milestone statuses, not raw transcripts or continuation instructions. Keep normal spaces and summarize detail instead of squeezing words into a ledger field. The text and successCriteria fields each allow 8192 UTF-8 bytes, progressNote allows 8192 UTF-8 bytes, rationale allows 2048 UTF-8 bytes, and evidence allows 8192 characters.",
     "",
     ...waitingGuidance,
     ...childNoticeGuidance,
@@ -1562,19 +1494,41 @@ function withFirstPartyTools(settings, tools) {
   }
   return mergeToolRefs(tools, [{ kind: "mcp", id: "opengeni" }]);
 }
-async function goalRunBudgetBlocked(services, input) {
-  const denial = await agentRunAdmissionDenial(services, { ...input, requestedAgentRuns: 1 });
-  if (denial === null) return null;
-  if (denial === "allowance_exhausted") {
-    return { pausedReason: "allowance", message: "OpenGeni usage allowance exhausted" };
-  }
-  const limits = configuredStaticUsageLimits2(services.settings);
-  const messages = {
-    insufficient_credits: "insufficient OpenGeni credits",
-    monthly_model_cost_limit: `monthly model cost limit reached (${limits.maxMonthlyCostMicrosPerAccount} micros)`,
-    monthly_agent_run_limit: `monthly agent run limit reached (${limits.maxMonthlyAgentRunsPerWorkspace})`
-  };
-  return { pausedReason: "limits", message: messages[denial] };
+async function goalRunBudgetBlocked(settings, db, accountId, workspaceId, fundedWithoutCredits) {
+  if (!fundedWithoutCredits && (settings.billingMode === "stripe" || settings.usageLimitsMode === "managed")) {
+    const balance = await getBillingBalance(db, accountId);
+    if (balance.balanceMicros <= 0) {
+      return "insufficient OpenGeni credits";
+    }
+  }
+  if (settings.usageLimitsMode === "static" || settings.usageLimitsMode === "managed") {
+    const limits = configuredStaticUsageLimits(settings);
+    if (!fundedWithoutCredits && limits.maxMonthlyCostMicrosPerAccount) {
+      const used = await sumUsageQuantity(db, {
+        accountId,
+        eventType: "model.cost",
+        since: startOfUtcMonth()
+      });
+      if (used >= limits.maxMonthlyCostMicrosPerAccount) {
+        return `monthly model cost limit reached (${limits.maxMonthlyCostMicrosPerAccount} micros)`;
+      }
+    }
+    if (limits.maxMonthlyAgentRunsPerWorkspace) {
+      const used = await sumUsageQuantity(db, {
+        workspaceId,
+        eventType: "agent_run.created",
+        since: startOfUtcMonth()
+      });
+      if (used + 1 > limits.maxMonthlyAgentRunsPerWorkspace) {
+        return `monthly agent run limit reached (${limits.maxMonthlyAgentRunsPerWorkspace})`;
+      }
+    }
+  }
+  return null;
+}
+function startOfUtcMonth() {
+  const now = /* @__PURE__ */ new Date();
+  return new Date(Date.UTC(now.getUTCFullYear(), now.getUTCMonth(), 1));
 }
 
 // src/activities/knowledge-source-sync.ts
@@ -6672,7 +6626,7 @@ function afterPersistableHomeMutation(services, ids, home) {
     if (!exactAdmission || typeof exactAdmission.id !== "string" || typeof exactAdmission.workspaceGeneration !== "number" || !("providerBinding" in boundAdmission)) {
       throw new Error("Persistable home mutation settlement lacked its bound admission");
     }
-    if ((outcome === "resolved" || outcome === "outcome_unknown") && retainedProcess) {
+    if (outcome === "resolved" && retainedProcess) {
       try {
         await retainWorkspaceProviderCommand(services.db, {
           accountId: fence.accountId,
@@ -6717,8 +6671,6 @@ function afterPersistableHomeMutation(services, ids, home) {
       }
       return;
     }
-    if (outcome === "outcome_unknown")
-      throw new Error("Outcome-unknown command settlement requires its exact retained invocation");
     await verifyWorkspaceMutationSettlement(services.db, {
       accountId: fence.accountId,
       workspaceId: ids.workspaceId,
@@ -8918,7 +8870,6 @@ import {
   isModalTaskExecStartPreDispatchUnavailableError,
   isRoutingMutationOutcomeUnknownError,
   RoutingWorkspaceRootChangedError,
-  ResponsesStreamingTerminalError,
   SandboxMaterializationVerificationError,
   materializationVerificationDiagnostic,
   PROVIDER_QUOTA_EXHAUSTED_CODE,
@@ -9883,7 +9834,6 @@ function shouldRecoverCompactionProviderFailure(error) {
   return agentRunFailurePayload(error).retryable === true;
 }
 function classifyContextWindowOverflowError(error) {
-  if (isProviderSafetyRefusal(error)) return null;
   const fields = collectErrorStrings(error);
   const matched = fields.find(
     (value) => /context[_\s-]*length[_\s-]*exceeded/i.test(value) || /exceeds?\s+(?:the\s+)?context\s+window/i.test(value) || /maximum\s+context\s+length/i.test(value) || /context\s+window[^.]*exceed/i.test(value)
@@ -9929,7 +9879,6 @@ function collectErrorStrings(value, seen = /* @__PURE__ */ new WeakSet()) {
   }
   seen.add(value);
   const out = [];
-  if (value instanceof ResponsesStreamingTerminalError) out.push(value.detail);
   const record = value;
   for (const key of ["message", "code", "type", "name", "param"]) {
     const field = record[key];
@@ -9947,9 +9896,6 @@ function isExactStatuslessUpstreamConnectivityMessage(message) {
   return message.trim().toLowerCase() === STATUSLESS_UPSTREAM_CONNECTIVITY_MESSAGE;
 }
 function providerSafetyRefusalDiagnostic(error) {
-  if (error instanceof ResponsesStreamingTerminalError) {
-    return error.category === "safety" ? error.detail : void 0;
-  }
   return collectErrorStrings(error).find(
     (value) => /^(?:content_policy_violation|content_filter|safety_violation|bio_policy|cyber_policy)$/.test(
       value
@@ -9960,9 +9906,6 @@ function isProviderSafetyRefusal(error) {
   return providerSafetyRefusalDiagnostic(error) !== void 0;
 }
 function isTransientProviderError(error) {
-  if (error instanceof ResponsesStreamingTerminalError) {
-    return error.category === "unavailable";
-  }
   if (isProviderSafetyRefusal(error)) return false;
   const status = typeof error === "object" && error !== null ? Number(
     error.status ?? error.statusCode
@@ -10078,28 +10021,6 @@ function baseAgentRunFailurePayload(error, options = {}) {
       detail: safetyRefusalDiagnostic
     };
   }
-  if (error instanceof ResponsesStreamingTerminalError) {
-    const quota2 = error.category === "unknown" || error.category === "rate_limit" ? classifyProviderQuotaExhaustionError({
-      code: error.code,
-      error: { code: error.code, type: error.type, message: error.detail },
-      retryAfterSeconds: error.retryAfterSeconds
-    }) : null;
-    if (quota2) {
-      return {
-        error: providerQuotaExhaustedMessage(quota2.scope),
-        code: PROVIDER_QUOTA_EXHAUSTED_CODE,
-        retryable: false,
-        quotaScope: quota2.scope,
-        detail: error.detail
-      };
-    }
-    return {
-      error: error.category === "rate_limit" ? "Model provider rate limit hit. Try again in a minute or lower the reasoning effort." : error.category === "unavailable" ? "The model provider is temporarily unavailable. The same turn will retry after a short delay." : "The model provider rejected the response. Automatic retries stopped.",
-      code: error.category === "rate_limit" ? "provider_rate_limited" : error.category === "unavailable" ? "provider_unavailable" : "provider_request_rejected",
-      retryable: error.category === "rate_limit" || error.category === "unavailable",
-      detail: error.detail
-    };
-  }
   const message = error instanceof Error ? error.message : String(error);
   const status = typeof error === "object" && error !== null ? Number(
     error.status ?? error.statusCode
@@ -10405,7 +10326,6 @@ export {
   signalPendingCodexCapacityWakeTargets,
   armAndReconcileCodexCapacityWait,
   createCodexCapacityActivities,
-  agentRunAdmissionDenial,
   createDocumentActivities,
   createGoalActivities,
   withFirstPartyTools,
@@ -10508,4 +10428,4 @@ export {
   codexUsageLimitFailurePayload,
   CODEX_USAGE_LIMIT_MAX_RESUME_MS
 };
-//# sourceMappingURL=chunk-ZCCLJK3V.js.map
+//# sourceMappingURL=chunk-RZAAGRFQ.js.map