diff --git a/src/providers/google-vertex-ai/chatComplete.ts b/src/providers/google-vertex-ai/chatComplete.ts index 66eee9d96..7cabaa402 100644 --- a/src/providers/google-vertex-ai/chatComplete.ts +++ b/src/providers/google-vertex-ai/chatComplete.ts @@ -99,15 +99,12 @@ export const VertexGoogleChatCompleteConfig: ProviderConfig = { }, }); }); - } else if ( - message.role === 'tool' && - typeof message.content === 'string' - ) { + } else if (message.role === 'tool') { parts.push({ functionResponse: { name: message.name ?? 'gateway-tool-filler-name', response: { - content: message.content, + output: message.content, }, }, }); diff --git a/src/providers/google/chatComplete.ts b/src/providers/google/chatComplete.ts index 5506195a3..ecd66fbc5 100644 --- a/src/providers/google/chatComplete.ts +++ b/src/providers/google/chatComplete.ts @@ -106,7 +106,7 @@ interface GoogleFunctionResponseMessagePart { name: string; response: { name?: string; - content: string; + content: string | ContentType[]; }; }; } @@ -200,15 +200,12 @@ export const GoogleChatCompleteConfig: ProviderConfig = { }, }); }); - } else if ( - message.role === 'tool' && - typeof message.content === 'string' - ) { + } else if (message.role === 'tool') { parts.push({ functionResponse: { name: message.name ?? 'gateway-tool-filler-name', response: { - content: message.content, + output: message.content, }, }, });