Merge pull request #267 from danbev/openai-api-base-url

openai: add support for setting the base url
sobelio · Feb 5, 2024 · bb57935 · bb57935
2 parents a416579 + 2ecfbff
commit bb57935
Showing 1 changed file with 5 additions and 0 deletions.
diff --git a/crates/llm-chain-openai/src/chatgpt/executor.rs b/crates/llm-chain-openai/src/chatgpt/executor.rs
@@ -94,6 +94,11 @@ impl traits::Executor for Executor {
         if let Ok(org_id) = std::env::var("OPENAI_ORG_ID") {
             cfg = cfg.with_org_id(org_id);
         }
+
+        if let Ok(base_url) = std::env::var("OPENAI_API_BASE_URL") {
+            cfg = cfg.with_api_base(base_url);
+        }
+
         let client = Arc::new(async_openai::Client::with_config(cfg));
         Ok(Self { client, options })
     }