Feat/psql

rig-postgres/Cargo.toml

@@ -12,6 +12,7 @@ serde = { version = "1.0.215", features = ["derive"] }
 serde_json = "1.0.133"
 tokio-postgres = "0.7.12"
 tracing = "0.1.40"
+dotenvy = "0.15.7"
 
 [dev-dependencies]
 anyhow = "1.0.94"

rig-postgres/README.md

@@ -7,7 +7,7 @@
     <span style="font-size: 48px; margin: 0 20px; font-weight: regular; font-family: Open Sans, sans-serif;"> + </span>
     <picture>
         <source srcset="https://www.postgresql.org/media/img/about/press/elephant.png">
-        <img src="https://www.postgresql.org/media/img/about/press/elephant.png" width="200" alt="SQLite logo">
+        <img src="https://www.postgresql.org/media/img/about/press/elephant.png" width="200" alt="Postgres logo">
     </picture>
 </div>
 

rig-postgres/examples/vector_search_postgres.rs

@@ -37,6 +37,9 @@ impl PostgresVectorStoreTable for Document {
 
 #[tokio::main]
 async fn main() -> Result<(), anyhow::Error> {
+    // use dotvenv to load environment variables
+    dotenvy::dotenv().ok();
+
     // set up postgres connection
     let database_url = std::env::var("DATABASE_URL").expect("DATABASE_URL not set");
     let db_config: tokio_postgres::Config = database_url.parse()?;