env support for most args

diff --git a/src/main.rs b/src/main.rs
index d0db01237ba6e441acf8dc899dcad79dc4530618..5b0d2802d4d436fbe265012dd24afa7316d88681 100644 (file)
--- a/src/main.rs
+++ b/src/main.rs
@@ -83,8 +83,7 @@
 //! 
 //! ## TODO
 //! 
-//! - model from main not PR
-//! - env support for all args
+//! - why model from PR not main?
 //! 
 use candle_transformers::models::bert::{BertModel, Config, HiddenAct, DTYPE};
 
@@ -104,13 +103,13 @@ enum Action {
     /// table doesn't exist already
     InitDatabase {
 
-       #[arg(long,default_value="vsearch")]
+       #[arg(long,env,default_value="vsearch")]
        dbname: String,
 
-       #[arg(long,default_value="localhost")]
+       #[arg(long,env,default_value="localhost")]
        host: String,
 
-       #[arg(long,default_value="cvmigrator")]
+       #[arg(long,env,default_value="cvmigrator")]
        user: String,
 
        #[arg(long,env)]
@@ -120,13 +119,13 @@ enum Action {
     /// Read one document and add it to the database index
     Index {
 
-       #[arg(long,default_value="vsearch")]
+       #[arg(long,env,default_value="vsearch")]
        dbname: String,
 
-       #[arg(long,default_value="localhost")]
+       #[arg(long,env,default_value="localhost")]
        host: String,
 
-       #[arg(long,default_value="cvmigrator")]
+       #[arg(long,env,default_value="cvmigrator")]
        user: String,
 
        #[arg(long,env)]
@@ -140,13 +139,13 @@ enum Action {
     /// Search the database for documents matching --search
     Search {
 
-       #[arg(long,default_value="vsearch")]
+       #[arg(long,env,default_value="vsearch")]
        dbname: String,
 
-       #[arg(long,default_value="localhost")]
+       #[arg(long,env,default_value="localhost")]
        host: String,
 
-       #[arg(long,default_value="cvmigrator")]
+       #[arg(long,env,default_value="cvmigrator")]
        user: String,
 
        #[arg(long,env)]
@@ -166,22 +165,22 @@ struct Args {
 
     /// The model to use, check out available models:
     /// <https://huggingface.co/models?library=sentence-transformers&sort=trending>
-    #[arg(long)]
+    #[arg(long,env)]
     model_id: Option<String>,
 
-    #[arg(long)]
+    #[arg(long,env)]
     revision: Option<String>,
 
     /// Use the pytorch weights rather than the safetensors ones
-    #[arg(long)]
+    #[arg(long,env)]
     use_pth: bool,
 
     /// L2 normalization for embeddings.
-    #[arg(long, default_value = "true")]
+    #[arg(long, env, default_value = "true")]
     normalize_embeddings: bool,
 
     /// Use tanh based approximation for Gelu instead of erf implementation.
-    #[arg(long, default_value = "false")]
+    #[arg(long, env, default_value = "false")]
     approximate_gelu: bool,
 
 }