From 83a073d44a9807226a8502073671d7a64dbe7e53 Mon Sep 17 00:00:00 2001 From: scosman Date: Sat, 1 Mar 2025 14:39:10 -0500 Subject: [PATCH] Add Grok 2. It's useful for generating biased content (for bias evals).. --- libs/core/kiln_ai/adapters/ml_model_list.py | 17 +++++++++++++++++ 1 file changed, 17 insertions(+) diff --git a/libs/core/kiln_ai/adapters/ml_model_list.py b/libs/core/kiln_ai/adapters/ml_model_list.py index 9c8b71fd..b2af1794 100644 --- a/libs/core/kiln_ai/adapters/ml_model_list.py +++ b/libs/core/kiln_ai/adapters/ml_model_list.py @@ -44,6 +44,7 @@ class ModelFamily(str, Enum): qwen = "qwen" deepseek = "deepseek" dolphin = "dolphin" + grok = "grok" # Where models have instruct and raw versions, instruct is default and raw is specified @@ -92,6 +93,7 @@ class ModelName(str, Enum): deepseek_r1_distill_qwen_7b = "deepseek_r1_distill_qwen_7b" deepseek_r1_distill_llama_8b = "deepseek_r1_distill_llama_8b" dolphin_2_9_8x22b = "dolphin_2_9_8x22b" + grok_2 = "grok_2" class ModelParserID(str, Enum): @@ -1016,4 +1018,19 @@ class KilnModel(BaseModel): ), ], ), + # Grok 2 + KilnModel( + family=ModelFamily.grok, + name=ModelName.grok_2, + friendly_name="Grok 2", + providers=[ + KilnModelProvider( + name=ModelProviderName.openrouter, + provider_options={"model": "x-ai/grok-2-1212"}, + supports_structured_output=True, + supports_data_gen=True, + structured_output_mode=StructuredOutputMode.json_schema, + ), + ], + ), ]