diff --git a/Cargo.lock b/Cargo.lock index ccbf54493..0a5402069 100644 --- a/Cargo.lock +++ b/Cargo.lock @@ -2084,6 +2084,12 @@ dependencies = [ "serde_json", ] +[[package]] +name = "dotenvy" +version = "0.15.7" +source = "registry+https://github.com/rust-lang/crates.io-index" +checksum = "1aaf95b3e5c8f23aa320147307562d361db0ae0d51242340f558153b4eb2439b" + [[package]] name = "downcast-rs" version = "1.2.1" @@ -5378,6 +5384,7 @@ name = "rig-postgres" version = "0.1.0" dependencies = [ "anyhow", + "dotenvy", "log", "pgvector", "rig-core", diff --git a/rig-postgres/Cargo.toml b/rig-postgres/Cargo.toml index a351f2dc6..577a9f5bb 100644 --- a/rig-postgres/Cargo.toml +++ b/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" diff --git a/rig-postgres/README.md b/rig-postgres/README.md index bc25a6ce8..71cbcd20b 100644 --- a/rig-postgres/README.md +++ b/rig-postgres/README.md @@ -7,7 +7,7 @@ + - SQLite logo + Postgres logo diff --git a/rig-postgres/examples/vector_search_postgres.rs b/rig-postgres/examples/vector_search_postgres.rs index d08850de5..188109b2c 100644 --- a/rig-postgres/examples/vector_search_postgres.rs +++ b/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()?;