fix(provider/amazon-bedrock): Mimic doGenerate reasoning response for…

…matting
vercel · Feb 28, 2025 · 468a750 · 468a750
1 parent 6f23643
commit 468a750
Showing 1 changed file with 30 additions and 29 deletions.
diff --git a/packages/amazon-bedrock/src/bedrock-chat-language-model.ts b/packages/amazon-bedrock/src/bedrock-chat-language-model.ts
@@ -223,6 +223,35 @@ export class BedrockChatLanguageModel implements LanguageModelV1 {
           }
         : undefined;
 
+    const reasoningDetails = (response.output?.message?.content || [])
+      .filter(
+        part =>
+          part.reasoningContent &&
+          (('reasoningText' in part.reasoningContent &&
+            part.reasoningContent.reasoningText.text != null) ||
+            ('redactedReasoning' in part.reasoningContent &&
+              part.reasoningContent.redactedReasoning.data != null)),
+      )
+      .map(part => {
+        const reasoningContent = part.reasoningContent!;
+
+        if ('reasoningText' in reasoningContent) {
+          return {
+            type: 'text' as const,
+            text: reasoningContent.reasoningText.text,
+            ...(reasoningContent.reasoningText.signature && {
+              signature: reasoningContent.reasoningText.signature,
+            }),
+          };
+        } else {
+          // Must be redactedReasoning
+          return {
+            type: 'redacted' as const,
+            data: reasoningContent.redactedReasoning.data,
+          };
+        }
+      });
+
     return {
       text:
         response.output?.message?.content
@@ -246,35 +275,7 @@ export class BedrockChatLanguageModel implements LanguageModelV1 {
       rawCall: { rawPrompt, rawSettings },
       rawResponse: { headers: responseHeaders },
       warnings,
-      reasoning:
-        response.output?.message?.content
-          ?.filter(
-            part =>
-              part.reasoningContent &&
-              (('reasoningText' in part.reasoningContent &&
-                part.reasoningContent.reasoningText.text != null) ||
-                ('redactedReasoning' in part.reasoningContent &&
-                  part.reasoningContent.redactedReasoning.data != null)),
-          )
-          ?.map(part => {
-            const reasoningContent = part.reasoningContent!;
-
-            if ('reasoningText' in reasoningContent) {
-              return {
-                type: 'text' as const,
-                text: reasoningContent.reasoningText.text,
-                ...(reasoningContent.reasoningText.signature && {
-                  signature: reasoningContent.reasoningText.signature,
-                }),
-              };
-            } else {
-              // Must be redactedReasoning
-              return {
-                type: 'redacted' as const,
-                data: reasoningContent.redactedReasoning.data,
-              };
-            }
-          }) || undefined,
+      reasoning: reasoningDetails,
       ...(providerMetadata && { providerMetadata }),
     };
   }