diff --git a/src/llm.py b/src/llm.py index 70937f9..8f48a65 100644 --- a/src/llm.py +++ b/src/llm.py @@ -98,7 +98,12 @@ 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) + if type(self._json[field]) is not list: + self._json[field] = [self._json[field]] + if type(parsed_value) is list: + self._json[field].extend(parsed_value) + else: + self._json[field].append(parsed_value) else: self._json[field] = parsed_value