Skip to content

Commit f3cb505

Browse files
authored
fix sorting table relevance scores (#515)
1 parent 7fcd9e8 commit f3cb505

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

services/engine/dataherald/sql_generator/dataherald_sqlagent.py

+2-2
Original file line numberDiff line numberDiff line change
@@ -294,8 +294,8 @@ def _run( # noqa: PLR0912
294294
df["similarities"] = df.table_embedding.apply(
295295
lambda x: self.cosine_similarity(x, question_embedding)
296296
)
297-
df = df.sort_values(by="similarities", ascending=True)
298-
df = df.tail(TOP_TABLES)
297+
df = df.sort_values(by="similarities", ascending=False)
298+
df = df.head(TOP_TABLES)
299299
most_similar_tables = self.similar_tables_based_on_few_shot_examples(df)
300300
table_relevance = ""
301301
for _, row in df.iterrows():

0 commit comments

Comments
 (0)