MyCustomHandler Class — langchain Architecture

Architecture documentation for the MyCustomHandler class in test_chat_models.py from the langchain codebase.

Class python

Entity Profile

Dependency Diagram

graph TD
  580b66bd_c504_4cf2_8626_12bc7df9a602["MyCustomHandler"]
  af283d05_75a6_5813_267b_f03a32501460["BaseCallbackHandler"]
  580b66bd_c504_4cf2_8626_12bc7df9a602 -->|extends| af283d05_75a6_5813_267b_f03a32501460
  aaa4b344_20ed_7fa3_4067_f8f05affc01f["test_chat_models.py"]
  580b66bd_c504_4cf2_8626_12bc7df9a602 -->|defined in| aaa4b344_20ed_7fa3_4067_f8f05affc01f
  c717b7d4_5f72_c9c6_7e65_92bdb3518815["on_llm_new_token()"]
  580b66bd_c504_4cf2_8626_12bc7df9a602 -->|method| c717b7d4_5f72_c9c6_7e65_92bdb3518815

Relationship Graph

Source Code

libs/partners/mistralai/tests/unit_tests/test_chat_models.py lines 144–148

class MyCustomHandler(BaseCallbackHandler):
    last_token: str = ""

    def on_llm_new_token(self, token: str, **kwargs: Any) -> None:
        self.last_token = token