test_astream_with_callback() — langchain Function Reference

Architecture documentation for the test_astream_with_callback() function in test_chat_models.py from the langchain codebase.

Function python LangChainCore LanguageModelBase

Entity Profile

LangChainCore→ LanguageModelBase→ test_astream_with_callback() — langchain Function Reference

Dependency Diagram

graph TD
  d78963d0_9ca1_f29b_640a_6b21096dea04["test_astream_with_callback()"]
  aaa4b344_20ed_7fa3_4067_f8f05affc01f["test_chat_models.py"]
  d78963d0_9ca1_f29b_640a_6b21096dea04 -->|defined in| aaa4b344_20ed_7fa3_4067_f8f05affc01f
  style d78963d0_9ca1_f29b_640a_6b21096dea04 fill:#6366f1,stroke:#818cf8,color:#fff

Relationship Graph

Source Code

libs/partners/mistralai/tests/unit_tests/test_chat_models.py lines 163–167

async def test_astream_with_callback() -> None:
    callback = MyCustomHandler()
    chat = ChatMistralAI(callbacks=[callback])
    async for token in chat.astream("Hello"):
        assert callback.last_token == token.content