get_model_class

rapidtools.models.get_model_class(provider: str) → type[BaseInferenceModel]

Import and return the wrapper class registered under provider.

Parameters:

provider – Registry key such as 'gemini' or 'qwen_vl'.

Returns:

type[BaseInferenceModel] – The wrapper class.

Raises:

KeyError – If provider is not registered.

Example

>>> from rapidtools.models import get_model_class
>>> get_model_class('claude').__name__
'ClaudeInference'