Support the `schema_generator` argument

[kshramt]

Support the schema_generator argument to allow users to utilize their own custom implementations.

[Jaza]

Description? Docs? Tests?

[kshramt]

Descriptin?

I have updated the description.

Docs?

Creating a new doc for this method might exceed the scope of this pull request. I suggest focusing on the current updates and addressing documentation separately if needed.

Tests?

Considering the simplicity of this change, adding a new test may not add significant value. Since there was no existing test, maintaining this status quo should suffice for now.