From cf1782715d7992fdd3b2ea6135c4d0771e32f9ca Mon Sep 17 00:00:00 2001 From: Satvik Singh Date: Mon, 23 Mar 2026 08:53:35 -0400 Subject: [PATCH] fix: :bug: handle plural values correctly in JSON response --- src/llm.py | 7 ++++++- 1 file changed, 6 insertions(+), 1 deletion(-) 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