# executor.py
from typing import List, Dict
from langchain_ollama import ChatOllama
from config import LLM_HOST, MODEL_NAME
import time

class ExecutorAgent:
    def __init__(self, tools, llm=None):
        self.tools = tools
        self.llm = llm or ChatOllama(
            model=MODEL_NAME, 
            temperature=0.0,
            base_url=LLM_HOST
        )

    def execute_plan(self, plan: List[Dict]) -> List[Dict]:
        """Выполняет план шаг за шагом через инструменты"""
        results = []
        total_steps = len(plan)
        
        print(f"\n{'='*60}")
        print(f"🚀 Начало выполнения плана ({total_steps} шагов)")
        print(f"{'='*60}")
        
        for i, step in enumerate(plan, 1):
            tool_name = step.get("tool", "unknown")
            args = step.get("args", {})
            description = step.get("description", "")
            
            print(f"\n{'─'*60}")
            print(f"📍 Шаг {i}/{total_steps}: {tool_name}")
            if description:
                print(f"   📝 Описание: {description}")
            print(f"   📦 Аргументы: {args}")
            print(f"   ─────────────────────────────")
            
            # Находим инструмент по имени
            tool = None
            for t in self.tools:
                if t.name == tool_name:
                    tool = t
                    break
            
            start_time = time.time()
            
            if tool:
                try:
                    result = tool.invoke(args)
                    elapsed = time.time() - start_time
                    
                    results.append({
                        "step": step,
                        "output": str(result),
                        "success": True,
                        "duration": elapsed
                    })
                    
                    print(f"   ✅ Успех ({elapsed:.2f}s)")
                    print(f"   📤 Результат: {str(result)[:100]}{'...' if len(str(result)) > 100 else ''}")
                    
                except Exception as e:
                    elapsed = time.time() - start_time
                    results.append({
                        "step": step,
                        "output": str(e),
                        "success": False,
                        "error": str(e),
                        "duration": elapsed
                    })
                    
                    print(f"   ❌ Ошибка ({elapsed:.2f}s)")
                    print(f"   💥 {type(e).__name__}: {str(e)[:100]}")
            else:
                results.append({
                    "step": step,
                    "output": f"Tool '{tool_name}' not found",
                    "success": False,
                    "duration": 0
                })
                
                print(f"   ❌ Инструмент '{tool_name}' не найден")
        
        # Итоговая статистика
        success_count = sum(1 for r in results if r.get("success", False))
        failed_count = total_steps - success_count
        total_duration = sum(r.get("duration", 0) for r in results)
        
        print(f"\n{'='*60}")
        print(f"📊 ИТОГИ ВЫПОЛНЕНИЯ")
        print(f"{'='*60}")
        print(f"✅ Успешно: {success_count}/{total_steps}")
        print(f"❌ Ошибки: {failed_count}")
        print(f"⏱️ Общее время: {total_duration:.2f}s")
        print(f"{'='*60}\n")
        
        return results