From 6958c8dd484d99757c7f6a882f88afcdb0515963 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Jo=C3=A3o=20Moura?= Date: Mon, 23 Sep 2024 23:39:04 -0300 Subject: [PATCH] adding OPENAI_BASE_URL as fallback --- src/crewai/agent.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/src/crewai/agent.py b/src/crewai/agent.py index a943c994d..b79d23a5f 100644 --- a/src/crewai/agent.py +++ b/src/crewai/agent.py @@ -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