test_fireworks_acall() — langchain Function Reference

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

Function python LangChainCore LanguageModelBase

Entity Profile

LangChainCore→ LanguageModelBase→ test_fireworks_acall() — langchain Function Reference

Dependency Diagram

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

Relationship Graph

Source Code

libs/partners/fireworks/tests/integration_tests/test_llms.py lines 33–45

async def test_fireworks_acall() -> None:
    """Test simple call to fireworks."""
    llm = Fireworks(
        model=_MODEL,
        temperature=0.2,
        max_tokens=250,
    )
    output = await llm.agenerate(["Say foo:"], stop=["bar"])

    assert llm._llm_type == "fireworks"
    output_text = output.generations[0][0].text
    assert isinstance(output_text, str)
    assert output_text.count("bar") <= 1