Spaces:
Running
Running
Fix evaluation database mapping
Browse files- src/database.py +19 -16
src/database.py
CHANGED
|
@@ -35,17 +35,27 @@ def init_db():
|
|
| 35 |
def save_evaluation(context, question, llm_response, result):
|
| 36 |
conn = sqlite3.connect(DB_PATH)
|
| 37 |
cursor = conn.cursor()
|
| 38 |
-
|
|
|
|
| 39 |
INSERT INTO evaluations (
|
| 40 |
-
context,
|
| 41 |
-
|
| 42 |
-
|
| 43 |
-
|
| 44 |
-
|
| 45 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 46 |
context,
|
| 47 |
question,
|
| 48 |
llm_response,
|
|
|
|
| 49 |
result["generation_evaluation"]["cosine"]["score"],
|
| 50 |
result["generation_evaluation"]["bert_score"]["score"],
|
| 51 |
result["generation_evaluation"]["nli"]["label"],
|
|
@@ -54,13 +64,6 @@ def save_evaluation(context, question, llm_response, result):
|
|
| 54 |
json.dumps(result),
|
| 55 |
datetime.now().isoformat()
|
| 56 |
))
|
| 57 |
-
conn.commit()
|
| 58 |
-
conn.close()
|
| 59 |
|
| 60 |
-
|
| 61 |
-
conn
|
| 62 |
-
cursor = conn.cursor()
|
| 63 |
-
cursor.execute("SELECT * FROM evaluations ORDER BY created_at DESC")
|
| 64 |
-
rows = cursor.fetchall()
|
| 65 |
-
conn.close()
|
| 66 |
-
return rows
|
|
|
|
| 35 |
def save_evaluation(context, question, llm_response, result):
|
| 36 |
conn = sqlite3.connect(DB_PATH)
|
| 37 |
cursor = conn.cursor()
|
| 38 |
+
|
| 39 |
+
cursor.execute("""
|
| 40 |
INSERT INTO evaluations (
|
| 41 |
+
context,
|
| 42 |
+
question,
|
| 43 |
+
llm_response,
|
| 44 |
+
final_verdict,
|
| 45 |
+
cosine_score,
|
| 46 |
+
bert_score,
|
| 47 |
+
nli_label,
|
| 48 |
+
nli_score,
|
| 49 |
+
fluency_verdict,
|
| 50 |
+
full_result,
|
| 51 |
+
created_at
|
| 52 |
+
)
|
| 53 |
+
VALUES (?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?)
|
| 54 |
+
""", (
|
| 55 |
context,
|
| 56 |
question,
|
| 57 |
llm_response,
|
| 58 |
+
result["final_verdict"],
|
| 59 |
result["generation_evaluation"]["cosine"]["score"],
|
| 60 |
result["generation_evaluation"]["bert_score"]["score"],
|
| 61 |
result["generation_evaluation"]["nli"]["label"],
|
|
|
|
| 64 |
json.dumps(result),
|
| 65 |
datetime.now().isoformat()
|
| 66 |
))
|
|
|
|
|
|
|
| 67 |
|
| 68 |
+
conn.commit()
|
| 69 |
+
conn.close()
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|