ai 4.3.9 → 4.3.10

package/dist/index.mjs

@@ -4106,6 +4106,7 @@ async function generateText({
   experimental_providerMetadata,
   providerOptions = experimental_providerMetadata,
   experimental_activeTools: activeTools,
+  experimental_prepareStep: prepareStep,
   experimental_repairToolCall: repairToolCall,
   _internal: {
     generateId: generateId3 = originalGenerateId3,
@@ -4148,6 +4149,9 @@ async function generateText({
           telemetry
         }),
         ...baseTelemetryAttributes,
+        // model:
+        "ai.model.provider": model.provider,
+        "ai.model.id": model.modelId,
         // specific settings that only make sense on the outer level:
         "ai.prompt": {
           input: () => JSON.stringify({ system, prompt, messages })
@@ -4157,11 +4161,7 @@ async function generateText({
     }),
     tracer,
     fn: async (span) => {
-      var _a18, _b, _c, _d, _e, _f, _g, _h, _i, _j, _k;
-      const mode = {
-        type: "regular",
-        ...prepareToolsAndToolChoice({ tools, toolChoice, activeTools })
-      };
+      var _a18, _b, _c, _d, _e, _f, _g, _h, _i, _j, _k, _l, _m, _n;
       const callSettings = prepareCallSettings(settings);
       let currentModelResponse;
       let currentToolCalls = [];
@@ -4184,16 +4184,33 @@ async function generateText({
           ...initialPrompt.messages,
           ...responseMessages
         ];
+        const prepareStepResult = await (prepareStep == null ? void 0 : prepareStep({
+          model,
+          steps,
+          maxSteps,
+          stepNumber: stepCount
+        }));
+        const stepToolChoice = (_a18 = prepareStepResult == null ? void 0 : prepareStepResult.toolChoice) != null ? _a18 : toolChoice;
+        const stepActiveTools = (_b = prepareStepResult == null ? void 0 : prepareStepResult.experimental_activeTools) != null ? _b : activeTools;
+        const stepModel = (_c = prepareStepResult == null ? void 0 : prepareStepResult.model) != null ? _c : model;
         const promptMessages = await convertToLanguageModelPrompt({
           prompt: {
             type: promptFormat,
             system: initialPrompt.system,
             messages: stepInputMessages
           },
-          modelSupportsImageUrls: model.supportsImageUrls,
-          modelSupportsUrl: (_a18 = model.supportsUrl) == null ? void 0 : _a18.bind(model)
+          modelSupportsImageUrls: stepModel.supportsImageUrls,
+          modelSupportsUrl: (_d = stepModel.supportsUrl) == null ? void 0 : _d.bind(stepModel)
           // support 'this' context
         });
+        const mode = {
+          type: "regular",
+          ...prepareToolsAndToolChoice({
+            tools,
+            toolChoice: stepToolChoice,
+            activeTools: stepActiveTools
+          })
+        };
         currentModelResponse = await retry(
           () => recordSpan({
             name: "ai.generateText.doGenerate",
@@ -4205,6 +4222,10 @@ async function generateText({
                   telemetry
                 }),
                 ...baseTelemetryAttributes,
+                // model:
+                "ai.model.provider": stepModel.provider,
+                "ai.model.id": stepModel.modelId,
+                // prompt:
                 "ai.prompt.format": { input: () => promptFormat },
                 "ai.prompt.messages": {
                   input: () => JSON.stringify(promptMessages)
@@ -4220,8 +4241,8 @@ async function generateText({
                   input: () => mode.toolChoice != null ? JSON.stringify(mode.toolChoice) : void 0
                 },
                 // standardized gen-ai llm span attributes:
-                "gen_ai.system": model.provider,
-                "gen_ai.request.model": model.modelId,
+                "gen_ai.system": stepModel.provider,
+                "gen_ai.request.model": stepModel.modelId,
                 "gen_ai.request.frequency_penalty": settings.frequencyPenalty,
                 "gen_ai.request.max_tokens": settings.maxTokens,
                 "gen_ai.request.presence_penalty": settings.presencePenalty,
@@ -4234,7 +4255,7 @@ async function generateText({
             tracer,
             fn: async (span2) => {
               var _a19, _b2, _c2, _d2, _e2, _f2;
-              const result = await model.doGenerate({
+              const result = await stepModel.doGenerate({
                 mode,
                 ...callSettings,
                 inputFormat: promptFormat,
@@ -4247,7 +4268,7 @@ async function generateText({
               const responseData = {
                 id: (_b2 = (_a19 = result.response) == null ? void 0 : _a19.id) != null ? _b2 : generateId3(),
                 timestamp: (_d2 = (_c2 = result.response) == null ? void 0 : _c2.timestamp) != null ? _d2 : currentDate(),
-                modelId: (_f2 = (_e2 = result.response) == null ? void 0 : _e2.modelId) != null ? _f2 : model.modelId
+                modelId: (_f2 = (_e2 = result.response) == null ? void 0 : _e2.modelId) != null ? _f2 : stepModel.modelId
               };
               span2.setAttributes(
                 selectTelemetryAttributes({
@@ -4279,7 +4300,7 @@ async function generateText({
           })
         );
         currentToolCalls = await Promise.all(
-          ((_b = currentModelResponse.toolCalls) != null ? _b : []).map(
+          ((_e = currentModelResponse.toolCalls) != null ? _e : []).map(
             (toolCall) => parseToolCall({
               toolCall,
               tools,
@@ -4314,7 +4335,7 @@ async function generateText({
             nextStepType = "tool-result";
           }
         }
-        const originalText = (_c = currentModelResponse.text) != null ? _c : "";
+        const originalText = (_f = currentModelResponse.text) != null ? _f : "";
         const stepTextLeadingWhitespaceTrimmed = stepType === "continue" && // only for continue steps
         text2.trimEnd() !== text2 ? originalText.trimStart() : originalText;
         const stepText = nextStepType === "continue" ? removeTextAfterLastWhitespace(stepTextLeadingWhitespaceTrimmed) : stepTextLeadingWhitespaceTrimmed;
@@ -4322,7 +4343,7 @@ async function generateText({
         currentReasoningDetails = asReasoningDetails(
           currentModelResponse.reasoning
         );
-        sources.push(...(_d = currentModelResponse.sources) != null ? _d : []);
+        sources.push(...(_g = currentModelResponse.sources) != null ? _g : []);
         if (stepType === "continue") {
           const lastMessage = responseMessages[responseMessages.length - 1];
           if (typeof lastMessage.content === "string") {
@@ -4354,18 +4375,18 @@ async function generateText({
           reasoning: asReasoningText(currentReasoningDetails),
           reasoningDetails: currentReasoningDetails,
           files: asFiles(currentModelResponse.files),
-          sources: (_e = currentModelResponse.sources) != null ? _e : [],
+          sources: (_h = currentModelResponse.sources) != null ? _h : [],
           toolCalls: currentToolCalls,
           toolResults: currentToolResults,
           finishReason: currentModelResponse.finishReason,
           usage: currentUsage,
           warnings: currentModelResponse.warnings,
           logprobs: currentModelResponse.logprobs,
-          request: (_f = currentModelResponse.request) != null ? _f : {},
+          request: (_i = currentModelResponse.request) != null ? _i : {},
           response: {
             ...currentModelResponse.response,
-            headers: (_g = currentModelResponse.rawResponse) == null ? void 0 : _g.headers,
-            body: (_h = currentModelResponse.rawResponse) == null ? void 0 : _h.body,
+            headers: (_j = currentModelResponse.rawResponse) == null ? void 0 : _j.headers,
+            body: (_k = currentModelResponse.rawResponse) == null ? void 0 : _k.body,
             // deep clone msgs to avoid mutating past messages in multi-step:
             messages: structuredClone(responseMessages)
           },
@@ -4417,11 +4438,11 @@ async function generateText({
         finishReason: currentModelResponse.finishReason,
         usage,
         warnings: currentModelResponse.warnings,
-        request: (_i = currentModelResponse.request) != null ? _i : {},
+        request: (_l = currentModelResponse.request) != null ? _l : {},
         response: {
           ...currentModelResponse.response,
-          headers: (_j = currentModelResponse.rawResponse) == null ? void 0 : _j.headers,
-          body: (_k = currentModelResponse.rawResponse) == null ? void 0 : _k.body,
+          headers: (_m = currentModelResponse.rawResponse) == null ? void 0 : _m.headers,
+          body: (_n = currentModelResponse.rawResponse) == null ? void 0 : _n.body,
           messages: responseMessages
         },
         logprobs: currentModelResponse.logprobs,