@probelabs/probe 0.6.0-rc137 → 0.6.0-rc139

package/cjs/index.cjs

@@ -1038,15 +1038,20 @@ async function extract(options) {
   if (!options) {
     throw new Error("Options object is required");
   }
-  if ((!options.files || !Array.isArray(options.files) || options.files.length === 0) && !options.inputFile) {
-    throw new Error("Either files array or inputFile must be provided");
+  const hasFiles = options.files && Array.isArray(options.files) && options.files.length > 0;
+  const hasInputFile = !!options.inputFile;
+  const hasContent = options.content !== void 0 && options.content !== null;
+  if (!hasFiles && !hasInputFile && !hasContent) {
+    throw new Error("Either files array, inputFile, or content must be provided");
   }
   const binaryPath = await getBinaryPath(options.binaryOptions || {});
-  const cliArgs = buildCliArgs(options, EXTRACT_FLAG_MAP);
+  const filteredOptions = { ...options };
+  delete filteredOptions.content;
+  const cliArgs = buildCliArgs(filteredOptions, EXTRACT_FLAG_MAP);
   if (options.json && !options.format) {
     cliArgs.push("--format", "json");
   }
-  if (options.files && Array.isArray(options.files) && options.files.length > 0) {
+  if (hasFiles) {
     for (const file of options.files) {
       cliArgs.push(escapeString(file));
     }
@@ -1058,66 +1063,115 @@ Extract:`;
       logMessage += ` files="${options.files.join(", ")}"`;
     }
     if (options.inputFile) logMessage += ` inputFile="${options.inputFile}"`;
+    if (options.content) logMessage += ` content=(${typeof options.content === "string" ? options.content.length : options.content.byteLength} bytes)`;
     if (options.allowTests) logMessage += " allowTests=true";
     if (options.contextLines) logMessage += ` contextLines=${options.contextLines}`;
     if (options.format) logMessage += ` format=${options.format}`;
     if (options.json) logMessage += " json=true";
     console.error(logMessage);
   }
+  if (hasContent) {
+    return extractWithStdin(binaryPath, cliArgs, options.content, options);
+  }
   const command = `${binaryPath} extract ${cliArgs.join(" ")}`;
   try {
     const { stdout, stderr } = await execAsync3(command);
     if (stderr) {
       console.error(`stderr: ${stderr}`);
     }
-    let tokenUsage = {
-      requestTokens: 0,
-      responseTokens: 0,
-      totalTokens: 0
-    };
-    if (options.files && Array.isArray(options.files)) {
-      tokenUsage.requestTokens = options.files.join(" ").length / 4;
-    } else if (options.inputFile) {
-      tokenUsage.requestTokens = options.inputFile.length / 4;
-    }
-    if (stdout.includes("Total tokens returned:")) {
-      const tokenMatch = stdout.match(/Total tokens returned: (\d+)/);
-      if (tokenMatch && tokenMatch[1]) {
-        tokenUsage.responseTokens = parseInt(tokenMatch[1], 10);
-        tokenUsage.totalTokens = tokenUsage.requestTokens + tokenUsage.responseTokens;
+    return processExtractOutput(stdout, options);
+  } catch (error2) {
+    const errorMessage = `Error executing extract command: ${error2.message}
+Command: ${command}`;
+    throw new Error(errorMessage);
+  }
+}
+function extractWithStdin(binaryPath, cliArgs, content, options) {
+  return new Promise((resolve4, reject2) => {
+    const process2 = (0, import_child_process4.spawn)(binaryPath, ["extract", ...cliArgs], {
+      stdio: ["pipe", "pipe", "pipe"]
+    });
+    let stdout = "";
+    let stderr = "";
+    process2.stdout.on("data", (data2) => {
+      stdout += data2.toString();
+    });
+    process2.stderr.on("data", (data2) => {
+      stderr += data2.toString();
+    });
+    process2.on("close", (code) => {
+      if (stderr && process2.env.DEBUG === "1") {
+        console.error(`stderr: ${stderr}`);
+      }
+      if (code !== 0) {
+        reject2(new Error(`Extract command failed with exit code ${code}: ${stderr}`));
+        return;
+      }
+      try {
+        const result = processExtractOutput(stdout, options);
+        resolve4(result);
+      } catch (error2) {
+        reject2(error2);
       }
+    });
+    process2.on("error", (error2) => {
+      reject2(new Error(`Failed to spawn extract process: ${error2.message}`));
+    });
+    if (typeof content === "string") {
+      process2.stdin.write(content);
+    } else {
+      process2.stdin.write(content);
     }
-    let output = stdout;
-    if (!output.includes("Token Usage:")) {
-      output += `
+    process2.stdin.end();
+  });
+}
+function processExtractOutput(stdout, options) {
+  let tokenUsage = {
+    requestTokens: 0,
+    responseTokens: 0,
+    totalTokens: 0
+  };
+  if (options.files && Array.isArray(options.files)) {
+    tokenUsage.requestTokens = options.files.join(" ").length / 4;
+  } else if (options.inputFile) {
+    tokenUsage.requestTokens = options.inputFile.length / 4;
+  } else if (options.content) {
+    const contentLength = typeof options.content === "string" ? options.content.length : options.content.byteLength;
+    tokenUsage.requestTokens = contentLength / 4;
+  }
+  if (stdout.includes("Total tokens returned:")) {
+    const tokenMatch = stdout.match(/Total tokens returned: (\d+)/);
+    if (tokenMatch && tokenMatch[1]) {
+      tokenUsage.responseTokens = parseInt(tokenMatch[1], 10);
+      tokenUsage.totalTokens = tokenUsage.requestTokens + tokenUsage.responseTokens;
+    }
+  }
+  let output = stdout;
+  if (!output.includes("Token Usage:")) {
+    output += `
 Token Usage:
   Request tokens: ${tokenUsage.requestTokens}
   Response tokens: ${tokenUsage.responseTokens}
   Total tokens: ${tokenUsage.totalTokens}
 `;
-    }
-    if (options.json || options.format === "json") {
-      try {
-        const jsonOutput = JSON.parse(stdout);
-        if (!jsonOutput.token_usage) {
-          jsonOutput.token_usage = {
-            request_tokens: tokenUsage.requestTokens,
-            response_tokens: tokenUsage.responseTokens,
-            total_tokens: tokenUsage.totalTokens
-          };
-        }
-        return jsonOutput;
-      } catch (error2) {
-        console.error("Error parsing JSON output:", error2);
-        return output;
+  }
+  if (options.json || options.format === "json") {
+    try {
+      const jsonOutput = JSON.parse(stdout);
+      if (!jsonOutput.token_usage) {
+        jsonOutput.token_usage = {
+          request_tokens: tokenUsage.requestTokens,
+          response_tokens: tokenUsage.responseTokens,
+          total_tokens: tokenUsage.totalTokens
+        };
       }
+      return jsonOutput;
+    } catch (error2) {
+      console.error("Error parsing JSON output:", error2);
+      return output;
     }
-    return output;
-  } catch (error2) {
-    const errorMessage = `Error executing extract command: ${error2.message}
-Command: ${command}`;
-    throw new Error(errorMessage);
   }
+  return output;
 }
 var import_child_process4, import_util4, execAsync3, EXTRACT_FLAG_MAP;
 var init_extract = __esm({
@@ -10788,22 +10842,13 @@ var require_dist_cjs18 = __commonJS({
     var sdkStreamMixin2 = require_sdk_stream_mixin();
     var splitStream = require_splitStream();
     var streamTypeCheck = require_stream_type_check();
-    function transformToString(payload2, encoding = "utf-8") {
-      if (encoding === "base64") {
-        return utilBase64.toBase64(payload2);
-      }
-      return utilUtf8.toUtf8(payload2);
-    }
-    function transformFromString(str, encoding) {
-      if (encoding === "base64") {
-        return Uint8ArrayBlobAdapter2.mutate(utilBase64.fromBase64(str));
-      }
-      return Uint8ArrayBlobAdapter2.mutate(utilUtf8.fromUtf8(str));
-    }
     var Uint8ArrayBlobAdapter2 = class _Uint8ArrayBlobAdapter extends Uint8Array {
       static fromString(source, encoding = "utf-8") {
         if (typeof source === "string") {
-          return transformFromString(source, encoding);
+          if (encoding === "base64") {
+            return _Uint8ArrayBlobAdapter.mutate(utilBase64.fromBase64(source));
+          }
+          return _Uint8ArrayBlobAdapter.mutate(utilUtf8.fromUtf8(source));
         }
         throw new Error(`Unsupported conversion from ${typeof source} to Uint8ArrayBlobAdapter.`);
       }
@@ -10812,7 +10857,10 @@ var require_dist_cjs18 = __commonJS({
         return source;
       }
       transformToString(encoding = "utf-8") {
-        return transformToString(this, encoding);
+        if (encoding === "base64") {
+          return utilBase64.toBase64(this);
+        }
+        return utilUtf8.toUtf8(this);
       }
     };
     exports2.Uint8ArrayBlobAdapter = Uint8ArrayBlobAdapter2;
@@ -10925,6 +10973,138 @@ var init_deref = __esm({
   }
 });
 
+// node_modules/@smithy/core/dist-es/submodules/schema/schemas/operation.js
+var operation;
+var init_operation = __esm({
+  "node_modules/@smithy/core/dist-es/submodules/schema/schemas/operation.js"() {
+    operation = (namespace, name14, traits, input, output) => ({
+      name: name14,
+      namespace,
+      traits,
+      input,
+      output
+    });
+  }
+});
+
+// node_modules/@smithy/core/dist-es/submodules/schema/middleware/schemaDeserializationMiddleware.js
+var import_protocol_http2, import_util_middleware3, schemaDeserializationMiddleware, findHeader;
+var init_schemaDeserializationMiddleware = __esm({
+  "node_modules/@smithy/core/dist-es/submodules/schema/middleware/schemaDeserializationMiddleware.js"() {
+    import_protocol_http2 = __toESM(require_dist_cjs2());
+    import_util_middleware3 = __toESM(require_dist_cjs7());
+    init_operation();
+    schemaDeserializationMiddleware = (config) => (next, context3) => async (args) => {
+      const { response } = await next(args);
+      const { operationSchema } = (0, import_util_middleware3.getSmithyContext)(context3);
+      const [, ns, n3, t3, i3, o3] = operationSchema ?? [];
+      try {
+        const parsed = await config.protocol.deserializeResponse(operation(ns, n3, t3, i3, o3), {
+          ...config,
+          ...context3
+        }, response);
+        return {
+          response,
+          output: parsed
+        };
+      } catch (error2) {
+        Object.defineProperty(error2, "$response", {
+          value: response
+        });
+        if (!("$metadata" in error2)) {
+          const hint = `Deserialization error: to see the raw response, inspect the hidden field {error}.$response on this object.`;
+          try {
+            error2.message += "\n  " + hint;
+          } catch (e3) {
+            if (!context3.logger || context3.logger?.constructor?.name === "NoOpLogger") {
+              console.warn(hint);
+            } else {
+              context3.logger?.warn?.(hint);
+            }
+          }
+          if (typeof error2.$responseBodyText !== "undefined") {
+            if (error2.$response) {
+              error2.$response.body = error2.$responseBodyText;
+            }
+          }
+          try {
+            if (import_protocol_http2.HttpResponse.isInstance(response)) {
+              const { headers = {} } = response;
+              const headerEntries = Object.entries(headers);
+              error2.$metadata = {
+                httpStatusCode: response.statusCode,
+                requestId: findHeader(/^x-[\w-]+-request-?id$/, headerEntries),
+                extendedRequestId: findHeader(/^x-[\w-]+-id-2$/, headerEntries),
+                cfId: findHeader(/^x-[\w-]+-cf-id$/, headerEntries)
+              };
+            }
+          } catch (e3) {
+          }
+        }
+        throw error2;
+      }
+    };
+    findHeader = (pattern, headers) => {
+      return (headers.find(([k3]) => {
+        return k3.match(pattern);
+      }) || [void 0, void 0])[1];
+    };
+  }
+});
+
+// node_modules/@smithy/core/dist-es/submodules/schema/middleware/schemaSerializationMiddleware.js
+var import_util_middleware4, schemaSerializationMiddleware;
+var init_schemaSerializationMiddleware = __esm({
+  "node_modules/@smithy/core/dist-es/submodules/schema/middleware/schemaSerializationMiddleware.js"() {
+    import_util_middleware4 = __toESM(require_dist_cjs7());
+    init_operation();
+    schemaSerializationMiddleware = (config) => (next, context3) => async (args) => {
+      const { operationSchema } = (0, import_util_middleware4.getSmithyContext)(context3);
+      const [, ns, n3, t3, i3, o3] = operationSchema ?? [];
+      const endpoint = context3.endpointV2?.url && config.urlParser ? async () => config.urlParser(context3.endpointV2.url) : config.endpoint;
+      const request = await config.protocol.serializeRequest(operation(ns, n3, t3, i3, o3), args.input, {
+        ...config,
+        ...context3,
+        endpoint
+      });
+      return next({
+        ...args,
+        request
+      });
+    };
+  }
+});
+
+// node_modules/@smithy/core/dist-es/submodules/schema/middleware/getSchemaSerdePlugin.js
+function getSchemaSerdePlugin(config) {
+  return {
+    applyToStack: (commandStack) => {
+      commandStack.add(schemaSerializationMiddleware(config), serializerMiddlewareOption2);
+      commandStack.add(schemaDeserializationMiddleware(config), deserializerMiddlewareOption);
+      config.protocol.setSerdeContext(config);
+    }
+  };
+}
+var deserializerMiddlewareOption, serializerMiddlewareOption2;
+var init_getSchemaSerdePlugin = __esm({
+  "node_modules/@smithy/core/dist-es/submodules/schema/middleware/getSchemaSerdePlugin.js"() {
+    init_schemaDeserializationMiddleware();
+    init_schemaSerializationMiddleware();
+    deserializerMiddlewareOption = {
+      name: "deserializerMiddleware",
+      step: "deserialize",
+      tags: ["DESERIALIZER"],
+      override: true
+    };
+    serializerMiddlewareOption2 = {
+      name: "serializerMiddleware",
+      step: "serialize",
+      tags: ["SERIALIZER"],
+      override: true
+    };
+  }
+});
+
 // node_modules/@smithy/core/dist-es/submodules/schema/schemas/Schema.js
 var Schema;
 var init_Schema = __esm({
@@ -10952,51 +11132,6 @@ var init_Schema = __esm({
   }
 });
 
-// node_modules/@smithy/core/dist-es/submodules/schema/schemas/StructureSchema.js
-var StructureSchema, struct;
-var init_StructureSchema = __esm({
-  "node_modules/@smithy/core/dist-es/submodules/schema/schemas/StructureSchema.js"() {
-    init_Schema();
-    StructureSchema = class _StructureSchema extends Schema {
-      static symbol = Symbol.for("@smithy/str");
-      name;
-      traits;
-      memberNames;
-      memberList;
-      symbol = _StructureSchema.symbol;
-    };
-    struct = (namespace, name14, traits, memberNames, memberList) => Schema.assign(new StructureSchema(), {
-      name: name14,
-      namespace,
-      traits,
-      memberNames,
-      memberList
-    });
-  }
-});
-
-// node_modules/@smithy/core/dist-es/submodules/schema/schemas/ErrorSchema.js
-var ErrorSchema, error;
-var init_ErrorSchema = __esm({
-  "node_modules/@smithy/core/dist-es/submodules/schema/schemas/ErrorSchema.js"() {
-    init_Schema();
-    init_StructureSchema();
-    ErrorSchema = class _ErrorSchema extends StructureSchema {
-      static symbol = Symbol.for("@smithy/err");
-      ctor;
-      symbol = _ErrorSchema.symbol;
-    };
-    error = (namespace, name14, traits, memberNames, memberList, ctor) => Schema.assign(new ErrorSchema(), {
-      name: name14,
-      namespace,
-      traits,
-      memberNames,
-      memberList,
-      ctor: null
-    });
-  }
-});
-
 // node_modules/@smithy/core/dist-es/submodules/schema/schemas/ListSchema.js
 var ListSchema, list;
 var init_ListSchema = __esm({
@@ -11064,29 +11199,47 @@ var init_OperationSchema = __esm({
   }
 });
 
-// node_modules/@smithy/core/dist-es/submodules/schema/schemas/SimpleSchema.js
-var SimpleSchema, sim, simAdapter;
-var init_SimpleSchema = __esm({
-  "node_modules/@smithy/core/dist-es/submodules/schema/schemas/SimpleSchema.js"() {
+// node_modules/@smithy/core/dist-es/submodules/schema/schemas/StructureSchema.js
+var StructureSchema, struct;
+var init_StructureSchema = __esm({
+  "node_modules/@smithy/core/dist-es/submodules/schema/schemas/StructureSchema.js"() {
     init_Schema();
-    SimpleSchema = class _SimpleSchema extends Schema {
-      static symbol = Symbol.for("@smithy/sim");
+    StructureSchema = class _StructureSchema extends Schema {
+      static symbol = Symbol.for("@smithy/str");
       name;
-      schemaRef;
       traits;
-      symbol = _SimpleSchema.symbol;
+      memberNames;
+      memberList;
+      symbol = _StructureSchema.symbol;
     };
-    sim = (namespace, name14, schemaRef, traits) => Schema.assign(new SimpleSchema(), {
+    struct = (namespace, name14, traits, memberNames, memberList) => Schema.assign(new StructureSchema(), {
       name: name14,
       namespace,
       traits,
-      schemaRef
+      memberNames,
+      memberList
     });
-    simAdapter = (namespace, name14, traits, schemaRef) => Schema.assign(new SimpleSchema(), {
+  }
+});
+
+// node_modules/@smithy/core/dist-es/submodules/schema/schemas/ErrorSchema.js
+var ErrorSchema, error;
+var init_ErrorSchema = __esm({
+  "node_modules/@smithy/core/dist-es/submodules/schema/schemas/ErrorSchema.js"() {
+    init_Schema();
+    init_StructureSchema();
+    ErrorSchema = class _ErrorSchema extends StructureSchema {
+      static symbol = Symbol.for("@smithy/err");
+      ctor;
+      symbol = _ErrorSchema.symbol;
+    };
+    error = (namespace, name14, traits, memberNames, memberList, ctor) => Schema.assign(new ErrorSchema(), {
       name: name14,
       namespace,
       traits,
-      schemaRef
+      memberNames,
+      memberList,
+      ctor: null
     });
   }
 });
@@ -11130,28 +11283,10 @@ function member(memberSchema, memberName) {
   const internalCtorAccess = NormalizedSchema;
   return new internalCtorAccess(memberSchema, memberName);
 }
-function hydrate(ss) {
-  const [id, ...rest] = ss;
-  return {
-    [0]: simAdapter,
-    [1]: list,
-    [2]: map,
-    [3]: struct,
-    [-3]: error,
-    [9]: op
-  }[id].call(null, ...rest);
-}
 var NormalizedSchema, isMemberSchema, isStaticSchema;
 var init_NormalizedSchema = __esm({
   "node_modules/@smithy/core/dist-es/submodules/schema/schemas/NormalizedSchema.js"() {
     init_deref();
-    init_ErrorSchema();
-    init_ListSchema();
-    init_MapSchema();
-    init_OperationSchema();
-    init_Schema();
-    init_SimpleSchema();
-    init_StructureSchema();
     init_translateTraits();
     NormalizedSchema = class _NormalizedSchema {
       ref;
@@ -11177,8 +11312,6 @@ var init_NormalizedSchema = __esm({
           schema = deref(_ref);
           this._isMemberSchema = true;
         }
-        if (isStaticSchema(schema))
-          schema = hydrate(schema);
         if (traitStack.length > 0) {
           this.memberTraits = {};
           for (let i3 = traitStack.length - 1; i3 >= 0; --i3) {
@@ -11197,18 +11330,24 @@ var init_NormalizedSchema = __esm({
           return;
         }
         this.schema = deref(schema);
-        if (this.schema && typeof this.schema === "object") {
-          this.traits = this.schema?.traits ?? {};
+        if (isStaticSchema(this.schema)) {
+          this.name = `${this.schema[1]}#${this.schema[2]}`;
+          this.traits = this.schema[3];
         } else {
+          this.name = this.memberName ?? String(schema);
           this.traits = 0;
         }
-        this.name = (this.schema instanceof Schema ? this.schema.getName?.() : void 0) ?? this.memberName ?? String(schema);
         if (this._isMemberSchema && !memberName) {
           throw new Error(`@smithy/core/schema - NormalizedSchema member init ${this.getName(true)} missing member name.`);
         }
       }
       static [Symbol.hasInstance](lhs) {
-        return Schema[Symbol.hasInstance].bind(this)(lhs);
+        const isPrototype2 = this.prototype.isPrototypeOf(lhs);
+        if (!isPrototype2 && typeof lhs === "object" && lhs !== null) {
+          const ns = lhs;
+          return ns.symbol === this.symbol;
+        }
+        return isPrototype2;
       }
       static of(ref) {
         const sc = deref(ref);
@@ -11226,7 +11365,11 @@ var init_NormalizedSchema = __esm({
         return new _NormalizedSchema(sc);
       }
       getSchema() {
-        return deref(this.schema?.schemaRef ?? this.schema);
+        const sc = this.schema;
+        if (sc[0] === 0) {
+          return sc[4];
+        }
+        return sc;
       }
       getName(withNamespace = false) {
         const { name: name14 } = this;
@@ -11241,15 +11384,15 @@ var init_NormalizedSchema = __esm({
       }
       isListSchema() {
         const sc = this.getSchema();
-        return typeof sc === "number" ? sc >= 64 && sc < 128 : sc instanceof ListSchema;
+        return typeof sc === "number" ? sc >= 64 && sc < 128 : sc[0] === 1;
       }
       isMapSchema() {
         const sc = this.getSchema();
-        return typeof sc === "number" ? sc >= 128 && sc <= 255 : sc instanceof MapSchema;
+        return typeof sc === "number" ? sc >= 128 && sc <= 255 : sc[0] === 2;
       }
       isStructSchema() {
         const sc = this.getSchema();
-        return sc !== null && typeof sc === "object" && "members" in sc || sc instanceof StructureSchema;
+        return sc[0] === 3 || sc[0] === -3;
       }
       isBlobSchema() {
         const sc = this.getSchema();
@@ -11307,13 +11450,13 @@ var init_NormalizedSchema = __esm({
           throw new Error(`@smithy/core/schema - cannot get key for non-map: ${this.getName(true)}`);
         }
         const schema = this.getSchema();
-        const memberSchema = isDoc ? 15 : schema?.keySchema ?? 0;
+        const memberSchema = isDoc ? 15 : schema[4] ?? 0;
         return member([memberSchema, 0], "key");
       }
       getValueSchema() {
         const sc = this.getSchema();
         const [isDoc, isMap2, isList] = [this.isDocumentSchema(), this.isMapSchema(), this.isListSchema()];
-        const memberSchema = typeof sc === "number" ? 63 & sc : sc && typeof sc === "object" && (isMap2 || isList) ? sc.valueSchema : isDoc ? 15 : void 0;
+        const memberSchema = typeof sc === "number" ? 63 & sc : sc && typeof sc === "object" && (isMap2 || isList) ? sc[3 + sc[0]] : isDoc ? 15 : void 0;
         if (memberSchema != null) {
           return member([memberSchema, 0], isMap2 ? "value" : "member");
         }
@@ -11321,9 +11464,9 @@ var init_NormalizedSchema = __esm({
       }
       getMemberSchema(memberName) {
         const struct2 = this.getSchema();
-        if (this.isStructSchema() && struct2.memberNames.includes(memberName)) {
-          const i3 = struct2.memberNames.indexOf(memberName);
-          const memberSchema = struct2.memberList[i3];
+        if (this.isStructSchema() && struct2[4].includes(memberName)) {
+          const i3 = struct2[4].indexOf(memberName);
+          const memberSchema = struct2[5][i3];
           return member(isMemberSchema(memberSchema) ? memberSchema : [memberSchema, 0], memberName);
         }
         if (this.isDocumentSchema()) {
@@ -11359,8 +11502,8 @@ var init_NormalizedSchema = __esm({
           throw new Error("@smithy/core/schema - cannot iterate non-struct schema.");
         }
         const struct2 = this.getSchema();
-        for (let i3 = 0; i3 < struct2.memberNames.length; ++i3) {
-          yield [struct2.memberNames[i3], member([struct2.memberList[i3], 0], struct2.memberNames[i3])];
+        for (let i3 = 0; i3 < struct2[4].length; ++i3) {
+          yield [struct2[4][i3], member([struct2[5][i3], 0], struct2[4][i3])];
         }
       }
     };
@@ -11369,125 +11512,30 @@ var init_NormalizedSchema = __esm({
   }
 });
 
-// node_modules/@smithy/core/dist-es/submodules/schema/middleware/schemaDeserializationMiddleware.js
-var import_protocol_http2, import_util_middleware3, schemaDeserializationMiddleware, findHeader;
-var init_schemaDeserializationMiddleware = __esm({
-  "node_modules/@smithy/core/dist-es/submodules/schema/middleware/schemaDeserializationMiddleware.js"() {
-    import_protocol_http2 = __toESM(require_dist_cjs2());
-    import_util_middleware3 = __toESM(require_dist_cjs7());
-    init_NormalizedSchema();
-    schemaDeserializationMiddleware = (config) => (next, context3) => async (args) => {
-      const { response } = await next(args);
-      let { operationSchema } = (0, import_util_middleware3.getSmithyContext)(context3);
-      if (isStaticSchema(operationSchema)) {
-        operationSchema = hydrate(operationSchema);
-      }
-      try {
-        const parsed = await config.protocol.deserializeResponse(operationSchema, {
-          ...config,
-          ...context3
-        }, response);
-        return {
-          response,
-          output: parsed
-        };
-      } catch (error2) {
-        Object.defineProperty(error2, "$response", {
-          value: response
-        });
-        if (!("$metadata" in error2)) {
-          const hint = `Deserialization error: to see the raw response, inspect the hidden field {error}.$response on this object.`;
-          try {
-            error2.message += "\n  " + hint;
-          } catch (e3) {
-            if (!context3.logger || context3.logger?.constructor?.name === "NoOpLogger") {
-              console.warn(hint);
-            } else {
-              context3.logger?.warn?.(hint);
-            }
-          }
-          if (typeof error2.$responseBodyText !== "undefined") {
-            if (error2.$response) {
-              error2.$response.body = error2.$responseBodyText;
-            }
-          }
-          try {
-            if (import_protocol_http2.HttpResponse.isInstance(response)) {
-              const { headers = {} } = response;
-              const headerEntries = Object.entries(headers);
-              error2.$metadata = {
-                httpStatusCode: response.statusCode,
-                requestId: findHeader(/^x-[\w-]+-request-?id$/, headerEntries),
-                extendedRequestId: findHeader(/^x-[\w-]+-id-2$/, headerEntries),
-                cfId: findHeader(/^x-[\w-]+-cf-id$/, headerEntries)
-              };
-            }
-          } catch (e3) {
-          }
-        }
-        throw error2;
-      }
-    };
-    findHeader = (pattern, headers) => {
-      return (headers.find(([k3]) => {
-        return k3.match(pattern);
-      }) || [void 0, void 0])[1];
-    };
-  }
-});
-
-// node_modules/@smithy/core/dist-es/submodules/schema/middleware/schemaSerializationMiddleware.js
-var import_util_middleware4, schemaSerializationMiddleware;
-var init_schemaSerializationMiddleware = __esm({
-  "node_modules/@smithy/core/dist-es/submodules/schema/middleware/schemaSerializationMiddleware.js"() {
-    import_util_middleware4 = __toESM(require_dist_cjs7());
-    init_NormalizedSchema();
-    schemaSerializationMiddleware = (config) => (next, context3) => async (args) => {
-      let { operationSchema } = (0, import_util_middleware4.getSmithyContext)(context3);
-      if (isStaticSchema(operationSchema)) {
-        operationSchema = hydrate(operationSchema);
-      }
-      const endpoint = context3.endpointV2?.url && config.urlParser ? async () => config.urlParser(context3.endpointV2.url) : config.endpoint;
-      const request = await config.protocol.serializeRequest(operationSchema, args.input, {
-        ...config,
-        ...context3,
-        endpoint
-      });
-      return next({
-        ...args,
-        request
-      });
-    };
-  }
-});
-
-// node_modules/@smithy/core/dist-es/submodules/schema/middleware/getSchemaSerdePlugin.js
-function getSchemaSerdePlugin(config) {
-  return {
-    applyToStack: (commandStack) => {
-      commandStack.add(schemaSerializationMiddleware(config), serializerMiddlewareOption2);
-      commandStack.add(schemaDeserializationMiddleware(config), deserializerMiddlewareOption);
-      config.protocol.setSerdeContext(config);
-    }
-  };
-}
-var deserializerMiddlewareOption, serializerMiddlewareOption2;
-var init_getSchemaSerdePlugin = __esm({
-  "node_modules/@smithy/core/dist-es/submodules/schema/middleware/getSchemaSerdePlugin.js"() {
-    init_schemaDeserializationMiddleware();
-    init_schemaSerializationMiddleware();
-    deserializerMiddlewareOption = {
-      name: "deserializerMiddleware",
-      step: "deserialize",
-      tags: ["DESERIALIZER"],
-      override: true
-    };
-    serializerMiddlewareOption2 = {
-      name: "serializerMiddleware",
-      step: "serialize",
-      tags: ["SERIALIZER"],
-      override: true
+// node_modules/@smithy/core/dist-es/submodules/schema/schemas/SimpleSchema.js
+var SimpleSchema, sim, simAdapter;
+var init_SimpleSchema = __esm({
+  "node_modules/@smithy/core/dist-es/submodules/schema/schemas/SimpleSchema.js"() {
+    init_Schema();
+    SimpleSchema = class _SimpleSchema extends Schema {
+      static symbol = Symbol.for("@smithy/sim");
+      name;
+      schemaRef;
+      traits;
+      symbol = _SimpleSchema.symbol;
     };
+    sim = (namespace, name14, schemaRef, traits) => Schema.assign(new SimpleSchema(), {
+      name: name14,
+      namespace,
+      traits,
+      schemaRef
+    });
+    simAdapter = (namespace, name14, traits, schemaRef) => Schema.assign(new SimpleSchema(), {
+      name: name14,
+      namespace,
+      traits,
+      schemaRef
+    });
   }
 });
 
@@ -11555,7 +11603,7 @@ var init_TypeRegistry = __esm({
       getErrorCtor(es) {
         const $error = es;
         const registry = _TypeRegistry.for($error[1]);
-        return registry.exceptions.get(es);
+        return registry.exceptions.get($error);
       }
       getBaseException() {
         for (const exceptionKey of this.exceptions.keys()) {
@@ -11603,11 +11651,11 @@ __export(schema_exports, {
   deserializerMiddlewareOption: () => deserializerMiddlewareOption,
   error: () => error,
   getSchemaSerdePlugin: () => getSchemaSerdePlugin,
-  hydrate: () => hydrate,
   isStaticSchema: () => isStaticSchema,
   list: () => list,
   map: () => map,
   op: () => op,
+  operation: () => operation,
   serializerMiddlewareOption: () => serializerMiddlewareOption2,
   sim: () => sim,
   simAdapter: () => simAdapter,
@@ -11621,6 +11669,7 @@ var init_schema = __esm({
     init_ListSchema();
     init_MapSchema();
     init_OperationSchema();
+    init_operation();
     init_ErrorSchema();
     init_NormalizedSchema();
     init_Schema();
@@ -12987,7 +13036,6 @@ var init_EventStreamSerde = __esm({
         const marshaller = this.marshaller;
         const eventStreamMember = requestSchema.getEventStreamMember();
         const unionSchema = requestSchema.getMemberSchema(eventStreamMember);
-        const memberSchemas = unionSchema.getMemberSchemas();
         const serializer = this.serializer;
         const defaultContentType = this.defaultContentType;
         const initialRequestMarker = Symbol("initialRequestMarker");
@@ -13098,7 +13146,7 @@ var init_EventStreamSerde = __esm({
         let explicitPayloadContentType;
         const isKnownSchema = (() => {
           const struct2 = unionSchema.getSchema();
-          return struct2.memberNames.includes(unionMember);
+          return struct2[4].includes(unionMember);
         })();
         const additionalHeaders = {};
         if (!isKnownSchema) {
@@ -13232,10 +13280,10 @@ var init_HttpProtocol = __esm({
         }
       }
       setHostPrefix(request, operationSchema, input) {
-        const operationNs = NormalizedSchema.of(operationSchema);
         const inputNs = NormalizedSchema.of(operationSchema.input);
-        if (operationNs.getMergedTraits().endpoint) {
-          let hostPrefix = operationNs.getMergedTraits().endpoint?.[0];
+        const opTraits = translateTraits(operationSchema.traits ?? {});
+        if (opTraits.endpoint) {
+          let hostPrefix = opTraits.endpoint?.[0];
           if (typeof hostPrefix === "string") {
             const hostLabelInputs = [...inputNs.structIterator()].filter(([, member2]) => member2.getMergedTraits().hostLabel);
             for (const [name14] of hostLabelInputs) {
@@ -14514,20 +14562,24 @@ var require_dist_cjs20 = __commonJS({
       if (typeof obj === "string") {
         return evaluateTemplate(obj, options);
       } else if (obj["fn"]) {
-        return callFunction(obj, options);
+        return group$2.callFunction(obj, options);
       } else if (obj["ref"]) {
         return getReferenceValue(obj, options);
       }
       throw new EndpointError(`'${keyName}': ${String(obj)} is not a string, function or reference.`);
     };
     var callFunction = ({ fn, argv }, options) => {
-      const evaluatedArgs = argv.map((arg) => ["boolean", "number"].includes(typeof arg) ? arg : evaluateExpression(arg, "arg", options));
+      const evaluatedArgs = argv.map((arg) => ["boolean", "number"].includes(typeof arg) ? arg : group$2.evaluateExpression(arg, "arg", options));
       const fnSegments = fn.split(".");
       if (fnSegments[0] in customEndpointFunctions3 && fnSegments[1] != null) {
         return customEndpointFunctions3[fnSegments[0]][fnSegments[1]](...evaluatedArgs);
       }
       return endpointFunctions[fn](...evaluatedArgs);
     };
+    var group$2 = {
+      evaluateExpression,
+      callFunction
+    };
     var evaluateCondition = ({ assign: assign2, ...fnArgs }, options) => {
       if (assign2 && assign2 in options.referenceRecord) {
         throw new EndpointError(`'${assign2}' is already defined in Reference Record.`);
@@ -14569,6 +14621,10 @@ var require_dist_cjs20 = __commonJS({
         return processedExpr;
       })
     }), {});
+    var getEndpointProperties = (properties, options) => Object.entries(properties).reduce((acc, [propertyKey, propertyVal]) => ({
+      ...acc,
+      [propertyKey]: group$1.getEndpointProperty(propertyVal, options)
+    }), {});
     var getEndpointProperty = (property2, options) => {
       if (Array.isArray(property2)) {
         return property2.map((propertyEntry) => getEndpointProperty(propertyEntry, options));
@@ -14580,17 +14636,17 @@ var require_dist_cjs20 = __commonJS({
           if (property2 === null) {
             throw new EndpointError(`Unexpected endpoint property: ${property2}`);
           }
-          return getEndpointProperties(property2, options);
+          return group$1.getEndpointProperties(property2, options);
         case "boolean":
           return property2;
         default:
           throw new EndpointError(`Unexpected endpoint property type: ${typeof property2}`);
       }
     };
-    var getEndpointProperties = (properties, options) => Object.entries(properties).reduce((acc, [propertyKey, propertyVal]) => ({
-      ...acc,
-      [propertyKey]: getEndpointProperty(propertyVal, options)
-    }), {});
+    var group$1 = {
+      getEndpointProperty,
+      getEndpointProperties
+    };
     var getEndpointUrl = (endpointUrl, options) => {
       const expression = evaluateExpression(endpointUrl, "Endpoint URL", options);
       if (typeof expression === "string") {
@@ -14636,17 +14692,6 @@ var require_dist_cjs20 = __commonJS({
         referenceRecord: { ...options.referenceRecord, ...referenceRecord }
       }));
     };
-    var evaluateTreeRule = (treeRule, options) => {
-      const { conditions, rules } = treeRule;
-      const { result, referenceRecord } = evaluateConditions(conditions, options);
-      if (!result) {
-        return;
-      }
-      return evaluateRules(rules, {
-        ...options,
-        referenceRecord: { ...options.referenceRecord, ...referenceRecord }
-      });
-    };
     var evaluateRules = (rules, options) => {
       for (const rule of rules) {
         if (rule.type === "endpoint") {
@@ -14657,7 +14702,7 @@ var require_dist_cjs20 = __commonJS({
         } else if (rule.type === "error") {
           evaluateErrorRule(rule, options);
         } else if (rule.type === "tree") {
-          const endpointOrUndefined = evaluateTreeRule(rule, options);
+          const endpointOrUndefined = group.evaluateTreeRule(rule, options);
           if (endpointOrUndefined) {
             return endpointOrUndefined;
           }
@@ -14667,6 +14712,21 @@ var require_dist_cjs20 = __commonJS({
       }
       throw new EndpointError(`Rules evaluation failed`);
     };
+    var evaluateTreeRule = (treeRule, options) => {
+      const { conditions, rules } = treeRule;
+      const { result, referenceRecord } = evaluateConditions(conditions, options);
+      if (!result) {
+        return;
+      }
+      return group.evaluateRules(rules, {
+        ...options,
+        referenceRecord: { ...options.referenceRecord, ...referenceRecord }
+      });
+    };
+    var group = {
+      evaluateRules,
+      evaluateTreeRule
+    };
     var resolveEndpoint3 = (ruleSetObject, options) => {
       const { endpointParams, logger: logger2 } = options;
       const { parameters, rules } = ruleSetObject;
@@ -17189,8 +17249,8 @@ var init_SmithyRpcV2CborProtocol = __esm({
           } catch (e3) {
           }
         }
-        const { service, operation } = (0, import_util_middleware5.getSmithyContext)(context3);
-        const path7 = `/service/${service}/operation/${operation}`;
+        const { service, operation: operation2 } = (0, import_util_middleware5.getSmithyContext)(context3);
+        const path7 = `/service/${service}/operation/${operation2}`;
         if (request.path.endsWith("/")) {
           request.path += path7.slice(1);
         } else {
@@ -17895,10 +17955,10 @@ var require_dist_cjs27 = __commonJS({
         this._middlewareFn = middlewareSupplier;
         return this;
       }
-      s(service, operation, smithyContext = {}) {
+      s(service, operation2, smithyContext = {}) {
         this._smithyContext = {
           service,
-          operation,
+          operation: operation2,
           ...smithyContext
         };
         return this;
@@ -17925,9 +17985,9 @@ var require_dist_cjs27 = __commonJS({
         this._deserializer = deserializer;
         return this;
       }
-      sc(operation) {
-        this._operationSchema = operation;
-        this._smithyContext.operationSchema = operation;
+      sc(operation2) {
+        this._operationSchema = operation2;
+        this._smithyContext.operationSchema = operation2;
         return this;
       }
       build() {
@@ -18779,7 +18839,7 @@ var init_AwsJsonRpcProtocol = __esm({
         }
         Object.assign(request.headers, {
           "content-type": `application/x-amz-json-${this.getJsonRpcVersion()}`,
-          "x-amz-target": `${this.serviceTarget}.${NormalizedSchema.of(operationSchema).getName()}`
+          "x-amz-target": `${this.serviceTarget}.${operationSchema.name}`
         });
         if (this.awsQueryCompatible) {
           request.headers["x-amzn-query-mode"] = "true";
@@ -22902,6 +22962,7 @@ var require_dist_cjs42 = __commonJS({
     var ENV_PROFILE = "AWS_PROFILE";
     var DEFAULT_PROFILE = "default";
     var getProfileName = (init) => init.profile || process.env[ENV_PROFILE] || DEFAULT_PROFILE;
+    var CONFIG_PREFIX_SEPARATOR = ".";
     var getConfigData = (data2) => Object.entries(data2).filter(([key]) => {
       const indexOfSeparator = key.indexOf(CONFIG_PREFIX_SEPARATOR);
       if (indexOfSeparator === -1) {
@@ -22968,7 +23029,6 @@ var require_dist_cjs42 = __commonJS({
       return map4;
     };
     var swallowError$1 = () => ({});
-    var CONFIG_PREFIX_SEPARATOR = ".";
     var loadSharedConfigFiles = async (init = {}) => {
       const { filepath = getCredentialsFilepath(), configFilepath = getConfigFilepath() } = init;
       const homeDir = getHomeDir.getHomeDir();
@@ -24173,7 +24233,7 @@ var require_package = __commonJS({
     module2.exports = {
       name: "@aws-sdk/client-bedrock-runtime",
       description: "AWS SDK for JavaScript Bedrock Runtime Client for Node.js, Browser and React Native",
-      version: "3.910.0",
+      version: "3.911.0",
       scripts: {
         build: "concurrently 'yarn:build:cjs' 'yarn:build:es' 'yarn:build:types'",
         "build:cjs": "node ../../scripts/compilation/inline client-bedrock-runtime",
@@ -24192,21 +24252,21 @@ var require_package = __commonJS({
       dependencies: {
         "@aws-crypto/sha256-browser": "5.2.0",
         "@aws-crypto/sha256-js": "5.2.0",
-        "@aws-sdk/core": "3.910.0",
-        "@aws-sdk/credential-provider-node": "3.910.0",
+        "@aws-sdk/core": "3.911.0",
+        "@aws-sdk/credential-provider-node": "3.911.0",
         "@aws-sdk/eventstream-handler-node": "3.910.0",
         "@aws-sdk/middleware-eventstream": "3.910.0",
         "@aws-sdk/middleware-host-header": "3.910.0",
         "@aws-sdk/middleware-logger": "3.910.0",
         "@aws-sdk/middleware-recursion-detection": "3.910.0",
-        "@aws-sdk/middleware-user-agent": "3.910.0",
+        "@aws-sdk/middleware-user-agent": "3.911.0",
         "@aws-sdk/middleware-websocket": "3.910.0",
         "@aws-sdk/region-config-resolver": "3.910.0",
-        "@aws-sdk/token-providers": "3.910.0",
+        "@aws-sdk/token-providers": "3.911.0",
         "@aws-sdk/types": "3.910.0",
         "@aws-sdk/util-endpoints": "3.910.0",
         "@aws-sdk/util-user-agent-browser": "3.910.0",
-        "@aws-sdk/util-user-agent-node": "3.910.0",
+        "@aws-sdk/util-user-agent-node": "3.911.0",
         "@smithy/config-resolver": "^4.3.2",
         "@smithy/core": "^3.16.1",
         "@smithy/eventstream-serde-browser": "^4.2.2",
@@ -24954,7 +25014,7 @@ var init_package = __esm({
   "node_modules/@aws-sdk/nested-clients/package.json"() {
     package_default = {
       name: "@aws-sdk/nested-clients",
-      version: "3.910.0",
+      version: "3.911.0",
       description: "Nested clients for AWS SDK packages.",
       main: "./dist-cjs/index.js",
       module: "./dist-es/index.js",
@@ -24983,16 +25043,16 @@ var init_package = __esm({
       dependencies: {
         "@aws-crypto/sha256-browser": "5.2.0",
         "@aws-crypto/sha256-js": "5.2.0",
-        "@aws-sdk/core": "3.910.0",
+        "@aws-sdk/core": "3.911.0",
         "@aws-sdk/middleware-host-header": "3.910.0",
         "@aws-sdk/middleware-logger": "3.910.0",
         "@aws-sdk/middleware-recursion-detection": "3.910.0",
-        "@aws-sdk/middleware-user-agent": "3.910.0",
+        "@aws-sdk/middleware-user-agent": "3.911.0",
         "@aws-sdk/region-config-resolver": "3.910.0",
         "@aws-sdk/types": "3.910.0",
         "@aws-sdk/util-endpoints": "3.910.0",
         "@aws-sdk/util-user-agent-browser": "3.910.0",
-        "@aws-sdk/util-user-agent-node": "3.910.0",
+        "@aws-sdk/util-user-agent-node": "3.911.0",
         "@smithy/config-resolver": "^4.3.2",
         "@smithy/core": "^3.16.1",
         "@smithy/fetch-http-handler": "^5.3.3",
@@ -26407,7 +26467,7 @@ var require_package2 = __commonJS({
     module2.exports = {
       name: "@aws-sdk/client-sso",
       description: "AWS SDK for JavaScript Sso Client for Node.js, Browser and React Native",
-      version: "3.910.0",
+      version: "3.911.0",
       scripts: {
         build: "concurrently 'yarn:build:cjs' 'yarn:build:es' 'yarn:build:types'",
         "build:cjs": "node ../../scripts/compilation/inline client-sso",
@@ -26426,16 +26486,16 @@ var require_package2 = __commonJS({
       dependencies: {
         "@aws-crypto/sha256-browser": "5.2.0",
         "@aws-crypto/sha256-js": "5.2.0",
-        "@aws-sdk/core": "3.910.0",
+        "@aws-sdk/core": "3.911.0",
         "@aws-sdk/middleware-host-header": "3.910.0",
         "@aws-sdk/middleware-logger": "3.910.0",
         "@aws-sdk/middleware-recursion-detection": "3.910.0",
-        "@aws-sdk/middleware-user-agent": "3.910.0",
+        "@aws-sdk/middleware-user-agent": "3.911.0",
         "@aws-sdk/region-config-resolver": "3.910.0",
         "@aws-sdk/types": "3.910.0",
         "@aws-sdk/util-endpoints": "3.910.0",
         "@aws-sdk/util-user-agent-browser": "3.910.0",
-        "@aws-sdk/util-user-agent-node": "3.910.0",
+        "@aws-sdk/util-user-agent-node": "3.911.0",
         "@smithy/config-resolver": "^4.3.2",
         "@smithy/core": "^3.16.1",
         "@smithy/fetch-http-handler": "^5.3.3",
@@ -74729,13 +74789,14 @@ You are working with a repository located at: ${searchDirectory}
                   maxTokens: maxResponseTokens,
                   temperature: 0.3
                 });
+                const usagePromise = result.usage;
                 for await (const delta of result.textStream) {
                   assistantResponseContent += delta;
                   if (options.onStream) {
                     options.onStream(delta);
                   }
                 }
-                const usage = await result.usage;
+                const usage = await usagePromise;
                 if (usage) {
                   this.tokenCounter.recordUsage(usage, result.experimental_providerMetadata);
                 }