mirror of
https://github.com/hwchase17/langchain.git
synced 2025-04-27 19:46:55 +00:00
fmt
This commit is contained in:
parent
eec4df0d0a
commit
fd1f6fb574
@ -262,13 +262,3 @@ async def test_json_mode_async(llm: AzureChatOpenAI) -> None:
|
||||
assert isinstance(full, AIMessageChunk)
|
||||
assert isinstance(full.content, str)
|
||||
assert json.loads(full.content) == {"a": 1}
|
||||
|
||||
|
||||
def test_o1_max_tokens() -> None:
|
||||
response = ChatOpenAI(model="o1-mini", max_tokens=10).invoke("how are you") # type: ignore[call-arg]
|
||||
assert isinstance(response, AIMessage)
|
||||
|
||||
response = ChatOpenAI(model="gpt-4o", max_completion_tokens=10).invoke(
|
||||
"how are you"
|
||||
)
|
||||
assert isinstance(response, AIMessage)
|
||||
|
Loading…
Reference in New Issue
Block a user