Skip to content

Commit 14c9fae

Browse files
committed
Fix reading of few-shot example file for LLM categoriser
1 parent 9465cc2 commit 14c9fae

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

processors/machine_learning/annotate_text.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -281,7 +281,7 @@ def validate_query(query, request, user):
281281
if not labels or len([l for l in labels if l.strip()]) < 2:
282282
raise QueryParametersException("At least two labels should be provided for text classification.")
283283
else:
284-
file = request.files["option-category_file"]
284+
file = request.files["option-category-file"]
285285
if not file:
286286
raise QueryParametersException(
287287
"No label file provided. A label file is required when using few-shot classification.")
@@ -296,7 +296,7 @@ def validate_query(query, request, user):
296296
has_header = csv.Sniffer().has_header(sample)
297297
dialect = csv.Sniffer().sniff(sample, delimiters=(",", ";", "\t"))
298298

299-
reader = csv.reader(wrapped_file) if not has_header else csv.DictReader(wrapped_file)
299+
reader = csv.reader(wrapped_file, dialect=dialect) if not has_header else csv.DictReader(wrapped_file)
300300
row = next(reader)
301301
if len(list(row)) != 2:
302302
raise QueryParametersException("The label file must have exactly two columns.")

0 commit comments

Comments
 (0)