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

infra: reenable min version testing 2, ci ignore ai21 (#25709)

Browse Source
This commit is contained in:
Erick Friis
2024-08-23 16:28:42 -07:00
committed by GitHub
parent f6491ceb7d
commit 3dc7d447aa
3 changed files with 26 additions and 17 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
libs/core/tests/unit_tests/runnables/test_runnable.py
View File

@@ -3133,7 +3133,7 @@ def test_map_stream() -> None:
assert streamed_chunks[0] in [
{"passthrough": prompt.invoke({"question": "What is your name?"})},
{"llm": "i"},
{"chat": AIMessageChunk(content="i")},
{"chat": _AnyIdAIMessageChunk(content="i")},
]
assert len(streamed_chunks) == len(chat_res) + len(llm_res) + 1
assert all(len(c.keys()) == 1 for c in streamed_chunks)
@@ -3191,7 +3191,7 @@ def test_map_stream() -> None:
assert streamed_chunks[0] in [
{"llm": "i"},
{"chat": AIMessageChunk(content="i")},
{"chat": _AnyIdAIMessageChunk(content="i")},
]
assert len(streamed_chunks) == len(llm_res) + len(chat_res)
@@ -3234,7 +3234,7 @@ def test_map_stream_iterator_input() -> None:
assert streamed_chunks[0] in [
{"passthrough": "i"},
{"llm": "i"},
{"chat": AIMessageChunk(content="i")},
{"chat": _AnyIdAIMessageChunk(content="i")},
]
assert len(streamed_chunks) == len(chat_res) + len(llm_res) + len(llm_res)
assert all(len(c.keys()) == 1 for c in streamed_chunks)