From 3f2d452ddd6e92108e05c543f983fb784089e896 Mon Sep 17 00:00:00 2001
From: Erik Mackdanz <erikmack@gmail.com>
Date: Thu, 28 Nov 2024 09:58:51 -0600
Subject: [PATCH] data is keyed by content

---
 src/main.rs | 5 +++--
 1 file changed, 3 insertions(+), 2 deletions(-)

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)],
     )?;
 
-- 
2.52.0