test_json_mode() — langchain Function Reference

Architecture documentation for the test_json_mode() function in test_azure.py from the langchain codebase.

Function python LangChainCore LanguageModelBase

Entity Profile

LangChainCore→ LanguageModelBase→ test_json_mode() — langchain Function Reference

Dependency Diagram

graph TD
  d477d4c7_49a7_a6a1_200d_ae568388d09a["test_json_mode()"]
  c413d48d_e43d_eae6_47cb_3eea9394c77c["test_azure.py"]
  d477d4c7_49a7_a6a1_200d_ae568388d09a -->|defined in| c413d48d_e43d_eae6_47cb_3eea9394c77c
  style d477d4c7_49a7_a6a1_200d_ae568388d09a fill:#6366f1,stroke:#818cf8,color:#fff

Relationship Graph

Source Code

libs/partners/openai/tests/integration_tests/chat_models/test_azure.py lines 234–249

def test_json_mode(llm: AzureChatOpenAI) -> None:
    response = llm.invoke(
        "Return this as json: {'a': 1}", response_format={"type": "json_object"}
    )
    assert isinstance(response.content, str)
    assert json.loads(response.content) == {"a": 1}

    # Test streaming
    full: BaseMessageChunk | None = None
    for chunk in llm.stream(
        "Return this as json: {'a': 1}", response_format={"type": "json_object"}
    ):
        full = chunk if full is None else full + chunk
    assert isinstance(full, AIMessageChunk)
    assert isinstance(full.content, str)
    assert json.loads(full.content) == {"a": 1}