andre/crewAI
1
0
Fork 0
mirror of https://github.com/crewAIInc/crewAI.git synced 2026-01-25 16:18:13 +00:00
Code Issues Actions 1 Packages Projects Releases Wiki Activity

adding OPENAI_BASE_URL as fallback

Browse Source
This commit is contained in:
João Moura
2024-09-23 23:39:04 -03:00
parent b6075f1a97
commit 7987bfee39
1 changed files with 3 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
src/crewai/agent.py
View File

@@ -132,7 +132,9 @@ class Agent(BaseAgent):
model_name = os.environ.get("OPENAI_MODEL_NAME", "gpt-4o-mini") model_name = os.environ.get("OPENAI_MODEL_NAME", "gpt-4o-mini")
llm_params = {"model": model_name} 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: if api_base:
llm_params["base_url"] = api_base llm_params["base_url"] = api_base