import os

from dotenv import load_dotenv
from smolagents import CodeAgent, LiteLLMModel, tool

load_dotenv()  # Charge les variables d'environnement
llm_model = LiteLLMModel(model_id="mistral/codestral-latest", api_key=os.getenv("MISTRAL_API_KEY"))


def create_code_analyzer_agent():
    """
    Crée un agent simple pour exécuter du code et obtenir sa sortie.

    Returns:
        CodeAgent: Agent configuré pour l'exécution de code
    """
    # Initialiser le modèle Mistral Codestral
    model = LiteLLMModel(
        model_id="mistral/codestral-latest",
        temperature=0.1,  # Déterministe pour l'exécution
        max_tokens=2000,
    )

    # Instructions système simples et directes
    system_prompt = """
Tu es un exécuteur de code Python. Ton rôle est simple : exécuter le code fourni et donner sa sortie finale.

Quand on te donne du code à analyser :
1. Exécute directement le code Python
2. Capture toute la sortie (print, résultats, etc.)
3. Retourne la sortie finale

Sois concis et direct dans tes réponses. Concentre-toi uniquement sur l'exécution et le résultat.
"""

    code_analyzer_agent = CodeAgent(
        name="code_analyzer_agent",
        model=model,
        tools=[],
        description=system_prompt,
        max_steps=3,  # Peu d'étapes nécessaires
    )

    return code_analyzer_agent