from gerenciador_agentes_ia.core import AgentManager
from gerenciador_agentes_ia.agents import Agent

# Inicializar o gerenciador de agentes
agent_manager = AgentManager()

# Criar um agente personalizado para análise de dados
data_agent = Agent(
    id="agent_data_analysis",
    name="Agente de Análise de Dados",
    description="Especialista em análise e visualização de dados",
    specialty="data_analysis",
    base_prompt="""
    Você é um especialista em análise de dados com ampla experiência em estatística,
    visualização de dados e interpretação de resultados. Seu objetivo é ajudar a extrair
    insights valiosos de conjuntos de dados e apresentá-los de forma clara e acionável.
    
    Ao analisar dados, você deve:
    1. Identificar tendências, padrões e anomalias
    2. Aplicar métodos estatísticos apropriados
    3. Criar visualizações informativas
    4. Fornecer interpretações claras dos resultados
    5. Sugerir ações baseadas nos insights
    
    Suas respostas devem ser estruturadas, precisas e focadas em insights acionáveis.
    """,
    tools=[
        "data_import",
        "statistical_analysis",
        "data_visualization",
        "report_generation"
    ],
    config={
        "model": "gpt-4",
        "temperature": 0.2,
        "max_tokens": 2000,
        "supported_formats": ["csv", "excel", "json", "sql"]
    }
)

# Registrar o agente no sistema
agent_manager.register_agent(data_agent)

# Verificar se o agente foi registrado corretamente
registered_agent = agent_manager.get_agent("agent_data_analysis")
print(f"Agente registrado: {registered_agent.name}")
                            

from gerenciador_agentes_ia.core import ToolRegistry
from gerenciador_agentes_ia.tools import Tool
import requests

# Criar uma ferramenta para integração com API de clima
class WeatherTool(Tool):
    def __init__(self, api_key):
        super().__init__(
            name="weather_tool",
            description="Obtém informações de clima para uma localização específica",
            parameters={
                "location": {
                    "type": "string",
                    "description": "Nome da cidade ou coordenadas (latitude,longitude)"
                },
                "units": {
                    "type": "string",
                    "description": "Unidades de medida (metric, imperial)",
                    "default": "metric"
                }
            }
        )
        self.api_key = api_key
        self.base_url = "https://api.openweathermap.org/data/2.5/weather"
    
    def execute(self, parameters):
        """
        Executa a ferramenta com os parâmetros fornecidos.
        
        Args:
            parameters (dict): Parâmetros para a execução
            
        Returns:
            dict: Resultado da execução
        """
        try:
            # Preparar parâmetros da requisição
            params = {
                "q": parameters.get("location"),
                "units": parameters.get("units", "metric"),
                "appid": self.api_key
            }
            
            # Fazer requisição à API
            response = requests.get(self.base_url, params=params)
            response.raise_for_status()
            
            # Processar resposta
            data = response.json()
            
            # Formatar resultado
            result = {
                "location": f"{data['name']}, {data['sys']['country']}",
                "temperature": {
                    "current": data["main"]["temp"],
                    "feels_like": data["main"]["feels_like"],
                    "min": data["main"]["temp_min"],
                    "max": data["main"]["temp_max"]
                },
                "humidity": data["main"]["humidity"],
                "pressure": data["main"]["pressure"],
                "wind": {
                    "speed": data["wind"]["speed"],
                    "direction": data["wind"]["deg"]
                },
                "weather": {
                    "main": data["weather"][0]["main"],
                    "description": data["weather"][0]["description"]
                }
            }
            
            return {
                "status": "success",
                "data": result
            }
            
        except Exception as e:
            return {
                "status": "error",
                "message": str(e)
            }

# Registrar a ferramenta no sistema
tool_registry = ToolRegistry()
weather_tool = WeatherTool(api_key="your_api_key_here")
tool_registry.register_tool(weather_tool)

# Atribuir a ferramenta a um agente
agent_manager = AgentManager()
agent = agent_manager.get_agent("agent_operations_1")
agent.add_tool("weather_tool")