test_generic_fake_chat_model_ainvoke() — langchain Function Reference

Architecture documentation for the test_generic_fake_chat_model_ainvoke() function in test_fake_chat_model.py from the langchain codebase.

Function python LangChainCore LanguageModelBase

Entity Profile

LangChainCore→ LanguageModelBase→ test_generic_fake_chat_model_ainvoke() — langchain Function Reference

Dependency Diagram

graph TD
  aba33f91_0496_3913_477e_47ea98fc9576["test_generic_fake_chat_model_ainvoke()"]
  06318702_96e4_1032_1475_4dde8fe1f643["test_fake_chat_model.py"]
  aba33f91_0496_3913_477e_47ea98fc9576 -->|defined in| 06318702_96e4_1032_1475_4dde8fe1f643
  style aba33f91_0496_3913_477e_47ea98fc9576 fill:#6366f1,stroke:#818cf8,color:#fff

Relationship Graph

Source Code

libs/core/tests/unit_tests/fake/test_fake_chat_model.py lines 38–47

async def test_generic_fake_chat_model_ainvoke() -> None:
    # Will alternate between responding with hello and goodbye
    infinite_cycle = cycle([AIMessage(content="hello"), AIMessage(content="goodbye")])
    model = GenericFakeChatModel(messages=infinite_cycle)
    response = await model.ainvoke("meow")
    assert response == _any_id_ai_message(content="hello")
    response = await model.ainvoke("kitty")
    assert response == _any_id_ai_message(content="goodbye")
    response = await model.ainvoke("meow")
    assert response == _any_id_ai_message(content="hello")