github/langchain
1
0
Fork 0
mirror of https://github.com/hwchase17/langchain.git synced 2025-09-02 03:26:17 +00:00
Code Issues Packages Projects Releases Wiki Activity

BUGFIX: llm backwards compat imports (#13698)

Browse Source
This commit is contained in:
Bagatur
2023-11-21 20:12:35 -08:00
committed by GitHub
parent ace9e64d62
commit a21e84faf7
6 changed files with 54 additions and 18 deletions
Download Patch File Download Diff File Expand all files Collapse all files

0
libs/core/tests/unit_tests/language_models/llms/__init__.py Normal file
View File

19
libs/core/tests/unit_tests/language_models/llms/test_base.py Normal file
View File

@@ -0,0 +1,19 @@
from tests.unit_tests.fake.llm import FakeListLLM
def test_batch() -> None:
llm = FakeListLLM(responses=["foo"] * 3)
output = llm.batch(["foo", "bar", "foo"])
assert output == ["foo"] * 3
output = llm.batch(["foo", "bar", "foo"], config={"max_concurrency": 2})
assert output == ["foo"] * 3
async def test_abatch() -> None:
llm = FakeListLLM(responses=["foo"] * 3)
output = await llm.abatch(["foo", "bar", "foo"])
assert output == ["foo"] * 3
output = await llm.abatch(["foo", "bar", "foo"], config={"max_concurrency": 2})
assert output == ["foo"] * 3