refactor: clean code for align

dsdanielpark · Mar 13, 2024 · e7e58b1 · e7e58b1
1 parent b36f1db
commit e7e58b1
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 3 deletions.
diff --git a/gemini/src/misc/utils.py b/gemini/src/misc/utils.py
@@ -16,8 +16,7 @@ def extract_code(text: str) -> str:
  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.
  """
- if not text:
- return []
+
  snippets = []
  start_pattern = "```"
  end_pattern = "```"

diff --git a/gemini/src/parser/response_parser.py b/gemini/src/parser/response_parser.py
@@ -130,7 +130,7 @@ def _parse_candidates(self, candidates_data: Dict) -> Dict:
  candidate_dict = {
  "rcid": candidate_data[0],
  "text": candidate_data[1][0],
- "code": extract_code(candidate_data[1][0]),
+ "code": self._parse_code(candidate_data[1][0]),
  "web_images": web_images,
  "generated_images": generated_images,
  }
@@ -181,4 +181,6 @@ def _parse_generated_images(self, images_data: Dict) -> Dict:
  ]
 
  def _parse_code(self, text):
+ if not text:
+ return []
  return extract_code(text)