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

Extend test

Browse Source
This commit is contained in:
Nuno Campos 2023-08-18 16:22:01 +01:00
parent 9777c2801d
commit fdcd50aab4
1 changed files with 7 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
libs/langchain/tests/unit_tests/schema/runnable/test_runnable.py
View File

@ -1113,10 +1113,16 @@ def test_each(snapshot: SnapshotAssertion) -> None:
+ "{question}" + "{question}"
) )
first_llm = FakeStreamingListLLM(responses=["first item, second item, third item"]) first_llm = FakeStreamingListLLM(responses=["first item, second item, third item"])
parser = FakeSplitIntoListParser()
second_llm = FakeStreamingListLLM(responses=["this", "is", "a", "test"]) second_llm = FakeStreamingListLLM(responses=["this", "is", "a", "test"])
chain = prompt | first_llm | FakeSplitIntoListParser() | second_llm.map() chain = prompt | first_llm | parser | second_llm.map()
assert dumps(chain, pretty=True) == snapshot assert dumps(chain, pretty=True) == snapshot
output = chain.invoke({"question": "What up"}) output = chain.invoke({"question": "What up"})
assert output == ["this", "is", "a"] assert output == ["this", "is", "a"]
assert (parser | second_llm.map()).invoke("first item, second item") == [
"test",
"this",
]