scriptoriumadmin/litellm
1
0
Fork 0
mirror of https://github.com/BerriAI/litellm.git synced 2026-09-19 00:01:29 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 88

test(chatgpt): inject the authenticator into the responses config so handler tests never log in

Browse source
This commit is contained in:
mateo-berri 2026-09-15 05:40:19 -07:00
parent 5e2e009d49
commit 04e164bcc2
2 changed files with 11 additions and 3 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

4
litellm/llms/chatgpt/responses/transformation.py
View file

@ -33,9 +33,9 @@ if TYPE_CHECKING:
class ChatGPTResponsesAPIConfig(OpenAIResponsesAPIConfig):
def __init__(self) -> None:
def __init__(self, authenticator: Authenticator | None = None) -> None:
super().__init__()
self.authenticator = Authenticator()
self.authenticator = authenticator if authenticator is not None else Authenticator()
@property
def custom_llm_provider(self) -> LlmProviders:

10
tests/test_litellm/llms/custom_httpx/test_llm_http_handler.py
View file

@ -25,6 +25,7 @@ from litellm.llms.base_llm.search.transformation import BaseSearchConfig, Search
from litellm.llms.bedrock.base_aws_llm import SignsRequestsWithAWS
from litellm.llms.brave.search.transformation import BraveSearchConfig
from litellm.llms.base_llm.image_edit.transformation import BaseImageEditConfig
from litellm.llms.chatgpt.authenticator import Authenticator as ChatGPTAuthenticator
from litellm.llms.chatgpt.responses.transformation import ChatGPTResponsesAPIConfig
from litellm.llms.custom_httpx.http_handler import AsyncHTTPHandler, HTTPHandler
from litellm.llms.custom_httpx.llm_http_handler import (
@ -473,11 +474,18 @@ def _chatgpt_responses_logging_obj():
return logging_obj
def _chatgpt_responses_config():
authenticator = Mock(spec=ChatGPTAuthenticator)
authenticator.get_access_token.return_value = "access-test"
authenticator.get_account_id.return_value = "acct-test"
return ChatGPTResponsesAPIConfig(authenticator=authenticator)
def _chatgpt_handler_kwargs(caller_params, client):
return {
"model": "gpt-5.3-codex",
"input": "hi",
"responses_api_provider_config": ChatGPTResponsesAPIConfig(),
"responses_api_provider_config": _chatgpt_responses_config(),
"response_api_optional_request_params": caller_params,
"custom_llm_provider": "chatgpt",
"litellm_params": GenericLiteLLMParams(api_key="sk-test", api_base="https://chatgpt.example.com"),