Skip to content

Commit f14dae8

Browse files
committed
fix(pgvectorrm): Fixed bug when include similarity = false
1 parent 3a47bc3 commit f14dae8

File tree

1 file changed

+4
-1
lines changed

1 file changed

+4
-1
lines changed

dspy/retrieve/pgvector_rm.py

+4-1
Original file line numberDiff line numberDiff line change
@@ -117,6 +117,9 @@ def forward(self, query: str):
117117
).format(embedding_field=sql.Identifier(self.embedding_field))
118118
)
119119
fields += similarity_field
120+
args = (query_embedding, query_embedding, self.k)
121+
else:
122+
args = (query_embedding, self.k)
120123

121124
sql_query = sql.SQL(
122125
"select {fields} from {table} order by {embedding_field} <=> %s::vector limit %s"
@@ -130,7 +133,7 @@ def forward(self, query: str):
130133
with conn.cursor() as cur:
131134
cur.execute(
132135
sql_query,
133-
(query_embedding, query_embedding, self.k))
136+
args)
134137
rows = cur.fetchall()
135138
columns = [descrip[0] for descrip in cur.description]
136139
for row in rows:

0 commit comments

Comments
 (0)