config.rs 353 B

12345678910111213141516
  1. use anyhow::{anyhow, Error};
  2. pub struct OpenAISetting {
  3. pub openai_api_key: String,
  4. }
  5. const OPENAI_API_KEY: &str = "OPENAI_API_KEY";
  6. impl OpenAISetting {
  7. pub fn from_env() -> Result<Self, Error> {
  8. let openai_api_key =
  9. std::env::var(OPENAI_API_KEY).map_err(|_| anyhow!("Missing OPENAI_API_KEY"))?;
  10. Ok(Self { openai_api_key })
  11. }
  12. }