🐛 (anthropic) Fix transform functions when response mapping has Messa…

…ge content
baptisteArno · Mar 15, 2024 · 7d11a7d · 7d11a7d
1 parent a8a9259
commit 7d11a7d
Show file tree

Hide file tree

Showing 5 changed files with 23 additions and 0 deletions.
diff --git 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
@@ -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
@@ -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
@@ -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
@@ -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
+        ),
       }),
     },
   ],