We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent 7fcd9e8 commit f3cb505Copy full SHA for f3cb505
services/engine/dataherald/sql_generator/dataherald_sqlagent.py
@@ -294,8 +294,8 @@ def _run( # noqa: PLR0912
294
df["similarities"] = df.table_embedding.apply(
295
lambda x: self.cosine_similarity(x, question_embedding)
296
)
297
- df = df.sort_values(by="similarities", ascending=True)
298
- df = df.tail(TOP_TABLES)
+ df = df.sort_values(by="similarities", ascending=False)
+ df = df.head(TOP_TABLES)
299
most_similar_tables = self.similar_tables_based_on_few_shot_examples(df)
300
table_relevance = ""
301
for _, row in df.iterrows():
0 commit comments