From 65c5653158285da74e0bf8ea1446628c9657f6a0 Mon Sep 17 00:00:00 2001
From: Will Hohyon Ryu <will@oxopolitics.com>
Date: Thu, 7 Mar 2024 18:54:20 -0800
Subject: [PATCH] feat(parseJsonForChatGPTResponse): implement function to
 parse JSON response from ChatGPT API

---
 package.json                       |  2 ++
 src/main.ts                        |  3 ++-
 src/parseJsonForChatGPTResponse.ts | 20 ++++++++++++++++++++
 yarn.lock                          | 10 ++++++++++
 4 files changed, 34 insertions(+), 1 deletion(-)
 create mode 100644 src/parseJsonForChatGPTResponse.ts

diff --git a/package.json b/package.json
index 5ec57c8..1a7a974 100644
--- a/package.json
+++ b/package.json
@@ -14,12 +14,14 @@
   "dependencies": {
     "@actions/core": "^1.10.0",
     "@octokit/rest": "^19.0.7",
+    "jsonic": "^1.0.1",
     "minimatch": "^7.4.2",
     "openai": "^4.20.1",
     "parse-diff": "^0.11.1",
     "ts-node": "^10.9.1"
   },
   "devDependencies": {
+    "@types/jsonic": "^0.3.3",
     "@types/node": "^18.15.5",
     "@vercel/ncc": "^0.36.1",
     "prettier": "^2.8.6",
diff --git a/src/main.ts b/src/main.ts
index 13677ae..f47a6b0 100644
--- a/src/main.ts
+++ b/src/main.ts
@@ -4,6 +4,7 @@ import OpenAI from "openai";
 import { Octokit } from "@octokit/rest";
 import parseDiff, { Chunk, File } from "parse-diff";
 import minimatch from "minimatch";
+import {parseJsonForChatGPTResponse} from "./parseJsonForChatGPTResponse";
 
 const GITHUB_TOKEN: string = core.getInput("GITHUB_TOKEN");
 const OPENAI_API_KEY: string = core.getInput("OPENAI_API_KEY");
@@ -139,7 +140,7 @@ async function getAIResponse(prompt: string): Promise<Array<{
     });
 
     const res = response.choices[0].message?.content?.trim() || "{}";
-    return JSON.parse(res).reviews;
+    return parseJsonForChatGPTResponse(res).reviews;
   } catch (error) {
     console.error("Error:", error);
     return null;
diff --git a/src/parseJsonForChatGPTResponse.ts b/src/parseJsonForChatGPTResponse.ts
new file mode 100644
index 0000000..183cb6d
--- /dev/null
+++ b/src/parseJsonForChatGPTResponse.ts
@@ -0,0 +1,20 @@
+import jsonic from "jsonic";
+
+export function parseJsonForChatGPTResponse(response: string) {
+  let jsonData: any = null;
+
+  const responseString = response
+    .replace(/```json/g, "")
+    .replace(/```/g, "");
+
+  try {
+    jsonData = jsonic(responseString);
+  } catch (e) {
+    try {
+      jsonData = JSON.parse(responseString);
+    } catch (e) {
+      console.error(e);
+    }
+  }
+  return jsonData;
+}
diff --git a/yarn.lock b/yarn.lock
index 0998992..8b3469b 100644
--- a/yarn.lock
+++ b/yarn.lock
@@ -163,6 +163,11 @@
   resolved "https://registry.yarnpkg.com/@tsconfig/node16/-/node16-1.0.3.tgz#472eaab5f15c1ffdd7f8628bd4c4f753995ec79e"
   integrity sha512-yOlFc+7UtL/89t2ZhjPvvB/DeAr3r+Dq58IgzsFkOAvVC6NMJXmCGjbptdXdR9qsX7pKcTL+s87FtYREi2dEEQ==
 
+"@types/jsonic@^0.3.3":
+  version "0.3.3"
+  resolved "https://registry.yarnpkg.com/@types/jsonic/-/jsonic-0.3.3.tgz#0f47f2ab95ce38647fc522d62104d411f004b7ba"
+  integrity sha512-QXYC0N2jOSGw21QkYRP4YyR9sFZkAPLjxn4PVswsgSz3r8SOBYQpQOFdRc8FbhWZ7pWPKEnq2u8tJCt2Qcn4oQ==
+
 "@types/node-fetch@^2.6.4":
   version "2.6.9"
   resolved "https://registry.yarnpkg.com/@types/node-fetch/-/node-fetch-2.6.9.tgz#15f529d247f1ede1824f7e7acdaa192d5f28071e"
@@ -340,6 +345,11 @@ is-plain-object@^5.0.0:
   resolved "https://registry.yarnpkg.com/is-plain-object/-/is-plain-object-5.0.0.tgz#4427f50ab3429e9025ea7d52e9043a9ef4159344"
   integrity sha512-VRSzKkbMm5jMDoKLbltAkFQ5Qr7VDiTFGXxYFXXowVj387GeGNOCsOH6Msy00SGZ3Fp84b1Naa1psqgcCIEP5Q==
 
+jsonic@^1.0.1:
+  version "1.0.1"
+  resolved "https://registry.yarnpkg.com/jsonic/-/jsonic-1.0.1.tgz#817054df7abfaf525ecf39d6647a2ed6d5fca7f1"
+  integrity sha512-6GitEN4plTuB/I1o9kDZl7Pgc+DvFG1BG88IqaUz4eQglCA1uAgxWdXhLNA6ffaYsmzPjOysDpp6CYTwRiuXLw==
+
 make-error@^1.1.1:
   version "1.3.6"
   resolved "https://registry.yarnpkg.com/make-error/-/make-error-1.3.6.tgz#2eb2e37ea9b67c4891f684a1394799af484cf7a2"