adds a fix for tool arguments

elastic · Jan 24, 2025 · 0abfe13 · 0abfe13
1 parent 9a3fc89
commit 0abfe13
Showing 1 changed file with 5 additions and 0 deletions.
diff --git a/x-pack/platform/plugins/shared/stack_connectors/server/connector_types/inference/helpers.ts b/x-pack/platform/plugins/shared/stack_connectors/server/connector_types/inference/helpers.ts
@@ -102,6 +102,11 @@ export function chunksIntoMessage(obs$: Observable<UnifiedChatCompleteResponse>)
           if (concatenatedChunk.choices[0].message.content === '') {
             concatenatedChunk.choices[0].message.content = null;
           }
+          concatenatedChunk.choices[0].message.tool_calls?.forEach((toolCall: any) => {
+            if (toolCall.function?.arguments?.trim() === '') {
+              toolCall.function.arguments = '{}';
+            }
+          });
           return concatenatedChunk;
         })
       )