common/hermes
2
0
Fork 0
Code Issues Pull Requests Actions 1 Packages Projects Releases Wiki Activity

feat: add grok to TOOL_USE_ENFORCEMENT_MODELS for direct xAI usage (#5595)

Browse Source 
Grok models (x-ai/grok-4.20-beta, grok-code-fast-1) now receive tool-use
enforcement guidance, steering them to actually call tools instead of
describing intended actions. Matches both OpenRouter (x-ai/grok-*) and
direct xAI API usage.
This commit is contained in:
Teknium
2026-04-06 11:22:07 -07:00
committed by GitHub
parent 0bac07ded3
commit 582dbbbbf7
2 changed files with 4 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
agent/prompt_builder.py
View File

@@ -187,7 +187,7 @@ TOOL_USE_ENFORCEMENT_GUIDANCE = (
# Model name substrings that trigger tool-use enforcement guidance. # Model name substrings that trigger tool-use enforcement guidance.
# Add new patterns here when a model family needs explicit steering. # Add new patterns here when a model family needs explicit steering.
TOOL_USE_ENFORCEMENT_MODELS = ("gpt", "codex", "gemini", "gemma") TOOL_USE_ENFORCEMENT_MODELS = ("gpt", "codex", "gemini", "gemma", "grok")
# OpenAI GPT/Codex-specific execution guidance. Addresses known failure modes # OpenAI GPT/Codex-specific execution guidance. Addresses known failure modes
# where GPT models abandon work on partial results, skip prerequisite lookups, # where GPT models abandon work on partial results, skip prerequisite lookups,

3
tests/agent/test_prompt_builder.py
View File

@@ -1018,6 +1018,9 @@ class TestToolUseEnforcementGuidance:
def test_enforcement_models_includes_codex(self): def test_enforcement_models_includes_codex(self):
assert "codex" in TOOL_USE_ENFORCEMENT_MODELS assert "codex" in TOOL_USE_ENFORCEMENT_MODELS
def test_enforcement_models_includes_grok(self):
assert "grok" in TOOL_USE_ENFORCEMENT_MODELS
def test_enforcement_models_is_tuple(self): def test_enforcement_models_is_tuple(self):
assert isinstance(TOOL_USE_ENFORCEMENT_MODELS, tuple) assert isinstance(TOOL_USE_ENFORCEMENT_MODELS, tuple)