sha6th commited on
Commit
f6a32f7
·
1 Parent(s): dd5aff2

Fix evaluation database mapping

Browse files
Files changed (1) hide show
  1. 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
- cursor.execute('''
 
39
  INSERT INTO evaluations (
40
- context, question, llm_response,
41
- final_verdict, cosine_score, bert_score,
42
- nli_label, nli_score, fluency_verdict,
43
- full_result, created_at
44
- ) VALUES (?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?)
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
- def get_all_evaluations():
61
- conn = sqlite3.connect(DB_PATH)
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()