mock_client() — langchain Function Reference

Architecture documentation for the mock_client() function in test_base.py from the langchain codebase.

Function python LangChainCore LanguageModelBase

Entity Profile

LangChainCore→ LanguageModelBase→ mock_client() — langchain Function Reference

Dependency Diagram

graph TD
  8e1e4e08_fc96_d93c_7dfa_44c8ab3996ac["mock_client()"]
  48232d20_f8c1_b597_14fa_7dc407e9bfe5["test_base.py"]
  8e1e4e08_fc96_d93c_7dfa_44c8ab3996ac -->|defined in| 48232d20_f8c1_b597_14fa_7dc407e9bfe5
  style 8e1e4e08_fc96_d93c_7dfa_44c8ab3996ac fill:#6366f1,stroke:#818cf8,color:#fff

Relationship Graph

Source Code

libs/partners/openai/tests/unit_tests/chat_models/test_base.py lines 635–647

def mock_client(mock_completion: dict) -> MagicMock:
    rtn = MagicMock()

    mock_create = MagicMock()

    mock_resp = MagicMock()
    mock_resp.headers = {"content-type": "application/json"}
    mock_resp.parse.return_value = mock_completion
    mock_create.return_value = mock_resp

    rtn.with_raw_response.create = mock_create
    rtn.create.return_value = mock_completion
    return rtn