From f238c27c26c4917930d8283bd331bd42135b1171 Mon Sep 17 00:00:00 2001
From: Jack Driscoll <jdriscoll98@ufl.edu>
Date: Sat, 27 Jul 2024 09:04:28 -0700
Subject: [PATCH] Fix edge case for json comment

---
 src/main.ts | 6 +++++-
 1 file changed, 5 insertions(+), 1 deletion(-)

diff --git a/src/main.ts b/src/main.ts
index beef8cb..43cbe46 100644
--- a/src/main.ts
+++ b/src/main.ts
@@ -156,7 +156,11 @@ async function getAIResponse(prompt: string): Promise<Array<{
     }
 
     const res = response.choices[0].message?.content?.trim() || "{}";
-    return JSON.parse(res).reviews;
+    if (res.startsWith("```json")) {
+      return JSON.parse(res.slice(7, -3)).reviews
+    } else {
+      return JSON.parse(res).reviews;
+    }
   } catch (error) {
     console.error("Error:", error, response?.choices[0].message?.content);
     return null;