From 7d11a7dbc881c66ee84731314879928a84b2e83d Mon Sep 17 00:00:00 2001
From: Baptiste Arnaud <baptiste.arnaud95@gmail.com>
Date: Fri, 15 Mar 2024 15:13:54 +0100
Subject: [PATCH] :bug: (anthropic) Fix transform functions when response
 mapping has Message content

---
 .../forge/blocks/anthropic/actions/createChatMessage.tsx     | 3 +++
 .../forge/blocks/mistral/actions/createChatCompletion.ts     | 5 +++++
 .../forge/blocks/openRouter/actions/createChatCompletion.tsx | 5 +++++
 .../forge/blocks/openai/actions/createChatCompletion.tsx     | 5 +++++
 .../forge/blocks/togetherAi/actions/createChatCompletion.tsx | 5 +++++
 5 files changed, 23 insertions(+)

diff --git a/packages/forge/blocks/anthropic/actions/createChatMessage.tsx b/packages/forge/blocks/anthropic/actions/createChatMessage.tsx
index e60389e87..dfb9d2c4d 100644
--- a/packages/forge/blocks/anthropic/actions/createChatMessage.tsx
+++ b/packages/forge/blocks/anthropic/actions/createChatMessage.tsx
@@ -79,6 +79,9 @@ export const options = option.object({
 const transformToChatCompletionOptions = (options: any) => ({
   ...options,
   action: 'Create chat completion',
+  responseMapping: options.responseMapping?.map((res: any) =>
+    res.item === 'Message Content' ? { ...res, item: 'Message content' } : res
+  ),
 })
 
 export const createChatMessage = createAction({
diff --git a/packages/forge/blocks/mistral/actions/createChatCompletion.ts b/packages/forge/blocks/mistral/actions/createChatCompletion.ts
index 4f330f6e2..c7c9836cc 100644
--- a/packages/forge/blocks/mistral/actions/createChatCompletion.ts
+++ b/packages/forge/blocks/mistral/actions/createChatCompletion.ts
@@ -80,6 +80,11 @@ export const createChatCompletion = createAction({
       transform: (options) => ({
         ...options,
         action: 'Create Chat Message',
+        responseMapping: options.responseMapping?.map((res: any) =>
+          res.item === 'Message content'
+            ? { ...res, item: 'Message Content' }
+            : res
+        ),
       }),
     },
   ],
diff --git a/packages/forge/blocks/openRouter/actions/createChatCompletion.tsx b/packages/forge/blocks/openRouter/actions/createChatCompletion.tsx
index c0bd48e03..fd9554fef 100644
--- a/packages/forge/blocks/openRouter/actions/createChatCompletion.tsx
+++ b/packages/forge/blocks/openRouter/actions/createChatCompletion.tsx
@@ -25,6 +25,11 @@ export const createChatCompletion = createAction({
       transform: (options) => ({
         ...options,
         action: 'Create Chat Message',
+        responseMapping: options.responseMapping?.map((res: any) =>
+          res.item === 'Message content'
+            ? { ...res, item: 'Message Content' }
+            : res
+        ),
       }),
     },
   ],
diff --git a/packages/forge/blocks/openai/actions/createChatCompletion.tsx b/packages/forge/blocks/openai/actions/createChatCompletion.tsx
index 141744886..20e7db840 100644
--- a/packages/forge/blocks/openai/actions/createChatCompletion.tsx
+++ b/packages/forge/blocks/openai/actions/createChatCompletion.tsx
@@ -32,6 +32,11 @@ export const createChatCompletion = createAction({
       transform: (options) => ({
         ...options,
         action: 'Create Chat Message',
+        responseMapping: options.responseMapping?.map((res: any) =>
+          res.item === 'Message content'
+            ? { ...res, item: 'Message Content' }
+            : res
+        ),
       }),
     },
   ],
diff --git a/packages/forge/blocks/togetherAi/actions/createChatCompletion.tsx b/packages/forge/blocks/togetherAi/actions/createChatCompletion.tsx
index f2432dcbd..1e4b97cbe 100644
--- a/packages/forge/blocks/togetherAi/actions/createChatCompletion.tsx
+++ b/packages/forge/blocks/togetherAi/actions/createChatCompletion.tsx
@@ -27,6 +27,11 @@ export const createChatCompletion = createAction({
       transform: (options) => ({
         ...options,
         action: 'Create Chat Message',
+        responseMapping: options.responseMapping?.map((res: any) =>
+          res.item === 'Message content'
+            ? { ...res, item: 'Message Content' }
+            : res
+        ),
       }),
     },
   ],