dify-plugin-sdks/tests/test_tool_call_model_init.py

import pytest

from dify_plugin.entities.model.message import AssistantPromptMessage

EMPTY_STRING = ""


def test_tool_call_model_init_with_explicit_none_fields() -> None:
    response_tool_call = {
        "function": {
            "name": None,
            "arguments": None,
        },
        "id": None,
        "type": None,
    }

    try:
        function = AssistantPromptMessage.ToolCall.ToolCallFunction(
            name=response_tool_call.get("function", {}).get("name", ""),
            arguments=response_tool_call.get("function", {}).get("arguments", ""),
        )

        tool_call = AssistantPromptMessage.ToolCall(
            id=response_tool_call.get("id", ""),
            type=response_tool_call.get("type", ""),
            function=function,
        )

    except Exception as ex:
        pytest.fail(f"failed to initialize tool call: {ex!s}")
    else:
        assert tool_call.id == EMPTY_STRING
        assert tool_call.type == EMPTY_STRING
        assert tool_call.function.name == EMPTY_STRING
        assert tool_call.function.arguments == EMPTY_STRING

    response_function_call = {"name": None, "arguments": None, "id": None}

    try:
        function = AssistantPromptMessage.ToolCall.ToolCallFunction(
            name=response_function_call.get("name", ""),
            arguments=response_function_call.get("arguments", ""),
        )

        tool_call = AssistantPromptMessage.ToolCall(
            id=response_function_call.get("id", ""),
            type="function",
            function=function,
        )

    except Exception as ex:
        pytest.fail(f"failed to initialize tool call: {ex!s}")
    else:
        assert tool_call.id == EMPTY_STRING
        assert tool_call.function.name == EMPTY_STRING
        assert tool_call.function.arguments == EMPTY_STRING