Adding print to debug

SeisoLLC · May 31, 2024 · c25379f · c25379f
1 parent 8cd3a0d
commit c25379f
Showing 1 changed file with 7 additions and 2 deletions.
diff --git a/code_review.py b/code_review.py
@@ -186,10 +186,15 @@ def submit_to_gpt(code: str, ai_client: OpenAI) -> dict:
             log.debug(f"Completion message content: {content}")
 
             # Strip triple backticks if they exist
-            if content.startswith("```json") and content.endswith("```"):
-                content = content.strip("```json").strip("```").strip()
+            if content.startswith("```") and content.endswith("```"):
+                content = content.strip("```").strip()
                 log.debug(f"Stripped content: {content}")
 
+            # Remove the `json` identifier if it exists
+            if content.startswith("json"):
+                content = content[4:].strip()
+                log.debug(f"Stripped content (json identifier): {content}")
+
             try:
                 review = json.loads(content)
             except json.JSONDecodeError as e: