@pytest.fixture()
def sample_tool_schema():
    return {
        "type": "function",
        "function": {
            "name": "calculate_sum",
            "description": "Calculate the sum of two numbers",
            "parameters": {
                "type": "object",
                "properties": {
                    "a": {"type": "number", "description": "First number"},
                    "b": {"type": "number", "description": "Second number"},
                },
                "required": ["a", "b"],
            },
        },
    }
    
@pytest.mark.order(3)
def test_tool_calling(get_client, sample_tool_schema):
    """Test tool calling functionality"""
    client, model, config = get_client
    if config["tool_calling"]:
        response = client.chat.completions.create(
            model=model,
            messages=[{"role": "user", "content": "Calculate the sum of 15 and 27"}],
            tools=[sample_tool_schema],
            max_tokens=300,
        )

        assert response is not None
        assert hasattr(response, "choices")
        assert len(response.choices) > 0
        choice = response.choices[0]

        # Check if tool was called
        assert choice.message.tool_calls
        tool_call = choice.message.tool_calls[0]
        assert tool_call.function.name == "calculate_sum"
        assert tool_call.function.arguments is not None
        args = json.loads(tool_call.function.arguments)
        assert "a" in args
        assert "b" in args
    else:
        pytest.skip(f"Model {model} does not support tool_calling functionality")