try.py

import os
import sys

from mlx_use.llm.factory import LLMFactory
from mlx_use.llm.providers import LLMConfig, LLM

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

from mlx_use import Agent
from mlx_use.controller.service import Controller

config = LLMConfig(
    provider_name=LLM.OPENAI,
    env_var="OPENAI_API_KEY",
    params={"model": "gpt-4o"}
)

provider = LLMFactory.create_provider(config)
llm = provider.get_llm()

controller = Controller()


async def main():
    agent_greeting = Agent(
        task='Say "Hi there $whoami,  What can I do for you today?"',
        llm=llm,
        controller=controller,
        use_vision=False,
        max_actions_per_step=1,
        max_failures=5
    )

    await agent_greeting.run(max_steps=25)
    task = input("Enter the task: ")

    agent_task = Agent(
        task=task,
        llm=llm,
        controller=controller,
        use_vision=False,
        max_actions_per_step=4,
        max_failures=5
    )

    await agent_task.run(max_steps=25)


asyncio.run(main())