Open AI env var fix (#2384)

This commit is contained in:
Pranav Puranik
2025-03-25 22:13:33 -05:00
committed by GitHub
parent 9cb2a13f3b
commit 4321d24284
5 changed files with 56 additions and 6 deletions

View File

@@ -1,5 +1,5 @@
import json
import os
import warnings
from typing import Dict, List, Optional
from openai import OpenAI
@@ -24,7 +24,19 @@ class OpenAILLM(LLMBase):
)
else:
api_key = self.config.api_key or os.getenv("OPENAI_API_KEY")
base_url = self.config.openai_base_url or os.getenv("OPENAI_API_BASE") or "https://api.openai.com/v1"
base_url = (
self.config.openai_base_url
or os.getenv("OPENAI_API_BASE")
or os.getenv("OPENAI_BASE_URL")
or "https://api.openai.com/v1"
)
if os.environ.get("OPENAI_API_BASE"):
warnings.warn(
"The environment variable 'OPENAI_API_BASE' is deprecated and will be removed in the 0.1.80. "
"Please use 'OPENAI_BASE_URL' instead.",
DeprecationWarning
)
self.client = OpenAI(api_key=api_key, base_url=base_url)
def _parse_response(self, response, tools):