main.py

import os
from datetime import datetime, timezone
from scrapybara import Scrapybara
from scrapybara.client import UbuntuInstance
from scrapybara.tools import ComputerTool
from scrapybara.openai import OpenAI, UBUNTU_SYSTEM_PROMPT


def main():
    client = Scrapybara(
        api_key=os.getenv("SCRAPYBARA_API_KEY"),
        base_url=os.getenv("SCRAPYBARA_BASE_URL"),
    )
    instance = UbuntuInstance(
        id="test-instance",
        launch_time=datetime.now(timezone.utc),
        status="running",
        client=client._base_client,
    )
    client.instance = instance
    client.act(
        model=OpenAI(),
        tools=[
            # BashTool() and EditTool() are not currently supported
            ComputerTool(instance=instance),
        ],
        system=UBUNTU_SYSTEM_PROMPT,
        prompt="Go to the top link on Hacker News",
        on_step=lambda step: print(step.text),
    )


if __name__ == "__main__":
    main()