Better response parsing

src/agents/action/action.py

@@ -25,11 +25,9 @@ def render(
 
     def validate_response(self, response: str):
         response = response.strip().replace("```json", "```")
-
-        if response.startswith("```") and response.endswith("```"):
-            response = response[3:-3].strip()
 
         try:
+            response = response.split("```")[1].split("```")[0]
             response = json.loads(response)
         except Exception as _:
             return False

src/agents/answer/answer.py

@@ -27,10 +27,8 @@ def render(
     def validate_response(self, response: str):
         response = response.strip().replace("```json", "```")
 
-        if response.startswith("```") and response.endswith("```"):
-            response = response[3:-3].strip()
-
         try:
+            response = response.split("```")[1].split("```")[0]
             response = json.loads(response)
         except Exception as _:
             return False

src/agents/decision/decision.py

@@ -17,11 +17,9 @@ def render(self, prompt: str) -> str:
 
     def validate_response(self, response: str):
         response = response.strip().replace("```json", "```")
-
-        if response.startswith("```") and response.endswith("```"):
-            response = response[3:-3].strip()
 
         try:
+            response = response.split("```")[1].split("```")[0]
             response = json.loads(response)
         except Exception as _:
             return False

src/agents/internal_monologue/internal_monologue.py

@@ -17,11 +17,9 @@ def render(self, current_prompt: str) -> str:
 
     def validate_response(self, response: str):
         response = response.strip().replace("```json", "```")
-
-        if response.startswith("```") and response.endswith("```"):
-            response = response[3:-3].strip()
 
         try:
+            response = response.split("```")[1].split("```")[0]
             response = json.loads(response)
         except Exception as _:
             return False

src/agents/researcher/researcher.py

@@ -25,9 +25,8 @@ def render(self, step_by_step_plan: str, contextual_keywords: str) -> str:
     def validate_response(self, response: str) -> dict | bool:
         response = response.strip().replace("```json", "```")
 
-        if response.startswith("```") and response.endswith("```"):
-            response = response[3:-3].strip()
         try:
+            response = response.split("```")[1].split("```")[0]
             response = json.loads(response)
         except Exception as _:
             return False

src/agents/runner/runner.py

@@ -53,11 +53,9 @@ def render_rerunner(
 
     def validate_response(self, response: str):
         response = response.strip().replace("```json", "```")
-
-        if response.startswith("```") and response.endswith("```"):
-            response = response[3:-3].strip()
 
         try:
+            response = response.split("```")[1].split("```")[0]
             response = json.loads(response)
         except Exception as _:
             return False
@@ -69,13 +67,11 @@ def validate_response(self, response: str):
 
     def validate_rerunner_response(self, response: str):
         response = response.strip().replace("```json", "```")
-
-        if response.startswith("```") and response.endswith("```"):
-            response = response[3:-3].strip()
 
         print(response)
 
         try:
+            response = response.split("```")[1].split("```")[0]
             response = json.loads(response)
         except Exception as _:
             return False