EvoAdvisor / logger.py
HemanM's picture
Update logger.py
bb51229 verified
raw
history blame contribute delete
657 Bytes
import csv
from datetime import datetime
import os
def log_feedback(query, rag_context, evo_output, feedback):
row = {
"timestamp": datetime.utcnow().isoformat(),
"query": query,
"context": rag_context,
"evo_output": evo_output,
"label": 1 if "πŸ‘" in feedback else 0 if "πŸ‘Ž" in feedback else "",
}
file_path = "feedback_log.csv"
file_exists = os.path.exists(file_path)
with open(file_path, mode="a", newline="", encoding="utf-8") as file:
writer = csv.DictWriter(file, fieldnames=row.keys())
if not file_exists:
writer.writeheader()
writer.writerow(row)