diff --git a/src/llm.py b/src/llm.py index 70937f9..894e3d1 100644 --- a/src/llm.py +++ b/src/llm.py @@ -98,7 +98,11 @@ def add_response_to_json(self, field, value): parsed_value = self.handle_plural_values(value) if field in self._json.keys(): - self._json[field].append(parsed_value) + current_value = self._json[field] + if isinstance(current_value, list): + self._json[field].append(parsed_value) + else: + self._json[field] = [current_value, parsed_value] else: self._json[field] = parsed_value