Skip to content

Commit 2363209

Browse files
committed
feat(export): include similarity search results in CSV export
* Added export of similarity search results with top_k in operation name. * Updated CSV writing logic to accommodate new results.
1 parent d097f9d commit 2363209

File tree

1 file changed

+18
-1
lines changed

1 file changed

+18
-1
lines changed

benchmarks/reporter.py

Lines changed: 18 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -114,7 +114,6 @@ def export_to_csv(
114114
"update_many",
115115
"get_all",
116116
"delete_many",
117-
"similarity_search",
118117
]
119118
for op in operations:
120119
matching = [r for r in mode_results[size] if r["operation"] == op]
@@ -125,4 +124,22 @@ def export_to_csv(
125124
else:
126125
writer.writerow([op, "N/A", "N/A"])
127126

127+
# Add similarity search results with top_k in operation name
128+
search_results = [
129+
r
130+
for r in mode_results[size]
131+
if r["operation"] == "similarity_search"
132+
]
133+
for result in search_results:
134+
top_k = result.get("top_k", 0)
135+
ops_per_sec = result.get("ops_per_sec", 0)
136+
time_val = result.get("time", 0)
137+
writer.writerow(
138+
[
139+
f"similarity_search_{top_k}",
140+
f"{ops_per_sec:.2f}",
141+
f"{time_val:.4f}",
142+
]
143+
)
144+
128145
print(f"Exported {db_mode} ({size} records) to: {filename}")

0 commit comments

Comments
 (0)