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.fireworks_finetune import FireworksFinetune
 5from kiln_ai.adapters.fine_tune.openai_finetune import OpenAIFinetune
 6from kiln_ai.adapters.ml_model_list import ModelProviderName
 7
 8finetune_registry: dict[ModelProviderName, Type[BaseFinetuneAdapter]] = {
 9    ModelProviderName.openai: OpenAIFinetune,
10    ModelProviderName.fireworks_ai: FireworksFinetune,
11}