after_model() — langchain Function Reference

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

Function python LangChainCore LanguageModelBase calls 1

Entity Profile

LangChainCore→ LanguageModelBase→ after_model() — langchain Function Reference

Dependency Diagram

graph TD
  d785fd03_0ba8_5b39_5297_f8614287fbc5["after_model()"]
  48713c67_3a9a_ec9e_ec74_46e6955f07bd["OpenAIModerationMiddleware"]
  d785fd03_0ba8_5b39_5297_f8614287fbc5 -->|defined in| 48713c67_3a9a_ec9e_ec74_46e6955f07bd
  171f9009_f0c5_1fb7_2224_0686c89f7a33["_moderate_output()"]
  d785fd03_0ba8_5b39_5297_f8614287fbc5 -->|calls| 171f9009_f0c5_1fb7_2224_0686c89f7a33
  style d785fd03_0ba8_5b39_5297_f8614287fbc5 fill:#6366f1,stroke:#818cf8,color:#fff

Relationship Graph

Source Code

libs/partners/openai/langchain_openai/middleware/openai_moderation.py lines 113–132

    def after_model(
        self, state: AgentState[Any], runtime: Runtime[Any]
    ) -> dict[str, Any] | None:  # type: ignore[override]
        """Moderate model output after the model is called.

        Args:
            state: Current agent state containing messages.
            runtime: Agent runtime context.

        Returns:
            Updated state with moderated messages, or `None` if no changes.
        """
        if not self.check_output:
            return None

        messages = list(state.get("messages", []))
        if not messages:
            return None

        return self._moderate_output(messages)