puuuuush

lib.py

@@ -55,7 +55,9 @@ def prompt_embedding(prompt):
     )
     cur = conn.cursor()
 
-    cur.execute(f"select id, filepath, embedding, embedding <-> {embedding} as distance from dbtable order by distance limit 1;")
+    cur.execute(f"select id, filepath, embedding <-> %s::vector as distance from dbtable order by distance limit 10;", (embedding["embedding"],))
 
     result = cur.fetchall()
+
+    conn.close()
     print(result)