EvoTransformer-v2.1 / logger.py
HemanM's picture
Update logger.py
8e0fcfe verified
raw
history blame
732 Bytes
import firebase_admin
from firebase_admin import credentials, firestore
from datetime import datetime
# Load credentials (ensure this JSON is uploaded as a private file on Hugging Face or locally)
cred = credentials.Certificate("firebase_key.json") # Keep this secure
firebase_admin.initialize_app(cred)
db = firestore.client()
def log_interaction(goal, sol1, sol2, evo, gpt, correct=None, source="user"):
doc = {
"timestamp": datetime.utcnow().isoformat(),
"goal": goal,
"solution_1": sol1,
"solution_2": sol2,
"evo_answer": evo,
"gpt_answer": gpt,
"correct_answer": correct if correct else gpt,
"source": source
}
db.collection("evo_logs").add(doc)