Add static BASE_URL getter/setter with tests

rmagatti · Oct 18, 2023 · 116b3b7 · 116b3b7
1 parent 6993f80
commit 116b3b7
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/examples/chat_cli.rs b/examples/chat_cli.rs
@@ -7,14 +7,15 @@ use dotenvy::dotenv;
 
 use openai::{
     chat::{ChatCompletion, ChatCompletionMessage, ChatCompletionMessageRole},
-    set_key,
+    set_base_url, set_key,
 };
 
 #[tokio::main]
 async fn main() {
     // Make sure you have a file named `.env` with the `OPENAI_KEY` environment variable defined!
     dotenv().unwrap();
     set_key(env::var("OPENAI_KEY").unwrap());
+    set_base_url(env::var("OPENAI_BASE_URL").unwrap_or_default());
 
     let mut messages = vec![ChatCompletionMessage {
         role: ChatCompletionMessageRole::System,