kiln_ai.adapters.fine_tune.finetune_registry

1from typing import Type
2
3from kiln_ai.adapters.fine_tune.base_finetune import BaseFinetuneAdapter
4from kiln_ai.adapters.fine_tune.openai_finetune import OpenAIFinetune
5from kiln_ai.adapters.ml_model_list import ModelProviderName
6
7finetune_registry: dict[ModelProviderName, Type[BaseFinetuneAdapter]] = {
8    ModelProviderName.openai: OpenAIFinetune,
9}