From 14c9fae409de993b6dd3a1b0de1d227cfe00cd2f Mon Sep 17 00:00:00 2001 From: Stijn Peeters Date: Wed, 8 Jan 2025 15:09:45 +0100 Subject: [PATCH] Fix reading of few-shot example file for LLM categoriser --- processors/machine_learning/annotate_text.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/processors/machine_learning/annotate_text.py b/processors/machine_learning/annotate_text.py index 26234a186..b8a3d7ebd 100644 --- a/processors/machine_learning/annotate_text.py +++ b/processors/machine_learning/annotate_text.py @@ -281,7 +281,7 @@ def validate_query(query, request, user): if not labels or len([l for l in labels if l.strip()]) < 2: raise QueryParametersException("At least two labels should be provided for text classification.") else: - file = request.files["option-category_file"] + file = request.files["option-category-file"] if not file: raise QueryParametersException( "No label file provided. A label file is required when using few-shot classification.") @@ -296,7 +296,7 @@ def validate_query(query, request, user): has_header = csv.Sniffer().has_header(sample) dialect = csv.Sniffer().sniff(sample, delimiters=(",", ";", "\t")) - reader = csv.reader(wrapped_file) if not has_header else csv.DictReader(wrapped_file) + reader = csv.reader(wrapped_file, dialect=dialect) if not has_header else csv.DictReader(wrapped_file) row = next(reader) if len(list(row)) != 2: raise QueryParametersException("The label file must have exactly two columns.")