ai 3.3.27 → 3.3.28

package/dist/index.mjs

@@ -4136,6 +4136,39 @@ var DefaultStreamTextResult = class {
 };
 var experimental_streamText = streamText;
 
+// core/middleware/wrap-language-model.ts
+var experimental_wrapLanguageModel = ({
+  model,
+  middleware: { transformParams, wrapGenerate, wrapStream },
+  modelId,
+  providerId
+}) => {
+  async function doTransform({
+    params,
+    type
+  }) {
+    return transformParams ? await transformParams({ params, type }) : params;
+  }
+  return {
+    specificationVersion: "v1",
+    provider: providerId != null ? providerId : model.provider,
+    modelId: modelId != null ? modelId : model.modelId,
+    defaultObjectGenerationMode: model.defaultObjectGenerationMode,
+    supportsImageUrls: model.supportsImageUrls,
+    supportsStructuredOutputs: model.supportsStructuredOutputs,
+    async doGenerate(params) {
+      const transformedParams = await doTransform({ params, type: "generate" });
+      const doGenerate = async () => model.doGenerate(transformedParams);
+      return wrapGenerate ? wrapGenerate({ doGenerate, params: transformedParams, model }) : doGenerate();
+    },
+    async doStream(params) {
+      const transformedParams = await doTransform({ params, type: "stream" });
+      const doStream = async () => model.doStream(transformedParams);
+      return wrapStream ? wrapStream({ doStream, params: transformedParams, model }) : doStream();
+    }
+  };
+};
+
 // core/prompt/attachments-to-parts.ts
 function attachmentsToParts(attachments) {
   var _a11, _b, _c;
@@ -5585,6 +5618,7 @@ export {
   experimental_generateText,
   experimental_streamObject,
   experimental_streamText,
+  experimental_wrapLanguageModel,
   formatStreamPart,
   generateId2 as generateId,
   generateObject,