From: Erik Mackdanz <erikmack@gmail.com>
Date: Thu, 28 Nov 2024 15:58:51 +0000 (-0600)
Subject: data is keyed by content
X-Git-Url: https://git.humopery.space/?a=commitdiff_plain;h=3f2d452ddd6e92108e05c543f983fb784089e896;p=vecsearch.git

data is keyed by content
---

diff --git a/src/main.rs b/src/main.rs
index 1def530..f85dae5 100644
--- a/src/main.rs
+++ b/src/main.rs
@@ -157,7 +157,7 @@ fn init_database(dbname: String, host: String, user: String, password: String) -
     client.execute("CREATE EXTENSION IF NOT EXISTS vector", &[])?;
     client.execute(
 	"CREATE TABLE if not exists documents \
-	 (id serial PRIMARY KEY, content text, \
+	 (content text PRIMARY KEY, \
 	 embedding vector(384))", &[])?;
 
     Ok(())
@@ -203,7 +203,8 @@ fn index(dbname: String, host: String, user: String, password: String,
 	.connect(NoTls)?;
 
     client.execute("INSERT INTO documents (content, embedding) \
-		    values ($1, $2)",
+		    values ($1, $2) \
+		    ON CONFLICT (content) DO UPDATE SET embedding = $2",
 		   &[&doc_content,&Vector::from(embeddings)],
     )?;