fix get_file_modifications detection issue

shroominic · Nov 7, 2024 · 619dc43 · 619dc43
1 parent 39f9519
commit 619dc43
Showing 1 changed file with 10 additions and 0 deletions.
diff --git a/src/codeinterpreterapi/chains/modifications_check.py b/src/codeinterpreterapi/chains/modifications_check.py
@@ -19,6 +19,11 @@ def get_file_modifications(
     result = llm.invoke(prompt)
 
     try:
+        if isinstance(result.content, str):
+            if result.content.endswith("```"):
+                result.content = result.content[:-3]
+            if result.content.startswith("```"):
+                result.content = result.content[3:]
         result = json.loads(result.content)
     except json.JSONDecodeError:
         result = ""
@@ -40,6 +45,11 @@ async def aget_file_modifications(
     result = await llm.ainvoke(prompt)
 
     try:
+        if isinstance(result.content, str):
+            if result.content.endswith("```"):
+                result.content = result.content[:-3]
+            if result.content.startswith("```"):
+                result.content = result.content[3:]
         result = json.loads(result.content)
     except json.JSONDecodeError:
         result = ""