aafter_model() — langchain Function Reference

Architecture documentation for the aafter_model() function in pii.py from the langchain codebase.

Function python LangChainCore LanguageModelBase calls 1

Entity Profile

LangChainCore→ LanguageModelBase→ aafter_model() — langchain Function Reference

Dependency Diagram

graph TD
  074d1c14_b6ba_2cd0_af55_0985e89d7adf["aafter_model()"]
  dc527b75_da40_4d1c_e472_82d1252ba70b["PIIMiddleware"]
  074d1c14_b6ba_2cd0_af55_0985e89d7adf -->|defined in| dc527b75_da40_4d1c_e472_82d1252ba70b
  27d4728e_2258_70e9_e8c8_f68e263ca8fb["after_model()"]
  074d1c14_b6ba_2cd0_af55_0985e89d7adf -->|calls| 27d4728e_2258_70e9_e8c8_f68e263ca8fb
  style 074d1c14_b6ba_2cd0_af55_0985e89d7adf fill:#6366f1,stroke:#818cf8,color:#fff

Relationship Graph

Source Code

libs/langchain_v1/langchain/agents/middleware/pii.py lines 346–364

    async def aafter_model(
        self,
        state: AgentState[Any],
        runtime: Runtime[ContextT],
    ) -> dict[str, Any] | None:
        """Async check AI messages for PII after model invocation.

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

        Returns:
            Updated state with PII handled according to strategy, or None if no PII
                detected.

        Raises:
            PIIDetectionError: If PII is detected and strategy is `'block'`.
        """
        return self.after_model(state, runtime)