Skip to content

Commit 3ac2aee

Browse files
AntoineRqeAntoine Rouquette64bit
authored andcommitted
feat: support for OPENAI_BASE_URL env var (64bit#480)
* retrieve openai_api_base from env var OPENAI_API_BASE * Update async-openai/src/config.rs * Update async-openai/README.md * Update async-openai/README.md * Update async-openai/src/config.rs * Update async-openai/README.md * Update async-openai/src/config.rs --------- Co-authored-by: Antoine Rouquette <[email protected]> Co-authored-by: Himanshu Neema <[email protected]> (cherry picked from commit 73b4735)
1 parent 1b12014 commit 3ac2aee

File tree

1 file changed

+3
-2
lines changed

1 file changed

+3
-2
lines changed

async-openai/src/config.rs

Lines changed: 3 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -68,7 +68,8 @@ pub struct OpenAIConfig {
6868
impl Default for OpenAIConfig {
6969
fn default() -> Self {
7070
Self {
71-
api_base: OPENAI_API_BASE.to_string(),
71+
api_base: std::env::var("OPENAI_BASE_URL")
72+
.unwrap_or_else(|_| OPENAI_API_BASE.to_string()),
7273
api_key: std::env::var("OPENAI_API_KEY")
7374
.or_else(|_| {
7475
std::env::var("OPENAI_ADMIN_KEY").inspect(|_admin_key| {
@@ -85,7 +86,7 @@ impl Default for OpenAIConfig {
8586
}
8687

8788
impl OpenAIConfig {
88-
/// Create client with default [OPENAI_API_BASE] url and default API key from OPENAI_API_KEY env var
89+
/// Create client with default [OPENAI_API_BASE] url (can also be changed with OPENAI_BASE_URL env var) and default API key from OPENAI_API_KEY env var
8990
pub fn new() -> Self {
9091
Default::default()
9192
}

0 commit comments

Comments
 (0)