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

fmt

Browse Source
This commit is contained in:
Bagatur
2024-11-26 15:57:05 -08:00
parent eec4df0d0a
commit fd1f6fb574
1 changed files with 0 additions and 10 deletions
Download Patch File Download Diff File Expand all files Collapse all files

10
libs/partners/openai/tests/integration_tests/chat_models/test_azure.py
View File

@@ -262,13 +262,3 @@ async def test_json_mode_async(llm: AzureChatOpenAI) -> None:
assert isinstance(full, AIMessageChunk) assert isinstance(full, AIMessageChunk)
assert isinstance(full.content, str) assert isinstance(full.content, str)
assert json.loads(full.content) == {"a": 1} 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)