From d4bea179ec7ca5e86e713ab0c663607c527cd464 Mon Sep 17 00:00:00 2001 From: Erik Mackdanz Date: Fri, 29 Nov 2024 09:03:48 -0600 Subject: [PATCH] consistent project name --- Cargo.lock | 38 +++++++++++++++++++------------------- Cargo.toml | 2 +- src/main.rs | 22 +++++++++++++++------- 3 files changed, 35 insertions(+), 27 deletions(-) diff --git a/Cargo.lock b/Cargo.lock index 055b283..2816045 100644 --- a/Cargo.lock +++ b/Cargo.lock @@ -321,25 +321,6 @@ dependencies = [ "tracing", ] -[[package]] -name = "candlevec" -version = "0.1.0" -dependencies = [ - "anyhow", - "candle-core", - "candle-examples", - "candle-nn", - "candle-transformers", - "clap", - "hf-hub", - "pgvector", - "postgres", - "serde_json", - "tokenizers 0.20.3", - "tracing-chrome", - "tracing-subscriber", -] - [[package]] name = "cc" version = "1.2.1" @@ -3029,6 +3010,25 @@ version = "0.2.15" source = "registry+https://github.com/rust-lang/crates.io-index" checksum = "accd4ea62f7bb7a82fe23066fb0957d48ef677f6eeb8215f372f52e48bb32426" +[[package]] +name = "vecsearch" +version = "0.1.0" +dependencies = [ + "anyhow", + "candle-core", + "candle-examples", + "candle-nn", + "candle-transformers", + "clap", + "hf-hub", + "pgvector", + "postgres", + "serde_json", + "tokenizers 0.20.3", + "tracing-chrome", + "tracing-subscriber", +] + [[package]] name = "version_check" version = "0.9.5" diff --git a/Cargo.toml b/Cargo.toml index 6fba559..f99a9e8 100644 --- a/Cargo.toml +++ b/Cargo.toml @@ -1,5 +1,5 @@ [package] -name = "candlevec" +name = "vecsearch" version = "0.1.0" edition = "2021" diff --git a/src/main.rs b/src/main.rs index 2d9d9d7..3331680 100644 --- a/src/main.rs +++ b/src/main.rs @@ -15,11 +15,11 @@ //! createdb permission. //! //! ```text -//! $ ./candlevec init-database --help +//! $ ./vecsearch init-database --help //! //! Initialize the database when the database or table doesn't exist already //! -//! Usage: candlevec init-database [OPTIONS] --password +//! Usage: vecsearch init-database [OPTIONS] --password //! //! Options: //! --dbname [default: vsearch] @@ -33,7 +33,7 @@ //! //! ```text //! $ export PASSWORD=$(gpg -d pw-cvmigrator.gpg) -//! $ candlevec init-database +//! $ vecsearch init-database //! maybe creating the database //! database vsearch exists already //! maybe creating database objects @@ -48,13 +48,13 @@ //! //! ```text //! $ export PASSWORD=$(gpg -d pw-cvmigrator.gpg) -//! $ candlevec index --file testdata/0 +//! $ vecsearch index --file testdata/0 //! indexing a file //! Loaded and encoded 59.479µs //! Took 14.982262ms -//! $ candlevec index --file testdata/1 +//! $ vecsearch index --file testdata/1 //! ... -//! $ candlevec index --file testdata/2 +//! $ vecsearch index --file testdata/2 //! ... //! ``` //! @@ -69,7 +69,7 @@ //! //! ```text //! $ export PASSWORD=$(gpg -d pw-cvmigrator.gpg) -//! $ candlevec search --search feline +//! $ vecsearch search --search feline //! searching for document matches //! Loaded and encoded 49.306µs //! Took 14.452557ms @@ -78,6 +78,14 @@ //! The dog is barking //! ``` //! +//! ## TODO +//! +//! - slightly better test data +//! - unit tests? +//! - index multiple files +//! - model from main not PR +//! - unpin libs in Cargo.toml +//! //! //! use candle_transformers::models::bert::{BertModel, Config, HiddenAct, DTYPE}; -- 2.52.0