github/langchain
1
0
Fork 0
mirror of https://github.com/hwchase17/langchain.git synced 2025-09-28 15:00:23 +00:00
Code Issues Packages Projects Releases Wiki Activity

test(openai): raise token limit for o1 test (#33118)

Browse Source 
`test_o1[False-False]` was sometimes failing because the OpenAI o1 model
was hitting a token limit with only 100 tokens
This commit is contained in:
Mason Daugherty
2025-09-25 12:57:33 -04:00
committed by GitHub
parent eaf8dce7c2
commit 12daba63ff
1 changed files with 4 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
libs/partners/openai/tests/integration_tests/chat_models/test_base.py
View File

@@ -1017,10 +1017,12 @@ async def test_astream_response_format() -> None:
@pytest.mark.parametrize("use_responses_api", [False, True])
@pytest.mark.parametrize("use_max_completion_tokens", [True, False])
def test_o1(use_max_completion_tokens: bool, use_responses_api: bool) -> None:
# o1 models need higher token limits for reasoning
o1_token_limit = 1000
if use_max_completion_tokens:
kwargs: dict = {"max_completion_tokens": MAX_TOKEN_COUNT}
kwargs: dict = {"max_completion_tokens": o1_token_limit}
else:
kwargs = {"max_tokens": MAX_TOKEN_COUNT}
kwargs = {"max_tokens": o1_token_limit}
response = ChatOpenAI(
model="o1",
reasoning_effort="low",