test_abatch_tags() — langchain Function Reference

Architecture documentation for the test_abatch_tags() function in test_llms.py from the langchain codebase.

Function python LangChainCore LanguageModelBase

Entity Profile

LangChainCore→ LanguageModelBase→ test_abatch_tags() — langchain Function Reference

Dependency Diagram

graph TD
  9b2b0304_128a_a0e8_2d65_59fc1f7f92e7["test_abatch_tags()"]
  15bd5c4f_97a3_2536_2af4_954ef3df2a7c["test_llms.py"]
  9b2b0304_128a_a0e8_2d65_59fc1f7f92e7 -->|defined in| 15bd5c4f_97a3_2536_2af4_954ef3df2a7c
  style 9b2b0304_128a_a0e8_2d65_59fc1f7f92e7 fill:#6366f1,stroke:#818cf8,color:#fff

Relationship Graph

Source Code

libs/partners/fireworks/tests/integration_tests/test_llms.py lines 73–81

async def test_abatch_tags() -> None:
    """Test batch tokens from Fireworks."""
    llm = Fireworks(model=_MODEL)

    result = await llm.abatch(
        ["I'm Pickle Rick", "I'm not Pickle Rick"], config={"tags": ["foo"]}
    )
    for token in result:
        assert isinstance(token, str)