# coding: utf-8
# Copyright (c) 2025 inclusionAI.
import asyncio

from aworld.agents.parallel_llm_agent import ParallelizableAgent
from aworld.config import RunConfig

from aworld.config.conf import AgentConfig
from aworld.core.task import Task
from aworld.runner import Runners
from examples.tools.common import Tools, Agents
from examples.gym_demo.agent import GymDemoAgent as GymAgent
from examples.tools.gym_tool.async_openai_gym import OpenAIGym


async def main():
    agent = GymAgent(name=Agents.GYM.value, conf=AgentConfig(), tool_names=[Tools.GYM.value])
    agent = ParallelizableAgent(name=agent.name(), conf=agent.conf,
                                tool_names=[Tools.GYM.value],
                                agents=[agent])
    # It can also be used `ToolFactory` for simplification.
    task = Task(agent=agent,
                tools_conf={Tools.GYM.value: {"env_id": "CartPole-v1", "render_mode": "human", "render": True}})
    res = await Runners.run_task(task=task, run_conf=RunConfig())
    return res


if __name__ == "__main__":
    # We use it as a showcase to demonstrate the framework's scalability.
    asyncio.run(main())