import logging

from askui import ComputerAgent
from askui.agent_settings import AgentSettings
from askui.model_providers.anthropic_vlm_provider import AnthropicVlmProvider
from askui.models.shared.agent_message_param import CacheControlEphemeralParam
from askui.models.shared.settings import ActSettings, MessageSettings
from askui.reporting import SimpleHtmlReporter

logging.basicConfig(level=logging.INFO)
logger = logging.getLogger(__name__)


# test and implement anthropic's automatic prompt caching, as explained here:
# https://platform.claude.com/docs/en/build-with-claude/prompt-caching#automatic-caching


def main() -> None:
    agent = ComputerAgent(
        display=1,
        reporters=[SimpleHtmlReporter()],
        settings=AgentSettings(vlm_provider=AnthropicVlmProvider()),
    )
    agent.act_settings.messages.provider_options = {
        "cache_control": CacheControlEphemeralParam()
    }
    with agent:
        agent.act("Open a new Chrome browser")

        agent.act("Navigate to www.askui.com")


if __name__ == "__main__":
    main()