Lint

Author: Joshmantova

dspy/retrieve/pgvector_rm.py

@@ -113,7 +113,7 @@ def forward(self, query: str):
             similarity_field = (
                 sql.SQL(',') +
                 sql.SQL(
-                    '1 - ({embedding_field} <=> %s) AS similarity'
+                    '1 - ({embedding_field} <=> %s) AS similarity',
                 ).format(embedding_field=sql.Identifier(self.embedding_field))
             )
             fields += similarity_field
@@ -122,7 +122,7 @@ def forward(self, query: str):
             args = (query_embedding, self.k)
 
         sql_query = sql.SQL(
-            "select {fields} from {table} order by {embedding_field} <=> %s::vector limit %s"
+            "select {fields} from {table} order by {embedding_field} <=> %s::vector limit %s",
         ).format(
             fields=fields,
             table=sql.Identifier(self.pg_table_name),