after_model() — langchain Function Reference

Architecture documentation for the after_model() function in types.py from the langchain codebase.

Function python LangChainCore LanguageModelBase calls 1 called by 1

Entity Profile

LangChainCore→ LanguageModelBase→ after_model() — langchain Function Reference

Dependency Diagram

graph TD
  77982dfb_d47a_e437_1eaf_8b14518844c3["after_model()"]
  949c7cf4_56fe_f3b4_cd89_9631a7e9cb1e["AgentMiddleware"]
  77982dfb_d47a_e437_1eaf_8b14518844c3 -->|defined in| 949c7cf4_56fe_f3b4_cd89_9631a7e9cb1e
  becf4ff8_a57a_4780_fc64_de5f6e35885f["after_model()"]
  becf4ff8_a57a_4780_fc64_de5f6e35885f -->|calls| 77982dfb_d47a_e437_1eaf_8b14518844c3
  becf4ff8_a57a_4780_fc64_de5f6e35885f["after_model()"]
  77982dfb_d47a_e437_1eaf_8b14518844c3 -->|calls| becf4ff8_a57a_4780_fc64_de5f6e35885f
  style 77982dfb_d47a_e437_1eaf_8b14518844c3 fill:#6366f1,stroke:#818cf8,color:#fff

Relationship Graph

Source Code

libs/langchain_v1/langchain/agents/middleware/types.py lines 454–463

    def after_model(self, state: StateT, runtime: Runtime[ContextT]) -> dict[str, Any] | None:
        """Logic to run after the model is called.

        Args:
            state: The current agent state.
            runtime: The runtime context.

        Returns:
            Agent state updates to apply after model call.
        """