async def test_toolkit_litellm_kwarg_passthrough(uc_client):
    """Test toolkit with keyword arguments for LitelLLM tool conversion."""
    mock_function_info = generate_function_info()

    with (
        mock.patch(
            "unitycatalog.ai.core.utils.client_utils.validate_or_set_default_client",
            return_value=uc_client,
        ),
        mock.patch.object(uc_client, "get_function", return_value=mock_function_info),
    ):
        tool = UCFunctionToolkit.uc_function_to_litellm_tool(
            function_name="catalog.schema.test",
            client=uc_client,
        )

        assert tool is not None
Test toolkit with keyword arguments for LitelLLM tool conversion.
ai/integrations/litellm/tests/test_litellm_toolkit_oss.py