@opengeni/runtime 3.3.0 → 3.4.0

package/dist/{chunk-7OUG7WBV.js → chunk-GWRBALGQ.js}

@@ -3,7 +3,7 @@ import {
 } from "./chunk-5ULGMFKR.js";
 import {
   MemorySession
-} from "./chunk-KCPW4K7L.js";
+} from "./chunk-PZQUUBEB.js";
 import {
   BatchTraceProcessor,
   ModelBehaviorError,
@@ -34,7 +34,7 @@ import {
   tool,
   withGenerationSpan,
   withResponseSpan
-} from "./chunk-7LJ7KDPI.js";
+} from "./chunk-ERD4ICVB.js";
 import {
   MCP_MAX_TOOL_DEFINITION_BYTES,
   MCP_MAX_TOOL_SEARCH_DISCLOSURE_BYTES,
@@ -13125,6 +13125,36 @@ function familyCompanions(disclosed, pool) {
   }
   return companions;
 }
+var INTERACTION_ENTRY_TOOLS = /* @__PURE__ */ new Set(["browser_open", "computer_open"]);
+function isInteractionEntryFunctionTool(tool2, mcpServerIds = NO_MCP_SERVER_IDS, modelServerIds) {
+  if (!tool2 || typeof tool2 !== "object" || tool2.type !== "function") {
+    return false;
+  }
+  const name = tool2.name;
+  if (typeof name !== "string" || !isInteractionEntryToolName(name)) return false;
+  const serverId = mcpServerIdForTool(tool2, mcpServerIds, modelServerIds);
+  return serverId === null || serverId === INTERACTION_MCP_SERVER_ID;
+}
+function isInteractionEntryToolName(name) {
+  const short = interactionToolName(name);
+  return short !== null && INTERACTION_ENTRY_TOOLS.has(short);
+}
+function interactionEntryCompanions(entryName, searchable) {
+  if (!isInteractionEntryToolName(entryName)) return [];
+  const pool = searchable.filter(
+    (tool2) => tool2.type === "function" && typeof tool2.name === "string"
+  );
+  const disclosed = [];
+  let disclosedBytes = 0;
+  for (const tool2 of familyCompanions([{ name: entryName }], pool)) {
+    const toolBytes = mcpSerializedSizeBytes(tool2);
+    if (toolBytes > MCP_MAX_TOOL_DEFINITION_BYTES) continue;
+    if (disclosedBytes + toolBytes > MCP_MAX_TOOL_SEARCH_DISCLOSURE_BYTES) continue;
+    disclosedBytes += toolBytes;
+    disclosed.push(tool2);
+  }
+  return disclosed;
+}
 function isHarnessControlMcpFunctionTool(tool2, mcpServerIds = NO_MCP_SERVER_IDS, modelServerIds) {
   const serverId = mcpServerIdForTool(tool2, mcpServerIds, modelServerIds);
   return serverId !== null && isHarnessControlMcpToolName(serverId, tool2.name);
@@ -13421,6 +13451,8 @@ var ALWAYS_VISIBLE_BASE_TOOL_NAMES = /* @__PURE__ */ new Set([
 ]);
 var DISPATCH_MARKER_KEY = "opengeni.lazy_dispatch.v1";
 var SEARCH_MARKER_KEY = "opengeni.lazy_search.v1";
+var FAMILY_DISCLOSURE_MARKER_KEY = "opengeni.family_disclosure.v1";
+var FAMILY_DISCLOSURE_CALL_PREFIX = "opengeni:family:";
 var INTERNAL_REGISTRATION_TOOL_MARKER_KEY = "opengeni.internal_lazy_registration.v1";
 var INTERNAL_DISPATCH_REGISTRATION_CALL_PREFIX = "opengeni:lazy-dispatch:register:";
 var SEARCH_DESCRIPTION = "Search the currently authorized tools by capability. Describe what you need to do in plain language. Returns matching tool names and input schemas. Search is ranked, not exhaustive: if relevant tools are missing, use tool_list to browse, then pass exact names here with query set to an empty string. Names selects exact tools only; it does not broaden authorization.";
@@ -13610,6 +13642,7 @@ var LazyToolRuntime = class {
       if (ALWAYS_VISIBLE_BASE_TOOL_NAMES.has(tool2.name)) continue;
       if (this.preparationIndependentToolNames.has(tool2.name)) continue;
       if (isHarnessControlMcpFunctionTool(tool2, this.mcpServerIds, modelServerIds)) continue;
+      if (isInteractionEntryFunctionTool(tool2, this.mcpServerIds, modelServerIds)) continue;
       const lazy = isSearchableMcpFunctionTool(tool2, this.deferredMcpServerIds, modelServerIds) || !isSearchableMcpFunctionTool(tool2, this.mcpServerIds, modelServerIds);
       if (lazy) {
         tool2.deferLoading = false;
@@ -13666,6 +13699,11 @@ var LazyToolRuntime = class {
   search(rawArguments) {
     return searchToolPool(this.searchableTools(this.currentTools), rawArguments);
   }
+  /** Model-visible schemas an entry-tool call discloses, or [] for any other tool. */
+  familyDisclosure(toolName) {
+    if (!isInteractionEntryToolName(toolName)) return [];
+    return interactionEntryCompanions(toolName, this.searchableTools(this.currentTools)).map((tool2) => modelVisibleToolDefinition(tool2)).filter((definition) => definition !== null);
+  }
   genericSearchOutput(rawArguments) {
     const definitions = [];
     for (const tool2 of this.search(rawArguments)) {
@@ -13876,6 +13914,7 @@ function restoredProviderData(providerData) {
   const restored = { ...providerData };
   delete restored[DISPATCH_MARKER_KEY];
   delete restored[SEARCH_MARKER_KEY];
+  delete restored[FAMILY_DISCLOSURE_MARKER_KEY];
   return Object.keys(restored).length > 0 ? restored : void 0;
 }
 function restoreGenericDispatchHistory(input) {
@@ -13900,7 +13939,8 @@ function restoreGenericDispatchHistoryItem(candidate) {
   const hasDispatchMarker = isRecord7(marker) && marker.version === 1 && typeof marker.arguments === "string";
   const searchMarker = providerData?.[SEARCH_MARKER_KEY];
   const hasSearchMarker = isRecord7(searchMarker) && searchMarker.version === 1 && typeof searchMarker.output === "string";
-  if (!hasDispatchMarker && !hasSearchMarker) {
+  const hasFamilyMarker = providerData !== void 0 && FAMILY_DISCLOSURE_MARKER_KEY in providerData;
+  if (!hasDispatchMarker && !hasSearchMarker && !hasFamilyMarker) {
     return candidate;
   }
   const restored = { ...candidate };
@@ -14012,14 +14052,140 @@ function restoreDirectDeferredCallArgumentTypes(candidate, runtime) {
   const restored = restoreDeferredToolArgumentTypes(candidate.arguments, tool2.parameters);
   return restored === null ? candidate : { ...candidate, arguments: restored };
 }
-function transformGenericDispatchResponse(response, runtime) {
+function transformResponseItem(item, runtime) {
+  const items = runtime.transport === "generic_dispatch" ? transformGenericDispatchCall(item, runtime) : [item];
+  return items.map((candidate) => withFamilyDisclosure(candidate, runtime));
+}
+function transformResponseOutput(output, runtime) {
+  const transformed = output.flatMap((item) => transformResponseItem(item, runtime));
+  const unchanged = transformed.length === output.length && transformed.every((item, index) => item === output[index]);
+  return unchanged ? output : transformed;
+}
+function withFamilyDisclosure(candidate, runtime) {
+  if (!isRecord7(candidate) || candidate.type !== "function_call") return candidate;
+  if (typeof candidate.name !== "string" || !isInteractionEntryToolName(candidate.name)) {
+    return candidate;
+  }
+  const providerData = isRecord7(candidate.providerData) ? candidate.providerData : {};
+  if (FAMILY_DISCLOSURE_MARKER_KEY in providerData) return candidate;
+  const tools = runtime.familyDisclosure(candidate.name);
+  if (tools.length === 0) return candidate;
   return {
-    ...response,
-    output: response.output.flatMap(
-      (item) => transformGenericDispatchCall(item, runtime)
-    )
+    ...candidate,
+    providerData: {
+      ...providerData,
+      [FAMILY_DISCLOSURE_MARKER_KEY]: {
+        version: 1,
+        format: runtime.transport === "generic_dispatch" ? "generic" : "native",
+        tools
+      }
+    }
   };
 }
+function familyDisclosureMarker(candidate) {
+  const providerData = isRecord7(candidate.providerData) ? candidate.providerData : null;
+  const marker = providerData?.[FAMILY_DISCLOSURE_MARKER_KEY];
+  if (!isRecord7(marker) || marker.version !== 1 || marker.format !== "native" && marker.format !== "generic" || !Array.isArray(marker.tools)) {
+    return null;
+  }
+  const tools = marker.tools.filter(
+    (tool2) => isRecord7(tool2) && typeof tool2.name === "string"
+  );
+  return { version: 1, format: marker.format, tools };
+}
+function disclosedToolNames(tools) {
+  if (!Array.isArray(tools)) return [];
+  return tools.flatMap(
+    (tool2) => isRecord7(tool2) && typeof tool2.name === "string" ? [tool2.name] : []
+  );
+}
+function genericSearchOutputNames(output) {
+  const parsed = parseJsonObject(output);
+  return parsed ? disclosedToolNames(parsed.tools) : [];
+}
+function expandInteractionFamilyDisclosures(input) {
+  const pending = /* @__PURE__ */ new Map();
+  const disclosed = /* @__PURE__ */ new Set();
+  let expanded = null;
+  for (let index = 0; index < input.length; index += 1) {
+    const original = input[index];
+    expanded?.push(original);
+    const candidate = original;
+    if (!isRecord7(candidate)) continue;
+    if (candidate.type === "tool_search_output") {
+      for (const name of disclosedToolNames(candidate.tools)) disclosed.add(name);
+      continue;
+    }
+    if (candidate.type === "function_call") {
+      const providerData = isRecord7(candidate.providerData) ? candidate.providerData : null;
+      const searchMarker = providerData?.[SEARCH_MARKER_KEY];
+      if (isRecord7(searchMarker) && typeof searchMarker.output === "string") {
+        for (const name of genericSearchOutputNames(searchMarker.output)) disclosed.add(name);
+      }
+      const id2 = callId2(candidate);
+      const marker2 = familyDisclosureMarker(candidate);
+      if (id2 && marker2) pending.set(id2, marker2);
+      continue;
+    }
+    if (candidate.type === "function_call_result" && candidate.name === TOOL_SEARCH_NAME) {
+      if (typeof candidate.output === "string") {
+        for (const name of genericSearchOutputNames(candidate.output)) disclosed.add(name);
+      }
+      continue;
+    }
+    if (candidate.type !== "function_call_result") continue;
+    const id = callId2(candidate);
+    const marker = id ? pending.get(id) : void 0;
+    if (!id || !marker) continue;
+    pending.delete(id);
+    const tools = marker.tools.filter((tool2) => !disclosed.has(String(tool2.name)));
+    if (tools.length === 0) continue;
+    for (const tool2 of tools) disclosed.add(String(tool2.name));
+    expanded ??= input.slice(0, index + 1);
+    expanded.push(...cachedFamilyDisclosureItems(candidate, id, marker.format, tools));
+  }
+  return expanded ?? input;
+}
+var familyDisclosureItemCache = /* @__PURE__ */ new WeakMap();
+function cachedFamilyDisclosureItems(result, id, format, tools) {
+  const key = `${format}\0${tools.map((tool2) => String(tool2.name)).join("\0")}`;
+  const cached = familyDisclosureItemCache.get(result);
+  if (cached?.key === key) return cached.items;
+  const items = familyDisclosureItems(`${FAMILY_DISCLOSURE_CALL_PREFIX}${id}`, format, tools);
+  familyDisclosureItemCache.set(result, { key, items });
+  return items;
+}
+function familyDisclosureItems(id, format, tools) {
+  const names = tools.map((tool2) => String(tool2.name));
+  if (format === "native") {
+    return [
+      {
+        type: "tool_search_call",
+        call_id: id,
+        status: "completed",
+        execution: "client",
+        arguments: { query: "", names }
+      },
+      { type: "tool_search_output", call_id: id, status: "completed", execution: "client", tools }
+    ];
+  }
+  return [
+    {
+      type: "function_call",
+      callId: id,
+      name: TOOL_SEARCH_NAME,
+      arguments: JSON.stringify({ query: "", names }),
+      status: "completed"
+    },
+    {
+      type: "function_call_result",
+      callId: id,
+      name: TOOL_SEARCH_NAME,
+      status: "completed",
+      output: JSON.stringify({ tools })
+    }
+  ];
+}
 function prepareLazyToolRequest(request, runtime) {
   const input = restoreGenericSearchResults(request.input);
   return {
@@ -14044,7 +14210,7 @@ var LazyToolModel = class {
     if (responseRequiresToolPreparation(response, this.runtime)) {
       await this.runtime.ensurePrepared();
     }
-    return this.runtime.transport === "generic_dispatch" ? transformGenericDispatchResponse(response, this.runtime) : response;
+    return { ...response, output: transformResponseOutput(response.output, this.runtime) };
   }
   async *getStreamedResponse(request) {
     await beforeModelRequest("model_entry");
@@ -14057,12 +14223,13 @@ var LazyToolModel = class {
           await this.runtime.ensurePrepared();
         }
       }
-      const delivered = this.runtime.transport === "generic_dispatch" && event.type === "response_done" ? {
+      const delivered = event.type === "response_done" ? {
         ...event,
         response: {
           ...event.response,
-          output: event.response.output.flatMap(
-            (item) => transformGenericDispatchCall(item, this.runtime)
+          output: transformResponseOutput(
+            event.response.output,
+            this.runtime
           )
         }
       } : event;
@@ -16192,6 +16359,7 @@ export {
   restoreGenericDispatchHistoryItem,
   restoreGenericDispatchHistoryItems,
   restoreDirectDeferredCallArgumentTypes,
+  expandInteractionFamilyDisclosures,
   LazyToolModelProvider,
   AssistantMessagePhaseTracker,
   HUMAN_INPUT_TOOL_NAME,
@@ -16218,4 +16386,4 @@ export {
   normalizeModelCallUsage,
   modelCallUsageTelemetry
 };
-//# sourceMappingURL=chunk-7OUG7WBV.js.map
+//# sourceMappingURL=chunk-GWRBALGQ.js.map