Fix edge case for json comment

aidar-freeed · Jul 27, 2024 · b051475 · b051475
1 parent 62fdb56
commit b051475
Showing 1 changed file with 5 additions and 1 deletion.
diff --git 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));
+    } else {
+      return JSON.parse(res).reviews;
+    }
   } catch (error) {
     console.error("Error:", error, response?.choices[0].message?.content);
     return null;