File size: 516 Bytes
a5d886c 12003d8 a5d886c 12003d8 a5d886c 12003d8 d74109c 12003d8 a5d886c 12003d8 d74109c | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 | import csv
from pathlib import Path
class CSVDataLoader:
def __init__(self, filename: str):
self.path = Path(filename)
def load_qa_pairs(self) -> list[str]:
docs = []
with self.path.open(encoding="utf-8") as f:
reader = csv.DictReader(f)
for row in reader:
q = row.get("question", "").strip()
a = row.get("answer", "").strip()
if q and a:
docs.append(f"Q: {q}\nA: {a}")
return docs
|