test_convert_to_openai_function_nested_strict_2() — langchain Function Reference

Source Code

libs/core/tests/unit_tests/utils/test_function_calling.py lines 1123–1158

def test_convert_to_openai_function_nested_strict_2() -> None:
    def my_function(arg1: dict, arg2: dict | None) -> None:
        """Dummy function."""

    expected: dict = {
        "name": "my_function",
        "description": "Dummy function.",
        "parameters": {
            "type": "object",
            "properties": {
                "arg1": {
                    "additionalProperties": False,
                    "type": "object",
                },
                "arg2": {
                    "anyOf": [
                        {"additionalProperties": False, "type": "object"},
                        {"type": "null"},
                    ],
                },
            },
            "required": ["arg1", "arg2"],
            "additionalProperties": False,
        },
        "strict": True,
    }

    # there will be no extra `"additionalProperties": False` when Pydantic < 2.11
    if parse(version("pydantic")) < parse("2.11"):
        del expected["parameters"]["properties"]["arg1"]["additionalProperties"]
        del expected["parameters"]["properties"]["arg2"]["anyOf"][0][
            "additionalProperties"
        ]

    actual = convert_to_openai_function(my_function, strict=True)
    assert actual == expected