fix(response_parser.py): fix parsing methods

gemini/src/model/parser/response_parser.py

@@ -1,6 +1,5 @@
 import json
 from typing import Dict
-from gemini.src.misc.utils import extract_code
 from gemini.src.model.parser.base import BaesParser
 
 
@@ -98,16 +97,12 @@ def __extract_strategy_1(self, response_text: str) -> Dict:
  body = json.loads(json.loads(response_text.split("\n")[3])[0][2])
  if not body[4]:
  body = json.loads(json.loads(response_text.split("\n")[3])[4][2])
- else:
- raise ValueError("Invalid response data received.")
  return body
 
  def __extract_strategy_2(self, response_text: str) -> Dict:
  body = json.loads(json.loads(response_text.split("\n")[2])[0][2])
  if not body[4]:
  body = json.loads(json.loads(response_text.split("\n")[2])[4][2])
- else:
- raise ValueError("Invalid response data received.")
  return body
 
  def __extract_strategy_3(self, response_text: str) -> Dict:
@@ -209,7 +204,7 @@ def _parse_code(self, text: str) -> Dict:
  """
  if not text:
  return {}
- extracted_code = extract_code(text)
+ extracted_code = self.extract_code(text)
  code_dict = {}
 
  if isinstance(extracted_code, str) and extracted_code != text:
@@ -221,3 +216,40 @@ def _parse_code(self, text: str) -> Dict:
  return {}
 
  return code_dict
+
+ @staticmethod
+ def extract_code(text: str) -> str:
+ """
+ Extracts code snippets from the given text.
+ If only one snippet is found, returns it directly instead of a list.
+ If no snippets are found, returns the original text.
+
+ Args:
+ text (str): The text containing mixed code snippets.
+
+ Returns:
+ str or list of str: A single code snippet string if only one is found, otherwise a list of all extracted code snippets. Returns the original text if no snippets are found.
+ """
+
+ snippets = []
+ start_pattern = "```"
+ end_pattern = "```"
+ start_idx = text.find(start_pattern)
+
+ while start_idx != -1:
+ end_idx = text.find(end_pattern, start_idx + len(start_pattern))
+ if end_idx != -1:
+ snippet = text[start_idx : end_idx + len(end_pattern)].strip()
+ snippets.append(snippet)
+ start_idx = text.find(start_pattern, end_idx + len(end_pattern))
+ else:
+ break
+
+ # Return directly if only one snippet is found
+ if len(snippets) == 1:
+ return snippets[0]
+ elif len(snippets) > 1:
+ return snippets
+ else:
+ # Return the original text if no snippets are found
+ return text