kenken999 commited on
Commit
1c6b863
β€’
1 Parent(s): ac64265
Files changed (1) hide show
  1. babyagi/classesa/diamond.py +1 -1
babyagi/classesa/diamond.py CHANGED
@@ -38,7 +38,7 @@ class ProductDatabase:
38
  cursor.execute("UPDATE diamondprice SET vector_col = %s WHERE id = %s", (vector, product_id))
39
  self.conn.commit()
40
 
41
- def search_similar_vectors(self, query_text, top_k=50):
42
  query_vector = self.get_embedding(query_text).tolist() # ndarray γ‚’γƒͺγ‚Ήγƒˆγ«ε€‰ζ›
43
  with self.conn.cursor() as cursor:
44
  cursor.execute("""
 
38
  cursor.execute("UPDATE diamondprice SET vector_col = %s WHERE id = %s", (vector, product_id))
39
  self.conn.commit()
40
 
41
+ def search_similar_vectors(self, query_text, top_k=10):
42
  query_vector = self.get_embedding(query_text).tolist() # ndarray γ‚’γƒͺγ‚Ήγƒˆγ«ε€‰ζ›
43
  with self.conn.cursor() as cursor:
44
  cursor.execute("""