Co-authored-by: Quentin Torroba <quentin.torroba@mistral.ai>
Co-authored-by: Clement Sirieix <clem.sirieix@gmail.com>
Co-authored-by: Kim-Adeline Miguel <kimadeline.miguel@mistral.ai>
Co-authored-by: Simon Van de Kerckhove <simon.vandekerckhove@mistral.ai>
Co-authored-by: Vincent Guilloux <vincent.guilloux@mistral.ai>
Co-authored-by: Michel Thomazo <michel.thomazo@mistral.ai>
Co-authored-by: Mistral Vibe <vibe@mistral.ai>
This commit is contained in:
Mathias Gesbert 2026-03-16 17:51:47 +01:00 committed by GitHub
parent 9421fbc08e
commit 5103019b01
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
104 changed files with 7277 additions and 691 deletions

View file

@ -41,6 +41,7 @@ class BuiltinAgentName(StrEnum):
ACCEPT_EDITS = "accept-edits"
AUTO_APPROVE = "auto-approve"
EXPLORE = "explore"
LEAN = "lean"
@dataclass(frozen=True)
@ -51,6 +52,7 @@ class AgentProfile:
safety: AgentSafety
agent_type: AgentType = AgentType.AGENT
overrides: dict[str, Any] = field(default_factory=dict)
install_required: bool = False
def apply_to_config(self, base: VibeConfig) -> VibeConfig:
from vibe.core.config import VibeConfig as VC
@ -134,10 +136,51 @@ EXPLORE = AgentProfile(
overrides={"enabled_tools": ["grep", "read_file"], "system_prompt_id": "explore"},
)
LEAN = AgentProfile(
name=BuiltinAgentName.LEAN,
display_name="Lean",
description="Specialized mode for Lean 4 code analysis, proof assistance, and theorem proving",
safety=AgentSafety.NEUTRAL,
agent_type=AgentType.AGENT,
install_required=True,
overrides={
"system_prompt_id": "lean",
"active_model": "leanstral",
"providers": [
{
"name": "mistral-testing",
"api_base": "https://api.mistral.ai/v1",
"api_key_env_var": "MISTRAL_API_KEY",
"api_style": "reasoning",
"backend": "generic",
}
],
"models": [
{
"name": "labs-leanstral-2603",
"provider": "mistral-testing",
"alias": "leanstral",
"thinking": "high",
"temperature": 1.0,
"auto_compact_threshold": 168_000,
}
],
"compaction_model": {
"name": "mistral-vibe-cli-latest",
"provider": "mistral-testing",
"alias": "devstral-compact",
"temperature": 0.2,
"thinking": "off",
},
"tools": {"bash": {"default_timeout": 1200}},
},
)
BUILTIN_AGENTS: dict[str, AgentProfile] = {
BuiltinAgentName.DEFAULT: DEFAULT,
BuiltinAgentName.PLAN: PLAN,
BuiltinAgentName.ACCEPT_EDITS: ACCEPT_EDITS,
BuiltinAgentName.AUTO_APPROVE: AUTO_APPROVE,
BuiltinAgentName.EXPLORE: EXPLORE,
BuiltinAgentName.LEAN: LEAN,
}