adding OPENAI_BASE_URL as fallback

This commit is contained in:
João Moura
2024-09-23 23:39:04 -03:00
parent 09eaa76b44
commit 6958c8dd48

View File

@@ -132,7 +132,9 @@ class Agent(BaseAgent):
model_name = os.environ.get("OPENAI_MODEL_NAME", "gpt-4o-mini")
llm_params = {"model": model_name}
api_base = os.environ.get("OPENAI_API_BASE")
api_base = os.environ.get("OPENAI_API_BASE") or os.environ.get(
"OPENAI_BASE_URL"
)
if api_base:
llm_params["base_url"] = api_base