#!/usr/bin/env python3
"""
Script para corrigir o model_identifier de agentes
"""

import sys
import os
from dotenv import load_dotenv

load_dotenv()

sys.path.append(os.path.dirname(os.path.abspath(__file__)))

from src import create_app
from src.models import Agente, db

def fix_agent_model():
    app = create_app()
    with app.app_context():
        print("=== CORRIGIR MODEL_IDENTIFIER ===")
        
        agentes_problema = Agente.query.filter(
            (Agente.nome.ilike('%llama%')) | 
            (Agente.nome.ilike('%5g%'))
        ).all()
        
        print(f"\nAgentes encontrados com nomes relacionados ao Llama:")
        for agente in agentes_problema:
            print(f"\n--- Agente ID: {agente.id} ---")
            print(f"Nome: {agente.nome}")
            print(f"Model Identifier atual: {agente.model_identifier}")
            print(f"User ID: {agente.user_id}")
            
            if agente.model_identifier != 'llama3:latest':
                resposta = input(f"\nDeseja alterar o model_identifier do agente '{agente.nome}' para 'llama3:latest'? (s/N): ")
                if resposta.lower() in ['s', 'sim', 'y', 'yes']:
                    agente.model_identifier = 'llama3:latest'
                    db.session.commit()
                    print(f"✅ Model identifier do agente '{agente.nome}' alterado para 'llama3:latest'")
                else:
                    print(f"❌ Não alterado")
            else:
                print(f"✅ Já está correto")

if __name__ == '__main__':
    fix_agent_model()