langchain Architecture Docs
Home / Function/ test_after_model_replaces_flagged_message() — langchain Function Reference

test_after_model_replaces_flagged_message() — langchain Function Reference

Architecture documentation for the test_after_model_replaces_flagged_message() function in test_openai_moderation_middleware.py from the langchain codebase.

Function python LangChainCore LanguageModelBase calls 2

Entity Profile

Dependency Diagram

graph TD
  44d03b6f_c14f_3e82_9042_f9c0eaf9b1e9["test_after_model_replaces_flagged_message()"]
  8b12907a_ea9d_32e0_a40a_d68cef74e4ab["test_openai_moderation_middleware.py"]
  44d03b6f_c14f_3e82_9042_f9c0eaf9b1e9 -->|defined in| 8b12907a_ea9d_32e0_a40a_d68cef74e4ab
  c1fe7511_88b4_fbbd_27e9_1aec3a6c7218["flagged_result()"]
  44d03b6f_c14f_3e82_9042_f9c0eaf9b1e9 -->|calls| c1fe7511_88b4_fbbd_27e9_1aec3a6c7218
  84d5741d_014f_d277_d96b_4ded50bb8f87["make_state()"]
  44d03b6f_c14f_3e82_9042_f9c0eaf9b1e9 -->|calls| 84d5741d_014f_d277_d96b_4ded50bb8f87
  style 44d03b6f_c14f_3e82_9042_f9c0eaf9b1e9 fill:#6366f1,stroke:#818cf8,color:#fff

Relationship Graph

Source Code

libs/partners/openai/tests/unit_tests/middleware/test_openai_moderation_middleware.py lines 148–159

def test_after_model_replaces_flagged_message() -> None:
    middleware = StubModerationMiddleware(
        {"unsafe": flagged_result()}, model="test", exit_behavior="replace"
    )
    state = make_state([AIMessage(content="unsafe", id="ai-1")])

    response = middleware.after_model(state, Mock())
    assert response is not None
    updated_messages = response["messages"]
    assert isinstance(updated_messages[-1], AIMessage)
    assert updated_messages[-1].id == "ai-1"
    assert "flagged" in updated_messages[-1].content

Domain

Subdomains

Defined In

Calls

Source

Frequently Asked Questions

What does test_after_model_replaces_flagged_message() do?
test_after_model_replaces_flagged_message() is a function in the langchain codebase, defined in libs/partners/openai/tests/unit_tests/middleware/test_openai_moderation_middleware.py.
Where is test_after_model_replaces_flagged_message() defined?
test_after_model_replaces_flagged_message() is defined in libs/partners/openai/tests/unit_tests/middleware/test_openai_moderation_middleware.py at line 148.
What does test_after_model_replaces_flagged_message() call?
test_after_model_replaces_flagged_message() calls 2 function(s): flagged_result, make_state.

Analyze Your Own Codebase

Get architecture documentation, dependency graphs, and domain analysis for your codebase in minutes.

Try Supermodel Free