mirror of
https://github.com/mistralai/mistral-vibe
synced 2026-04-25 17:14:55 +02:00
e9a9217cc8
Co-authored-by: Clément Sirieix <clement.sirieix@mistral.ai> Co-authored-by: Kim-Adeline Miguel <kimadeline.miguel@mistral.ai> Co-authored-by: Lucas Marandat <31749711+lucasmrdt@users.noreply.github.com> Co-authored-by: Michel Thomazo <51709227+michelTho@users.noreply.github.com> Co-authored-by: Paul Cacheux <paul.cacheux@mistral.ai> Co-authored-by: Peter Evers <pevers90@gmail.com> Co-authored-by: Pierre Rossinès <pierre.rossines@mistral.ai> Co-authored-by: Pierre Rossinès <pierre.rossines@protonmail.com> Co-authored-by: Quentin <quentin.torroba@mistral.ai> Co-authored-by: Simon Van de Kerckhove <simon.vandekerckhove@mistral.ai> Co-authored-by: Val <102326092+vdeva@users.noreply.github.com> Co-authored-by: Vincent G <10739306+VinceOPS@users.noreply.github.com> Co-authored-by: Mistral Vibe <vibe@mistral.ai>
103 lines
2.8 KiB
Python
103 lines
2.8 KiB
Python
from __future__ import annotations
|
|
|
|
import argparse
|
|
from dataclasses import dataclass
|
|
import os
|
|
import sys
|
|
|
|
import tomli_w
|
|
|
|
from vibe import __version__
|
|
from vibe.core.config import VibeConfig
|
|
from vibe.core.config.harness_files import (
|
|
get_harness_files_manager,
|
|
init_harness_files_manager,
|
|
)
|
|
from vibe.core.logger import logger
|
|
from vibe.core.paths import HISTORY_FILE
|
|
|
|
# Configure line buffering for subprocess communication
|
|
sys.stdout.reconfigure(line_buffering=True) # pyright: ignore[reportAttributeAccessIssue]
|
|
sys.stderr.reconfigure(line_buffering=True) # pyright: ignore[reportAttributeAccessIssue]
|
|
sys.stdin.reconfigure(line_buffering=True) # pyright: ignore[reportAttributeAccessIssue]
|
|
|
|
|
|
@dataclass
|
|
class Arguments:
|
|
setup: bool
|
|
|
|
|
|
def parse_arguments() -> Arguments:
|
|
parser = argparse.ArgumentParser(description="Run Mistral Vibe in ACP mode")
|
|
parser.add_argument(
|
|
"-v", "--version", action="version", version=f"%(prog)s {__version__}"
|
|
)
|
|
parser.add_argument("--setup", action="store_true", help="Setup API key and exit")
|
|
args = parser.parse_args()
|
|
return Arguments(setup=args.setup)
|
|
|
|
|
|
def bootstrap_config_files() -> None:
|
|
mgr = get_harness_files_manager()
|
|
config_file = mgr.user_config_file
|
|
if not config_file.exists():
|
|
try:
|
|
config_file.parent.mkdir(parents=True, exist_ok=True)
|
|
with config_file.open("wb") as f:
|
|
tomli_w.dump(VibeConfig.create_default(), f)
|
|
except Exception as e:
|
|
logger.error(f"Could not create default config file: {e}")
|
|
raise
|
|
|
|
history_file = HISTORY_FILE.path
|
|
if not history_file.exists():
|
|
try:
|
|
history_file.parent.mkdir(parents=True, exist_ok=True)
|
|
history_file.write_text("Hello Vibe!\n", "utf-8")
|
|
except Exception as e:
|
|
logger.error(f"Could not create history file: {e}")
|
|
raise
|
|
|
|
|
|
def handle_debug_mode() -> None:
|
|
if os.environ.get("DEBUG_MODE") != "true":
|
|
return
|
|
|
|
try:
|
|
import debugpy
|
|
except ImportError:
|
|
return
|
|
|
|
debugpy.listen(("localhost", 5678))
|
|
# uncomment this to wait for the debugger to attach
|
|
# debugpy.wait_for_client()
|
|
|
|
|
|
def main() -> None:
|
|
handle_debug_mode()
|
|
init_harness_files_manager("user", "project")
|
|
|
|
from vibe.acp.acp_agent_loop import run_acp_server
|
|
from vibe.core.config import VibeConfig, load_dotenv_values
|
|
from vibe.core.tracing import setup_tracing
|
|
from vibe.setup.onboarding import run_onboarding
|
|
|
|
load_dotenv_values()
|
|
bootstrap_config_files()
|
|
args = parse_arguments()
|
|
if args.setup:
|
|
run_onboarding()
|
|
sys.exit(0)
|
|
|
|
try:
|
|
config = VibeConfig.load()
|
|
setup_tracing(config)
|
|
except Exception:
|
|
pass # tracing disabled
|
|
|
|
run_acp_server()
|
|
|
|
|
|
if __name__ == "__main__":
|
|
main()
|