env guarding

Signed-off-by: Alex Jones <[email protected]>

Cargo.toml

@@ -44,6 +44,8 @@ tokio = { version = "1", features = ["full"] }
 unescape = "0.1.0"
 v = "0.1.0"
 chrono = "0.4.31"
+log = "0.4.20"
+env_logger = "0.10.1"
 
 # Config for 'cargo dist'
 [workspace.metadata.dist]

src/analyze/mod.rs

@@ -12,6 +12,7 @@ use std::error::Error;
 use std::sync::mpsc;
 use std::sync::mpsc::{Receiver, Sender};
 use std::time::Duration;
+use log::warn;
 
 pub async fn list_analyzers() -> Result<(), Box<dyn Error>> {
     // Setup available providers
@@ -149,7 +150,9 @@ pub async fn run_analysis(
                             // check if the processed results analyzer exists as key
                             // upsert the analysis result into the vector
                         }
-                        Err(_e) => (),
+                        Err(e) => (
+                            warn!("{}", e)
+                            ),
                     }
                 }
             }

src/bedrock/mod.rs

@@ -32,25 +32,27 @@ impl From<ClaudParams> for Blob {
         Blob::new(serde_json::to_string(&val).unwrap())
     }
 }
-pub struct BedrockClient {}
+pub struct BedrockClient {
+    region: Region,
+    bedrock_model: String
+}
 
 impl BedrockClient {
     pub fn new() -> Self {
-        Self {}
+        Self {
+            region: Region::new(env::var("BEDROCK_REGION").unwrap().clone()),
+            bedrock_model: env::var("BEDROCK_MODEL").unwrap().clone()
+        }
     }
     pub async fn enrich(&self, prompt: String) -> Result<String, Box<dyn Error>> {
         // force the config rejoin be set
-        let bedrock_region = env::var("BEDROCK_REGION")?.clone();
-        let region = Region::new(bedrock_region);
-        let config = aws_config::from_env().region(region).load().await;
+        let config = aws_config::from_env().region(self.region.clone()).load().await;
         let client = aws_sdk_bedrockruntime::Client::new(&config);
-        let bedrock_model = env::var("BEDROCK_MODEL")?;
-
         let question = format!("{} {}", prompt::DEFAULT_PROMPT, prompt.as_str());
 
         let response = client
             .invoke_model()
-            .model_id(bedrock_model)
+            .model_id(self.bedrock_model.clone())
             .content_type("application/json")
             .body(ClaudParams::new(question.as_str()).into())
             .send()

src/main.rs

@@ -1,3 +1,5 @@
+use std::env;
+use std::process::exit;
 use clap::{Parser, Subcommand};
 mod analyze;
 mod analyzer;
@@ -40,8 +42,27 @@ enum Commands {
         resource: String,
     },
 }
+
+const ENVS: &'static [&'static str] = &["BEDROCK_REGION",
+    "BEDROCK_MODEL",
+"AWS_REGION","AWS_ACCESS_KEY","AWS_SECRET_ACCESS_KEY"];
+
+
 #[tokio::main]
 async fn main() {
+    env_logger::init();
+
+    for e in ENVS {
+        let r = env::var(e);
+        match r {
+            Ok(x) => {},
+            Err(_e) => {
+                    println!("ENV: {} not found",e);
+                    exit(1);
+            },
+        }
+    }
+
     let args = Args::parse();
 
     match &args.command {