test_openai_ainvoke() — langchain Function Reference

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

Function python LangChainCore LanguageModelBase

Entity Profile

LangChainCore→ LanguageModelBase→ test_openai_ainvoke() — langchain Function Reference

Dependency Diagram

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

Relationship Graph

Source Code

libs/partners/openai/tests/unit_tests/chat_models/test_base.py lines 676–685

async def test_openai_ainvoke(mock_async_client: AsyncMock) -> None:
    llm = ChatOpenAI()

    with patch.object(llm, "async_client", mock_async_client):
        res = await llm.ainvoke("bar")
        assert res.content == "Bar Baz"

        # headers are not in response_metadata if include_response_headers not set
        assert "headers" not in res.response_metadata
    assert mock_async_client.with_raw_response.create.called