Skip to content

Commit

Permalink
Add Grok 2. It's useful for generating biased content (for bias evals)..
Browse files Browse the repository at this point in the history
  • Loading branch information
scosman committed Mar 1, 2025
1 parent 76bfd3b commit 83a073d
Showing 1 changed file with 17 additions and 0 deletions.
17 changes: 17 additions & 0 deletions libs/core/kiln_ai/adapters/ml_model_list.py
Original file line number Diff line number Diff line change
Expand Up @@ -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
Expand Down Expand Up @@ -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):
Expand Down Expand Up @@ -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,
),
],
),
]

0 comments on commit 83a073d

Please sign in to comment.