From a31be9e5d650f1516b00b3590e584ebfb5eed332 Mon Sep 17 00:00:00 2001
From: Jack Driscoll <jdriscoll98@ufl.edu>
Date: Fri, 26 Jul 2024 12:25:41 -0700
Subject: [PATCH] add check for finsih reason

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

diff --git a/src/main.ts b/src/main.ts
index 540fe5a..0990c3d 100644
--- a/src/main.ts
+++ b/src/main.ts
@@ -136,6 +136,12 @@ async function getAIResponse(prompt: string): Promise<Array<{
       ],
     });
 
+    const finish_response = response.choices[0].finish_reason;
+    if (finish_response === 'length') {
+      console.log('The maximum context length has been exceeded. Please reduce the length of the code snippets.');
+      return null;
+    }
+
     const res = response.choices[0].message?.content?.trim() || "{}";
     return JSON.parse(res).reviews;
   } catch (error) {