chore: upgrade scrapybara

Author: justinsunyt

.DS_Store

@@ -3,4 +3,5 @@
 .env
 .venv/
 .DS_Store
-keys/
+keys/
+.DS_Store

.gitignore

@@ -8,7 +8,7 @@ authors = ["Justin Sun <justinsunyt@gmail.com>"]
 python = "^3.9"
 PyGithub = "^2.1.1"
 python-dotenv = "^1.0.0"
-scrapybara = "2.1.3"
+scrapybara = "^2.2.9"
 pydantic = "^2.5.3"
 python-jose = {extras = ["cryptography"], version = "^3.3.0"}
 openai = "^1.59.9"

poetry.lock

@@ -5,7 +5,7 @@
 from scrapybara import Scrapybara
 from scrapybara.anthropic import Anthropic
 from scrapybara.tools import ComputerTool, BashTool, EditTool
-from scrapybara.client import Step, Instance
+from scrapybara.client import Step, UbuntuInstance
 from ..config import settings
 from ..github import add_pr_comment, edit_pr_comment
 from ..generate import GenerateResponse
@@ -38,7 +38,7 @@ class ExecuteAgent:
     def __init__(self, config: ExecuteConfig):
         self.config = config
         self.scrapybara_client = Scrapybara(api_key=settings.scrapybara_api_key)
-        self.instance: Optional[Instance] = None
+        self.instance: Optional[UbuntuInstance] = None
         self._review: Optional[Review] = None
 
     def handle_setup_step(
@@ -143,7 +143,7 @@ async def execute_tests(
             review = upsert_review(review)
 
         try:
-            self.instance = self.scrapybara_client.start(instance_type="large")
+            self.instance = self.scrapybara_client.start_ubuntu()
 
             if review:
                 review.instance_id = self.instance.id