Spaces:
Running
on
Zero
Running
on
Zero
fda
Browse files
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=
|
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("""
|