From 2818cd91a9ed69a7819d55e5ad4d84aa14e75c31 Mon Sep 17 00:00:00 2001
From: Waleed Latif <walif6@gmail.com>
Date: Wed, 12 Feb 2025 13:37:51 -0800
Subject: [PATCH] Added code to trim gemini response of any markdown/whitespace

---
 providers/google/index.ts | 9 ++++++++-
 1 file changed, 8 insertions(+), 1 deletion(-)

diff --git a/providers/google/index.ts b/providers/google/index.ts
index de78d36809..897a12bef9 100644
--- a/providers/google/index.ts
+++ b/providers/google/index.ts
@@ -182,8 +182,15 @@ export const googleProvider: ProviderConfig = {
   },
 
   transformResponse: (response: any) => {
+    let content = response.candidates?.[0]?.content?.parts?.[0]?.text || ''
+
+    if (content) {
+      content = content.replace(/```json\n?|\n?```/g, '')
+      content = content.trim()
+    }
+
     return {
-      content: response.candidates?.[0]?.content?.parts?.[0]?.text || '',
+      content,
       tokens: response.usageMetadata && {
         prompt: response.usageMetadata.promptTokenCount,
         completion: response.usageMetadata.candidatesTokenCount,