Refactor code to load API keys from JSON config instead of env variables

src/bin/websocket.rs

@@ -31,14 +31,11 @@ async fn main() {
     tools.push(tool);
   }
 
-  // read /data/options.json as json
-  let options = std::fs::read_to_string("/data/options.json").expect("Error reading options");
-  let options: serde_json::Value = serde_json::from_str(&options).expect("Error parsing options");
-  println!("Loaded options: {}", options);
-  println!("Loaded {} tools", tools.len());
   println!("Supervisor token: {}", *shared::SUPERVISOR_TOKEN);
+  println!("Loaded options: {}", *shared::OPENAI_API_KEY);
   println!("OpenAi API key: {}", *shared::OPENAI_API_KEY);
   println!("Socket port: {}", *shared::SOCKET_PORT);
+  println!("Loaded {} tools", tools.len());
 
   let instructions = std::fs::read_to_string("resources/instructions.txt").expect("Error reading instructions");
   let env = Environment::new(tools, instructions);

src/lib.rs

@@ -2,3 +2,4 @@ pub mod openai;
 pub mod socket;
 pub mod files;
 pub mod shared;
+

src/shared.rs

@@ -1,9 +1,36 @@
 use std::env::var;
 
+#[derive(Debug, serde::Deserialize)]
+struct Options {
+  pub openai_api_key: String
+}
+
 lazy_static::lazy_static! {
   pub static ref SUPERVISOR_TOKEN: String = var("SUPERVISOR_TOKEN").expect("please set up the SUPERVISOR_TOKEN env variable before running this");
-  pub static ref OPENAI_API_KEY: String = var("OPTIONS_OPENAI_API_KEY").expect("please set up the OPENAI_API_KEY env variable before running this");
   pub static ref SOCKET_PORT: String = var("SOCKET_PORT").expect("please set up the SOCKET_PORT env variable before running this");
+
+  pub static ref OPENAI_API_KEY: String = {
+    if let Ok(key) = var("OPENAI_API_KEY") {
+      return key
+    }
+
+    match std::fs::read_to_string("/data/options.json") {
+      Ok(content) => {
+        match serde_json::from_str::<Options>(&content) {
+          Ok(options) => {
+            std::env::set_var("OPENAI_API_KEY", &options.openai_api_key);
+            return options.openai_api_key;
+          },
+          Err(err) => {
+            panic!("Error parsing /data/options.json file: {}", err);
+          }
+        }
+      },
+      Err(err) => {
+        panic!("Error reading /data/options.json file: {}", err);
+      }
+    }
+  };
 }
 
 pub static MODEL: &str = "gpt-3.5-turbo-1106";