diff --git a/align_system/algorithms/llm_chat_baseline.py b/align_system/algorithms/llm_chat_baseline.py index 34d1c6bb..4e93d2c7 100644 --- a/align_system/algorithms/llm_chat_baseline.py +++ b/align_system/algorithms/llm_chat_baseline.py @@ -320,7 +320,7 @@ def attempt_generic_parse(generated_output, fields_of_interest): else: # Special handling of common "Index" field (should be # an integer) - if field == 'Index': + if field == 'Answer': if m := re.search(r'\d+', parsed_field): # noqa parsed_field = m.group(0)