@theokit/sdk 2.11.3 → 2.13.0

package/dist/index.cjs

@@ -13428,6 +13428,35 @@ function toOllamaTools(tools) {
   }));
 }
 
+// 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;
+function extractHermesToolCalls(content, makeId) {
+  const toolCalls = [];
+  for (const block of content.matchAll(HERMES_BLOCK)) {
+    const name = (block[1] ?? "").trim();
+    if (name.length === 0) continue;
+    toolCalls.push({
+      type: "tool_use",
+      id: makeId(),
+      name,
+      input: parseHermesParams(block[2] ?? "")
+    });
+  }
+  const residualText = toolCalls.length === 0 ? content : content.replace(HERMES_BLOCK, "").trim();
+  return { toolCalls, residualText };
+}
+function parseHermesParams(inner) {
+  const input = {};
+  for (const param of inner.matchAll(HERMES_PARAM)) {
+    const key2 = param[1];
+    const value = param[2];
+    if (key2 === void 0 || value === void 0) continue;
+    input[key2.trim()] = value;
+  }
+  return input;
+}
+
 // src/internal/llm/openai.ts
 var OpenAIClient = class {
   constructor(options) {
@@ -13439,6 +13468,14 @@ var OpenAIClient = class {
   name = "openai";
   baseUrl;
   fetchImpl;
+  /**
+   * Whether this client recovers leaked Hermes tool-call dialect from assistant
+   * text (theokit#58 follow-up). Observability for the route→client wiring of
+   * `extractToolCallsFromContent`. @internal
+   */
+  get recoversLeakedToolCalls() {
+    return this.options.extractToolCallsFromContent ?? false;
+  }
   // biome-ignore lint/complexity/noExcessiveCognitiveComplexity: HTTP+SSE handshake + accumulator is intentionally one block
   async *stream(request, signal) {
     const headers = {
@@ -13489,7 +13526,10 @@ var OpenAIClient = class {
         endpoint: "/v1/chat/completions"
       });
     }
-    const accumulator = new OpenAIStreamAccumulator();
+    const accumulator = new OpenAIStreamAccumulator(
+      this.options.extractToolCallsFromContent ?? false,
+      providerId
+    );
     for await (const record of parseSseStream(response.body, signal)) {
       if (record.data === "[DONE]") break;
       let chunk;
@@ -13515,6 +13555,16 @@ var OpenAIClient = class {
   }
 };
 var OpenAIStreamAccumulator = class {
+  /**
+   * @param extractFromContent opt-in leaked-dialect safe-parse (theokit#58). Default false.
+   * @param providerName provider id, used only to label the recovery log line.
+   */
+  constructor(extractFromContent = false, providerName = "openai") {
+    this.extractFromContent = extractFromContent;
+    this.providerName = providerName;
+  }
+  extractFromContent;
+  providerName;
   text = "";
   stopReason = "end_turn";
   inputTokens;
@@ -13580,9 +13630,26 @@ var OpenAIStreamAccumulator = class {
       const input = parseToolArguments(call.args);
       toolCalls.push({ type: "tool_use", id: call.id, name: call.name, input });
     }
+    let text = this.text;
+    let stopReason = this.stopReason;
+    if (this.extractFromContent && toolCalls.length === 0) {
+      const recovered = extractHermesToolCalls(
+        this.text,
+        () => `hermes-${globalThis.crypto.randomUUID()}`
+      );
+      if (recovered.toolCalls.length > 0) {
+        toolCalls.push(...recovered.toolCalls);
+        text = recovered.residualText;
+        stopReason = "tool_use";
+        process.stderr.write(
+          `[theokit-sdk] recovered ${recovered.toolCalls.length} leaked tool call(s) from assistant content (provider="${this.providerName}", names=${recovered.toolCalls.map((c) => c.name).join(",")})
+`
+        );
+      }
+    }
     return makeLlmFinish({
-      stopReason: this.stopReason,
-      text: this.text,
+      stopReason,
+      text,
       toolCalls,
       inputTokens: this.inputTokens,
       outputTokens: this.outputTokens,
@@ -14059,8 +14126,9 @@ function buildChain(options) {
   return clients;
 }
 function buildClient(name, routerOptions) {
-  const profile = getProviderProfile(name);
-  if (profile === void 0) return void 0;
+  const baseProfile = getProviderProfile(name);
+  if (baseProfile === void 0) return void 0;
+  const profile = routerOptions.extractToolCallsFromContent === true && baseProfile.extractToolCallsFromContent !== true ? { ...baseProfile, extractToolCallsFromContent: true } : baseProfile;
   const ambient = currentCredentialPool(name);
   if (ambient !== void 0) {
     return new PoolAwareLlmClient(ambient, (apiKey) => selectTransport(profile, apiKey));
@@ -14162,6 +14230,9 @@ function selectTransport(profile, apiKey) {
     const opts = { apiKey };
     opts.baseUrl = profile.baseUrl;
     opts.providerName = profile.name;
+    if (profile.extractToolCallsFromContent === true) {
+      opts.extractToolCallsFromContent = true;
+    }
     if (profile.name === "openai" && process.env.OPENAI_ORGANIZATION !== void 0) {
       opts.organization = process.env.OPENAI_ORGANIZATION;
     }
@@ -14459,11 +14530,13 @@ function buildLoopInputs(options, runId, userText) {
   const fallback = options.agentOptions.providers?.fallback;
   const apiKeys = options.agentOptions.providers?.apiKeys;
   const credentialPoolStrategy = options.agentOptions.providers?.credentialPoolStrategy;
+  const extractToolCallsFromContent = options.agentOptions.providers?.routes?.[0]?.extractToolCallsFromContent;
   const chain = resolveProviderChain({
     primary,
     ...fallback !== void 0 ? { fallback } : {},
     ...apiKeys !== void 0 ? { apiKeys } : {},
-    ...credentialPoolStrategy !== void 0 ? { credentialPoolStrategy } : {}
+    ...credentialPoolStrategy !== void 0 ? { credentialPoolStrategy } : {},
+    ...extractToolCallsFromContent === true ? { extractToolCallsFromContent: true } : {}
   });
   const llm = chain.length === 1 ? chain[0] : new FallbackLlmClient(chain);
   return {