File size: 556 Bytes
4af310b
 
 
 
 
 
 
 
 
5bf211e
 
4af310b
 
 
 
 
 
5bf211e
 
4af310b
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
import json
from time import time

LOG_FILE = "rag_eval_logs.jsonl"

def log_eval(
    query: str,
    retrieved_count: int,
    confidence: float,
    answer_known: bool,
    source_type: str = "internal_pdf"  # Added source_type
):
    record = {
        "timestamp": time(),
        "query": query,
        "retrieved_count": retrieved_count,
        "confidence": confidence,
        "answer_known": answer_known,
        "source_type": source_type
    }

    with open(LOG_FILE, "a", encoding="utf-8") as f:
        f.write(json.dumps(record) + "\n")