scriptoriumadmin/litellm
1
0
Fork 0
mirror of https://github.com/BerriAI/litellm.git synced 2026-10-05 02:41:56 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 49

fix(chatgpt): ignore request api_base for image generation

Browse source
This commit is contained in:
CrystalVibe28 2026-05-14 23:12:52 +08:00
parent 4f42ead4ca
commit 4b168229b5
2 changed files with 39 additions and 6 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

2
litellm/llms/chatgpt/image_generation/transformation.py
View file

@ -120,7 +120,7 @@ class ChatGPTImageGenerationConfig(BaseImageGenerationConfig):
litellm_params: dict,
stream: Optional[bool] = None,
) -> str:
api_base = api_base or self.authenticator.get_api_base() or CHATGPT_API_BASE
api_base = self.authenticator.get_api_base() or CHATGPT_API_BASE
api_base = self._canonicalize_codex_api_base(api_base)
return f"{api_base}/responses"

43
tests/image_gen_tests/test_chatgpt_image_generation.py
View file

@ -254,7 +254,7 @@ def test_chatgpt_image_generation_validate_environment_auth_error(
@pytest.mark.parametrize(
"api_base, expected",
"server_api_base, expected",
[
(
"https://chatgpt.com/backend-api",
@ -264,18 +264,27 @@ def test_chatgpt_image_generation_validate_environment_auth_error(
"https://chatgpt.com/backend-api/responses",
"https://chatgpt.com/backend-api/codex/responses",
),
("https://example.test/custom/", "https://example.test/custom/responses"),
(
"https://example.test/custom/",
"https://example.test/custom/responses",
),
],
)
def test_chatgpt_image_generation_get_complete_url_canonicalizes_api_base(
monkeypatch, tmp_path, api_base, expected
def test_chatgpt_image_generation_get_complete_url_canonicalizes_server_api_base(
monkeypatch, tmp_path, server_api_base, expected
):
monkeypatch.setenv("CHATGPT_TOKEN_DIR", str(tmp_path))
config = ChatGPTImageGenerationConfig()
class FakeAuthenticator:
def get_api_base(self):
return server_api_base
config.authenticator = cast(Any, FakeAuthenticator())
assert (
config.get_complete_url(
api_base=api_base,
api_base=None,
api_key=None,
model="gpt-image-2",
optional_params={},
@ -285,6 +294,30 @@ def test_chatgpt_image_generation_get_complete_url_canonicalizes_api_base(
)
def test_chatgpt_image_generation_get_complete_url_ignores_request_api_base(
monkeypatch, tmp_path
):
monkeypatch.setenv("CHATGPT_TOKEN_DIR", str(tmp_path))
config = ChatGPTImageGenerationConfig()
class FakeAuthenticator:
def get_api_base(self):
return "https://chatgpt.com/backend-api"
config.authenticator = cast(Any, FakeAuthenticator())
assert (
config.get_complete_url(
api_base="https://attacker.test/collect",
api_key=None,
model="gpt-image-2",
optional_params={},
litellm_params={},
)
== "https://chatgpt.com/backend-api/codex/responses"
)
def test_chatgpt_image_generation_get_complete_url_uses_authenticator_api_base(
monkeypatch, tmp_path
):