@theokit/sdk 5.15.0 → 5.15.1

package/dist/{chunk-DJKGDLYZ.cjs → chunk-3E3VSMB6.cjs}

@@ -814,7 +814,7 @@ async function resolveBedrockToken(region) {
     return cachedToken.value;
   }
   try {
-    const r = module$1.createRequire((typeof document === 'undefined' ? require('u' + 'rl').pathToFileURL(__filename).href : (_documentCurrentScript && _documentCurrentScript.tagName.toUpperCase() === 'SCRIPT' && _documentCurrentScript.src || new URL('chunk-DJKGDLYZ.cjs', document.baseURI).href)));
+    const r = module$1.createRequire((typeof document === 'undefined' ? require('u' + 'rl').pathToFileURL(__filename).href : (_documentCurrentScript && _documentCurrentScript.tagName.toUpperCase() === 'SCRIPT' && _documentCurrentScript.src || new URL('chunk-3E3VSMB6.cjs', document.baseURI).href)));
     const gen = r("@aws/bedrock-token-generator");
     if (typeof gen.getToken !== "function") return void 0;
     const providers = r("@aws-sdk/credential-providers");
@@ -1542,16 +1542,38 @@ function pickString(record, key) {
 // src/internal/llm/hermes-tool-extract.ts
 var HERMES_BLOCK = /<function=\s*([^>\s]+)\s*>([\s\S]*?)<\/tool_call>/g;
 var HERMES_PARAM = /<parameter=\s*([^>\s]+)\s*>([\s\S]*?)<\/parameter>/g;
+var CTRL_CALL_SRC = String.raw`print\(\s*default_api\.([A-Za-z_]\w*)\(([\s\S]*?)\)\s*\)(?=\s*(?:print\(\s*default_api\.|<ctrl\d+>|$))`;
+var CTRL_BLOCK = new RegExp(`<ctrl42>call\\s*((?:${CTRL_CALL_SRC}\\s*)+)(?=<ctrl\\d+>|$)`, "g");
+var CTRL_MARKER = "<ctrl42>call";
+var CTRL_PREFIX = "print(default_api.";
 function extractHermesToolCalls(content, makeId, allowedToolNames) {
   const isPromoted = (name) => name.length > 0 && (allowedToolNames === void 0 || allowedToolNames.has(name));
   const toolCalls = [];
   const droppedNames = [];
+  const formats = /* @__PURE__ */ new Set();
+  for (const block of content.matchAll(CTRL_BLOCK)) {
+    for (const call of (block[1] ?? "").matchAll(new RegExp(CTRL_CALL_SRC, "g"))) {
+      const name = call[1] ?? "";
+      if (!isPromoted(name)) {
+        if (allowedToolNames !== void 0) droppedNames.push(name);
+        continue;
+      }
+      toolCalls.push({
+        type: "tool_use",
+        id: makeId(),
+        name,
+        input: parseCtrlArgs(call[2] ?? "")
+      });
+      formats.add("gemini-ctrl42");
+    }
+  }
   for (const block of content.matchAll(HERMES_BLOCK)) {
     const name = (block[1] ?? "").trim();
     if (!isPromoted(name)) {
       if (name.length > 0 && allowedToolNames !== void 0) droppedNames.push(name);
       continue;
     }
+    formats.add("hermes");
     toolCalls.push({
       type: "tool_use",
       id: makeId(),
@@ -1560,10 +1582,37 @@ function extractHermesToolCalls(content, makeId, allowedToolNames) {
     });
   }
   const residualText = toolCalls.length === 0 ? content : content.replace(
+    CTRL_BLOCK,
+    (_full, calls) => keepUnpromotedCtrl(calls, isPromoted)
+  ).replace(
     HERMES_BLOCK,
     (full, rawName) => isPromoted((rawName ?? "").trim()) ? "" : full
   ).trim();
-  return { toolCalls, residualText, droppedNames };
+  return { toolCalls, residualText, droppedNames, formats: [...formats] };
+}
+function keepUnpromotedCtrl(calls, isPromoted) {
+  const kept = [...calls.matchAll(new RegExp(CTRL_CALL_SRC, "g"))].filter((c) => !isPromoted(c[1] ?? "")).map((c) => c[0]);
+  return kept.length === 0 ? "" : `<ctrl42>call
+${kept.join("\n")}`;
+}
+var CTRL_ARG = /\s*([A-Za-z_]\w*)\s*=\s*("(?:[^"\\]|\\.)*"|'(?:[^'\\]|\\.)*'|[^,]+)\s*(?:,|$)/y;
+function parseCtrlArgs(raw) {
+  const input = {};
+  CTRL_ARG.lastIndex = 0;
+  for (let m = CTRL_ARG.exec(raw); m !== null; m = CTRL_ARG.exec(raw)) {
+    input[m[1]] = parseCtrlValue(m[2].trim());
+  }
+  return chunkYSTXFWQE_cjs.sanitizeToolInput(input, { trim: true }).value;
+}
+function parseCtrlValue(v) {
+  const q = v[0];
+  if ((q === '"' || q === "'") && v.length >= 2 && v.endsWith(q)) {
+    return v.slice(1, -1).replace(/\\(.)/g, (_, c) => c === "n" ? "\n" : c === "t" ? "	" : c);
+  }
+  if (v === "True") return true;
+  if (v === "False") return false;
+  if (v === "None") return null;
+  return /^-?\d+(\.\d+)?$/.test(v) ? Number(v) : v;
 }
 function parseHermesParams(inner) {
   const input = {};
@@ -1581,6 +1630,10 @@ var isStreamWs = (c) => c === " " || c === "	" || c === "\n" || c === "\r";
 function streamToolCallBufferState(held, allowedToolNames, cap = DEFAULT_STREAM_BUFFER_CAP) {
   if (allowedToolNames.size === 0) return "impossible";
   const t = held.trimStart();
+  if (t.startsWith("<c")) return ctrlState(t, held.length, allowedToolNames, cap);
+  return functionMarkerState(held, t, allowedToolNames, cap);
+}
+function functionMarkerState(held, t, allowedToolNames, cap) {
   if (t.length < STREAM_MARKER.length) {
     return STREAM_MARKER.startsWith(t) ? "possible" : "impossible";
   }
@@ -1592,6 +1645,20 @@ function streamToolCallBufferState(held, allowedToolNames, cap = DEFAULT_STREAM_
   if (!nameOk) return "impossible";
   return held.length > cap ? "impossible" : "possible";
 }
+function ctrlState(t, heldLength, allowedToolNames, cap) {
+  if (t.length < CTRL_MARKER.length) return CTRL_MARKER.startsWith(t) ? "possible" : "impossible";
+  if (!t.startsWith(CTRL_MARKER)) return "impossible";
+  const rest = t.slice(CTRL_MARKER.length).trimStart();
+  if (rest.length <= CTRL_PREFIX.length)
+    return CTRL_PREFIX.startsWith(rest) ? "possible" : "impossible";
+  if (!rest.startsWith(CTRL_PREFIX)) return "impossible";
+  const afterPrefix = rest.slice(CTRL_PREFIX.length);
+  const paren = afterPrefix.indexOf("(");
+  const name = paren === -1 ? afterPrefix : afterPrefix.slice(0, paren);
+  if (!/^\w*$/.test(name)) return "impossible";
+  const nameOk = paren === -1 ? someToolNameStartsWith(allowedToolNames, name) : allowedToolNames.has(name);
+  return nameOk && heldLength <= cap ? "possible" : "impossible";
+}
 function parseStreamMarkerName(t) {
   let cursor = STREAM_MARKER.length;
   while (cursor < t.length && isStreamWs(t[cursor])) cursor += 1;
@@ -1947,7 +2014,7 @@ var OpenAIStreamAccumulator = class {
         text = recovered.residualText;
         stopReason = "tool_use";
         chunkRNXXYF22_cjs.diag(
-          `[theokit-sdk] recovered ${recovered.toolCalls.length} leaked tool call(s) from assistant content (provider="${this.providerName}", names=${recovered.toolCalls.map((c) => c.name).join(",")})
+          `[theokit-sdk] recovered ${recovered.toolCalls.length} leaked tool call(s) from assistant content (provider="${this.providerName}", format=${recovered.formats.join(",") || "hermes"}, names=${recovered.toolCalls.map((c) => c.name).join(",")})
 `
         );
       }
@@ -2708,7 +2775,7 @@ async function resolveVertexAccessToken() {
 async function getAuthClientOrThrow() {
   let mod;
   try {
-    const r = module$1.createRequire((typeof document === 'undefined' ? require('u' + 'rl').pathToFileURL(__filename).href : (_documentCurrentScript && _documentCurrentScript.tagName.toUpperCase() === 'SCRIPT' && _documentCurrentScript.src || new URL('chunk-DJKGDLYZ.cjs', document.baseURI).href)));
+    const r = module$1.createRequire((typeof document === 'undefined' ? require('u' + 'rl').pathToFileURL(__filename).href : (_documentCurrentScript && _documentCurrentScript.tagName.toUpperCase() === 'SCRIPT' && _documentCurrentScript.src || new URL('chunk-3E3VSMB6.cjs', document.baseURI).href)));
     mod = r("google-auth-library");
   } catch (err) {
     const isModuleMissing = err instanceof Error && (err.message.includes("Cannot find module") || err.code === "MODULE_NOT_FOUND");
@@ -3561,5 +3628,5 @@ exports.shouldUseRealLocalRuntime = shouldUseRealLocalRuntime;
 exports.trustFilePath = trustFilePath;
 exports.tryFirstEvent = tryFirstEvent;
 exports.validateApiKeyShape = validateApiKeyShape;
-//# sourceMappingURL=chunk-DJKGDLYZ.cjs.map
-//# sourceMappingURL=chunk-DJKGDLYZ.cjs.map
+//# sourceMappingURL=chunk-3E3VSMB6.cjs.map
+//# sourceMappingURL=chunk-3E3VSMB6.cjs.map