fixed extra fetches

Author: anngoroshi

test2text/pages/reports/report_by_tc.py

@@ -77,11 +77,11 @@ def write_requirements(current_requirements: set[tuple]):
             if distance_sql:
                 tc_dict = {
                     f"{test_case} [smart search d={round_distance(distance)}]": tc_id
-                    for (tc_id, _, test_case, distance) in data.fetchall()
+                    for (tc_id, _, test_case, distance) in data
                 }
             else:
                 tc_dict = {
-                    test_case: tc_id for (tc_id, _, test_case) in data.fetchall()
+                    test_case: tc_id for (tc_id, _, test_case) in data
                 }
 
             st.subheader("Choose ONE of filtered test cases")

test2text/services/visualisation/visualize_vectors.py

@@ -15,11 +15,11 @@
 
 def extract_annotation_vectors(db: DbClient):
     vectors = []
-    embeddings = db.get_column_values("embedding", "Annotations")
-    if embeddings.fetchone() is None:
+    embeddings = db.get_column_values("embedding", from_table="Annotations")
+    if not embeddings:
         st.error("Embeddings is empty. Please fill embeddings in annotations.")
         return None
-    for row in embeddings.fetchall():
+    for row in embeddings:
         if row[0] is not None:
             vectors.append(np.array(unpack_float32(row[0])))
     return np.array(vectors)
@@ -38,11 +38,11 @@ def extract_closest_annotation_vectors(db: DbClient):
 
 def extract_requirement_vectors(db: DbClient):
     vectors = []
-    embeddings = db.get_column_values("embedding", "Requirements")
+    embeddings = db.get_column_values("embedding", from_table="Requirements")
     if embeddings.fetchone() is None:
         st.error("Embeddings is empty. Please fill embeddings in requirements.")
         return None
-    for row in embeddings.fetchall():
+    for row in embeddings:
         vectors.append(np.array(unpack_float32(row[0])))
     return np.array(vectors)