@opengeni/worker-bundle 2.0.0 → 2.0.1

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

@@ -13,6 +13,7 @@ import type { loadWorkspaceEnvironmentForRunWithCredentials, sandboxEnvironmentF
 import type { AttemptIdentityState, EventingState, ProviderTurnState, SandboxRuntimeState } from "./turn-context.js";
 export type BuildTurnAgentDeps = {
     skillCatalog: NonNullable<BuildAgentOptions["skillCatalog"]>;
+    mcpServers: Settings["mcpServers"];
     input: RunAgentTurnInput;
     db: ActivityServices["db"];
     runtime: ActivityServices["runtime"];

package/dist/activities/agent-turn/tool-environment.d.ts

@@ -125,6 +125,40 @@ export declare function prepareTurnToolPolicy(deps: PrepareTurnToolPolicyDeps):
 export declare function prepareTurnToolRuntime(deps: PrepareTurnToolRuntimeDeps): Promise<{
     attemptConnectorActionBindings: AttemptConnectorActionBinding[];
     connectorActionPolicy: ConnectorActionPolicyHooks;
+    mcpServers: {
+        id: string;
+        name?: string | undefined;
+        url: string;
+        allowedTools?: string[] | undefined;
+        timeoutMs?: number | undefined;
+        cacheToolsList: boolean;
+        requireApproval?: boolean | string[] | undefined;
+        headers?: Record<string, string> | undefined;
+        connectionRef?: {
+            connectionId?: string | undefined;
+            accountSelection?: "all_eligible" | undefined;
+            authoritySource?: "host" | undefined;
+            hostBinding?: {
+                bindingId: string;
+                generation: number;
+            } | {
+                selection: "accepted_turn";
+            } | undefined;
+            provider?: string | undefined;
+            providerDomain: string;
+            kind?: "api_key" | "app_install" | "delegated" | "oauth2" | undefined;
+            scopes?: string[] | undefined;
+            resource?: string | undefined;
+            selectedResources?: {
+                id: string;
+                kind: "repository";
+            }[] | undefined;
+            subjectScope?: "subject" | "workspace" | undefined;
+        } | undefined;
+        operationRecovery?: Record<string, {
+            observerTool: string;
+        }> | undefined;
+    }[];
     generateSessionTitleInParallel: boolean;
     postToolPreparationStartedAt: number;
     preparationIndependentToolNames: string[];

package/dist/activities/codemode-dispatcher.d.ts

@@ -1,4 +1,5 @@
 import { type AttemptToolEnvironment } from "@opengeni/codemode";
+import { type ToolDisplayMetadata } from "@opengeni/contracts";
 import { type Database } from "@opengeni/db";
 import { type EventBus } from "@opengeni/events";
 export type CodemodeDispatcherScope = {
@@ -25,6 +26,7 @@ export declare class CodemodeAttemptDispatcher {
     private readonly scope;
     private readonly turnSignal?;
     private readonly maxConcurrentCalls;
+    private readonly toolDisplayMetadata?;
     private readonly stopController;
     private readonly inFlight;
     private pendingClaims;
@@ -32,7 +34,7 @@ export declare class CodemodeAttemptDispatcher {
     private closing;
     private readonly claimLeaseMs;
     private readonly claimHeartbeatMs;
-    constructor(db: Database, bus: EventBus, environment: AttemptToolEnvironment, scope: CodemodeDispatcherScope, turnSignal?: AbortSignal | undefined, maxConcurrentCalls?: number, timings?: CodemodeDispatcherTimings);
+    constructor(db: Database, bus: EventBus, environment: AttemptToolEnvironment, scope: CodemodeDispatcherScope, turnSignal?: AbortSignal | undefined, maxConcurrentCalls?: number, timings?: CodemodeDispatcherTimings, toolDisplayMetadata?: ((modelName: string) => ToolDisplayMetadata | undefined) | undefined);
     start(): void;
     close(reason?: string): Promise<void>;
     private execute;

package/dist/{activities-control-BOLIY3H4.js → activities-control-QA6OLRQY.js}

@@ -33,7 +33,7 @@ import {
   warnDrainSnapshotFailure,
   withFirstPartyTools,
   workflowIdForSession
-} from "./chunk-TVEE2UUC.js";
+} from "./chunk-M3NVMF2V.js";
 import {
   inspectOpenSandboxKubernetesInventory,
   recordCreditBalanceGauges,
@@ -7498,4 +7498,4 @@ export {
   createControlActivities,
   createControlActivitiesFromServices
 };
-//# sourceMappingURL=activities-control-BOLIY3H4.js.map
+//# sourceMappingURL=activities-control-QA6OLRQY.js.map

package/dist/{activities-turn-2B3WMHBB.js → activities-turn-YVVKCGAK.js}

@@ -86,7 +86,7 @@ import {
   withFirstPartyTools,
   wrapLazyTurnBoxWithRouting,
   wrapTurnBoxWithRouting
-} from "./chunk-TVEE2UUC.js";
+} from "./chunk-M3NVMF2V.js";
 import {
   StreamTimingMetrics,
   makeMachineOpObserver,
@@ -6994,6 +6994,7 @@ import { publishDurableSessionEvents as publishDurableSessionEvents3 } from "@op
 import {
   normalizeModelCallUsage as normalizeModelCallUsage2,
   normalizeSdkEvent,
+  withMcpToolDisplayMetadata,
   extractOpenSuffixFromRunState,
   assertOpenSuffixResumable,
   interruptionKindForCallItem,
@@ -8824,6 +8825,11 @@ async function runTurnStreamAttempt(deps) {
           } : {}
         );
         for (const event of normalized) {
+          if (event.type === "agent.toolCall.created")
+            event.payload = withMcpToolDisplayMetadata(
+              eventing.preparedTools?.mcpServers ?? [],
+              event.payload
+            );
           streamTiming.onEvent(event.type);
           await eventing.batcher.push(event);
         }
@@ -9113,7 +9119,14 @@ async function runTurnStreamAttempt(deps) {
           ...approvals.length > 0 ? [
             {
               type: "session.requiresAction",
-              payload: { approvals }
+              payload: {
+                approvals: approvals.map(
+                  (approval) => withMcpToolDisplayMetadata(
+                    eventing.preparedTools?.mcpServers ?? [],
+                    approval
+                  )
+                )
+              }
             }
           ] : [],
           {
@@ -9407,6 +9420,7 @@ import { appendAndPublishTurnEventsFenced as appendAndPublishTurnEventsFenced2,
 import { linkCurrentSpanToAdmission, turnExecutionTelemetryKey } from "@opengeni/observability";
 import {
   assertTurnExecutionPolicyMatchesConfigV1,
+  settingsForAcceptedSubscriptionTurn,
   resolveTurnExecutionPolicyV1 as resolveTurnExecutionPolicyV12
 } from "@opengeni/config";
 
@@ -9730,7 +9744,7 @@ async function claimTurnAttempt(deps) {
     gatewaySettings,
     claimedPolicy.kind === "valid" ? claimedPolicy.policy.productModelId : turn.model
   );
-  const capabilitySettings = await settingsWithOrganizationProviderCredentials(
+  let capabilitySettings = await settingsWithOrganizationProviderCredentials(
     db,
     input.accountId,
     input.workspaceId,
@@ -9738,7 +9752,6 @@ async function claimTurnAttempt(deps) {
     claimedPolicy.kind === "valid" ? claimedPolicy.policy.productModelId : turn.model
   );
   const codexAppsCredentialId = capabilitySettings.codexConnectedAppsEnabled ? await resolveCodexAppsCredentialIdForRun(db, input.workspaceId) : null;
-  runtime.configure(capabilitySettings);
   const policyForAbsent = claimedPolicy.kind === "valid" ? claimedPolicy.policy : resolveTurnExecutionPolicyV12(capabilitySettings, legacyTurnExecutionPolicyInput(turn));
   const installedPolicy = await installOrReadTurnExecutionPolicyForAttempt(db, {
     accountId: input.accountId,
@@ -9752,6 +9765,16 @@ async function claimTurnAttempt(deps) {
   if (!installedPolicy.accepted) {
     throw new TurnAttemptFencedError(`turn execution policy was fenced: ${installedPolicy.reason}`);
   }
+  capabilitySettings = settingsForAcceptedSubscriptionTurn(
+    capabilitySettings,
+    installedPolicy.policy,
+    {
+      modelId: turn.model,
+      reasoningEffort: turn.reasoningEffort,
+      latencyMode: turn.latencyMode
+    }
+  );
+  runtime.configure(capabilitySettings);
   const verifiedExecutionPolicy = assertTurnExecutionPolicyMatchesConfigV1(
     capabilitySettings,
     installedPolicy.policy,
@@ -14684,7 +14707,8 @@ import {
 } from "@opengeni/db";
 import { publishDurableSessionEvents as publishDurableSessionEvents9 } from "@opengeni/events";
 import {
-  createFirstPartyInteractionAttemptToolDefinitions
+  createFirstPartyInteractionAttemptToolDefinitions,
+  mcpToolDisplayMetadata
 } from "@opengeni/runtime";
 
 // src/activities/google-drive-publication.ts
@@ -30932,13 +30956,14 @@ import {
   publishDurableSessionEvents as publishDurableSessionEvents8
 } from "@opengeni/events";
 var CodemodeAttemptDispatcher = class {
-  constructor(db, bus, environment, scope, turnSignal, maxConcurrentCalls = CODEMODE_MAX_CONCURRENT_CALLS_PER_ATTEMPT, timings = {}) {
+  constructor(db, bus, environment, scope, turnSignal, maxConcurrentCalls = CODEMODE_MAX_CONCURRENT_CALLS_PER_ATTEMPT, timings = {}, toolDisplayMetadata) {
     this.db = db;
     this.bus = bus;
     this.environment = environment;
     this.scope = scope;
     this.turnSignal = turnSignal;
     this.maxConcurrentCalls = maxConcurrentCalls;
+    this.toolDisplayMetadata = toolDisplayMetadata;
     if (environment.catalog.accountId !== scope.accountId || environment.catalog.workspaceId !== scope.workspaceId || environment.catalog.sessionId !== scope.sessionId || environment.catalog.turnId !== scope.turnId || environment.catalog.attemptId !== scope.attemptId || environment.catalog.executionGeneration !== scope.executionGeneration) {
       throw new Error("Codemode dispatcher scope does not match its tool environment");
     }
@@ -31196,6 +31221,10 @@ var CodemodeAttemptDispatcher = class {
             payload: {
               id: operation.operationId,
               name: entry.modelName,
+              display: this.toolDisplayMetadata?.(entry.modelName) ?? {
+                toolName: entry.identity.toolName,
+                ...entry.title ? { title: entry.title } : {}
+              },
               arguments: operation.arguments,
               origin: "codemode",
               subjectId: operation.caller.subjectId,
@@ -32957,7 +32986,10 @@ async function prepareTurnToolRuntime(deps) {
         attemptId: input.attemptId,
         executionGeneration: attempt.executionGeneration
       },
-      cancellationSignal
+      cancellationSignal,
+      void 0,
+      {},
+      (name) => mcpToolDisplayMetadata(tools.mcpServers, name)
     );
     eventing.codemodeDispatcher.start();
   };
@@ -32973,6 +33005,7 @@ async function prepareTurnToolRuntime(deps) {
   return {
     attemptConnectorActionBindings,
     connectorActionPolicy,
+    mcpServers: runSettings.mcpServers,
     generateSessionTitleInParallel: titleToolPlan.generateTitleInParallel,
     postToolPreparationStartedAt,
     preparationIndependentToolNames: [
@@ -34348,6 +34381,7 @@ async function resolveVideoReferenceSandboxAccess(request, sandboxState, sdkOwne
 // src/activities/agent-turn/agent-build.ts
 async function buildTurnAgent(deps) {
   const {
+    mcpServers,
     input,
     db,
     runtime,
@@ -34765,7 +34799,7 @@ async function buildTurnAgent(deps) {
     const agentConstructionStartedAt = performance.now();
     let agentConstructionOutcome = "completed";
     try {
-      return runtime.buildAgent(eventing.modelRunSettings, runtimeResources, {
+      return runtime.buildAgent({ ...eventing.modelRunSettings, mcpServers }, runtimeResources, {
         ...linkedToolAuthority ? {
           authorizeAttemptExecution: async () => {
             const current = await getSessionTurnForAttempt2(
@@ -35942,6 +35976,7 @@ These are publication receipts for the human's earlier answer. Do not call the r
           } = toolRuntime;
           const builtAgent = await buildTurnAgent({
             skillCatalog: toolRuntime.skillCatalog,
+            mcpServers: toolRuntime.mcpServers,
             input,
             db,
             runtime,
@@ -37498,4 +37533,4 @@ export {
   createTurnActivities,
   createTurnActivitiesFromServices
 };
-//# sourceMappingURL=activities-turn-2B3WMHBB.js.map
+//# sourceMappingURL=activities-turn-YVVKCGAK.js.map