From a488b9b4f54e5f0f0c63c4ee9516bce6fbfe4dea Mon Sep 17 00:00:00 2001 From: Guoli Yin <67664443+gyin94@users.noreply.github.com> Date: Wed, 28 Aug 2024 08:14:41 -0700 Subject: [PATCH] [Feature] Make OPENAI_API_BASE compatible with openai default env (#1461) * Make OPENAI_API_BASE compatible with openai default env * Make OPENAI_API_BASE compatible with openai default env --------- Co-authored-by: Guoli Yin --- opencompass/models/openai_api.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/opencompass/models/openai_api.py b/opencompass/models/openai_api.py index c03d9650..1ae48006 100644 --- a/opencompass/models/openai_api.py +++ b/opencompass/models/openai_api.py @@ -15,7 +15,9 @@ from opencompass.utils.prompt import PromptList from .base_api import BaseAPIModel PromptType = Union[PromptList, str] -OPENAI_API_BASE = 'https://api.openai.com/v1/chat/completions' +OPENAI_API_BASE = os.path.join( + os.environ.get('OPENAI_BASE_URL', 'https://api.openai.com/v1/'), + 'chat/completions') @MODELS.register_module()