puuuush

2024-12-11 22:21:14 +01:00
parent 9377937347
commit fde7e7570a
2 changed files with 14 additions and 12 deletions
@@ -55,7 +55,7 @@ def prompt_embedding(prompt):
    )
    cur = conn.cursor()

-    cur.execute(f"select id, filepath, embedding <-> %s::vector as distance from dbtable order by distance limit 10;", (embedding["embedding"],))
+    cur.execute(f"select id, doc_id, filepath, embedding <-> %s::vector as distance from dbtable order by distance limit 3;", (embedding["embedding"],))

    result = cur.fetchall()