fix: add OPENAI_BASE_URL support

2025-04-10 21:05:28 +08:00
parent eb6c834063
commit a401359b1b
1 changed files with 2 additions and 1 deletions
--- a/minirag/llm/openai.py
+++ b/minirag/llm/openai.py
@@ -101,7 +101,8 @@ async def openai_complete_if_cache(
 ) -> str:
    if api_key:
        os.environ["OPENAI_API_KEY"] = api_key
-
+    if base_url==None:
+        base_url = os.environ["OPENAI_API_BASE"]
    openai_async_client = (
        AsyncOpenAI() if base_url is None else AsyncOpenAI(base_url=base_url)
    )