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.fine_tune.together_finetune import TogetherFinetune
 7from kiln_ai.adapters.ml_model_list import ModelProviderName
 8
 9finetune_registry: dict[ModelProviderName, Type[BaseFinetuneAdapter]] = {
10    ModelProviderName.openai: OpenAIFinetune,
11    ModelProviderName.fireworks_ai: FireworksFinetune,
12    ModelProviderName.together_ai: TogetherFinetune,
13}
finetune_registry: dict[kiln_ai.adapters.ml_model_list.ModelProviderName, typing.Type[kiln_ai.adapters.fine_tune.base_finetune.BaseFinetuneAdapter]] = {<ModelProviderName.openai: 'openai'>: <class 'kiln_ai.adapters.fine_tune.openai_finetune.OpenAIFinetune'>, <ModelProviderName.fireworks_ai: 'fireworks_ai'>: <class 'kiln_ai.adapters.fine_tune.fireworks_finetune.FireworksFinetune'>, <ModelProviderName.together_ai: 'together_ai'>: <class 'kiln_ai.adapters.fine_tune.together_finetune.TogetherFinetune'>}