from loguru import logger
from dotenv import load_dotenv
from swarms import create_agents_from_yaml

# Load environment variables
load_dotenv()

# Path to your YAML file
yaml_file = 'agents_config.yaml'

try:
    # Create agents and run tasks (using 'both' to return agents and task results)
    agents, task_results = create_agents_from_yaml(yaml_file, return_type="both")

    # Print the results of the tasks
    for result in task_results:
        print(f"Agent: {result['agent_name']} | Task: {result['task']} | Output: {result.get('output', 'Error encountered')}")

except Exception as e:
    logger.error(f"An error occurred: {e}")